package gates

import (
	"go/ast"
	"go/parser"
	"go/token"
	"io/fs"
	"os"
	"os/exec"
	"path/filepath"
	"regexp"
	"sort"
	"strconv"
	"strings"
	"testing"
)

// The battery's hint under its skips must name EVERY host condition the tests read, must say for each
// whether THIS host meets it, and must name what it opens (PD-374): a hint that names one condition of
// several sends a session to a knob that is already on, and the skips it still gets read as the zone's
// normal. The Makefile's `conditions` target derives that list from the test sources; this test holds
// it to that.
//
// ⚠ The two derive the same facts by DIFFERENT MEANS, and that is the point rather than a detail: the
// recipe greps, this gate PARSES the sources and looks at call expressions. A name that lives only in
// a comment is a reader to the grep and not to the parser, and a recipe narrowed to fewer idioms than
// the tests use makes the two disagree — the disagreement is what goes red. Holding both to one
// anchor would make the gate a mirror of the recipe: green whatever either says.
//
// ⚠ WHAT THE WALK DOES NOT SEE, said here because the next author reads this file and not the report:
// it matches a call with a LITERAL argument — `os.Getenv("TM_PLATFORM_TEST_…")`, its `os.LookupEnv`
// twin, `exec.LookPath("…")` — in a `_test.go` of this module. A name arriving through a constant, a
// variable or a helper's parameter, an aliased or dot-imported `os`, or a helper living in a non-test
// file, is invisible to the walk AND to the recipe, so both fall silent together and PD-374 returns at
// full strength. `systemdOrSkip` is exactly that shape and is why the systemd row is hand-written on
// both sides. The class is narrowed here, not closed.
//
// What it asserts, for EVERY row and not for a sample: the NAME appears; the STATE is the host's
// answer, taken here independently (the environment for a variable, `exec.LookPath` for a binary, the
// same systemctl probe `systemdOrSkip` makes); the PACKAGES are exactly those whose tests read it; and
// nothing is printed that the sources do not read.
//
// Mutation caught: replacing the derivation with a literal list; freezing any state column to a word;
// printing an invented condition; attributing a condition to a package that does not read it; dropping
// the systemd probe.
func TestTheBatteryNamesEveryHostConditionItsTestsRead(t *testing.T) {
	if _, err := exec.LookPath("make"); err != nil {
		t.Skip("make is not on this host: the battery's hint cannot be exercised")
	}
	env, bin := conditionsInTheSources(t)
	if len(env) < 2 || len(bin) < 2 {
		t.Fatalf("parsed %d environment conditions and %d binaries out of the test sources, fewer than the battery is known to read: this walk reads the wrong tree", len(env), len(bin))
	}
	// One variable is forced in both directions so the state column is proven to follow the host
	// rather than to print a word; every other row is checked against the host as it stands.
	unset, unsetEnv := conditions(t, "TM_PLATFORM_TEST_DSN=")
	set, _ := conditions(t, "TM_PLATFORM_TEST_DSN=postgres://somewhere")

	printedEnv := map[string][]string{}
	printedBin := map[string][]string{}
	for _, line := range strings.Split(unset, "\n") {
		if m := envRow.FindStringSubmatch(line); m != nil {
			printedEnv[m[1]] = strings.Fields(m[3])
			// The state, against the environment THE RECIPE WAS GIVEN — not this process's own, which
			// differs the moment a variable is forced for the run, and not "is it exported": NON-EMPTY
			// is the fact. That is what the recipe asks (`[ -n "$(printenv …)" ]`) and, more to the
			// point, what every gated test asks — they skip on `os.Getenv(…) == ""`. A wrapper that
			// always exports a knob and leaves it empty (`TM_PLATFORM_TEST_DSN="${DSN:-}"` in a CI
			// script) is a host under which the live suite skips, and a gate that called that "set"
			// would demand the hint lie about it: red for a state the zone did not cause.
			live := unsetEnv[m[1]] != ""
			if want := map[bool]string{true: "set", false: "UNSET"}[live]; m[2] != want {
				t.Errorf("%s is printed as %q and this host says %q", m[1], m[2], want)
			}
			continue
		}
		if m := binRow.FindStringSubmatch(line); m != nil {
			printedBin[m[1]] = strings.Fields(m[3])
			_, err := exec.LookPath(m[1])
			if want := map[bool]string{true: "present", false: "MISSING"}[err == nil]; m[2] != want {
				t.Errorf("binary %s is printed as %q and this host says %q", m[1], m[2], want)
			}
		}
	}
	// Both directions: everything the sources read is printed, and everything printed is read.
	for name, want := range env {
		got, ok := printedEnv[name]
		if !ok {
			t.Errorf("the list does not name %s, which a test reads:\n%s", name, unset)
			continue
		}
		if !equal(got, want) {
			t.Errorf("%s is announced as read by %v, and the sources say %v", name, got, want)
		}
	}
	for name := range printedEnv {
		if _, ok := env[name]; !ok {
			t.Errorf("the list names %s, and no test reads it: a condition nobody reads sends a session to a knob that opens nothing", name)
		}
	}
	for name, want := range bin {
		got, ok := printedBin[name]
		if !ok {
			t.Errorf("the list does not name the binary %s, which a test helper looks up before it runs:\n%s", name, unset)
			continue
		}
		if !equal(got, want) {
			t.Errorf("binary %s is announced as read by %v, and the sources say %v", name, got, want)
		}
	}
	for name := range printedBin {
		if _, ok := bin[name]; !ok {
			t.Errorf("the list names the binary %s, and no test helper looks it up", name)
		}
	}
	if !regexp.MustCompile(`(?m)^\s*TM_PLATFORM_TEST_DSN\s+set\b`).MatchString(set) {
		t.Errorf("with the variable set the list did not say set:\n%s", set)
	}
	// The systemd row carries no name a walk can find — it is a helper's own probe — so it is checked
	// against that same probe, run here.
	m := regexp.MustCompile(`(?m)^\s*systemctl --user\s+(reachable|UNREACHABLE)\s+read by: (\S+).*systemdOrSkip`).FindStringSubmatch(unset)
	if m == nil {
		t.Fatalf("the reachable-user-systemd condition (systemdOrSkip) is not in the list:\n%s", unset)
	}
	reachable := exec.CommandContext(t.Context(), "systemctl", "--user", "show", "--property=Version").Run() == nil
	if want := map[bool]string{true: "reachable", false: "UNREACHABLE"}[reachable]; m[1] != want {
		t.Errorf("the systemd condition is printed as %q and this host answers %q", m[1], want)
	}
	if !strings.Contains(m[2], "internal/runner") {
		t.Errorf("the systemd condition is announced as read by %q, and systemdOrSkip lives in internal/runner", m[2])
	}
	// The idioms themselves, because a set comparison can only speak about conditions that EXIST: the
	// day a test starts reading one through an idiom the recipe does not grep for, that condition
	// vanishes from the hint and BOTH sides fall silent together — the walk would not see it either if
	// the walk were narrowed to match. So the recipe's own anchors are read here and held to the ones
	// this gate understands; narrowing either is what goes red, before any test uses the idiom.
	recipe, err := os.ReadFile(filepath.Join(zoneRoot, "Makefile"))
	if err != nil {
		t.Fatal(err)
	}
	// The recipe is cut the way MAKE cuts it — the target line, then every line that is blank or
	// tab-indented — and not at the first blank line: a recipe split into two indented lines with a
	// blank between them is legal make with identical output, and a gate that failed on it would send
	// the next person who tidies this file looking for a defect that is not there.
	lines := strings.Split(string(recipe), "\n")
	start := -1
	for i, l := range lines {
		if strings.HasPrefix(l, "conditions:") {
			start = i
			break
		}
	}
	if start < 0 {
		t.Fatal("the Makefile declares no `conditions` target: the battery's hint has no source")
	}
	var recipeLines []string
	for _, l := range lines[start+1:] {
		if l != "" && !strings.HasPrefix(l, "\t") {
			break
		}
		recipeLines = append(recipeLines, l)
	}
	// The GREP EXPRESSIONS the recipe runs, not words that appear near them: an idiom named in a
	// comment inside the recipe would satisfy a substring test while the grep beside it had been
	// narrowed back, which is the hole this check exists to close.
	//
	// LOGICAL commands, joined the way the shell joins them: a pipeline split across a backslash
	// continuation is one command however many lines it occupies, and a check that judged physical
	// lines would call a legal reformat a defect — the thing the slice above is careful not to do.
	commands := []string{}
	pending := ""
	for _, l := range recipeLines {
		code := strings.TrimPrefix(l, "\t")
		if head, _, found := strings.Cut(code, "#"); found {
			code = head
		}
		trimmed := strings.TrimRight(code, " \t")
		if strings.HasSuffix(trimmed, `\`) {
			pending += strings.TrimSuffix(trimmed, `\`) + " "
			continue
		}
		commands = append(commands, pending+code)
		pending = ""
	}
	if pending != "" {
		commands = append(commands, pending)
	}
	// EVERY grep that looks for an environment condition must carry BOTH idioms, and the recipe runs
	// more than one of them: the names come from one grep and the packages that read each name from
	// another. An idiom dropped from either is a condition or an attribution that quietly goes
	// missing, and counting whole commands would not see it — the recipe's loops put several greps in
	// one command. So the unit is the grep INVOCATION: each one is taken from its `grep` to the next,
	// and recognised by what it hunts (`TM_PLATFORM_TEST_`, an env idiom, `LookPath`) rather than by
	// the idiom it carries, so narrowing one cannot hide it from this check.
	envGreps, binGreps := 0, 0
	for _, cmd := range commands {
		rest := cmd
		for {
			i := strings.Index(rest, "grep")
			if i < 0 {
				break
			}
			rest = rest[i+len("grep"):]
			invocation := rest
			if j := strings.Index(rest, "grep"); j >= 0 {
				invocation = rest[:j]
			}
			// Only a grep that reads the SOURCES carries an idiom; the pipeline's second grep merely
			// cuts the name out of the first one's output and has no files to look in.
			if !strings.Contains(invocation, "--include") {
				continue
			}
			switch {
			case strings.Contains(invocation, "TM_PLATFORM_TEST_") ||
				strings.Contains(invocation, "Getenv") || strings.Contains(invocation, "LookupEnv"):
				envGreps++
				for _, idiom := range []string{"Getenv", "LookupEnv"} {
					if !strings.Contains(invocation, idiom) {
						t.Errorf("a grep of the `conditions` recipe hunts an environment condition and does not look for %s, which this gate reads out of the sources — a test using that idiom would be a host condition the hint never names:\n\tgrep%s",
							idiom, strings.TrimRight(invocation, " \t"))
					}
				}
			case strings.Contains(invocation, "LookPath"):
				binGreps++
			}
		}
	}
	if envGreps < 2 || binGreps < 1 {
		t.Errorf("the `conditions` recipe runs %d environment greps and %d binary greps; it is known to need at least two and one (names, packages, binaries), so a grep has gone missing or this check reads the wrong lines", envGreps, binGreps)
	}
	// Every printed row must be one of the shapes this gate can judge. A row in a third shape is
	// invisible to both directions above — it can be deleted, or replaced by an invented condition,
	// with the gate green — so the shapes themselves are the assertion.
	for _, line := range strings.Split(unset, "\n") {
		if strings.TrimSpace(line) == "" {
			continue
		}
		switch {
		case envRow.MatchString(line), binRow.MatchString(line),
			strings.Contains(line, "systemctl --user"), strings.Contains(line, "CREATEDB"):
		default:
			t.Errorf("the hint prints a row in a shape this gate cannot judge, so nothing checks it in either direction: %q", line)
		}
	}
}

var (
	envRow = regexp.MustCompile(`^\s*(TM_PLATFORM_TEST_[A-Z_]+)\s+(set|UNSET)\s+read by: (.*)$`)
	binRow = regexp.MustCompile(`^\s*(\S+) \(on PATH\)\s+(present|MISSING)\s+read by: (.*)$`)
)

// conditions runs the target the way `check` does, with some variables forced, and returns BOTH the
// output and the environment the recipe actually saw. The second half is what makes the state column
// checkable: this process's own environment is not the recipe's the moment anything is forced, and a
// gate that compared the two would fail under `make check` (where the live knobs are set) while
// passing on a bare `go test`.
func conditions(t *testing.T, env ...string) (string, map[string]string) {
	t.Helper()
	cmd := exec.CommandContext(t.Context(), "make", "--no-print-directory", "-s", "conditions")
	cmd.Dir = zoneRoot
	cmd.Env = append(os.Environ(), env...)
	out, err := cmd.CombinedOutput()
	if err != nil {
		t.Fatalf("make conditions failed: %v\n%s", err, out)
	}
	// Later entries win in a process environment, so the map is built in the same order.
	effective := map[string]string{}
	for _, kv := range cmd.Env {
		if k, v, ok := strings.Cut(kv, "="); ok {
			effective[k] = v
		}
	}
	return string(out), effective
}

// conditionsInTheSources reads the battery's host conditions out of the test files BY PARSING them:
// the environment variables and the binaries, each mapped to the packages whose tests read it.
//
// A parser rather than a regexp, and that is the whole of its value here: it sees CALLS, so a name in
// a comment or in a string that happens to look like one is not a reader, and a helper that switches
// idioms (`os.Getenv` → `os.LookupEnv`) is still seen. The recipe greps; if the recipe's anchors and
// the language disagree, these two sets disagree and the gate says so.
func conditionsInTheSources(t *testing.T) (env, bin map[string][]string) {
	t.Helper()
	env, bin = map[string][]string{}, map[string][]string{}
	fset := token.NewFileSet()
	err := filepath.WalkDir(zoneRoot, func(path string, d fs.DirEntry, err error) error {
		if err != nil {
			return err
		}
		if d.IsDir() || !strings.HasSuffix(path, "_test.go") {
			return nil
		}
		file, perr := parser.ParseFile(fset, path, nil, 0)
		if perr != nil {
			t.Fatalf("a test source of this zone does not parse (%s): %v", path, perr)
		}
		pkg, rerr := filepath.Rel(zoneRoot, filepath.Dir(path))
		if rerr != nil {
			return rerr
		}
		add := func(m map[string][]string, name string) {
			for _, p := range m[name] {
				if p == pkg {
					return
				}
			}
			m[name] = append(m[name], pkg)
		}
		ast.Inspect(file, func(n ast.Node) bool {
			call, ok := n.(*ast.CallExpr)
			if !ok || len(call.Args) != 1 {
				return true
			}
			sel, ok := call.Fun.(*ast.SelectorExpr)
			if !ok {
				return true
			}
			pkgIdent, ok := sel.X.(*ast.Ident)
			if !ok {
				return true
			}
			lit, ok := call.Args[0].(*ast.BasicLit)
			if !ok || lit.Kind != token.STRING {
				return true
			}
			arg, uerr := strconv.Unquote(lit.Value)
			if uerr != nil {
				return true
			}
			switch {
			case pkgIdent.Name == "os" && (sel.Sel.Name == "Getenv" || sel.Sel.Name == "LookupEnv") &&
				strings.HasPrefix(arg, "TM_PLATFORM_TEST_"):
				add(env, arg)
			case pkgIdent.Name == "exec" && sel.Sel.Name == "LookPath":
				add(bin, arg)
			}
			return true
		})
		return nil
	})
	if err != nil {
		t.Fatal(err)
	}
	return env, bin
}

func equal(got, want []string) bool {
	if len(got) != len(want) {
		return false
	}
	g, w := append([]string(nil), got...), append([]string(nil), want...)
	sort.Strings(g)
	sort.Strings(w)
	for i := range g {
		if g[i] != w[i] {
			return false
		}
	}
	return true
}