package search

import (
	"path/filepath"
	"strings"
	"testing"

	"github.com/aisim/kb-cli/internal/graph"
	"github.com/aisim/kb-cli/internal/index"
)

func TestSearch(t *testing.T) {
	// 创建临时数据库
	tmpDir := t.TempDir()
	dbPath := filepath.Join(tmpDir, "test.db")

	store, err := index.Open(dbPath)
	if err != nil {
		t.Fatalf("Open failed: %v", err)
	}
	defer store.Close()

	// 插入测试数据
	nodes := []*graph.Node{
		{
			ID:      1,
			Path:    "FAQ/充装类/001-test.md",
			Title:   "充装问题排查",
			Section: "FAQ",
			Tags:    []string{"充装"},
			Content: "关于智能枪充装问题的排查方法",
		},
		{
			ID:      2,
			Path:    "知识/002-config.md",
			Title:   "充装规格配置",
			Section: "知识",
			Tags:    []string{"配置"},
			Content: "充装规格配置说明",
		},
	}

	for _, n := range nodes {
		_, err := store.InsertNode(n)
		if err != nil {
			t.Fatalf("InsertNode failed: %v", err)
		}
	}

	// 创建 FTS 索引
	if err := store.CreateFTS(); err != nil {
		t.Fatalf("CreateFTS failed: %v", err)
	}
	if err := store.PopulateFTS(); err != nil {
		t.Fatalf("PopulateFTS failed: %v", err)
	}

	// 测试搜索
	results, err := Search(store, []string{"充装"}, SearchOptions{})
	if err != nil {
		t.Fatalf("Search failed: %v", err)
	}

	if len(results) == 0 {
		t.Error("Expected results, got none")
	}

	// 验证结果包含关键词
	found := false
	for _, r := range results {
		if strings.Contains(r.Title, "充装") || strings.Contains(r.Path, "充装") {
			found = true
			break
		}
	}
	if !found {
		t.Error("Expected results to contain '充装'")
	}
}

func TestIsGenericWord(t *testing.T) {
	tests := []struct {
		word     string
		expected bool
	}{
		{"问题", true},
		{"故障", true},
		{"充装", false},
		{"智能枪", false},
	}

	for _, tt := range tests {
		result := IsGenericWord(tt.word)
		if result != tt.expected {
			t.Errorf("IsGenericWord(%q) = %v, want %v", tt.word, result, tt.expected)
		}
	}
}