store

type MetricSnapshot struct {
	// Pool metrics
	CollectedAt       time.Time
	TotalWorkers      int
	ActiveWorkers     int
	IdleWorkers       int
	QueueDepth        int
	TasksCompleted    int64
	TasksFailed       int64
	AvgTaskDurationMs float64
	P95TaskDurationMs float64

	// Dispatcher metrics
	DispCollectedAt      time.Time
	ClaimedCount         int
	TotalRouted          int64
	TotalRequeued        int64
	TotalEventsProcessed int64
	AvgPollLatencyMs     float64
	LastPollAt           time.Time
}

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

type Store interface {
	// Sessions
	CreateSession(ctx context.Context, s *models.Session) error
	GetSession(ctx context.Context, id string) (*models.Session, error)
	UpdateSession(ctx context.Context, s *models.Session) error
	ListSessions(ctx context.Context, status models.SessionStatus) ([]*models.Session, error)

	// Cost tracking
	RecordCost(ctx context.Context, r *models.CostRecord) error
	ComputeCostSince(ctx context.Context, since time.Time) (*models.CostSummary, error)
	ComputeCostForIssue(ctx context.Context, issueNumber int) (*models.CostSummary, error)
	GetBudgetStatus(ctx context.Context, dailyBudgetUSD float64) (spent float64, remaining float64, err error)

	// Scaling events (written by dispatch, read by serve and MCP)
	SaveScalingEvent(ctx context.Context, e models.ScalingEvent) error
	ListScalingEvents(ctx context.Context, limit int) ([]*models.ScalingEvent, error)

	// Scaling control (written by serve/CLI, read by dispatch autoscaler)
	GetScalingControl(ctx context.Context) (*models.ScalingControl, error)
	UpsertScalingControl(ctx context.Context, c models.ScalingControl) error

	// Task profiles (updated after each completed session, read by policy engine and metrics API)
	UpdateTaskProfile(ctx context.Context, agentType, provider string) error
	ListTaskProfiles(ctx context.Context) ([]*models.TaskProfile, error)
	GetTaskProfile(ctx context.Context, agentType, provider string) (*models.TaskProfile, error)

	// Metric snapshots (written by dispatch, read by serve for cross-process metrics)
	SaveMetricSnapshot(ctx context.Context, m MetricSnapshot) error
	LatestMetricSnapshot(ctx context.Context) (*MetricSnapshot, error)

	// Failure events (written by dispatcher and runner, read by API, MCP, and CLI)
	SaveFailureEvent(ctx context.Context, e *models.FailureEvent) error
	ListFailureEvents(ctx context.Context, since time.Time, limit int) ([]*models.FailureEvent, error)
	CountFailuresByIssue(ctx context.Context, issueNumber int) (int, error)
	GetFailureSummary(ctx context.Context, since time.Time) (*models.FailureSummary, error)

	// Persisted issue failure counter (survives restarts, unlike in-memory map)
	GetIssueFailureCount(ctx context.Context, issueNumber int) (int, error)
	IncrementIssueFailureCount(ctx context.Context, issueNumber int) (int, error)
	ClearIssueFailureCount(ctx context.Context, issueNumber int) error

	// Correction events (written by correction engine, read by API and diagnostics)
	SaveCorrectionEvent(ctx context.Context, e *models.CorrectionEvent) error
	ListCorrectionEvents(ctx context.Context, issueNumber int) ([]*models.CorrectionEvent, error)

	// Provider audits (written by audit CLI, read by API and diagnostics)
	SaveAuditResults(ctx context.Context, results []audit.AuditResult) error
	GetLastAudit(ctx context.Context) ([]audit.AuditResult, error)

	// Check-in state (persists the last get_digest call time for accurate "away" duration)
	GetLastCheckIn(ctx context.Context) (time.Time, error)
	SetLastCheckIn(ctx context.Context, t time.Time) error

	Close() error
}