store

type AlreadyDecidedError struct {
	ID     string
	Status string // current status
}

type Approval struct {
	ID          string          `json:"id"`
	RunID       string          `json:"run_id"`
	SessionID   string          `json:"session_id"`
	ToolUseID   *string         `json:"tool_use_id,omitempty"`
	Status      ApprovalStatus  `json:"status"`
	CreatedAt   time.Time       `json:"created_at"`
	RespondedAt *time.Time      `json:"responded_at,omitempty"`
	ToolName    string          `json:"tool_name"`
	ToolInput   json.RawMessage `json:"tool_input"`
	Comment     string          `json:"comment,omitempty"`
}

type ApprovalStatus string

type ConversationEvent struct {
	ID              int64
	SessionID       string
	ClaudeSessionID string
	Sequence        int
	EventType       string // 'message', 'tool_call', 'tool_result', 'system'
	CreatedAt       time.Time

	// Message fields
	Role    string // user, assistant, system
	Content string

	// Tool call fields
	ToolID          string
	ToolName        string
	ToolInputJSON   string
	ParentToolUseID string

	// Tool result fields
	ToolResultForID   string
	ToolResultContent string

	// Tool call tracking
	IsCompleted    bool   // TRUE when tool result received
	ApprovalStatus string // NULL, 'pending', 'approved', 'denied'
	ApprovalID     string // HumanLayer approval ID when correlated
}

type ConversationStore interface {
	// Session operations
	CreateSession(ctx context.Context, session *Session) error
	UpdateSession(ctx context.Context, sessionID string, updates SessionUpdate) error
	HardDeleteSession(ctx context.Context, sessionID string) error
	GetSession(ctx context.Context, sessionID string) (*Session, error)
	GetSessionByRunID(ctx context.Context, runID string) (*Session, error)
	ListSessions(ctx context.Context) ([]*Session, error)
	SearchSessionsByTitle(ctx context.Context, query string, limit int) ([]*Session, error)
	// GetExpiredDangerousPermissionsSessions returns sessions where dangerous permissions have expired
	GetExpiredDangerousPermissionsSessions(ctx context.Context) ([]*Session, error)

	// Conversation operations
	AddConversationEvent(ctx context.Context, event *ConversationEvent) error
	GetConversation(ctx context.Context, claudeSessionID string) ([]*ConversationEvent, error)
	GetSessionConversation(ctx context.Context, sessionID string) ([]*ConversationEvent, error)

	// Tool call operations
	GetPendingToolCall(ctx context.Context, sessionID string, toolName string) (*ConversationEvent, error)
	GetUncorrelatedPendingToolCall(ctx context.Context, sessionID string, toolName string) (*ConversationEvent, error)
	GetPendingToolCalls(ctx context.Context, sessionID string) ([]*ConversationEvent, error)
	GetToolCallByID(ctx context.Context, toolID string) (*ConversationEvent, error)
	MarkToolCallCompleted(ctx context.Context, toolID string, sessionID string) error
	CorrelateApproval(ctx context.Context, sessionID string, toolName string, approvalID string) error
	LinkConversationEventToApprovalUsingToolID(ctx context.Context, sessionID string, toolID string, approvalID string) error
	UpdateApprovalStatus(ctx context.Context, approvalID string, status string) error

	// MCP server operations
	StoreMCPServers(ctx context.Context, sessionID string, servers []MCPServer) error
	GetMCPServers(ctx context.Context, sessionID string) ([]MCPServer, error)

	// Raw event storage (for debugging)
	StoreRawEvent(ctx context.Context, sessionID string, eventJSON string) error

	// Approval operations for local approvals
	CreateApproval(ctx context.Context, approval *Approval) error
	GetApproval(ctx context.Context, id string) (*Approval, error)
	GetPendingApprovals(ctx context.Context, sessionID string) ([]*Approval, error)
	UpdateApprovalResponse(ctx context.Context, id string, status ApprovalStatus, comment string) error

	// File snapshot operations
	CreateFileSnapshot(ctx context.Context, snapshot *FileSnapshot) error
	GetFileSnapshots(ctx context.Context, sessionID string) ([]FileSnapshot, error)
	// Recent paths operations
	GetRecentWorkingDirs(ctx context.Context, limit int) ([]RecentPath, error)

	// User settings operations
	GetUserSettings(ctx context.Context) (*UserSettings, error)
	UpdateUserSettings(ctx context.Context, settings UserSettings) error

	// Database lifecycle
	Close() error
}

type FileSnapshot struct {
	ID        int64
	ToolID    string
	SessionID string
	FilePath  string // Relative path from tool call
	Content   string
	CreatedAt time.Time
}

type MCPServer struct {
	ID        int64
	SessionID string
	Name      string
	Command   string
	ArgsJSON  string // JSON array
	EnvJSON   string // JSON object
}

type NotFoundError struct {
	Type string // e.g., "approval", "session"
	ID   string
}

type RecentPath struct {
	Path       string    `json:"path"`
	LastUsed   time.Time `json:"last_used"`
	UsageCount int       `json:"usage_count"`
}

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

type Session struct {
	ID                                  string
	RunID                               string
	ClaudeSessionID                     string
	ParentSessionID                     string
	Query                               string
	Summary                             string
	Title                               string // New field for user-editable title
	Model                               string
	ModelID                             string // Full model identifier (e.g., "claude-opus-4-1-20250805")
	WorkingDir                          string
	MaxTurns                            int
	SystemPrompt                        string
	AppendSystemPrompt                  string // NEW: Append to system prompt
	CustomInstructions                  string
	PermissionPromptTool                string // NEW: MCP tool for permission prompts
	AllowedTools                        string // NEW: JSON array of allowed tools
	DisallowedTools                     string // NEW: JSON array of disallowed tools
	AdditionalDirectories               string // NEW: JSON array of additional directories
	Status                              string
	CreatedAt                           time.Time
	LastActivityAt                      time.Time
	CompletedAt                         *time.Time
	CostUSD                             *float64
	InputTokens                         *int `db:"input_tokens"`
	OutputTokens                        *int `db:"output_tokens"`
	CacheCreationInputTokens            *int `db:"cache_creation_input_tokens"`
	CacheReadInputTokens                *int `db:"cache_read_input_tokens"`
	EffectiveContextTokens              *int `db:"effective_context_tokens"`
	DurationMS                          *int
	NumTurns                            *int
	ResultContent                       string
	ErrorMessage                        string
	AutoAcceptEdits                     bool       `db:"auto_accept_edits"`
	DangerouslySkipPermissions          bool       `db:"dangerously_skip_permissions"`
	DangerouslySkipPermissionsExpiresAt *time.Time `db:"dangerously_skip_permissions_expires_at"`
	DangerouslySkipPermissionsTimeoutMs *int64     `db:"dangerously_skip_permissions_timeout_ms"`
	Archived                            bool       // New field for session archiving

	// Proxy configuration
	ProxyEnabled       bool   `db:"proxy_enabled"`
	ProxyBaseURL       string `db:"proxy_base_url"`
	ProxyModelOverride string `db:"proxy_model_override"`
	ProxyAPIKey        string `db:"proxy_api_key"`

	// Editor state for draft sessions (JSON blob)
	EditorState *string `db:"editor_state"`
}

type SessionUpdate struct {
	ClaudeSessionID                     *string
	Query                               *string // For updating draft session queries
	Summary                             *string
	Title                               *string // New field for updating title
	Status                              *string
	LastActivityAt                      *time.Time
	CompletedAt                         *time.Time
	CostUSD                             *float64
	InputTokens                         *int
	OutputTokens                        *int
	CacheCreationInputTokens            *int
	CacheReadInputTokens                *int
	EffectiveContextTokens              *int
	DurationMS                          *int
	NumTurns                            *int
	ResultContent                       *string
	ErrorMessage                        *string
	AutoAcceptEdits                     *bool       `db:"auto_accept_edits"`
	DangerouslySkipPermissions          *bool       `db:"dangerously_skip_permissions"`
	DangerouslySkipPermissionsExpiresAt **time.Time `db:"dangerously_skip_permissions_expires_at"`
	DangerouslySkipPermissionsTimeoutMs *int64      `db:"dangerously_skip_permissions_timeout_ms"`
	Model                               *string
	ModelID                             *string // Full model identifier
	Archived                            *bool   // New field for updating archived status
	AdditionalDirectories               *string `db:"additional_directories"`
	// New proxy fields
	ProxyEnabled       *bool   `db:"proxy_enabled"`
	ProxyBaseURL       *string `db:"proxy_base_url"`
	ProxyModelOverride *string `db:"proxy_model_override"`
	ProxyAPIKey        *string `db:"proxy_api_key"`
	// Working directory field
	WorkingDir *string `db:"working_dir"`
	// Editor state field (JSON blob)
	EditorState *string `db:"editor_state"`
}

type UserSettings struct {
	AdvancedProviders bool      `json:"advanced_providers"`
	OptInTelemetry    *bool     `json:"opt_in_telemetry"` // Pointer to handle NULL (unset)
	CreatedAt         time.Time `json:"created_at"`
	UpdatedAt         time.Time `json:"updated_at"`
}