store

type AIConversation struct {
	UID       string
	Title     string
	ParrotID  string
	RowStatus RowStatus
	CreatedTs int64
	UpdatedTs int64
	ID        int32
	CreatorID int32
	Pinned    bool
}

type AIMessage struct {
	UID            string
	Type           AIMessageType
	Role           AIMessageRole
	Content        string
	Metadata       string
	CreatedTs      int64
	ID             int32
	ConversationID int32
}

type AIMessageRole string

type AIMessageType string

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 CostStats struct {
	TotalCostUSD         float64
	DailyAverageUSD      float64
	SessionCount         int64
	MostExpensiveSession *AgentSessionStats
	DailyBreakdown       []*DailyCostData
}

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

type DeleteAIConversation struct {
	ID int32
}

type DeleteAIMessage struct {
	ID             *int32
	ConversationID *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

	// AIMessage model related methods.
	CreateAIMessage(ctx context.Context, create *AIMessage) (*AIMessage, error)
	ListAIMessages(ctx context.Context, find *FindAIMessage) ([]*AIMessage, error)
	DeleteAIMessage(ctx context.Context, delete *DeleteAIMessage) 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

	// 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 FindAIConversation struct {
	ID        *int32
	UID       *string
	CreatorID *int32
	Pinned    *bool
	RowStatus *RowStatus
}

type FindAIMessage struct {
	ID             *int32
	UID            *string
	ConversationID *int32
}

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 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 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 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 Store struct {

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

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

type UpdateAIConversation struct {
	Title     *string
	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 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 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
}

type Visibility string