package host

import (
	"errors"
	"testing"
)

func envFunc(m map[string]string) func(string) string {
	return func(k string) string { return m[k] }
}

func TestResolveFleetIdentityNotAFleetBox(t *testing.T) {
	_, err := resolveFleetIdentity(envFunc(map[string]string{}))
	if !errors.Is(err, errNotFleetBox) {
		t.Fatalf("err = %v, want errNotFleetBox", err)
	}
	// Explicitly empty is the same as unset.
	_, err = resolveFleetIdentity(envFunc(map[string]string{"CARLOS_FLEET": ""}))
	if !errors.Is(err, errNotFleetBox) {
		t.Fatalf("err = %v, want errNotFleetBox for empty CARLOS_FLEET", err)
	}
}

func TestResolveFleetIdentityExplicitBoxLabel(t *testing.T) {
	id, err := resolveFleetIdentity(envFunc(map[string]string{
		"CARLOS_FLEET":     "pi",
		"CARLOS_FLEET_BOX": "amber-gliding",
	}))
	if err != nil {
		t.Fatalf("resolveFleetIdentity: %v", err)
	}
	if id.Fleet != "pi" || id.Label != "amber-gliding" {
		t.Fatalf("id = %+v, want {pi amber-gliding}", id)
	}
}

func TestResolveFleetIdentityDefaultsLabelFromTunnelName(t *testing.T) {
	id, err := resolveFleetIdentity(envFunc(map[string]string{
		"CARLOS_FLEET":       "pi",
		"CARLOS_TUNNEL_NAME": "pi/amber-gliding",
	}))
	if err != nil {
		t.Fatalf("resolveFleetIdentity: %v", err)
	}
	if id.Fleet != "pi" || id.Label != "amber-gliding" {
		t.Fatalf("id = %+v, want {pi amber-gliding}", id)
	}
}

// An explicit CARLOS_FLEET_BOX must win over CARLOS_TUNNEL_NAME's derived
// value — the explicit override is the reliable path the doc comment names
// for anything the standard "<fleet>/<label>" tunnel-name shape can't cover.
func TestResolveFleetIdentityExplicitBoxLabelWinsOverTunnelName(t *testing.T) {
	id, err := resolveFleetIdentity(envFunc(map[string]string{
		"CARLOS_FLEET":       "pi",
		"CARLOS_FLEET_BOX":   "explicit-label",
		"CARLOS_TUNNEL_NAME": "pi/tunnel-derived-label",
	}))
	if err != nil {
		t.Fatalf("resolveFleetIdentity: %v", err)
	}
	if id.Label != "explicit-label" {
		t.Fatalf("Label = %q, want explicit-label (CARLOS_FLEET_BOX must win)", id.Label)
	}
}

// Fix round 2 (Opus re-review): a bare legacy tunnel name — the LIVE shape
// in today's runbook, e.g. CARLOS_TUNNEL_NAME="pi-box" with no "<fleet>/"
// prefix at all — must be refused LOUDLY when no CARLOS_FLEET_BOX is set,
// not silently accepted as the label. Before this fix, TrimPrefix would
// no-op (the string doesn't start with "pi/") and leave "pi-box" itself as
// the "label" — which still passes ValidTunnelName, so it slipped through
// silently. That would have published a Box the edge could never dial:
// fleetChannelFor's rule 1 checks hub.State("pi" + "/" + "pi-box"), but the
// tunnel hub only ever knows this box's channel by its raw
// CARLOS_TUNNEL_NAME, "pi-box" — never "pi/pi-box".
func TestResolveFleetIdentityRefusesBareLegacyTunnelNameWithoutFleetPrefix(t *testing.T) {
	_, err := resolveFleetIdentity(envFunc(map[string]string{
		"CARLOS_FLEET":       "pi",
		"CARLOS_TUNNEL_NAME": "pi-box", // no "pi/" prefix — the legacy shape
	}))
	if err == nil || errors.Is(err, errNotFleetBox) {
		t.Fatalf("err = %v, want a loud, non-errNotFleetBox refusal", err)
	}
}

// An explicit CARLOS_FLEET_BOX must still work regardless of
// CARLOS_TUNNEL_NAME's shape — including the bare legacy form the previous
// test refuses to derive a label from. The explicit override never goes
// anywhere near the prefix-stripping path.
func TestResolveFleetIdentityExplicitBoxLabelWorksWithLegacyTunnelName(t *testing.T) {
	id, err := resolveFleetIdentity(envFunc(map[string]string{
		"CARLOS_FLEET":       "pi",
		"CARLOS_FLEET_BOX":   "amber-gliding",
		"CARLOS_TUNNEL_NAME": "pi-box", // legacy shape; must be ignored entirely
	}))
	if err != nil {
		t.Fatalf("resolveFleetIdentity: %v", err)
	}
	if id.Label != "amber-gliding" {
		t.Fatalf("Label = %q, want amber-gliding (explicit CARLOS_FLEET_BOX must work regardless of tunnel-name form)", id.Label)
	}
}

func TestResolveFleetIdentityInvalidFleetNameRefuses(t *testing.T) {
	_, err := resolveFleetIdentity(envFunc(map[string]string{
		"CARLOS_FLEET":     "Not Valid!",
		"CARLOS_FLEET_BOX": "amber-gliding",
	}))
	if err == nil || errors.Is(err, errNotFleetBox) {
		t.Fatalf("err = %v, want a non-nil, non-errNotFleetBox validation error", err)
	}
}

func TestResolveFleetIdentityMissingLabelRefuses(t *testing.T) {
	// CARLOS_FLEET set, neither CARLOS_FLEET_BOX nor CARLOS_TUNNEL_NAME set
	// at all — nothing to derive a label from.
	_, err := resolveFleetIdentity(envFunc(map[string]string{
		"CARLOS_FLEET": "pi",
	}))
	if err == nil || errors.Is(err, errNotFleetBox) {
		t.Fatalf("err = %v, want a non-nil, non-errNotFleetBox validation error", err)
	}
}

func TestResolveFleetIdentityInvalidLabelRefuses(t *testing.T) {
	_, err := resolveFleetIdentity(envFunc(map[string]string{
		"CARLOS_FLEET":     "pi",
		"CARLOS_FLEET_BOX": "not a valid label!",
	}))
	if err == nil || errors.Is(err, errNotFleetBox) {
		t.Fatalf("err = %v, want a non-nil, non-errNotFleetBox validation error", err)
	}
}