store

type AIBlock struct {
	ID                 int64
	UID                string
	ConversationID     int32
	RoundNumber        int32
	BlockType          AIBlockType
	Mode               AIBlockMode
	UserInputs         []UserInput
	AssistantContent   string
	AssistantTimestamp int64
	EventStream        []BlockEvent
	SessionStats       *SessionStats
	CCSessionID        string
	Status             AIBlockStatus
	Metadata           map[string]any
	ParentBlockID      *int64 // Parent block ID for tree branching (null for root blocks)
	BranchPath         string // Branch path for ordering (e.g., "0/1/3")
	// Token usage statistics (P1-A006: LLM Stats Collection)
	TokenUsage *TokenUsage
	// Cost estimation (ai-block-fields-extension)
	CostEstimate      int64  // In milli-cents (1/1000 of a US cent)
	ModelVersion      string // LLM model used (e.g., "deepseek-chat")
	UserFeedback      string // User feedback: "thumbs_up", "thumbs_down", or custom
	RegenerationCount int32  // Number of times this block was regenerated
	ErrorMessage      string // Error message if status = ERROR
	ArchivedAt        *int64 // Timestamp when block was archived (null if active)
	CreatedTs         int64
	UpdatedTs         int64
}

type AIBlockMode string

type AIBlockStatus string

type AIBlockStore interface {
	// CreateBlock creates a new block
	CreateBlock(ctx context.Context, create *CreateAIBlock) (*AIBlock, error)

	// GetBlock retrieves a block by ID
	GetBlock(ctx context.Context, id int64) (*AIBlock, error)

	// ListBlocks retrieves blocks for a conversation
	ListBlocks(ctx context.Context, find *FindAIBlock) ([]*AIBlock, error)

	// UpdateBlock updates a block
	UpdateBlock(ctx context.Context, update *UpdateAIBlock) (*AIBlock, error)

	// AppendUserInput appends a user input to an existing block
	AppendUserInput(ctx context.Context, blockID int64, input UserInput) error

	// AppendEvent appends an event to the event stream
	AppendEvent(ctx context.Context, blockID int64, event BlockEvent) error

	// AppendEventsBatch appends multiple events to the event stream in a single query
	AppendEventsBatch(ctx context.Context, blockID int64, events []BlockEvent) error

	// UpdateStatus updates the block status
	UpdateStatus(ctx context.Context, blockID int64, status AIBlockStatus) error

	// DeleteBlock deletes a block
	DeleteBlock(ctx context.Context, id int64) error

	// GetLatestBlock retrieves the latest block for a conversation
	GetLatestBlock(ctx context.Context, conversationID int32) (*AIBlock, error)

	// GetPendingBlocks retrieves all pending/streaming blocks for cleanup
	GetPendingBlocks(ctx context.Context) ([]*AIBlock, error)

	// ForkBlock creates a new block as a branch from an existing block.
	// The new block inherits the parent's conversation. User inputs can be optionally replaced.
	// If replaceUserInputs is nil, inherits parent's user inputs.
	// If replaceUserInputs is provided, uses the new user inputs (for message editing).
	ForkBlock(ctx context.Context, parentID int64, reason string, replaceUserInputs []UserInput) (*AIBlock, error)

	// ListChildBlocks lists all direct children of a block.
	ListChildBlocks(ctx context.Context, parentID int64) ([]*AIBlock, error)

	// GetActivePath retrieves the currently active branch path for a conversation.
	GetActivePath(ctx context.Context, conversationID int32) ([]*AIBlock, error)

	// DeleteBranch deletes a block and all its descendants.
	DeleteBranch(ctx context.Context, blockID int64, cascade bool) error

	// ArchiveInactiveBranches archives blocks not on the active branch path.
	// This sets archived_at timestamp for blocks that don't match the target path.
	ArchiveInactiveBranches(ctx context.Context, conversationID int32, targetPath string, archivedAt int64) error
}

type AIBlockType string

type AIConversation struct {
	UID         string
	Title       string
	TitleSource TitleSource // Indicates how the title was created
	ParrotID    string
	RowStatus   RowStatus
	CreatedTs   int64
	UpdatedTs   int64
	ID          int32
	CreatorID   int32
	Pinned      bool
	BlockCount  int32 // Number of blocks in this conversation (populated by ListAIConversations with JOIN)
}

type Activity struct {
	Payload   *storepb.ActivityPayload
	Type      ActivityType
	Level     ActivityLevel
	CreatedTs int64
	ID        int32
	CreatorID int32
}

type ActivityLevel string

type ActivityType string

type AgentMetrics struct {
	HourBucket   time.Time
	AgentType    string
	Errors       string
	ID           int64
	RequestCount int64
	SuccessCount int64
	LatencySumMs int64
	LatencyP50Ms int32
	LatencyP95Ms int32
}

type AgentSessionStats struct {
	ID                   int64
	SessionID            string
	ConversationID       int64
	UserID               int32
	AgentType            string
	StartedAt            time.Time
	EndedAt              time.Time
	TotalDurationMs      int64
	ThinkingDurationMs   int64
	ToolDurationMs       int64
	GenerationDurationMs int64
	InputTokens          int32
	OutputTokens         int32
	CacheWriteTokens     int32
	CacheReadTokens      int32
	TotalTokens          int32
	TotalCostUSD         float64
	ToolCallCount        int32
	ToolsUsed            []string
	FilesModified        int32
	FilePaths            []string
	ModelUsed            string
	IsError              bool
	ErrorMessage         string
	CreatedAt            time.Time
	UpdatedAt            time.Time
}

type AgentStatsStore interface {
	// SaveSessionStats saves a session statistics record.
	SaveSessionStats(ctx context.Context, stats *AgentSessionStats) error

	// GetSessionStats retrieves stats by session ID.
	GetSessionStats(ctx context.Context, sessionID string) (*AgentSessionStats, error)

	// ListSessionStats retrieves stats for a user with pagination.
	ListSessionStats(ctx context.Context, userID int32, limit, offset int) ([]*AgentSessionStats, int64, error)

	// GetDailyCostUsage retrieves total cost for a user in a date range.
	GetDailyCostUsage(ctx context.Context, userID int32, startDate, endDate time.Time) (float64, error)

	// GetCostStats retrieves aggregated cost statistics.
	GetCostStats(ctx context.Context, userID int32, days int) (*CostStats, error)

	// GetUserCostSettings retrieves or creates user cost settings.
	GetUserCostSettings(ctx context.Context, userID int32) (*UserCostSettings, error)

	// SetUserCostSettings updates user cost settings.
	SetUserCostSettings(ctx context.Context, settings *UserCostSettings) error
}

type Attachment struct {
	Payload       *storepb.AttachmentPayload
	MemoUID       *string
	MemoID        *int32
	Type          string
	FilePath      string
	RowStatus     string
	Filename      string
	UID           string
	OCRText       string
	ExtractedText string
	ThumbnailPath string
	Reference     string
	Blob          []byte
	CreatedTs     int64
	Size          int64
	UpdatedTs     int64
	StorageType   storepb.AttachmentStorageType
	ID            int32
	CreatorID     int32
}

type BM25Result struct {
	Memo  *Memo
	Score float32 // BM25 relevance score
}

type BM25SearchOptions struct {
	Query    string
	Limit    int
	UserID   int32
	MinScore float32
}

type BlockEvent struct {
	Type      string         `json:"type"` // "thinking", "tool_use", "tool_result", "answer", "error"
	Content   string         `json:"content,omitempty"`
	Timestamp int64          `json:"timestamp"`
	Meta      map[string]any `json:"meta,omitempty"`
}

type CostStats struct {
	TotalCostUSD         float64
	DailyAverageUSD      float64
	SessionCount         int64
	MostExpensiveSession *AgentSessionStats
	DailyBreakdown       []*DailyCostData
}

type CreateAIBlock struct {
	UID            string
	ConversationID int32
	BlockType      AIBlockType
	Mode           AIBlockMode
	UserInputs     []UserInput
	Metadata       map[string]any
	CCSessionID    string
	Status         AIBlockStatus
	ParentBlockID  *int64 // Parent block ID for forking (null for new root)
	// New fields for block extensions
	TokenUsage   *TokenUsage
	CostEstimate int64
	ModelVersion string
	CreatedTs    int64
	UpdatedTs    int64
}

type CreateRouterFeedback struct {
	UserID    int32
	Input     string
	Predicted string
	Actual    string
	Feedback  string
	Timestamp int64
	Source    string
}

type DailyCostData struct {
	Date         string // YYYY-MM-DD
	CostUSD      float64
	SessionCount int64
}

type DeleteAIConversation struct {
	ID int32
}

type DeleteAgentMetrics struct {
	BeforeTime *time.Time // Delete records older than this time
}

type DeleteAttachment struct {
	MemoID *int32
	ID     int32
}

type DeleteEpisodicMemory struct {
	ID     *int64
	UserID *int32
}

type DeleteIdentityProvider struct {
	ID int32
}

type DeleteInbox struct {
	ID int32
}

type DeleteInstanceSetting struct {
	Name string
}

type DeleteMemo struct {
	ID int32
}

type DeleteMemoRelation struct {
	MemoID        *int32
	RelatedMemoID *int32
	Type          *MemoRelationType
}

type DeleteReaction struct {
	ID int32
}

type DeleteSchedule struct {
	ID int32
}

type DeleteToolMetrics struct {
	BeforeTime *time.Time // Delete records older than this time
}

type DeleteUser struct {
	ID int32
}

type Driver interface {
	GetDB() *sql.DB
	Close() error

	IsInitialized(ctx context.Context) (bool, error)

	// Activity model related methods.
	CreateActivity(ctx context.Context, create *Activity) (*Activity, error)
	ListActivities(ctx context.Context, find *FindActivity) ([]*Activity, error)

	// Attachment model related methods.
	CreateAttachment(ctx context.Context, create *Attachment) (*Attachment, error)
	ListAttachments(ctx context.Context, find *FindAttachment) ([]*Attachment, error)
	UpdateAttachment(ctx context.Context, update *UpdateAttachment) error
	DeleteAttachment(ctx context.Context, delete *DeleteAttachment) error

	// Memo model related methods.
	CreateMemo(ctx context.Context, create *Memo) (*Memo, error)
	ListMemos(ctx context.Context, find *FindMemo) ([]*Memo, error)
	UpdateMemo(ctx context.Context, update *UpdateMemo) error
	DeleteMemo(ctx context.Context, delete *DeleteMemo) error

	// UpdateMemoEmbedding updates the embedding vector for a memo.
	UpdateMemoEmbedding(ctx context.Context, id int32, embedding []float32) error

	// SearchMemosByVector performs semantic search using vector similarity.
	// Returns memos and their similarity scores.
	SearchMemosByVector(ctx context.Context, embedding []float32, limit int) ([]*Memo, []float32, error)

	// MemoRelation model related methods.
	UpsertMemoRelation(ctx context.Context, create *MemoRelation) (*MemoRelation, error)
	ListMemoRelations(ctx context.Context, find *FindMemoRelation) ([]*MemoRelation, error)
	DeleteMemoRelation(ctx context.Context, delete *DeleteMemoRelation) error

	// InstanceSetting model related methods.
	UpsertInstanceSetting(ctx context.Context, upsert *InstanceSetting) (*InstanceSetting, error)
	ListInstanceSettings(ctx context.Context, find *FindInstanceSetting) ([]*InstanceSetting, error)
	DeleteInstanceSetting(ctx context.Context, delete *DeleteInstanceSetting) error

	// User model related methods.
	CreateUser(ctx context.Context, create *User) (*User, error)
	UpdateUser(ctx context.Context, update *UpdateUser) (*User, error)
	ListUsers(ctx context.Context, find *FindUser) ([]*User, error)
	DeleteUser(ctx context.Context, delete *DeleteUser) error

	// UserSetting model related methods.
	UpsertUserSetting(ctx context.Context, upsert *UserSetting) (*UserSetting, error)
	ListUserSettings(ctx context.Context, find *FindUserSetting) ([]*UserSetting, error)
	GetUserByPATHash(ctx context.Context, tokenHash string) (*PATQueryResult, error)

	// IdentityProvider model related methods.
	CreateIdentityProvider(ctx context.Context, create *IdentityProvider) (*IdentityProvider, error)
	ListIdentityProviders(ctx context.Context, find *FindIdentityProvider) ([]*IdentityProvider, error)
	UpdateIdentityProvider(ctx context.Context, update *UpdateIdentityProvider) (*IdentityProvider, error)
	DeleteIdentityProvider(ctx context.Context, delete *DeleteIdentityProvider) error

	// Inbox model related methods.
	CreateInbox(ctx context.Context, create *Inbox) (*Inbox, error)
	ListInboxes(ctx context.Context, find *FindInbox) ([]*Inbox, error)
	UpdateInbox(ctx context.Context, update *UpdateInbox) (*Inbox, error)
	DeleteInbox(ctx context.Context, delete *DeleteInbox) error

	// Reaction model related methods.
	UpsertReaction(ctx context.Context, create *Reaction) (*Reaction, error)
	ListReactions(ctx context.Context, find *FindReaction) ([]*Reaction, error)
	GetReaction(ctx context.Context, find *FindReaction) (*Reaction, error)
	DeleteReaction(ctx context.Context, delete *DeleteReaction) error

	// MemoEmbedding model related methods.
	UpsertMemoEmbedding(ctx context.Context, embedding *MemoEmbedding) (*MemoEmbedding, error)
	ListMemoEmbeddings(ctx context.Context, find *FindMemoEmbedding) ([]*MemoEmbedding, error)
	DeleteMemoEmbedding(ctx context.Context, memoID int32) error
	FindMemosWithoutEmbedding(ctx context.Context, find *FindMemosWithoutEmbedding) ([]*Memo, error)
	VectorSearch(ctx context.Context, opts *VectorSearchOptions) ([]*MemoWithScore, error)
	BM25Search(ctx context.Context, opts *BM25SearchOptions) ([]*BM25Result, error)

	// Schedule model related methods.
	CreateSchedule(ctx context.Context, create *Schedule) (*Schedule, error)
	ListSchedules(ctx context.Context, find *FindSchedule) ([]*Schedule, error)
	UpdateSchedule(ctx context.Context, update *UpdateSchedule) error
	DeleteSchedule(ctx context.Context, delete *DeleteSchedule) error

	// AIConversation model related methods.
	CreateAIConversation(ctx context.Context, create *AIConversation) (*AIConversation, error)
	ListAIConversations(ctx context.Context, find *FindAIConversation) ([]*AIConversation, error)
	UpdateAIConversation(ctx context.Context, update *UpdateAIConversation) (*AIConversation, error)
	DeleteAIConversation(ctx context.Context, delete *DeleteAIConversation) error

	// AIBlock model related methods (Unified Block Model).
	CreateAIBlock(ctx context.Context, create *CreateAIBlock) (*AIBlock, error)
	GetAIBlock(ctx context.Context, id int64) (*AIBlock, error)
	ListAIBlocks(ctx context.Context, find *FindAIBlock) ([]*AIBlock, error)
	UpdateAIBlock(ctx context.Context, update *UpdateAIBlock) (*AIBlock, error)
	DeleteAIBlock(ctx context.Context, id int64) error
	AppendUserInput(ctx context.Context, blockID int64, input UserInput) error
	AppendEvent(ctx context.Context, blockID int64, event BlockEvent) error
	AppendEventsBatch(ctx context.Context, blockID int64, events []BlockEvent) error
	UpdateAIBlockStatus(ctx context.Context, blockID int64, status AIBlockStatus) error
	GetLatestAIBlock(ctx context.Context, conversationID int32) (*AIBlock, error)
	GetPendingAIBlocks(ctx context.Context) ([]*AIBlock, error)
	CreateAIBlockWithRound(ctx context.Context, create *CreateAIBlock) (*AIBlock, error)

	// CompleteBlock atomically marks a block as completed with content and stats.
	CompleteBlock(ctx context.Context, blockID int64, assistantContent string, sessionStats *SessionStats) error

	// EpisodicMemory model related methods.
	CreateEpisodicMemory(ctx context.Context, create *EpisodicMemory) (*EpisodicMemory, error)
	ListEpisodicMemories(ctx context.Context, find *FindEpisodicMemory) ([]*EpisodicMemory, error)
	ListActiveUserIDs(ctx context.Context, cutoff time.Time) ([]int32, error)
	DeleteEpisodicMemory(ctx context.Context, delete *DeleteEpisodicMemory) error

	// UserPreferences model related methods.
	UpsertUserPreferences(ctx context.Context, upsert *UpsertUserPreferences) (*UserPreferences, error)
	GetUserPreferences(ctx context.Context, find *FindUserPreferences) (*UserPreferences, error)

	// AgentMetrics model related methods.
	UpsertAgentMetrics(ctx context.Context, upsert *UpsertAgentMetrics) (*AgentMetrics, error)
	ListAgentMetrics(ctx context.Context, find *FindAgentMetrics) ([]*AgentMetrics, error)
	DeleteAgentMetrics(ctx context.Context, delete *DeleteAgentMetrics) error

	// ToolMetrics model related methods.
	UpsertToolMetrics(ctx context.Context, upsert *UpsertToolMetrics) (*ToolMetrics, error)
	ListToolMetrics(ctx context.Context, find *FindToolMetrics) ([]*ToolMetrics, error)
	DeleteToolMetrics(ctx context.Context, delete *DeleteToolMetrics) error

	// ========== Tree Branching Methods (tree-conversation-branching) ==========
	// ForkBlock creates a new block as a branch from an existing block.
	// The new block inherits the parent's conversation. User inputs can be optionally replaced.
	// If replaceUserInputs is nil, inherits parent's user inputs.
	// If replaceUserInputs is provided, uses the new user inputs (for message editing).
	ForkBlock(ctx context.Context, parentID int64, reason string, replaceUserInputs []UserInput) (*AIBlock, error)
	// ListChildBlocks lists all direct children of a block.
	ListChildBlocks(ctx context.Context, parentID int64) ([]*AIBlock, error)
	// GetActivePath retrieves the currently active branch path for a conversation.
	GetActivePath(ctx context.Context, conversationID int32) ([]*AIBlock, error)
	// DeleteBranch deletes a block and all its descendants.
	DeleteBranch(ctx context.Context, blockID int64, cascade bool) error
	// ArchiveInactiveBranches archives blocks not on the active branch path.
	ArchiveInactiveBranches(ctx context.Context, conversationID int32, targetPath string, archivedAt int64) error

	// AgentStatsStore provides session statistics persistence.
	AgentStatsStore() AgentStatsStore

	// SecurityAuditStore provides security audit logging.
	SecurityAuditStore() SecurityAuditStore
}

type EpisodicMemory struct {
	Timestamp  time.Time
	AgentType  string
	UserInput  string
	Outcome    string
	Summary    string
	ID         int64
	CreatedTs  int64
	UserID     int32
	Importance float32
}

type FindAIBlock struct {
	ID             *int64
	UID            *string
	ConversationID *int32
	Status         *AIBlockStatus
	Mode           *AIBlockMode
	CCSessionID    *string
	ParentBlockID  *int64 // Filter by parent block (for branch queries)
}

type FindAIConversation struct {
	ID        *int32
	UID       *string
	CreatorID *int32
	Pinned    *bool
	RowStatus *RowStatus
}

type FindActivity struct {
	ID   *int32
	Type *ActivityType
}

type FindAgentMetrics struct {
	AgentType *string
	StartTime *time.Time
	EndTime   *time.Time
	Limit     int
}

type FindAttachment struct {
	MemoID         *int32
	ID             *int32
	UID            *string
	CreatorID      *int32
	Filename       *string
	FilenameSearch *string
	StorageType    *storepb.AttachmentStorageType
	Limit          *int
	Offset         *int
	MemoIDList     []int32
	Filters        []string
	GetBlob        bool
	HasRelatedMemo bool
}

type FindEpisodicMemory struct {
	ID        *int64
	UserID    *int32
	AgentType *string
	Query     *string // For text search in user_input and summary
	Limit     int
	Offset    int
}

type FindIdentityProvider struct {
	ID *int32
}

type FindInbox struct {
	ID          *int32
	SenderID    *int32
	ReceiverID  *int32
	Status      *InboxStatus
	MessageType *storepb.InboxMessage_Type

	// Pagination
	Limit  *int
	Offset *int
}

type FindInstanceSetting struct {
	Name string
}

type FindMemo struct {
	ID               *int32
	UID              *string
	Offset           *int
	Limit            *int
	RowStatus        *RowStatus
	CreatorID        *int32
	VisibilityList   []Visibility
	Filters          []string
	UIDList          []string
	IDList           []int32
	ExcludeContent   bool
	ExcludeComments  bool
	OrderByPinned    bool
	OrderByUpdatedTs bool
	OrderByTimeAsc   bool
}

type FindMemoEmbedding struct {
	MemoID *int32
	Model  *string
}

type FindMemoPayload struct {
	Raw                *string
	TagSearch          []string
	HasLink            bool
	HasTaskList        bool
	HasCode            bool
	HasIncompleteTasks bool
}

type FindMemoRelation struct {
	MemoID        *int32
	RelatedMemoID *int32
	Type          *MemoRelationType
	MemoFilter    *string
}

type FindMemosWithoutEmbedding struct {
	Model string // Embedding model to check
	Limit int    // Maximum number of memos to return
}

type FindPromptVersionMetrics struct {
	AgentType     *string
	PromptVersion *string
	StartTime     *time.Time
	EndTime       *time.Time
	Limit         int
}

type FindReaction struct {
	ID            *int32
	CreatorID     *int32
	ContentID     *string
	ContentIDList []string
}

type FindRouterFeedback struct {
	UserID    *int32
	StartTime *int64
	EndTime   *int64
	Feedback  *string
	Limit     int
}

type FindRouterWeight struct {
	UserID   *int32
	Category *string
}

type FindSchedule struct {
	ID        *int32
	UID       *string
	CreatorID *int32

	// Time range filters
	StartTs *int64
	EndTs   *int64

	// Status filter
	RowStatus *RowStatus

	// P1: Schedule query mode
	// 0 = AUTO (auto-select based on query type)
	// 1 = STANDARD (return schedules with any part in range)
	// 2 = STRICT (return only schedules completely in range)
	QueryMode *int32

	// Pagination
	Limit  *int
	Offset *int
}

type FindToolMetrics struct {
	ToolName  *string
	StartTime *time.Time
	EndTime   *time.Time
	Limit     int
}

type FindUser struct {
	ID        *int32
	RowStatus *RowStatus
	Username  *string
	Role      *Role
	Email     *string
	Nickname  *string
	Limit     *int
	Filters   []string
}

type FindUserPreferences struct {
	UserID *int32
}

type FindUserSetting struct {
	UserID *int32
	Key    storepb.UserSetting_Key
}

type GetRouterStats struct {
	UserID    int32
	TimeRange time.Duration
}

type IdentityProvider struct {
	Name             string
	IdentifierFilter string
	Config           string
	ID               int32
	Type             storepb.IdentityProvider_Type
}

type Inbox struct {
	Message    *storepb.InboxMessage
	Status     InboxStatus
	CreatedTs  int64
	ID         int32
	SenderID   int32
	ReceiverID int32
}

type InboxStatus string

type InstanceSetting struct {
	Name        string
	Value       string
	Description string
}

type Memo struct {
	Payload    *storepb.MemoPayload
	ParentUID  *string
	UID        string
	RowStatus  RowStatus
	Content    string
	Visibility Visibility
	CreatedTs  int64
	UpdatedTs  int64
	ID         int32
	CreatorID  int32
	Pinned     bool
}

type MemoEmbedding struct {
	Model     string
	Embedding []float32
	CreatedTs int64
	UpdatedTs int64
	ID        int32
	MemoID    int32
}

type MemoRelation struct {
	Type          MemoRelationType
	MemoID        int32
	RelatedMemoID int32
}

type MemoRelationType string

type MemoWithScore struct {
	Memo  *Memo
	Score float32 // Similarity score (0-1, higher is more similar)
}

type PATQueryResult struct {
	User   *User
	PAT    *storepb.PersonalAccessTokensUserSetting_PersonalAccessToken
	UserID int32
}

type PromptExperimentSummary struct {
	AgentType                  string
	ControlVersion             string
	TreatmentVersion           string
	Recommendation             string
	ControlRequests            int64
	TreatmentRequests          int64
	ControlSuccessRate         float64
	TreatmentSuccessRate       float64
	ControlAvgLatencyMs        int64
	TreatmentAvgLatencyMs      int64
	ImprovementRate            float64
	IsStatisticallySignificant bool
}

type PromptVersionMetrics struct {
	HourBucket       time.Time
	AgentType        string
	PromptVersion    string
	ID               int64
	RequestCount     int64
	SuccessCount     int64
	AvgLatencyMs     int64
	UserSatisfaction float32
}

type Reaction struct {
	ContentID    string
	ReactionType string
	CreatedTs    int64
	ID           int32
	CreatorID    int32
}

type RefreshTokenQueryResult struct {
	RefreshToken *storepb.RefreshTokensUserSetting_RefreshToken
	UserID       int32
}

type Role string

type RouterFeedback struct {
	ID        int64  `json:"id"`
	UserID    int32  `json:"user_id"`
	Input     string `json:"input"`
	Predicted string `json:"predicted"` // What the router predicted (as Intent string)
	Actual    string `json:"actual"`    // What the user actually wanted
	Feedback  string `json:"feedback"`  // FeedbackType: positive, rephrase, switch
	Timestamp int64  `json:"timestamp"`
	Source    string `json:"source"` // "rule", "history", "llm"
}

type RouterStats struct {
	TotalPredictions int64            `json:"total_predictions"`
	CorrectCount     int64            `json:"correct_count"`
	IncorrectCount   int64            `json:"incorrect_count"`
	Accuracy         float64          `json:"accuracy"`
	ByIntent         map[string]int64 `json:"by_intent"`
	BySource         map[string]int64 `json:"by_source"`
	LastUpdated      int64            `json:"last_updated"`
}

type RouterWeight struct {
	UserID    int32  `json:"user_id"`
	Category  string `json:"category"` // "schedule", "memo", "amazing"
	Keyword   string `json:"keyword"`
	Weight    int    `json:"weight"`
	CreatedTs int64  `json:"created_ts"`
	UpdatedTs int64  `json:"updated_ts"`
}

type RowStatus string

type Schedule struct {
	EndTs           *int64
	Payload         *string
	Reminders       *string
	RecurrenceEndTs *int64
	RecurrenceRule  *string
	Timezone        string
	Title           string
	Description     string
	Location        string
	RowStatus       RowStatus
	UID             string
	StartTs         int64
	UpdatedTs       int64
	CreatedTs       int64
	ID              int32
	CreatorID       int32
	AllDay          bool
}

type ScheduleService interface {
	CreateSchedule(ctx context.Context, create *Schedule) (*Schedule, error)
	ListSchedules(ctx context.Context, find *FindSchedule) ([]*Schedule, error)
	UpdateSchedule(ctx context.Context, update *UpdateSchedule) error
	DeleteSchedule(ctx context.Context, delete *DeleteSchedule) error
}

type SecurityAuditEvent struct {
	ID                    int64
	SessionID             string
	UserID                int32
	AgentType             string
	OperationType         string
	OperationName         string
	RiskLevel             string
	CommandInput          string
	CommandMatchedPattern string
	ActionTaken           string
	Reason                string
	FilePath              string
	ToolID                string
	OccurredAt            time.Time
}

type SecurityAuditStore interface {
	// LogSecurityEvent saves a security event.
	LogSecurityEvent(ctx context.Context, event *SecurityAuditEvent) error

	// ListSecurityEvents retrieves security events for a user with pagination.
	ListSecurityEvents(ctx context.Context, userID int32, limit, offset int) ([]*SecurityAuditEvent, int64, error)

	// ListSecurityEventsByRisk retrieves events filtered by risk level.
	ListSecurityEventsByRisk(ctx context.Context, userID int32, riskLevel string, limit, offset int) ([]*SecurityAuditEvent, int64, error)
}

type SessionStats struct {
	SessionID            string   `json:"session_id"`
	UserID               int32    `json:"user_id"`
	AgentType            string   `json:"agent_type"`
	TotalDurationMs      int64    `json:"total_duration_ms"`
	ThinkingDurationMs   int64    `json:"thinking_duration_ms"`
	ToolDurationMs       int64    `json:"tool_duration_ms"`
	GenerationDurationMs int64    `json:"generation_duration_ms"`
	InputTokens          int      `json:"input_tokens"`
	OutputTokens         int      `json:"output_tokens"`
	CacheWriteTokens     int      `json:"cache_write_tokens"`
	CacheReadTokens      int      `json:"cache_read_tokens"`
	TotalTokens          int      `json:"total_tokens"`
	TotalCostUsd         float64  `json:"total_cost_usd"`
	ToolCallCount        int      `json:"tool_call_count"`
	ToolsUsed            []string `json:"tools_used,omitempty"`
	FilesModified        int      `json:"files_modified"`
	FilePaths            []string `json:"file_paths,omitempty"`
	ModelUsed            string   `json:"model_used,omitempty"`
	IsError              bool     `json:"is_error"`
	ErrorMessage         string   `json:"error_message,omitempty"`
}

type Store struct {

	// Store services
	AgentStatsStore    AgentStatsStore    // session statistics persistence
	SecurityAuditStore SecurityAuditStore // security audit logging
	// contains filtered or unexported fields
}

type TitleSource string

type TokenUsage struct {
	PromptTokens     int32 `json:"prompt_tokens"`
	CompletionTokens int32 `json:"completion_tokens"`
	TotalTokens      int32 `json:"total_tokens"`
	CacheReadTokens  int32 `json:"cache_read_tokens,omitempty"`
	CacheWriteTokens int32 `json:"cache_write_tokens,omitempty"`
}

type ToolMetrics struct {
	HourBucket   time.Time
	ToolName     string
	ID           int64
	CallCount    int64
	SuccessCount int64
	LatencySumMs int64
}

type UpdateAIBlock struct {
	ID               int64
	UserInputs       *[]UserInput   // Replace user inputs
	AssistantContent *string        // Update AI response
	EventStream      *[]BlockEvent  // Replace event stream
	SessionStats     *SessionStats  // Update session stats
	CCSessionID      *string        // Update CC session ID
	Status           *AIBlockStatus // Update status
	Metadata         map[string]any // Merge metadata
	UpdatedTs        *int64         // Update timestamp
	// New fields for block extensions
	TokenUsage        *TokenUsage // Update token usage
	CostEstimate      *int64      // Update cost estimate
	ModelVersion      *string     // Update model version
	UserFeedback      *string     // Update user feedback
	RegenerationCount *int32      // Update regeneration count
	ErrorMessage      *string     // Update error message
	ArchivedAt        *int64      // Archive this block
}

type UpdateAIConversation struct {
	Title       *string
	TitleSource *TitleSource
	ParrotID    *string
	Pinned      *bool
	RowStatus   *RowStatus
	UpdatedTs   *int64
	ID          int32
}

type UpdateAttachment struct {
	UID           *string
	UpdatedTs     *int64
	Filename      *string
	MemoID        *int32
	Reference     *string
	Payload       *storepb.AttachmentPayload
	RowStatus     *string
	ExtractedText *string
	OCRText       *string
	ThumbnailPath *string
	ID            int32
}

type UpdateIdentityProvider struct {
	Name             *string
	IdentifierFilter *string
	Config           *string
	ID               int32
}

type UpdateIdentityProviderV1 struct {
	Name             *string
	IdentifierFilter *string
	Config           *storepb.IdentityProviderConfig
	ID               int32
	Type             storepb.IdentityProvider_Type
}

type UpdateInbox struct {
	Status InboxStatus
	ID     int32
}

type UpdateMemo struct {
	UID        *string
	CreatedTs  *int64
	UpdatedTs  *int64
	RowStatus  *RowStatus
	Content    *string
	Visibility *Visibility
	Pinned     *bool
	Payload    *storepb.MemoPayload
	ID         int32
}

type UpdateSchedule struct {
	StartTs         *int64
	EndTs           *int64
	CreatedTs       *int64
	UpdatedTs       *int64
	RowStatus       *RowStatus
	Title           *string
	UID             *string
	Description     *string
	Payload         *string
	Location        *string
	AllDay          *bool
	Timezone        *string
	RecurrenceRule  *string
	RecurrenceEndTs *int64
	Reminders       *string
	ID              int32
}

type UpdateUser struct {
	UpdatedTs    *int64
	RowStatus    *RowStatus
	Username     *string
	Role         *Role
	Email        *string
	Nickname     *string
	Password     *string
	AvatarURL    *string
	PasswordHash *string
	Description  *string
	ID           int32
}

type UpsertAgentMetrics struct {
	HourBucket   time.Time
	AgentType    string
	Errors       string
	RequestCount int64
	SuccessCount int64
	LatencySumMs int64
	LatencyP50Ms int32
	LatencyP95Ms int32
}

type UpsertPromptVersionMetrics struct {
	HourBucket    time.Time
	AgentType     string
	PromptVersion string
	RequestCount  int64
	SuccessCount  int64
	AvgLatencyMs  int64
}

type UpsertRouterWeight struct {
	UserID   int32
	Category string
	Keyword  string
	Weight   int
}

type UpsertToolMetrics struct {
	HourBucket   time.Time
	ToolName     string
	CallCount    int64
	SuccessCount int64
	LatencySumMs int64
}

type UpsertUserPreferences struct {
	Preferences string
	UserID      int32
}

type User struct {
	RowStatus    RowStatus
	Username     string
	Role         Role
	Email        string
	Nickname     string
	PasswordHash string
	AvatarURL    string
	Description  string
	CreatedTs    int64
	UpdatedTs    int64
	ID           int32
}

type UserCostSettings struct {
	UserID                 int32
	DailyBudgetUSD         *float64 // NULL = unlimited
	PerSessionThresholdUSD float64
	AlertEnabled           bool
	AlertEmail             bool
	AlertInApp             bool
	BudgetResetAt          *time.Time
}

type UserInput struct {
	Content   string         `json:"content"`
	Timestamp int64          `json:"timestamp"`
	Metadata  map[string]any `json:"metadata,omitempty"`
}

type UserPreferences struct {
	Preferences string
	CreatedTs   int64
	UpdatedTs   int64
	UserID      int32
}

type UserSetting struct {
	Value  string
	UserID int32
	Key    storepb.UserSetting_Key
}

type VectorSearchOptions struct {
	Vector        []float32
	Limit         int
	UserID        int32
	CreatedAfter  int64 // Optional: only search memos created after this timestamp
	MaxCandidates int   // Optional: maximum candidates to fetch (default: limit * 10)
}

type Visibility string