search

type Analyzer struct {
	Type        string   `json:"type"` // standard, simple, whitespace, keyword, pattern
	Tokenizer   string   `json:"tokenizer,omitempty"`
	Filters     []string `json:"filters,omitempty"`
	CharFilters []string `json:"char_filters,omitempty"`
}

type AutocompleteQuery struct {
	Index string `json:"index"`
	Query string `json:"query"`
	Field string `json:"field"`
	Limit int    `json:"limit,omitempty"`
}

type AutocompleteResults struct {
	Completions    []Completion  `json:"completions"`
	ProcessingTime time.Duration `json:"processing_time_ms"`
}

type Completion struct {
	Text  string  `json:"text"`
	Score float64 `json:"score"`
}

type Config struct {
	// Driver specifies the search backend: "inmemory", "elasticsearch", "meilisearch", "typesense"
	Driver string `json:"driver" mapstructure:"driver" yaml:"driver"`

	// Connection settings
	URL      string   `json:"url,omitempty"      mapstructure:"url"      yaml:"url,omitempty"`
	Hosts    []string `json:"hosts,omitempty"    mapstructure:"hosts"    yaml:"hosts,omitempty"`
	Username string   `json:"username,omitempty" mapstructure:"username" yaml:"username,omitempty"`
	Password string   `json:"password,omitempty" mapstructure:"password" yaml:"password,omitempty"`
	APIKey   string   `json:"api_key,omitempty"  mapstructure:"api_key"  yaml:"api_key,omitempty"`

	// Connection pool
	MaxConnections     int           `json:"max_connections"      mapstructure:"max_connections"      yaml:"max_connections"`
	MaxIdleConnections int           `json:"max_idle_connections" mapstructure:"max_idle_connections" yaml:"max_idle_connections"`
	ConnectTimeout     time.Duration `json:"connect_timeout"      mapstructure:"connect_timeout"      yaml:"connect_timeout"`
	RequestTimeout     time.Duration `json:"request_timeout"      mapstructure:"request_timeout"      yaml:"request_timeout"`
	KeepAlive          time.Duration `json:"keep_alive"           mapstructure:"keep_alive"           yaml:"keep_alive"`

	// Retry policy
	MaxRetries     int           `json:"max_retries"      mapstructure:"max_retries"      yaml:"max_retries"`
	RetryBackoff   time.Duration `json:"retry_backoff"    mapstructure:"retry_backoff"    yaml:"retry_backoff"`
	RetryOnTimeout bool          `json:"retry_on_timeout" mapstructure:"retry_on_timeout" yaml:"retry_on_timeout"`

	// Default search settings
	DefaultLimit    int     `json:"default_limit"     mapstructure:"default_limit"     yaml:"default_limit"`
	MaxLimit        int     `json:"max_limit"         mapstructure:"max_limit"         yaml:"max_limit"`
	DefaultMinScore float64 `json:"default_min_score" mapstructure:"default_min_score" yaml:"default_min_score"`
	EnableHighlight bool    `json:"enable_highlight"  mapstructure:"enable_highlight"  yaml:"enable_highlight"`
	EnableFacets    bool    `json:"enable_facets"     mapstructure:"enable_facets"     yaml:"enable_facets"`

	// Performance
	BulkSize          int           `json:"bulk_size"          mapstructure:"bulk_size"          yaml:"bulk_size"`
	FlushInterval     time.Duration `json:"flush_interval"     mapstructure:"flush_interval"     yaml:"flush_interval"`
	EnableCompression bool          `json:"enable_compression" mapstructure:"enable_compression" yaml:"enable_compression"`

	// Security
	EnableTLS          bool   `json:"enable_tls"              mapstructure:"enable_tls"           yaml:"enable_tls"`
	TLSCertFile        string `json:"tls_cert_file,omitempty" mapstructure:"tls_cert_file"        yaml:"tls_cert_file,omitempty"`
	TLSKeyFile         string `json:"tls_key_file,omitempty"  mapstructure:"tls_key_file"         yaml:"tls_key_file,omitempty"`
	TLSCAFile          string `json:"tls_ca_file,omitempty"   mapstructure:"tls_ca_file"          yaml:"tls_ca_file,omitempty"`
	InsecureSkipVerify bool   `json:"insecure_skip_verify"    mapstructure:"insecure_skip_verify" yaml:"insecure_skip_verify"`

	// Monitoring
	EnableMetrics bool `json:"enable_metrics" mapstructure:"enable_metrics" yaml:"enable_metrics"`
	EnableTracing bool `json:"enable_tracing" mapstructure:"enable_tracing" yaml:"enable_tracing"`

	// Config loading flags (not serialized)
	RequireConfig bool `json:"-" mapstructure:"-" yaml:"-"`
}

type ConfigOption func(*Config)

type Document struct {
	ID     string         `json:"id"`
	Fields map[string]any `json:"fields"`
}

type ElasticsearchSearch struct {
	// contains filtered or unexported fields
}

type Extension struct {
	*forge.BaseExtension
	// contains filtered or unexported fields
}

type Facet struct {
	Value string `json:"value"`
	Count int64  `json:"count"`
}

type FacetingConfig struct {
	MaxValues int    `json:"max_values"`
	Sort      string `json:"sort"` // count, name
}

type FieldSchema struct {
	Name         string  `json:"name"`
	Type         string  `json:"type"` // text, keyword, integer, float, boolean, date, geo_point
	Required     bool    `json:"required"`
	Searchable   bool    `json:"searchable"`
	Filterable   bool    `json:"filterable"`
	Sortable     bool    `json:"sortable"`
	Faceted      bool    `json:"faceted"`
	Stored       bool    `json:"stored"`
	Index        bool    `json:"index"`
	Boost        float64 `json:"boost,omitempty"`
	Analyzer     string  `json:"analyzer,omitempty"`
	Format       string  `json:"format,omitempty"` // For dates
	Locale       string  `json:"locale,omitempty"`
	DefaultValue any     `json:"default_value,omitempty"`
}

type Filter struct {
	Field    string `json:"field"`
	Operator string `json:"operator"` // =, !=, >, >=, <, <=, IN, NOT IN, BETWEEN, EXISTS
	Value    any    `json:"value"`
}

type HighlightConfig struct {
	PreTag  string `json:"pre_tag"`
	PostTag string `json:"post_tag"`
}

type Hit struct {
	ID         string              `json:"id"`
	Score      float64             `json:"score"`
	Document   map[string]any      `json:"document"`
	Highlights map[string][]string `json:"highlights,omitempty"`
}

type InMemorySearch struct {
	// contains filtered or unexported fields
}

type IndexInfo struct {
	Name          string      `json:"name"`
	DocumentCount int64       `json:"document_count"`
	IndexSize     int64       `json:"index_size"`
	CreatedAt     time.Time   `json:"created_at"`
	UpdatedAt     time.Time   `json:"updated_at"`
	Schema        IndexSchema `json:"schema"`
}

type IndexSchema struct {
	Fields       []FieldSchema       `json:"fields"`
	Settings     map[string]any      `json:"settings,omitempty"`
	Synonyms     []Synonym           `json:"synonyms,omitempty"`
	StopWords    []string            `json:"stop_words,omitempty"`
	Analyzers    map[string]Analyzer `json:"analyzers,omitempty"`
	Ranking      *RankingConfig      `json:"ranking,omitempty"`
	Faceting     *FacetingConfig     `json:"faceting,omitempty"`
	Highlighting *HighlightConfig    `json:"highlighting,omitempty"`
}

type MeilisearchSearch struct {
	// contains filtered or unexported fields
}

type RankingConfig struct {
	Rules   []string           `json:"rules"`
	Weights map[string]float64 `json:"weights,omitempty"`
}

type Search interface {
	// Connection management
	Connect(ctx context.Context) error
	Disconnect(ctx context.Context) error
	Ping(ctx context.Context) error

	// Index management
	CreateIndex(ctx context.Context, name string, schema IndexSchema) error
	DeleteIndex(ctx context.Context, name string) error
	ListIndexes(ctx context.Context) ([]string, error)
	GetIndexInfo(ctx context.Context, name string) (*IndexInfo, error)

	// Document operations
	Index(ctx context.Context, index string, doc Document) error
	BulkIndex(ctx context.Context, index string, docs []Document) error
	Get(ctx context.Context, index string, id string) (*Document, error)
	Delete(ctx context.Context, index string, id string) error
	Update(ctx context.Context, index string, id string, doc Document) error

	// Search operations
	Search(ctx context.Context, query SearchQuery) (*SearchResults, error)
	Suggest(ctx context.Context, query SuggestQuery) (*SuggestResults, error)
	Autocomplete(ctx context.Context, query AutocompleteQuery) (*AutocompleteResults, error)

	// Analytics
	Stats(ctx context.Context) (*SearchStats, error)
}

type SearchQuery struct {
	Index           string             `json:"index"`
	Query           string             `json:"query"`
	Filters         []Filter           `json:"filters,omitempty"`
	Sort            []SortField        `json:"sort,omitempty"`
	Facets          []string           `json:"facets,omitempty"`
	Offset          int                `json:"offset,omitempty"`
	Limit           int                `json:"limit,omitempty"`
	Highlight       bool               `json:"highlight,omitempty"`
	HighlightFields []string           `json:"highlight_fields,omitempty"`
	Fields          []string           `json:"fields,omitempty"` // Fields to return
	MinScore        float64            `json:"min_score,omitempty"`
	BoostFields     map[string]float64 `json:"boost_fields,omitempty"`
	FuzzyLevel      int                `json:"fuzzy_level,omitempty"` // 0=exact, 1-2=fuzzy
	Options         map[string]any     `json:"options,omitempty"`
}

type SearchResults struct {
	Hits           []Hit              `json:"hits"`
	Total          int64              `json:"total"`
	Offset         int                `json:"offset"`
	Limit          int                `json:"limit"`
	ProcessingTime time.Duration      `json:"processing_time_ms"`
	Facets         map[string][]Facet `json:"facets,omitempty"`
	Query          string             `json:"query"`
	Exhaustive     bool               `json:"exhaustive"`
}

type SearchStats struct {
	IndexCount    int64          `json:"index_count"`
	DocumentCount int64          `json:"document_count"`
	TotalSize     int64          `json:"total_size"`
	Queries       int64          `json:"queries"`
	AvgLatency    time.Duration  `json:"avg_latency_ms"`
	Uptime        time.Duration  `json:"uptime"`
	Version       string         `json:"version"`
	Extra         map[string]any `json:"extra,omitempty"`
}

type SortField struct {
	Field string `json:"field"`
	Order string `json:"order"` // asc, desc
}

type SuggestQuery struct {
	Index string `json:"index"`
	Query string `json:"query"`
	Field string `json:"field"`
	Limit int    `json:"limit,omitempty"`
	Fuzzy bool   `json:"fuzzy,omitempty"`
}

type SuggestResults struct {
	Suggestions    []Suggestion  `json:"suggestions"`
	ProcessingTime time.Duration `json:"processing_time_ms"`
}

type Suggestion struct {
	Text  string  `json:"text"`
	Score float64 `json:"score"`
}

type Synonym struct {
	Terms []string `json:"terms"`
}

type TypesenseSearch struct {
	// contains filtered or unexported fields
}