package store

import (
	"context"
	"testing"
	"time"

	"gitea.stevedudenhoeffer.com/steve/executus/skill"
)

func TestSQLiteSkillStore(t *testing.T) {
	ctx := context.Background()
	db, err := Open(":memory:")
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()
	st := db.Skills()
	if err := st.Initialize(ctx); err != nil {
		t.Fatal(err)
	}

	pub := &skill.Skill{ID: "a", Name: "pub", OwnerID: "o1", Visibility: skill.VisibilityPublic,
		Tools: []string{"summarize"}, ExposeAsChatbotTool: true}
	shared := &skill.Skill{ID: "b", Name: "shr", OwnerID: "o1", Visibility: skill.VisibilityShared, SharedWith: []string{"bob"}}
	if err := st.Save(ctx, pub); err != nil {
		t.Fatal(err)
	}
	if err := st.Save(ctx, shared); err != nil {
		t.Fatal(err)
	}

	got, err := st.Get(ctx, "a")
	if err != nil || len(got.Tools) != 1 || !got.ExposeAsChatbotTool {
		t.Fatalf("round-trip: %v %+v", err, got)
	}
	if ps, _ := st.ListPublic(ctx); len(ps) != 1 || ps[0].ID != "a" {
		t.Errorf("ListPublic = %+v", ps)
	}
	if ss, _ := st.ListSharedWith(ctx, "bob"); len(ss) != 1 || ss[0].ID != "b" {
		t.Errorf("ListSharedWith(bob) = %+v", ss)
	}
	if ss, _ := st.ListSharedWith(ctx, "carol"); len(ss) != 0 {
		t.Errorf("ListSharedWith(carol) should be empty: %+v", ss)
	}
	if ce, _ := st.ListChatbotExposed(ctx); len(ce) != 1 {
		t.Errorf("ListChatbotExposed = %d, want 1", len(ce))
	}

	// Versions newest-first + by id.
	st.AppendVersion(ctx, skill.SkillVersion{ID: "v1", SkillID: "a", Version: "1.0.0"})
	st.AppendVersion(ctx, skill.SkillVersion{ID: "v2", SkillID: "a", Version: "1.1.0"})
	vs, _ := st.ListVersionsBySkill(ctx, "a", 10)
	if len(vs) != 2 || vs[0].ID != "v2" {
		t.Errorf("versions newest-first: %+v", vs)
	}
	if gv, err := st.GetVersionByID(ctx, "v1"); err != nil || gv.Version != "1.0.0" {
		t.Errorf("GetVersionByID: %v %+v", err, gv)
	}

	// Scheduling.
	now := time.Now().UTC()
	cron := &skill.Skill{ID: "c", Name: "cron", OwnerID: "o1", Schedule: "0 * * * *", NextRunAt: now.Add(-time.Minute)}
	st.Save(ctx, cron)
	if due, _ := st.ListDueScheduled(ctx, now); len(due) != 1 || due[0].ID != "c" {
		t.Fatalf("ListDueScheduled = %+v", due)
	}
	st.MarkScheduledRun(ctx, "c", now, now.Add(time.Hour))
	if due, _ := st.ListDueScheduled(ctx, now); len(due) != 0 {
		t.Errorf("after MarkScheduledRun nothing due: %+v", due)
	}
}