package index

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

func TestReconcileAdd(t *testing.T) {
	dir := t.TempDir()
	if err := os.MkdirAll(filepath.Join(dir, "FAQ"), 0755); err != nil {
		t.Fatal(err)
	}
	content := "---\ntitle: 测试文档\ntags: [t1]\nstatus: 已解决\n---\n正文\n"
	if err := os.WriteFile(filepath.Join(dir, "FAQ", "001-测试文档.md"), []byte(content), 0644); err != nil {
		t.Fatal(err)
	}

	store, err := Open(filepath.Join(t.TempDir(), "kb.db"))
	if err != nil {
		t.Fatal(err)
	}
	defer store.Close()

	res, err := Reconcile(store, dir)
	if err != nil {
		t.Fatal(err)
	}
	if res.Added != 1 || res.Unchanged != 0 {
		t.Errorf("新增场景: %+v", res)
	}
	var title, status string
	if err := store.db.QueryRow(`SELECT title, status FROM nodes WHERE path='FAQ/001-测试文档.md'`).Scan(&title, &status); err != nil {
		t.Fatal(err)
	}
	if title != "测试文档" || status != "已解决" {
		t.Errorf("节点字段错误: %s / %s", title, status)
	}
}

func TestReconcileModify(t *testing.T) {
	dir := t.TempDir()
	if err := os.MkdirAll(filepath.Join(dir, "FAQ"), 0755); err != nil {
		t.Fatal(err)
	}
	p := filepath.Join(dir, "FAQ", "001-测试文档.md")
	if err := os.WriteFile(p, []byte("---\ntitle: 旧标题\n---\n旧内容\n"), 0644); err != nil {
		t.Fatal(err)
	}

	store, err := Open(filepath.Join(t.TempDir(), "kb.db"))
	if err != nil {
		t.Fatal(err)
	}
	defer store.Close()
	if _, err := Reconcile(store, dir); err != nil {
		t.Fatal(err)
	}

	// 修改内容（title 变新标题，内容加"补气"）
	time.Sleep(1100 * time.Millisecond) // mtime 秒级精度，确保 mtime 变化
	if err := os.WriteFile(p, []byte("---\ntitle: 新标题\n---\n新内容补气\n"), 0644); err != nil {
		t.Fatal(err)
	}

	res, err := Reconcile(store, dir)
	if err != nil {
		t.Fatal(err)
	}
	if res.Modified != 1 || res.Added != 0 {
		t.Errorf("修改场景: %+v", res)
	}
	var title string
	if err := store.db.QueryRow(`SELECT title FROM nodes WHERE path='FAQ/001-测试文档.md'`).Scan(&title); err != nil {
		t.Fatal(err)
	}
	if title != "新标题" {
		t.Errorf("标题未更新: %s", title)
	}
}

func TestReconcileDelete(t *testing.T) {
	dir := t.TempDir()
	if err := os.MkdirAll(filepath.Join(dir, "FAQ"), 0755); err != nil {
		t.Fatal(err)
	}
	p := filepath.Join(dir, "FAQ", "001-测试文档.md")
	if err := os.WriteFile(p, []byte("---\ntitle: 测试文档\n---\n内容\n"), 0644); err != nil {
		t.Fatal(err)
	}

	store, err := Open(filepath.Join(t.TempDir(), "kb.db"))
	if err != nil {
		t.Fatal(err)
	}
	defer store.Close()
	if _, err := Reconcile(store, dir); err != nil {
		t.Fatal(err)
	}
	if err := os.Remove(p); err != nil {
		t.Fatal(err)
	}

	res, err := Reconcile(store, dir)
	if err != nil {
		t.Fatal(err)
	}
	if res.Deleted != 1 {
		t.Errorf("删除场景: %+v", res)
	}
	var n int
	if err := store.db.QueryRow(`SELECT COUNT(*) FROM nodes`).Scan(&n); err != nil {
		t.Fatal(err)
	}
	if n != 0 {
		t.Errorf("节点未删净: %d", n)
	}
}

func TestReconcileUnchanged(t *testing.T) {
	dir := t.TempDir()
	if err := os.MkdirAll(filepath.Join(dir, "FAQ"), 0755); err != nil {
		t.Fatal(err)
	}
	if err := os.WriteFile(filepath.Join(dir, "FAQ", "001-测试文档.md"),
		[]byte("---\ntitle: 测试文档\n---\n内容\n"), 0644); err != nil {
		t.Fatal(err)
	}

	store, err := Open(filepath.Join(t.TempDir(), "kb.db"))
	if err != nil {
		t.Fatal(err)
	}
	defer store.Close()
	if _, err := Reconcile(store, dir); err != nil {
		t.Fatal(err)
	}

	res, err := Reconcile(store, dir)
	if err != nil {
		t.Fatal(err)
	}
	if res.Unchanged != 1 || res.Added != 0 || res.Modified != 0 {
		t.Errorf("未变场景: %+v", res)
	}
}

func TestQuickCheck(t *testing.T) {
	dir := t.TempDir()
	if err := os.MkdirAll(filepath.Join(dir, "FAQ"), 0755); err != nil {
		t.Fatal(err)
	}
	if err := os.WriteFile(filepath.Join(dir, "FAQ", "001-测试文档.md"),
		[]byte("---\ntitle: 测试文档\n---\n内容\n"), 0644); err != nil {
		t.Fatal(err)
	}

	store, err := Open(filepath.Join(t.TempDir(), "kb.db"))
	if err != nil {
		t.Fatal(err)
	}
	defer store.Close()
	if _, err := Reconcile(store, dir); err != nil {
		t.Fatal(err)
	}

	dirty, err := QuickCheck(store, dir)
	if err != nil {
		t.Fatal(err)
	}
	if dirty {
		t.Error("无差异时 QuickCheck 应 false")
	}
}