package host

import (
	"context"
	"os"
	"path/filepath"
	"testing"
)

func TestPlan14ParticipationFlagOffUpdate(t *testing.T) {
	t.Setenv("CARLOS_HOST_CLAIM", "")
	for _, region := range []bool{false, true} {
		for _, mirror := range []string{"absent", "empty", "pending", "symlink"} {
			t.Run(map[bool]string{false: "region-off", true: "region-on"}[region]+"/"+mirror, func(t *testing.T) {
				f, _ := applyHappy(t)
				f.a.regionHA = region
				if err := f.a.participation.State.Remove("census.json"); err != nil {
					t.Fatal(err)
				}
				f.a.participation.VerifyLaunches = nil
				switch mirror {
				case "empty":
					hostexec(t, f)
				case "pending":
					if err := os.WriteFile(filepath.Join(hostexec(t, f), "pending.json"), []byte(`{}`), 0600); err != nil {
						t.Fatal(err)
					}
				case "symlink":
					if err := os.Symlink(t.TempDir(), filepath.Join(f.a.dataDir, hostexecDir)); err != nil {
						t.Fatal(err)
					}
				}
				if err := f.a.Run(t.Context(), false); err != nil {
					t.Fatal(err)
				}
				want := systemOutcomeApplied
				if region && (mirror == "pending" || mirror == "symlink") {
					want = systemOutcomeDeferred
				}
				if got := f.result(); got == nil || got.Outcome != want {
					t.Fatalf("result=%+v want=%s", got, want)
				}
				if _, err := f.a.participation.UpdateID(); !os.IsNotExist(err) {
					t.Fatalf("legacy update created root reservation: %v", err)
				}
			})
		}
	}
}

func TestPlan14ParticipationFlagOffRecovery(t *testing.T) {
	t.Setenv("CARLOS_HOST_CLAIM", "")
	for _, journal := range []bool{false, true} {
		t.Run(map[bool]string{false: "marker", true: "journal"}[journal], func(t *testing.T) {
			f := newApplyFixture(t, "stable", "v1")
			if err := f.a.participation.State.Remove("census.json"); err != nil {
				t.Fatal(err)
			}
			f.a.participation.VerifyLaunches = nil
			if err := os.MkdirAll(systemGuardDir(f.a.optDir), 0755); err != nil {
				t.Fatal(err)
			}
			if err := os.WriteFile(updateBegunPath(f.a.optDir), []byte(`{"target":"v2"}`), 0644); err != nil {
				t.Fatal(err)
			}
			if journal {
				if err := f.a.journal(systemJournal{Phase: systemPhaseBackingUp, Target: "v2", PrevBuild: "v1", StartedAt: f.now}); err != nil {
					t.Fatal(err)
				}
			}
			old := recoverPendingControllers
			t.Cleanup(func() { recoverPendingControllers = old })
			recoverPendingControllers = func(context.Context, string, string) error {
				t.Fatal("legacy recovery called root participation recovery")
				return nil
			}
			if err := f.a.Run(t.Context(), true); err != nil {
				t.Fatal(err)
			}
			if _, err := os.Stat(updateBegunPath(f.a.optDir)); !os.IsNotExist(err) {
				t.Fatalf("legacy marker retained: %v", err)
			}
			if j, err := readSystemJournal(f.a.optDir); err != nil || j != nil {
				t.Fatalf("journal=%+v err=%v", j, err)
			}
		})
	}
}

func TestPlan14ParticipationFlagAndCensusExclusion(t *testing.T) {
	for _, flag := range []string{"", "1"} {
		for _, census := range []bool{false, true} {
			t.Run(flag+map[bool]string{false: "/missing", true: "/present"}[census], func(t *testing.T) {
				t.Setenv("CARLOS_HOST_CLAIM", flag)
				f := newApplyFixture(t, "stable", "old")
				if !census {
					if err := f.a.participation.State.Remove("census.json"); err != nil {
						t.Fatal(err)
					}
				}
				f.a.participation.VerifyLaunches = nil
				err := f.a.beginExecutableChange(t.Context(), "new", f.now)
				if (err != nil) != (flag != "" || census) {
					t.Fatalf("admission error=%v flag=%q census=%t", err, flag, census)
				}
			})
		}
	}
}

func TestPlan14ParticipationFlagOffControllersRecovered(t *testing.T) {
	t.Setenv("CARLOS_HOST_CLAIM", "")
	f := newApplyFixture(t, "stable", "v1")
	if err := f.a.participation.State.Remove("census.json"); err != nil {
		t.Fatal(err)
	}
	old := recoverPendingControllers
	t.Cleanup(func() { recoverPendingControllers = old })
	recoverPendingControllers = func(context.Context, string, string) error {
		t.Fatal("legacy controller invoked root recovery")
		return nil
	}
	dir := hostexec(t, f)
	if err := f.a.controllersRecovered(t.Context()); err != nil {
		t.Fatal(err)
	}
	if err := os.WriteFile(filepath.Join(dir, "pending.json"), []byte(`{}`), 0600); err != nil {
		t.Fatal(err)
	}
	if err := f.a.controllersRecovered(t.Context()); err == nil {
		t.Fatal("malformed legacy controller admitted")
	}
}

func TestPlan14ParticipationFlagOffGuardRecovery(t *testing.T) {
	t.Setenv("CARLOS_HOST_CLAIM", "")
	f, id := sharedGuardFixture(t)
	a := f.a
	if err := a.participation.State.Remove("census.json"); err != nil {
		t.Fatal(err)
	}
	a.participation.VerifyLaunches = nil
	if err := a.floor.Begin(t.Context(), id, a.dataDir); err != nil {
		t.Fatal(err)
	}
	if err := a.beginExecutableChange(t.Context(), "new-build", f.now); err != nil {
		t.Fatal(err)
	}
	if err := a.journal(systemJournal{Phase: systemPhaseBackingUp, Target: "new-build", PrevBuild: "old-build", StartedAt: f.now}); err != nil {
		t.Fatal(err)
	}
	if err := a.Run(t.Context(), true); err != nil {
		t.Fatal(err)
	}
	completed, err := os.ReadFile(filepath.Join(a.floor.Dir, "completed.json"))
	if err != nil || string(completed) != id {
		t.Fatalf("completion=%q err=%v", completed, err)
	}
	if _, err := os.Stat(updateBegunPath(a.optDir)); !os.IsNotExist(err) {
		t.Fatalf("legacy floor marker retained: %v", err)
	}
	if _, err := a.participation.UpdateID(); !os.IsNotExist(err) {
		t.Fatalf("legacy recovery created root reservation: %v", err)
	}
}