package store

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

	core "git-ssh.u1680832.nyat.app/weichen/Agent-System-Core.git"
)

func TestPlatformTables(t *testing.T) {
	st, err := OpenSQLite(filepath.Join(t.TempDir(), "p.db"))
	if err != nil {
		t.Fatal(err)
	}
	t.Cleanup(func() { st.Close() })

	if err := st.SetPluginEnabled("acme", "rdm", false); err != nil {
		t.Fatal(err)
	}
	on, err := st.PluginEnabled("acme", "rdm")
	if err != nil || on {
		t.Fatalf("disabled grant: on=%v err=%v", on, err)
	}
	on, err = st.PluginEnabled("acme", "other")
	if err != nil || !on {
		t.Fatalf("missing grant defaults to enabled: on=%v err=%v", on, err)
	}

	if err := st.SetPluginSchemaVersion("rdm", 2); err != nil {
		t.Fatal(err)
	}
	if v, err := st.PluginSchemaVersion("rdm"); err != nil || v != 2 {
		t.Fatalf("schema: %d %v", v, err)
	}
	if err := st.PutPluginKV("rdm", "acme", "slice", "s1"); err != nil {
		t.Fatal(err)
	}
	got, ok, err := st.GetPluginKV("rdm", "acme", "slice")
	if err != nil || !ok || got != "s1" {
		t.Fatalf("kv: %q ok=%v err=%v", got, ok, err)
	}

	task, err := st.CreateBackgroundTask("acme", "ingest", "入库", "")
	if err != nil {
		t.Fatal(err)
	}
	if task.ConversationID != "" || task.Status != "queued" {
		t.Fatalf("task should not require a conversation: %+v", task)
	}
	if err := st.UpdateBackgroundTask("acme", task.ID, "done", 100, ""); err != nil {
		t.Fatal(err)
	}
	task, err = st.GetBackgroundTask("acme", task.ID)
	if err != nil || task.Status != "done" || task.Progress != 100 {
		t.Fatalf("updated: %+v %v", task, err)
	}

	exp := time.Now().UTC().Add(time.Hour)
	mem, err := st.PutMemory("acme", core.MemoryEntry{Scope: "user", Type: "preference", Content: "用中文", Source: "manual", ExpiresAt: &exp})
	if err != nil || mem.ID == "" {
		t.Fatal(err)
	}
	list, err := st.ListMemory("acme", "user", "")
	if err != nil || len(list) != 1 || list[0].Content != "用中文" || list[0].ExpiresAt == nil {
		t.Fatalf("memory: %+v %v", list, err)
	}

	if err := st.AppendAuditEvent(AuditEvent{Owner: "acme", Action: "upload", Target: "f", Evidence: `{"ok":true}`, TraceID: "tr"}); err != nil {
		t.Fatal(err)
	}
	evs, err := st.ListAuditEvents("acme", 10)
	if err != nil || len(evs) != 1 || evs[0].Action != "upload" || evs[0].TraceID != "tr" {
		t.Fatalf("audit: %+v %v", evs, err)
	}

	if err := st.SavePendingConfirm("card1", "acme", `{"id":"x"}`); err != nil {
		t.Fatal(err)
	}
	owner, payload, ok, err := st.TakePendingConfirm("card1")
	if err != nil || !ok || owner != "acme" || payload != `{"id":"x"}` {
		t.Fatalf("take: %s %s %v %v", owner, payload, ok, err)
	}
	_, _, ok, err = st.TakePendingConfirm("card1")
	if err != nil || ok {
		t.Fatalf("second take must miss: ok=%v err=%v", ok, err)
	}

	st1, err := st.AccountTokens("acme", 30, 40)
	if err != nil || st1.Total != 30 || st1.Over {
		t.Fatalf("under budget: %+v %v", st1, err)
	}
	st2, err := st.AccountTokens("acme", 20, 40)
	if err != nil || st2.Total != 50 || !st2.Over {
		t.Fatalf("over budget: %+v %v", st2, err)
	}
}