config

type AlchemyConfig struct {
	APIKey         string
	DefaultNetwork string
	TimeoutSeconds int
}

type BlockmonConfig struct {
	AlchemyAPIKey       string
	WebhookURL          string
	DefaultNetwork      string
	PollIntervalSeconds int
}

type CollectionsConfig struct {
	Enabled           bool     `json:"enabled"`
	ActiveCollections []string `json:"active_collections"`
	AutoEnable        bool     `json:"auto_enable"`
}

type Config struct {
	// API settings
	APIKey  string `json:"api_key"`
	BaseURL string `json:"base_url"`
	Model   string `json:"model"`
	// AgentModel is the model used for agent / orchestrate / subagent work
	// (free tool-selection). Chat/TTS use Model. Empty falls back to Model.
	// Lets you pin a reasoning/tool-capable model for agent work while keeping a
	// cheap non-reasoning model for chat (model-router guardrail, task e8775b91).
	AgentModel string `json:"agent_model,omitempty"`
	// SmallModel is a cheaper model for housekeeping calls such as context
	// compaction summaries (#174). Empty falls back to Model.
	SmallModel string `json:"small_model,omitempty"`
	// PinModel turns live model resolution off: the configured models are
	// sent as written even if the provider no longer lists them. Same as
	// CELESTE_PIN_MODEL=1.
	PinModel bool `json:"pin_model,omitempty"`
	// WebFetchAllowPrivate lets web_fetch reach loopback, private-network,
	// link-local and other non-public addresses, which it refuses by default.
	// For local docs servers. Same as CELESTE_WEB_FETCH_ALLOW_PRIVATE=1.
	WebFetchAllowPrivate bool `json:"web_fetch_allow_private,omitempty"`
	// JevPrune turns on TypeSafe Jev as a judge for context pruning (#175).
	// "shadow" asks Jev in the background and only logs what it would have
	// pruned; "on" (2.0 W3) elides the least-needed results first, asking
	// Jev inside the loop (2.5 s cap, rules on any error); anything else is
	// off. Redacted excerpts of old tool results go to TypeSafe. The key
	// comes from TYPESAFE_API_KEY or ~/.celeste/typesafe.key.
	JevPrune string `json:"jev_prune,omitempty"`
	// JevGate asks Jev whether a tool call is destructive, exfiltrates data
	// or goes beyond what was asked (2.0 W3): "shadow" logs, "on" turns an
	// allowed call into an Ask (never the reverse). Off by default.
	JevGate string `json:"jev_gate,omitempty"`
	// JevRoute picks /orchestrate's lane with a Jev choice question (2.0
	// W3): "shadow" logs, "on" uses it. Off by default.
	JevRoute string `json:"jev_route,omitempty"`
	// Oracle picks the judge for steering questions (2.0 W3):
	// "heuristic" (default), "llm" (the small model) or "jev". The
	// watchdog ballot asks it.
	Oracle string `json:"oracle,omitempty"`
	// StreamRules: "shadow" (default) matches stream rules and logs, "on"
	// acts, "off" skips them (2.0 W3).
	StreamRules string `json:"stream_rules,omitempty"`
	// Watchdog: the ballot every 3 turns (2.0 W3): "off" (default),
	// "shadow" (asked and logged) or "on" (steers).
	Watchdog string `json:"watchdog,omitempty"`
	// CompletionGate (agent runs, 2.0 W3): "shadow" (default) keeps the
	// substring TASK_COMPLETE check and logs where the gate differs; "on"
	// lets the gate decide.
	CompletionGate string `json:"completion_gate,omitempty"`
	Timeout        int    `json:"timeout"`                 // seconds
	ContextLimit   int    `json:"context_limit,omitempty"` // Optional: Override context window size

	// Google Cloud authentication (for Gemini/Vertex AI)
	GoogleCredentialsFile string `json:"google_credentials_file,omitempty"` // Path to service account JSON file
	GoogleUseADC          bool   `json:"google_use_adc,omitempty"`          // Use Application Default Credentials

	// Runtime-detected provider (not persisted to config file)
	Provider string `json:"-"` // Detected from BaseURL at runtime

	// Default marks this named config as the one loaded when no -config flag
	// is given. Exactly one file should set it; the first match wins.
	Default bool `json:"default,omitempty"`

	// Confirm mode: when true, Celeste proposes actions before executing
	// write/generate operations. When false, she auto-executes.
	ConfirmActions bool `json:"confirm_actions,omitempty"`

	// ElevenLabs TTS settings
	ElevenLabsAPIKey  string `json:"elevenlabs_api_key,omitempty"`
	ElevenLabsVoiceID string `json:"elevenlabs_voice_id,omitempty"`

	// Streaming settings
	SimulateTyping bool `json:"simulate_typing"`
	TypingSpeed    int  `json:"typing_speed"` // chars per second

	// MaxToolIterations caps the model turns in one chat turn's tool loop.
	MaxToolIterations int `json:"max_tool_iterations,omitempty"`

	// Venice.ai settings (for NSFW mode)
	VeniceAPIKey     string `json:"venice_api_key,omitempty"`
	VeniceBaseURL    string `json:"venice_base_url,omitempty"`
	VeniceModel      string `json:"venice_model,omitempty"`       // Chat model (venice-uncensored)
	VeniceImageModel string `json:"venice_image_model,omitempty"` // Image model (lustify-sdxl)

	// Tarot settings
	TarotFunctionURL string `json:"tarot_function_url,omitempty"`
	TarotAuthToken   string `json:"tarot_auth_token,omitempty"`

	// Twitter settings
	TwitterBearerToken       string `json:"twitter_bearer_token,omitempty"`
	TwitterAPIKey            string `json:"twitter_api_key,omitempty"`
	TwitterAPISecret         string `json:"twitter_api_secret,omitempty"`
	TwitterAccessToken       string `json:"twitter_access_token,omitempty"`
	TwitterAccessTokenSecret string `json:"twitter_access_token_secret,omitempty"`

	// Weather settings
	WeatherDefaultZipCode string `json:"weather_default_zip_code,omitempty"`

	// Twitch settings
	TwitchClientID        string `json:"twitch_client_id,omitempty"`
	TwitchClientSecret    string `json:"twitch_client_secret,omitempty"`
	TwitchDefaultStreamer string `json:"twitch_default_streamer,omitempty"`

	// YouTube settings
	YouTubeAPIKey         string `json:"youtube_api_key,omitempty"`
	YouTubeDefaultChannel string `json:"youtube_default_channel,omitempty"`

	// IPFS settings
	IPFSProvider       string `json:"ipfs_provider,omitempty"` // "infura", "pinata", "custom"
	IPFSAPIKey         string `json:"ipfs_api_key,omitempty"`
	IPFSAPISecret      string `json:"ipfs_api_secret,omitempty"`
	IPFSProjectID      string `json:"ipfs_project_id,omitempty"` // Infura specific
	IPFSGatewayURL     string `json:"ipfs_gateway_url,omitempty"`
	IPFSTimeoutSeconds int    `json:"ipfs_timeout_seconds,omitempty"`

	// Alchemy settings
	AlchemyAPIKey         string `json:"alchemy_api_key,omitempty"`
	AlchemyDefaultNetwork string `json:"alchemy_default_network,omitempty"`
	AlchemyTimeoutSeconds int    `json:"alchemy_timeout_seconds,omitempty"`

	// Blockchain monitoring settings
	BlockmonAlchemyAPIKey       string `json:"blockmon_alchemy_api_key,omitempty"`
	BlockmonWebhookURL          string `json:"blockmon_webhook_url,omitempty"`
	BlockmonDefaultNetwork      string `json:"blockmon_default_network,omitempty"`
	BlockmonPollIntervalSeconds int    `json:"blockmon_poll_interval_seconds,omitempty"`

	// Wallet security settings
	WalletSecurityEnabled      bool   `json:"wallet_security_enabled,omitempty"`
	WalletSecurityPollInterval int    `json:"wallet_security_poll_interval,omitempty"` // seconds
	WalletSecurityAlertLevel   string `json:"wallet_security_alert_level,omitempty"`   // "low", "medium", "high", "critical"

	// Collections configuration (xAI only)
	XAIManagementAPIKey string             `json:"xai_management_api_key,omitempty"`
	Collections         *CollectionsConfig `json:"collections,omitempty"`
	XAIFeatures         *XAIFeaturesConfig `json:"xai_features,omitempty"`

	// Orchestrator settings
	Orchestrator *OrchestratorConfig `json:"orchestrator,omitempty"`

	// Sandbox is the OS sandbox for bash (2.0 W4); see Sandbox.
	Sandbox *Sandbox `json:"sandbox,omitempty"`
	// contains filtered or unexported fields
}

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

type ContextTracker struct {
	Session          *Session
	Model            string
	MaxTokens        int
	CurrentTokens    int
	PromptTokens     int
	CompletionTokens int

	// Thresholds (kept for backward compatibility)
	WarnThreshold     float64 // 0.75
	CautionThreshold  float64 // 0.85
	CriticalThreshold float64 // 0.95

	// Tracking
	LastWarningLevel string
	CompactionCount  int
	TruncationCount  int

	// Underlying budget (exported so callers can use it directly if needed)
	Budget *ctxmgr.TokenBudget
}

type DailyStats struct {
	Date         string  `json:"date"`
	SessionCount int     `json:"session_count"`
	MessageCount int     `json:"message_count"`
	TokenCount   int     `json:"token_count"`
	Cost         float64 `json:"cost"`
}

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

type GlobalAnalytics struct {
	TotalSessions int     `json:"total_sessions"`
	TotalMessages int     `json:"total_messages"`
	TotalTokens   int     `json:"total_tokens"`
	TotalCost     float64 `json:"total_cost"`

	// Per-provider breakdown
	ProviderUsage map[string]*ProviderStats `json:"provider_usage"`

	// Per-model breakdown
	ModelUsage map[string]*ModelStats `json:"model_usage"`

	// Time-based tracking (YYYY-MM-DD -> stats)
	DailyUsage map[string]*DailyStats `json:"daily_usage"`

	LastUpdated time.Time `json:"last_updated"`
}

type IPFSConfig struct {
	Provider       string
	APIKey         string
	APISecret      string
	ProjectID      string
	GatewayURL     string
	TimeoutSeconds int
}

type LaneConfig struct {
	Primary         string `json:"primary"`
	PrimaryBaseURL  string `json:"primary_base_url,omitempty"`
	PrimaryAPIKey   string `json:"primary_api_key,omitempty"`
	Reviewer        string `json:"reviewer,omitempty"`
	ReviewerBaseURL string `json:"reviewer_base_url,omitempty"`
	ReviewerAPIKey  string `json:"reviewer_api_key,omitempty"`
}

type ModelStats struct {
	SessionCount int     `json:"session_count"`
	MessageCount int     `json:"message_count"`
	InputTokens  int     `json:"input_tokens"`
	OutputTokens int     `json:"output_tokens"`
	Cost         float64 `json:"cost"`
}

type OrchestratorConfig struct {
	Lanes        map[string]LaneConfig `json:"lanes,omitempty"`
	DefaultLane  string                `json:"default_lane,omitempty"`
	DebateRounds int                   `json:"debate_rounds,omitempty"`
}

type PricingTier struct {
	InputCostPerMillion  float64
	OutputCostPerMillion float64
}

type ProviderBlocks struct {
	Provider string            `json:"provider"`
	Digest   string            `json:"digest,omitempty"`
	Blocks   []json.RawMessage `json:"blocks"`
}

type ProviderStats struct {
	SessionCount int     `json:"session_count"`
	MessageCount int     `json:"message_count"`
	TokenCount   int     `json:"token_count"`
	Cost         float64 `json:"cost"`
}

type Sandbox struct {
	Enabled  *bool    `json:"enabled,omitempty"`  // run bash under the OS sandbox
	Writable []string `json:"writable,omitempty"` // more writable directories (relative: to the workspace)
	Network  *bool    `json:"network,omitempty"`  // false cuts the network
}

type Session struct {
	ID         string           `json:"id"`
	Name       string           `json:"name,omitempty"`
	CreatedAt  time.Time        `json:"created_at"`
	UpdatedAt  time.Time        `json:"updated_at"`
	Messages   []SessionMessage `json:"messages"`
	NSFWMode   bool             `json:"nsfw_mode,omitempty"`
	Metadata   map[string]any   `json:"metadata,omitempty"`
	TokenCount int              `json:"token_count,omitempty"` // Estimated token count
	Model      string           `json:"model,omitempty"`       // Track model for limits

	// NEW: Enhanced tracking
	UsageMetrics *UsageMetrics `json:"usage_metrics,omitempty"` // Detailed usage tracking
	Provider     string        `json:"provider,omitempty"`      // Provider (openai, venice, etc)
	MaxContext   int           `json:"max_context,omitempty"`   // Model's max context window

	// Workspace is the absolute directory the chat ran in (2.0 W4 ruling
	// 1): `celeste resume` and /session list show this project's
	// sessions first. Older sessions have none.
	Workspace string `json:"workspace,omitempty"`
}

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

type SessionMessage struct {
	Role      string    `json:"role"`
	Content   string    `json:"content"`
	Timestamp time.Time `json:"timestamp"`

	// Tool traffic (#174): without it a resumed session replays only the
	// prose, so the model loses what it read and did.
	ToolCalls  []SessionToolCall `json:"tool_calls,omitempty"`   // assistant: calls made
	ToolCallID string            `json:"tool_call_id,omitempty"` // tool: call answered
	Name       string            `json:"name,omitempty"`         // tool: function name

	// Hidden messages are sent to the model but not rendered (directives,
	// compaction summaries). Compacted messages were replaced by a summary:
	// rendered in the scrollback but no longer sent.
	Hidden    bool `json:"hidden,omitempty"`
	Compacted bool `json:"compacted,omitempty"`

	// ProviderBlocks is the message as the provider returned it (2.0 F3),
	// saved only while it still matches Content and ToolCalls.
	ProviderBlocks *ProviderBlocks `json:"provider_blocks,omitempty"`
}

type SessionSummary struct {
	ID           string         `json:"id"`
	Name         string         `json:"name,omitempty"`
	MessageCount int            `json:"message_count"`
	CreatedAt    time.Time      `json:"created_at"`
	UpdatedAt    time.Time      `json:"updated_at"`
	FirstMessage string         `json:"first_message,omitempty"`
	Metadata     map[string]any `json:"metadata,omitempty"`
	Workspace    string         `json:"workspace,omitempty"`
}

type SessionToolCall struct {
	ID               string `json:"id"`
	Name             string `json:"name"`
	Arguments        string `json:"arguments"`
	ThoughtSignature []byte `json:"thought_signature,omitempty"`
}

type SliderConfig struct {
	// Flirt controls how forward / teasing she is.
	// 0=professional, 3=playful, 7=flirty, 10=aggressive-flirt
	Flirt int `json:"flirt"`

	// Warmth controls affection level.
	// 0=cold/distant, 3=polite, 7=warm, 10=openly-affectionate
	Warmth int `json:"warmth"`

	// Register controls speech style.
	// 0=clipped-operator, 3=standard, 7=theatrical, 10=uwu/baby-talk
	Register int `json:"register"`

	// Lewdness controls content eligibility. Only has effect when
	// R18Enabled is true — without R18, lewdness > 0 is a no-op.
	// 0=SFW, 3=suggestive, 7=explicit-tease, 10=R18
	Lewdness int `json:"lewdness"`

	// R18Enabled is the independent toggle. Provider-agnostic — it
	// controls what Celeste ATTEMPTS, not what the backend allows.
	// Default false. Orthogonal to every slider.
	R18Enabled bool `json:"r18_enabled"`

	// Presets stores user-saved named slider combos.
	Presets map[string]SliderPreset `json:"presets,omitempty"`
}

type SliderPreset struct {
	Flirt      int  `json:"flirt"`
	Warmth     int  `json:"warmth"`
	Register   int  `json:"register"`
	Lewdness   int  `json:"lewdness"`
	R18Enabled bool `json:"r18_enabled"`
}

type TarotConfig struct {
	FunctionURL string
	AuthToken   string
}

type TwitchConfig struct {
	ClientID        string
	ClientSecret    string
	DefaultStreamer string
}

type UsageMetrics struct {
	TotalInputTokens  int       `json:"total_input_tokens"`
	TotalOutputTokens int       `json:"total_output_tokens"`
	TotalTokens       int       `json:"total_tokens"`
	EstimatedCost     float64   `json:"estimated_cost"`
	CompactionCount   int       `json:"compaction_count"`
	TruncationCount   int       `json:"truncation_count"`
	MessageCount      int       `json:"message_count"`
	ConversationStart time.Time `json:"conversation_start"`
	ConversationEnd   time.Time `json:"conversation_end"`
}

type UserIdentity struct {
	// Name is the user's display name. Empty means use the default title.
	Name string `json:"name"`

	// Title is the abyss-themed fallback when Name is empty.
	// Default: "Summoner" — they invoked Celeste via the CLI.
	Title string `json:"title,omitempty"`
}

type VeniceConfig struct {
	APIKey     string
	BaseURL    string
	Model      string // Chat model (venice-uncensored)
	ImageModel string // Image generation model (lustify-sdxl, animewan, hidream, wai-Illustrious)
	Upscaler   string
}

type WalletSecuritySettingsConfig struct {
	Enabled      bool
	PollInterval int    // seconds
	AlertLevel   string // minimum severity to alert on
}

type WeatherConfig struct {
	DefaultZipCode string
}

type XAIFeaturesConfig struct {
	EnableWebSearch bool `json:"enable_web_search"`
	EnableXSearch   bool `json:"enable_x_search"`
}

type YouTubeConfig struct {
	APIKey         string
	DefaultChannel string
}