package main

import (
	"sort"
	"strings"

	"git.zb-server.de/Sebi/ssh-netbox-wrapper/internal/cache"
	"git.zb-server.de/Sebi/ssh-netbox-wrapper/internal/config"
	"git.zb-server.de/Sebi/ssh-netbox-wrapper/internal/netbox"
	"git.zb-server.de/Sebi/ssh-netbox-wrapper/internal/shortcuts"
)

func newNetBoxClient(cfg *config.Config) *netbox.Client {
	if len(cfg.NetBoxes) > 0 {
		sources := make([]netbox.Source, 0, len(cfg.NetBoxes))
		for _, source := range cfg.NetBoxes {
			sources = append(sources, netbox.Source{Name: source.Name, URL: source.URL, Token: source.Token, TokenVersion: source.TokenVersion})
		}
		return netbox.NewMultiClient(sources)
	}
	if cfg.NetBox.URL == "" {
		return nil
	}
	return netbox.NewClient(cfg.NetBox.URL, cfg.NetBox.Token, cfg.NetBox.TokenVersion)
}

func normalizeTarget(cfg config.ShortcutsConfig, qualified bool) func(string) string {
	return func(target string) string {
		source, name := netbox.SplitTarget(target)
		name = shortcuts.Normalize(name, cfg)
		if qualified && source != "" {
			return source + "/" + name
		}
		return name
	}
}

// hostCompletions returns the cached hosts matching toComplete in a form that
// starts with the typed text, because bash and fish only accept such
// candidates. Without an instance prefix, a hostname that is unique across
// all instances is offered bare ("web01"), so it completes like with a single
// NetBox; connecting to it still checks every instance for ambiguity.
// Names present in several instances stay qualified ("prod/web01").
func hostCompletions(entries []cache.Entry, toComplete string) []string {
	prefix := strings.ToLower(toComplete)
	qualifiedInput := strings.Contains(prefix, "/")
	instances := make(map[string]int)
	for _, e := range entries {
		_, name := netbox.SplitTarget(e.Name)
		instances[strings.ToLower(name)]++
	}
	var out []string
	for _, e := range entries {
		source, name := netbox.SplitTarget(e.Name)
		target := strings.ToLower(e.Name)
		switch {
		case source != "" && !qualifiedInput && strings.HasPrefix(strings.ToLower(name), prefix):
			if instances[strings.ToLower(name)] == 1 {
				out = append(out, name)
			} else {
				out = append(out, e.Name)
			}
		case strings.HasPrefix(target, prefix):
			out = append(out, e.Name)
		}
	}
	sort.Strings(out)
	return out
}