package routing_test

import (
	"context"
	"encoding/json"
	"os"
	"sort"
	"testing"

	"github.com/mathiasbq/supervisor/internal/registry"
	"github.com/mathiasbq/supervisor/internal/skills/debug"
	"github.com/mathiasbq/supervisor/internal/skills/retrospective"
	"github.com/mathiasbq/supervisor/internal/skills/review"
	"github.com/mathiasbq/supervisor/internal/skills/trainer"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

// TestToolsListMatchesSupervisorSnapshot pins the four routed skills' tool
// definitions to the supervisor's current advertisement. A deliberate schema
// change must be reflected here by updating testdata/tools_list.snapshot.json.
func TestToolsListMatchesSupervisorSnapshot(t *testing.T) {
	complete := func(_ context.Context, _, _, _ string) (string, int64, error) {
		return "", 0, nil
	}

	reg := registry.New()
	reg.Register(review.New(review.Config{
		SkillPrompt:  "stub",
		DefaultModel: "stub",
		CompleteFunc: complete,
	}))
	reg.Register(debug.New(debug.Config{
		SkillPrompt:  "stub",
		DefaultModel: "stub",
		CompleteFunc: complete,
	}))
	reg.Register(retrospective.New(retrospective.Config{
		SkillPrompt:  "stub",
		DefaultModel: "stub",
		CompleteFunc: complete,
	}))
	reg.Register(trainer.New(trainer.Config{
		ReaderPrompt: "stub",
		WriterPrompt: "stub",
		DefaultModel: "stub",
		CompleteFunc: complete,
	}))

	wanted := map[string]bool{
		"review":        true,
		"debug":         true,
		"retrospective": true,
		"trainer":       true,
	}
	var routed []registry.ToolDef
	for _, td := range reg.Tools() {
		if wanted[td.Name] {
			routed = append(routed, td)
		}
	}
	sort.Slice(routed, func(i, j int) bool { return routed[i].Name < routed[j].Name })

	got, err := json.MarshalIndent(routed, "", "  ")
	require.NoError(t, err)

	want, err := os.ReadFile("testdata/tools_list.snapshot.json")
	require.NoError(t, err)

	// Normalize both via re-encode so whitespace differences don't dominate.
	var gotV, wantV any
	require.NoError(t, json.Unmarshal(got, &gotV))
	require.NoError(t, json.Unmarshal(want, &wantV))

	gotN, _ := json.MarshalIndent(gotV, "", "  ")
	wantN, _ := json.MarshalIndent(wantV, "", "  ")

	assert.Equal(t, string(wantN), string(gotN),
		"tool advertisement drifted from supervisor snapshot — update testdata/tools_list.snapshot.json deliberately if the schema change is intentional")
}