package store_test

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

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

func TestQueuedTurnKeepsImages(t *testing.T) {
	ctx := context.Background()
	st, err := store.Open(filepath.Join(t.TempDir(), "byre.db"), 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: "t1", WorktreePath: "/nonexistent", BaseRef: "x", Status: api.TaskStatusActive, CreatedBy: "alice", CreatedAt: now})
	if err != nil {
		t.Fatalf("create task: %v", err)
	}
	err = st.CreateSession(ctx, api.AgentSession{Id: "s1", TaskId: "t1", Vendor: api.Claude, VendorSessionId: "s1", State: api.AgentSessionStateIdle, Role: api.Implementer, CreatedBy: "alice", CreatedAt: now}, "p1")
	if err != nil {
		t.Fatalf("create session: %v", err)
	}

	b, err := st.CreateBlob(ctx, "image/png", []byte("not really a png"), "alice")
	if err != nil {
		t.Fatalf("create blob: %v", err)
	}
	got, data, err := st.ReadBlob(ctx, b.Id)
	if err != nil {
		t.Fatalf("read blob: %v", err)
	}
	if got.ContentType != "image/png" || got.CreatedBy != "alice" || string(data) != "not really a png" {
		t.Fatalf("blob round trip: %+v %q", got, data)
	}

	err = st.EnqueueTurn(ctx, store.QueuedTurn{SessionID: "s1", Actor: "alice", Text: "look", Images: []string{b.Id}, CreatedAt: now})
	if err != nil {
		t.Fatalf("enqueue: %v", err)
	}
	queued, err := st.QueuedTurns(ctx)
	if err != nil {
		t.Fatalf("queued turns: %v", err)
	}
	if len(queued) != 1 || !slices.Equal(queued[0].Images, []string{b.Id}) {
		t.Fatalf("queued turn lost its images: %+v", queued)
	}
}