search

type ClauseFuzzy struct {
	Field     string
	Term      string
	Fuzziness int // 0,1,2…
	Prefix    int // 前缀长
	Boost     *float64
}

type ClauseMatch struct {
	Field    string
	Query    string
	Boost    *float64
	Operator string // "and"/"or"，默认 or
}

type ClausePhrase struct {
	Field  string
	Phrase string
	Slop   int // 词间隔
	Boost  *float64
}

type ClausePrefix struct {
	Field  string
	Prefix string
	Boost  *float64
}

type ClauseQueryString struct {
	Query  string   // 直接使用 QueryString 语法
	Fields []string // 如果不为空，会转换成 field:(q) OR ...
	Boost  *float64
}

type ClauseRegex struct {
	Field   string
	Pattern string
	Boost   *float64
}

type ClauseWildcard struct {
	Field   string
	Pattern string
	Boost   *float64
}

type Config struct {
	IndexPath           string
	DefaultAnalyzer     string
	DefaultSearchFields []string
	OpenTimeout         time.Duration
	QueryTimeout        time.Duration
	BatchSize           int
}

type Doc struct {
	ID     string                 `json:"id"`
	Type   string                 `json:"type"`
	Fields map[string]interface{} `json:"fields"` // 使用 interface{} 来处理任何类型的数据
}

type Engine interface {
	Index(ctx context.Context, doc Doc) error
	IndexBatch(ctx context.Context, docs []Doc) error
	Delete(ctx context.Context, id string) error
	Search(ctx context.Context, req SearchRequest) (SearchResult, error)
	Get(ctx context.Context, id string) (Hit, error)
	List(ctx context.Context, from int, size int) (SearchResult, error)
	GetAutoCompleteSuggestions(ctx context.Context, keyword string) ([]string, error)
	GetSearchSuggestions(ctx context.Context, keyword string) ([]string, error)
	Close() error
}

type FacetRequest struct {
	Name  string // 返回名
	Field string // 字段
	Size  int    // Top N

}

type FacetResult struct {
	Total int         `json:"total"`
	Terms []FacetTerm `json:"terms"`
}

type FacetTerm struct {
	Term  string `json:"term"`
	Count int    `json:"count"`
}

type Hit struct {
	ID        string              `json:"id"`
	Score     float64             `json:"score"`
	Fields    map[string]any      `json:"fields"`
	Fragments map[string][]string `json:"fragments,omitempty"`
}

type NumericRangeFilter struct {
	Field   string
	GTE, GT *float64
	LTE, LT *float64
}

type SearchRequest struct {
	// 关键字（保留老接口）
	Keyword      string
	SearchFields []string

	// 结构化 Term
	MustTerms    map[string][]string
	MustNotTerms map[string][]string
	ShouldTerms  map[string][]string

	// 数值/时间过滤
	NumericRanges []NumericRangeFilter
	TimeRanges    []TimeRangeFilter

	// 高级查询子句（新增）
	QueryString *ClauseQueryString
	Matches     []ClauseMatch
	Phrases     []ClausePhrase
	Prefixes    []ClausePrefix
	Wildcards   []ClauseWildcard
	Regexps     []ClauseRegex
	Fuzzies     []ClauseFuzzy

	// 布尔控制
	MinShould int // 至少满足多少个 should（对 ShouldTerms + 高级 should 子句生效）

	// Facet 聚合
	Facets []FacetRequest

	// 排序与分页
	SortBy []string
	From   int
	Size   int

	// 字段返回与高亮
	IncludeFields   []string
	Highlight       bool
	HighlightFields []string // 指定需要高亮的字段，默认全部 text 字段
	FragmentSize    int      // 片段长度
	MaxFragments    int      // 每字段片段数
}

type SearchResult struct {
	Total  uint64                 `json:"total"`
	Took   time.Duration          `json:"took"`
	Hits   []Hit                  `json:"hits"`
	Facets map[string]FacetResult `json:"facets,omitempty"`
}

type TimeRangeFilter struct {
	Field   string
	From    *time.Time
	To      *time.Time
	IncFrom bool
	IncTo   bool
}