session

type Checkpoint struct {
	// Version is the checkpoint format version for compatibility.
	Version string `json:"version" yaml:"version"`

	// ExportedAt is when the checkpoint was created.
	ExportedAt time.Time `json:"exported_at" yaml:"exported_at"`

	// ExportedBy is the user who exported the checkpoint (optional).
	ExportedBy string `json:"exported_by,omitempty" yaml:"exported_by,omitempty"`

	// Session contains session metadata.
	Session CheckpointSession `json:"session" yaml:"session"`

	// Messages contains the complete conversation history.
	Messages []CheckpointMessage `json:"messages" yaml:"messages"`

	// Context contains project-specific context.
	Context *CheckpointContext `json:"context,omitempty" yaml:"context,omitempty"`

	// Statistics contains session statistics.
	Statistics CheckpointStatistics `json:"statistics" yaml:"statistics"`
}

type CheckpointContext struct {
	// ProjectInstructions is the ATMOS.md content at export time (optional).
	ProjectInstructions string `json:"project_instructions,omitempty" yaml:"project_instructions,omitempty"`

	// FilesAccessed is a list of files accessed during the session (optional).
	FilesAccessed []string `json:"files_accessed,omitempty" yaml:"files_accessed,omitempty"`

	// WorkingDirectory is the working directory at export time (optional).
	WorkingDirectory string `json:"working_directory,omitempty" yaml:"working_directory,omitempty"`
}

type CheckpointMessage struct {
	// Role is the message role (user, assistant, system).
	Role string `json:"role" yaml:"role"`

	// Content is the message content.
	Content string `json:"content" yaml:"content"`

	// CreatedAt is when the message was created.
	CreatedAt time.Time `json:"created_at" yaml:"created_at"`

	// Archived indicates if the message was compacted.
	Archived bool `json:"archived,omitempty" yaml:"archived,omitempty"`
}

type CheckpointSession struct {
	// Name is the session name.
	Name string `json:"name" yaml:"name"`

	// Provider is the AI provider (anthropic, openai, etc.).
	Provider string `json:"provider" yaml:"provider"`

	// Model is the AI model used.
	Model string `json:"model" yaml:"model"`

	// Skill is the AI skill name (optional).
	Skill string `json:"skill,omitempty" yaml:"skill,omitempty"`

	// ProjectPath is the project path where session was created.
	ProjectPath string `json:"project_path" yaml:"project_path"`

	// CreatedAt is when the session was created.
	CreatedAt time.Time `json:"created_at" yaml:"created_at"`

	// UpdatedAt is when the session was last updated.
	UpdatedAt time.Time `json:"updated_at" yaml:"updated_at"`

	// Metadata contains custom metadata.
	Metadata map[string]interface{} `json:"metadata,omitempty" yaml:"metadata,omitempty"`
}

type CheckpointStatistics struct {
	// MessageCount is the total number of messages.
	MessageCount int `json:"message_count" yaml:"message_count"`

	// UserMessages is the number of user messages.
	UserMessages int `json:"user_messages" yaml:"user_messages"`

	// AssistantMessages is the number of assistant messages.
	AssistantMessages int `json:"assistant_messages" yaml:"assistant_messages"`

	// TotalTokens is the total token count (if tracked).
	TotalTokens int64 `json:"total_tokens,omitempty" yaml:"total_tokens,omitempty"`

	// ToolCalls is the number of tool executions (if tracked).
	ToolCalls int `json:"tool_calls,omitempty" yaml:"tool_calls,omitempty"`
}

type CompactConfig struct {
	Enabled            bool    `json:"enabled" yaml:"enabled"`
	TriggerThreshold   float64 `json:"trigger_threshold" yaml:"trigger_threshold"`       // 0.0-1.0, default 0.75
	CompactRatio       float64 `json:"compact_ratio" yaml:"compact_ratio"`               // 0.0-1.0, default 0.4
	PreserveRecent     int     `json:"preserve_recent" yaml:"preserve_recent"`           // Default 10
	UseAISummary       bool    `json:"use_ai_summary" yaml:"use_ai_summary"`             // Default true
	ShowSummaryMarkers bool    `json:"show_summary_markers" yaml:"show_summary_markers"` // Default false
}

type CompactPlan struct {
	SessionID         string
	TotalMessages     int
	MessagesToCompact []*Message
	MessagesToKeep    []*Message
	EstimatedSavings  int // Token savings estimate
	Reason            string
}

type CompactResult struct {
	SummaryID          string
	OriginalMessageIDs []int64
	SummaryContent     string
	TokenCount         int
	CompactedAt        time.Time
	Success            bool
	Error              error
}

type CompactStatus struct {
	Stage            string // "starting", "summarizing", "completed", "failed"
	MessageCount     int    // Number of messages being compacted
	EstimatedSavings int    // Estimated token savings
	Error            error  // Error if failed
}

type CompactStatusCallback func(status CompactStatus)

type Compactor interface {
	// ShouldCompact determines if compaction is needed based on current history.
	ShouldCompact(messages []*Message, maxMessages int, config *CompactConfig) (bool, *CompactPlan)

	// Compact performs the compaction operation.
	Compact(ctx context.Context, plan *CompactPlan, config *CompactConfig) (*CompactResult, error)
}

type ContextItem struct {
	SessionID    string `json:"session_id"`
	ContextType  string `json:"context_type"`  // stack_file, component, setting
	ContextKey   string `json:"context_key"`   // e.g., "vpc", "prod-use1"
	ContextValue string `json:"context_value"` // JSON or text value
}

type CreateSessionParams struct {
	Name     string
	Model    string
	Provider string
	Skill    string
	Metadata map[string]interface{}
}

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

type ExportOptions struct {
	// IncludeContext includes project instructions and file access history.
	IncludeContext bool

	// IncludeMetadata includes session metadata.
	IncludeMetadata bool

	// Format is the export format (json, yaml, markdown).
	Format string
}

type ImportOptions struct {
	// Name is the name for the imported session.
	// If empty, uses the checkpoint's session name.
	Name string

	// ProjectPath is the project path for the imported session.
	// If empty, uses the current project path.
	ProjectPath string

	// OverwriteExisting allows overwriting an existing session with the same name.
	OverwriteExisting bool

	// IncludeContext includes project context in the import.
	IncludeContext bool
}

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

type Message struct {
	ID        int64     `json:"id"`
	SessionID string    `json:"session_id"`
	Role      string    `json:"role"` // user, assistant, system
	Content   string    `json:"content"`
	CreatedAt time.Time `json:"created_at"`
	Archived  bool      `json:"archived"`   // True if message has been compacted into a summary
	IsSummary bool      `json:"is_summary"` // True if this is a summary message (for display purposes)
}

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

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

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

type Session struct {
	ID           string                 `json:"id"`
	Name         string                 `json:"name"`
	ProjectPath  string                 `json:"project_path"`
	Model        string                 `json:"model"`
	Provider     string                 `json:"provider"`
	Skill        string                 `json:"skill,omitempty"` // AI skill name (e.g., "general", "stack-analyzer")
	CreatedAt    time.Time              `json:"created_at"`
	UpdatedAt    time.Time              `json:"updated_at"`
	Metadata     map[string]interface{} `json:"metadata,omitempty"`
	MessageCount int                    `json:"message_count,omitempty"`
}

type Storage interface {
	// Session operations
	CreateSession(ctx context.Context, session *Session) error
	GetSession(ctx context.Context, id string) (*Session, error)
	GetSessionByName(ctx context.Context, projectPath, name string) (*Session, error)
	ListSessions(ctx context.Context, projectPath string, limit int) ([]*Session, error)
	UpdateSession(ctx context.Context, session *Session) error
	DeleteSession(ctx context.Context, id string) error
	DeleteOldSessions(ctx context.Context, olderThan time.Time) (int, error)

	// Message operations
	AddMessage(ctx context.Context, message *Message) error
	GetMessages(ctx context.Context, sessionID string, limit int) ([]*Message, error)
	GetMessageCount(ctx context.Context, sessionID string) (int, error)

	// Context operations
	AddContext(ctx context.Context, item *ContextItem) error
	GetContext(ctx context.Context, sessionID string) ([]*ContextItem, error)
	DeleteContext(ctx context.Context, sessionID string) error

	// Summary operations (auto-compact)
	StoreSummary(ctx context.Context, summary *Summary) error
	GetSummaries(ctx context.Context, sessionID string) ([]*Summary, error)
	ArchiveMessages(ctx context.Context, messageIDs []int64) error
	GetActiveMessages(ctx context.Context, sessionID string, limit int) ([]*Message, error)

	// Database management
	Close() error
	Migrate() error
}

type Summary struct {
	ID                 string    `json:"id"`
	SessionID          string    `json:"session_id"`
	Provider           string    `json:"provider"`
	OriginalMessageIDs []int64   `json:"original_message_ids"`
	MessageRange       string    `json:"message_range"` // Human-readable: "Messages 1-20"
	SummaryContent     string    `json:"summary_content"`
	TokenCount         int       `json:"token_count"`
	CompactedAt        time.Time `json:"compacted_at"`
}