package agents

import (
	"context"
	"encoding/json"
	"path/filepath"
	"strings"
	"testing"
	"time"

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

func TestExposePortRefusals(t *testing.T) {
	ctx := context.Background()
	st, err := store.Open(filepath.Join(t.TempDir(), "byre.db"), store.DefaultPortFirst, func(api.Event) {})
	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: "b", WorktreePath: "/nonexistent", BaseRef: "x", Status: api.TaskStatusActive, CreatedBy: "alice", CreatedAt: now}, "alice")
	if err != nil {
		t.Fatalf("create task: %v", err)
	}

	m := New(st, DefaultLimits, t.TempDir(), t.TempDir())
	l := &live{id: "s1", vendor: api.Claude, taskID: "t1", projectID: "p1"}
	call := func(name string, port int) error {
		raw, err := json.Marshal(exposePortArgs{Name: name, Port: port})
		if err != nil {
			t.Fatalf("marshal: %v", err)
		}
		_, err = m.exposePort(ctx, l, raw)
		return err
	}

	cases := []struct {
		name string
		port int
		want string
	}{
		{"", 41000, "name must be"},
		{"two\nlines", 41000, "name must be"},
		{strings.Repeat("x", maxPreviewName+1), 41000, "name must be"},
		{"web", 40999, "outside this task's block, 41000 to 41009"},
		{"web", 41010, "outside this task's block, 41000 to 41009"},
		{"web", 41009, "nothing is listening on 127.0.0.1:41009"},
	}
	for _, c := range cases {
		err = call(c.name, c.port)
		if err == nil || !strings.Contains(err.Error(), c.want) {
			t.Fatalf("expose_port(%q, %d) = %v, want an error containing %q", c.name, c.port, err, c.want)
		}
	}

	previews, err := st.ListPreviews(ctx, "t1")
	if err != nil || len(previews) != 0 {
		t.Fatalf("previews after refusals = %v, %v", previews, err)
	}
}