package handlers

import (
	"context"
	"testing"

	"task-dashboard/internal/models"
	"task-dashboard/internal/store"
)

type mockClaudomatorClient struct {
	stories []models.ClaudomatorStory
}

func (m *mockClaudomatorClient) GetActiveStories(ctx context.Context) ([]models.ClaudomatorStory, error) {
	return m.stories, nil
}

func TestBuildUnifiedAtomList_WithNativeTasks(t *testing.T) {
	s, err := store.New(":memory:", "../../migrations")
	if err != nil {
		t.Fatalf("failed to create in-memory store: %v", err)
	}
	defer s.Close()

	if err := s.CreateNativeTask(models.Task{ID: "native-1", Content: "Buy milk"}); err != nil {
		t.Fatalf("failed to create native task: %v", err)
	}

	atoms, _, err := BuildUnifiedAtomList(s, nil)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}

	found := false
	for _, a := range atoms {
		if a.Source == "doot" && a.Title == "Buy milk" {
			found = true
			break
		}
	}
	if !found {
		t.Errorf("expected atom with Source='doot' and Title='Buy milk'; got atoms: %+v", atoms)
	}
}

func TestBuildUnifiedAtomList_ExcludesLockedChainTasks(t *testing.T) {
	s, err := store.New(":memory:", "../../migrations")
	if err != nil {
		t.Fatalf("failed to create in-memory store: %v", err)
	}
	defer s.Close()

	if _, err := s.CreateChain("Track", []models.ChainTaskInput{
		{Content: "Step 1"}, {Content: "Step 2"}, {Content: "Step 3"},
	}); err != nil {
		t.Fatal(err)
	}

	atoms, _, err := BuildUnifiedAtomList(s, nil)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	for _, a := range atoms {
		if a.Title == "Step 1" || a.Title == "Step 2" || a.Title == "Step 3" {
			t.Errorf("chain task %q should not appear in the flat atom list (chains get their own Chains section)", a.Title)
		}
	}
}

func TestBuildUnifiedAtomList_ExcludesDormantBucketItems(t *testing.T) {
	s, err := store.New(":memory:", "../../migrations")
	if err != nil {
		t.Fatalf("failed to create in-memory store: %v", err)
	}
	defer s.Close()

	bucket, err := s.CreateBucket("Gutters", 30, 1)
	if err != nil {
		t.Fatal(err)
	}
	if err := s.CreateNativeTask(models.Task{ID: "pool-item", Content: "Clean gutters", Priority: 1}); err != nil {
		t.Fatal(err)
	}
	if err := s.AddBucketItem(bucket.ID, "pool-item"); err != nil {
		t.Fatal(err)
	}

	atoms, _, err := BuildUnifiedAtomList(s, nil)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	for _, a := range atoms {
		if a.Title == "Clean gutters" {
			t.Error("dormant bucket-pool item should not appear in the flat atom list")
		}
	}
}

func TestBuildUnifiedAtomList_WithClaudomator(t *testing.T) {
	s, err := store.New(":memory:", "../../migrations")
	if err != nil {
		t.Fatalf("failed to create in-memory store: %v", err)
	}
	defer s.Close()

	mock := &mockClaudomatorClient{
		stories: []models.ClaudomatorStory{
			{ID: "s1", Title: "My story", Status: "IN_PROGRESS", ProjectID: "nav"},
		},
	}

	atoms, _, err := BuildUnifiedAtomList(s, mock)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}

	found := false
	for _, a := range atoms {
		if a.Source == "claudomator" && a.Title == "My story" {
			found = true
			break
		}
	}
	if !found {
		t.Error("expected atom with Source='claudomator' and Title='My story'")
	}
}