config

type Agent struct {
	ID           string `json:"id"`
	Name         string `json:"name"`
	Model        string `json:"model"`
	ModelStrong  string `json:"model_strong"`
	ModelFast    string `json:"model_fast"`
	SystemPrompt string `json:"system_prompt"`
	Workspace    string `json:"workspace"`
	Enabled      bool   `json:"enabled"`
}

type Channel struct {
	ID      string `json:"id"`
	Enabled bool   `json:"enabled"`
	Config  string `json:"config"`
}

type CompactionConfig struct {
	// MaxTokens triggers compaction when the estimated token count exceeds this.
	// 0 (or omitted) uses the default of 80000. Negative values disable
	// automatic compaction. Manual /compact still works.
	MaxTokens int `json:"max_tokens"`
	// KeepTail is the number of recent message entries to preserve verbatim
	// after compaction. Default: 20.
	KeepTail int `json:"keep_tail"`
}

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

type HeartbeatConfig struct {
	Enabled *bool  `json:"enabled"`
	Every   string `json:"every"`
	File    string `json:"file"`
}

type ModelConfig struct {
	ID            string            `json:"id"`
	Name          string            `json:"name"`
	API           string            `json:"api"`
	Reasoning     bool              `json:"reasoning"`
	Input         []string          `json:"input"`
	ContextWindow int               `json:"context_window"`
	MaxTokens     int               `json:"max_tokens"`
	Headers       map[string]string `json:"headers"`
	Cost          *ModelCostConfig  `json:"cost"`
}

type ModelCostConfig struct {
	Input      float64 `json:"input"`
	Output     float64 `json:"output"`
	CacheRead  float64 `json:"cache_read"`
	CacheWrite float64 `json:"cache_write"`
}

type PluginConfig struct {
	Path   string         `json:"path"`
	Config map[string]any `json:"config"`
}

type Provider struct {
	ID      string `json:"id"`
	Name    string `json:"name"`
	APIKey  string `json:"api_key"`
	BaseURL string `json:"base_url"`
}

type ProviderCreds struct {
	APIKey  string
	BaseURL string
}

type RunnerConfig struct {
	Type        string           `json:"type"`
	System      string           `json:"system"`
	IdleTimeout int              `json:"idle_timeout"`
	Compaction  CompactionConfig `json:"compaction"`
}

type SchedulerConfig struct {
	Enabled *bool  `json:"enabled"`
	DataDir string `json:"data_dir"`
}

type Snapshot struct {
	// Provider, APIKey, BaseURL are the default provider credentials derived
	// from the Model field's provider prefix. Kept for backward compatibility.
	Provider     string
	Model        string
	ModelStrong  string
	ModelFast    string
	Workspace    string
	APIKey       string
	BaseURL      string
	SystemPrompt string // agent's soul/personality from DB
	Runner       RunnerConfig
	Compaction   CompactionConfig
	Heartbeat    HeartbeatConfig
	Scheduler    SchedulerConfig
	Plugins      []PluginConfig

	// Providers maps provider ID to credentials, enabling per-tier provider
	// resolution when model_strong or model_fast use a different provider.
	Providers map[string]ProviderCreds
}

type Store interface {
	// Providers
	ListProviders(ctx context.Context) ([]Provider, error)
	GetProvider(ctx context.Context, id string) (Provider, error)
	CreateProvider(ctx context.Context, p Provider) error
	UpdateProvider(ctx context.Context, p Provider) error
	DeleteProvider(ctx context.Context, id string) error

	// Agents
	ListAgents(ctx context.Context) ([]Agent, error)
	ListEnabledAgents(ctx context.Context) ([]Agent, error)
	GetAgent(ctx context.Context, id string) (Agent, error)
	CreateAgent(ctx context.Context, a Agent) error
	UpdateAgent(ctx context.Context, a Agent) error
	DeleteAgent(ctx context.Context, id string) error

	// Channels
	ListChannels(ctx context.Context) ([]Channel, error)
	GetChannel(ctx context.Context, id string) (Channel, error)
	UpsertChannel(ctx context.Context, ch Channel) error

	// Users
	ListUsers(ctx context.Context) ([]User, error)
	GetUser(ctx context.Context, id int64) (User, error)
	UpsertUser(ctx context.Context, externalID, platform, name string) (User, error)
	UpdateUserDefaultAgent(ctx context.Context, userID int64, agentID string) error

	// Chat Agents (group -> agent mapping)
	GetChatAgent(ctx context.Context, platform, chatID string) (string, error) // returns agentID
	SetChatAgent(ctx context.Context, platform, chatID, agentID string) error
	DeleteChatAgent(ctx context.Context, platform, chatID string) error

	// User Agent Memory
	GetUserAgentMemory(ctx context.Context, userID int64, agentID string) (string, error)
	SetUserAgentMemory(ctx context.Context, userID int64, agentID, content string) error
	ListUserMemories(ctx context.Context, userID int64) ([]UserAgentMemory, error)
	DeleteUserAgentMemory(ctx context.Context, userID int64, agentID string) error

	// Settings (key-value JSON)
	GetSetting(ctx context.Context, key string) (string, error) // returns JSON string
	SetSetting(ctx context.Context, key, value string) error

	// Snapshot assembles a read-only config snapshot for an agent.
	Snapshot(ctx context.Context, agentID string) (*Snapshot, error)

	// Bootstrap
	SeedDefaults(ctx context.Context) error
}

type User struct {
	ID             int64  `json:"id"`
	ExternalID     string `json:"external_id"`
	Platform       string `json:"platform"`
	Name           string `json:"name"`
	DefaultAgentID string `json:"default_agent_id"`
}

type UserAgentMemory struct {
	UserID    int64  `json:"user_id"`
	AgentID   string `json:"agent_id"`
	Content   string `json:"content"`
	UpdatedAt string `json:"updated_at"`
}