package store_test

import (
	"context"
	"errors"
	"fmt"
	"path/filepath"
	"slices"
	"testing"
	"time"

	"g.fc.run/misc/byre/internal/api"
	"g.fc.run/misc/byre/internal/store"
)

func TestSetPreviewReplacesNameAndPort(t *testing.T) {
	ctx := context.Background()
	events := []api.Event{}
	st, err := store.Open(filepath.Join(t.TempDir(), "byre.db"), store.DefaultPortFirst, func(ev api.Event) { events = append(events, ev) })
	if err != nil {
		t.Fatalf("open store: %v", err)
	}
	t.Cleanup(func() { st.Close() })

	now := time.Now().UTC()
	_, err = st.EnsureUser(ctx, "alice", "Alice")
	if err != nil {
		t.Fatalf("ensure user: %v", err)
	}
	err = st.CreateProject(ctx, api.Project{Id: "p1", Name: "demo", RepoPath: "/nonexistent", DefaultBranch: "main", Remote: "none", CreatedBy: "alice", CreatedAt: now})
	if err != nil {
		t.Fatalf("create project: %v", err)
	}
	err = st.CreateTask(ctx, api.Task{Id: "t1", ProjectId: "p1", Title: "t1", Branch: "t1", WorktreePath: "/nonexistent", BaseRef: "x", Status: api.TaskStatusActive, CreatedBy: "alice", CreatedAt: now}, "alice")
	if err != nil {
		t.Fatalf("create task: %v", err)
	}

	set := func(name string, port int) []int {
		t.Helper()
		dropped, err := st.SetPreview(ctx, api.Preview{TaskId: "t1", Name: name, Port: port, CreatedAt: now}, "agent")
		if err != nil {
			t.Fatalf("set preview %s:%d: %v", name, port, err)
		}
		return dropped
	}
	list := func() []string {
		t.Helper()
		previews, err := st.ListPreviews(ctx, "t1")
		if err != nil {
			t.Fatalf("list previews: %v", err)
		}
		out := []string{}
		for _, p := range previews {
			out = append(out, fmt.Sprintf("%s:%d", p.Name, p.Port))
		}
		return out
	}

	if d := set("web", 41001); len(d) != 0 {
		t.Fatalf("first preview dropped %v", d)
	}
	if d := set("api", 41002); len(d) != 0 {
		t.Fatalf("second preview dropped %v", d)
	}
	if d := set("web", 41001); len(d) != 0 {
		t.Fatalf("re-exposing the same name and port dropped %v", d)
	}
	if d := set("web", 41003); !slices.Equal(d, []int{41001}) {
		t.Fatalf("moving web to 41003 dropped %v, want [41001]", d)
	}
	if d := set("storybook", 41002); len(d) != 0 {
		t.Fatalf("renaming 41002 dropped %v, want none", d)
	}
	if got, want := list(), []string{"storybook:41002", "web:41003"}; !slices.Equal(got, want) {
		t.Fatalf("previews = %v, want %v", got, want)
	}
	if last := events[len(events)-1]; last.Kind != "preview.exposed" || last.Payload["name"] != "storybook" {
		t.Fatalf("last event = %s %v", last.Kind, last.Payload)
	}

	_, err = st.DiscardTask(ctx, "t1", "alice")
	if err != nil {
		t.Fatalf("discard: %v", err)
	}
	_, err = st.SetPreview(ctx, api.Preview{TaskId: "t1", Name: "late", Port: 41004, CreatedAt: now}, "agent")
	if !errors.Is(err, store.ErrNotFound) {
		t.Fatalf("set preview on a closed task: %v, want ErrNotFound", err)
	}
}