types

type MessageDocument struct {
	// Telegram message ID
	ID   int64 `json:"id"`
	Type int   `json:"type"`
	// The original text of the message
	Message string `json:"message"`
	// The OCRed text of the message
	Ocred string `json:"ocred"`
	// [TODO] The AI generated text of the message(summarization, caption, tagging, etc.)
	AIGenerated string `json:"aigenerated"`
	// The ID of the user who sent the message
	UserID    int64 `json:"user_id"`
	ChatID    int64 `json:"chat_id"`
	Timestamp int64 `json:"timestamp"`
}

type MessageType int

type SearchHit struct {
	MessageDocument
	Formatted SearchHitFormatted `json:"_formatted"`
}

type SearchHitFormatted struct {
	ID          string `json:"id"`
	Type        string `json:"type"`
	Message     string `json:"message"`
	Ocred       string `json:"ocred"`
	AIGenerated string `json:"aigenerated"`
	UserID      string `json:"user_id"`
	ChatID      string `json:"chat_id"`
	Timestamp   string `json:"timestamp"`
}

type SearchRequest struct {
	ChatID            int64         `json:"chat_id"`
	Query             string        `json:"query"`
	ChatIDs           []int64       `json:"chat_ids"`
	AllChats          bool          `json:"all_chats"`
	TypeFilters       []MessageType `json:"type_filters"`
	UserFilters       []int64       `json:"user_filters"`
	DisableOcred      bool          `json:"disable_ocred"`      // 不搜索 OCR 文字
	EnableAIGenerated bool          `json:"enable_aigenerated"` // [TODO] 搜索 AI 生成的内容(not implemented yet)
	Limit             int64         `json:"limit"`
	Offset            int64         `json:"offset"`
}

type SearchResponse struct {
	Hits               []SearchHit `json:"hits,omitempty"`
	ProcessingTimeMs   int64       `json:"processingTimeMs,omitempty"`
	Offset             int64       `json:"offset,omitempty"`
	Limit              int64       `json:"limit,omitempty"`
	EstimatedTotalHits int64       `json:"estimatedTotalHits,omitempty"`
	SemanticHitCount   int64       `json:"semanticHitCount,omitempty"`
	Raw                any
}