package search

import (
	"github.com/blevesearch/bleve/v2"
	"github.com/gin-gonic/gin"
	"log"
	"toutoukan/model/article"
)

var GlobalIndex bleve.Index

func DataSearch(c *gin.Context) {
	if GlobalIndex == nil {
		c.JSON(500, gin.H{"error": "索引未初始化，请先调用InitIndex()"})
		return
	}

	keyword := c.Query("keyword")
	if keyword == "" {
		c.JSON(400, gin.H{"error": "请提供搜索关键词（?keyword=xxx）"})
		return
	}

	query := bleve.NewQueryStringQuery(`title:"` + keyword + `" OR content:"` + keyword + `"`)
	searchRequest := bleve.NewSearchRequest(query)

	// 只返回必要字段减少资源消耗
	searchRequest.Fields = []string{"title", "content"}

	searchResult, err := GlobalIndex.Search(searchRequest)
	if err != nil {
		c.JSON(500, gin.H{"error": "搜索失败: " + err.Error()})
		return
	}

	type Result struct {
		ID    string            `json:"id"`
		Score float64           `json:"score"`
		Doc   *article.Document `json:"doc"`
	}
	var results []Result

	for _, hit := range searchResult.Hits {
		doc := &article.Document{}
		hasField := false

		// 宽松的字段处理逻辑
		if title, ok := hit.Fields["title"].(string); ok {
			doc.Title = title
			hasField = true
		} else {
			log.Printf("文档 %s 缺少title字段", hit.ID)
		}

		if content, ok := hit.Fields["content"].(string); ok {
			doc.Content = content
			hasField = true
		} else {
			log.Printf("文档 %s 缺少content字段", hit.ID)
		}

		// 至少有一个字段存在才包含结果
		if hasField {
			results = append(results, Result{
				ID:    hit.ID,
				Score: hit.Score,
				Doc:   doc,
			})
		}
	}

	// 添加调试信息
	if len(results) == 0 {
		log.Printf("关键词'%s'搜索返回%d个文档，但有效结果为0", keyword, searchResult.Total)
	} else {
		log.Printf("关键词'%s'搜索返回%d个文档", keyword, searchResult.Total)
	}

	c.JSON(200, gin.H{
		"total":  len(results),
		"took":   searchResult.Took,
		"result": results,
	})
}