Cyrene_v2/internal/config/config.go

// Package config loads, defaults, and validates the Cyrene v2 configuration
// from environment variables and an optional .env file (development only).
//
// Every setting is sourced from an environment variable (no hard-coded values
// in code). Nested groups map to struct fields tagged with `env:"NAME"`.
// Missing required values cause Load/MustLoad to return an error so startup
// can fail fast.
package config

import (
	"fmt"
	"reflect"
	"strings"
	"time"
)

// Config is the root configuration tree.
type Config struct {
	Env            string   `env:"ENV" default:"development"`
	LogLevel       string   `env:"LOG_LEVEL" default:"debug"`
	Timezone       string   `env:"TZ" default:"Asia/Shanghai"`
	DataDir        string   `env:"DATA_DIR" default:"./data"`
	AllowedOrigins []string `env:"ALLOWED_ORIGINS" default:"http://localhost:5173,http://localhost:3000"`

	Server      ServerConfig
	Auth        AuthConfig
	DB          DBConfig
	LLM         LLMConfig
	Embedding   EmbeddingConfig
	IoT         IoTConfig
	Voice       VoiceConfig
	Background  BackgroundConfig
	WSHub       WSHubConfig
	ExecSandbox ExecSandbox
	MCP         MCPConfig
	Capacity    CapacityConfig
	Channels    ChannelsConfig
}

// ServerConfig holds the single gateway listener settings.
type ServerConfig struct {
	Host string `env:"HOST" default:"0.0.0.0"`
	Port int    `env:"PORT" default:"8080"`
}

// AuthConfig holds authentication and admin credential settings.
type AuthConfig struct {
	JWTSecret           string `env:"JWT_SECRET" validate:"required,min=32"`
	AdminUser           string `env:"ADMIN_USERNAME" default:"admin"`
	AdminPass           string `env:"ADMIN_PASSWORD" default:"admin12345"`
	RegistrationEnabled bool   `env:"REGISTRATION_ENABLED" default:"true"`

	// RefreshTTL is how long an issued opaque refresh token stays valid. It is a
	// Go duration string (e.g. "720h" = 30 days). Access tokens keep their 24h
	// lifetime; the refresh token is the long-lived credential that renews them.
	RefreshTTL time.Duration `env:"AUTH_REFRESH_TTL" default:"720h"`

	// RefreshRotateDisabled opts out of single-use refresh rotation. Rotation is
	// on by default (the safe default): a successful /auth/refresh revokes the
	// presented token and returns a fresh one, so a replayed token fails. Set
	// AUTH_REFRESH_ROTATE_DISABLED=true only for clients that cannot persist a
	// rotated token. The field is negative-named so its zero value (an
	// unset/undecoded config) means rotation is enabled.
	RefreshRotateDisabled bool `env:"AUTH_REFRESH_ROTATE_DISABLED" default:"false"`
}

// DBConfig holds database connection settings. SQLite is the default; PGURL is
// the optional PostgreSQL+pgvector enhancement.
type DBConfig struct {
	SQLitePath string `env:"DB_SQLITE_PATH" default:"./data/cyrene.db"`
	PGURL      string `env:"DB_PG_URL"`
}

// LLMConfig holds the multi-supplier LLM adapter settings. Provider selects the
// wire protocol; every provider has its own independent base/key/model/fallback
// so multiple vendors can coexist.
type LLMConfig struct {
	// Provider declares which protocol to use: "openai" (OpenAI-compatible
	// /chat/completions) or "anthropic" (Anthropic Messages /v1/messages).
	Provider string `env:"LLM_PROVIDER" default:"openai"`

	// OpenAI-compatible provider (default). Kept flat for backward compatibility
	// with the historic LLM_BASE_URL/API_KEY/MODEL/FALLBACK_MODEL env names.
	BaseURL  string `env:"LLM_BASE_URL"`
	APIKey   string `env:"LLM_API_KEY"`
	Model    string `env:"LLM_MODEL"`
	Fallback string `env:"LLM_FALLBACK_MODEL"`

	// VisionModel is the dedicated model used for image/vision requests. When
	// set, the model router routes image content to it (see model.Router.WithVisionModel).
	VisionModel string `env:"LLM_VISION_MODEL"`

	// FallbackMaxTokens caps the last-resort fallback model in a group's chain
	// (primary → fallback). It is the final safety budget: it only applies when
	// the chain actually has a fallback step, that group sets no per-group
	// MaxTokens (models.json maxTokens) and the request carries no explicit
	// budget. Zero disables the cap (the adapter default then applies).
	FallbackMaxTokens int `env:"LLM_FALLBACK_MAX_TOKENS" default:"8192"`

	// Anthropic native provider, used when Provider=anthropic. Independent
	// config block so each vendor is a self-contained config.
	AnthropicBaseURL  string `env:"ANTHROPIC_BASE_URL"`
	AnthropicAPIKey   string `env:"ANTHROPIC_API_KEY"`
	AnthropicModel    string `env:"ANTHROPIC_MODEL"`
	AnthropicFallback string `env:"ANTHROPIC_FALLBACK_MODEL"`
}

// EmbeddingConfig holds the optional semantic embedding API settings. When
// unset, the runtime falls back to built-in n-gram/FTS vectors.
type EmbeddingConfig struct {
	APIURL string `env:"EMBEDDING_API_URL"`
	APIKey string `env:"EMBEDDING_API_KEY"`
	Model  string `env:"EMBEDDING_MODEL"`
}

// IoTConfig selects the IoT device provider.
type IoTConfig struct {
	Provider string `env:"IOT_PROVIDER" default:"mock"`
}

// VoiceConfig holds TTS/STT provider settings and the Alibaba Bailian DashScope
// credentials/models that back the cloud rung of the voice chain.
type VoiceConfig struct {
	// TTSProvider selects the non-cloud local TTS rung: "edge-tts" or "espeak".
	TTSProvider string `env:"TTS_PROVIDER" default:"edge-tts"`
	// STTProvider selects the cloud STT provider ("dashscope").
	STTProvider string `env:"STT_PROVIDER" default:"dashscope"`
	// DashscopeAPIKey is the Bailian API key backing STT + TTS. When empty the
	// dashscope adapters return ErrNotConfigured and the chain degrades.
	DashscopeAPIKey string `env:"DASHSCOPE_API_KEY"`

	// DashscopeBaseURL is the Bailian API base; the STT/TTS service paths are
	// appended to it.
	DashscopeBaseURL string `env:"DASHSCOPE_BASE_URL" default:"https://dashscope.aliyuncs.com/api/v1"`

	// STTModel is the DashScope speech-transcription model.
	STTModel    string `env:"VOICE_STT_MODEL" default:"qwen-audio-3.0-asr-flash"`
	STTLanguage string `env:"VOICE_STT_LANGUAGE" default:"zh"`

	// TTSModel is the DashScope speech-synthesis model. cosyvoice-v3.5-plus and
	// cosyvoice-v2 reject the HTTP SpeechSynthesizer call (engine 418); verified
	// working is cosyvoice-v3-flash.
	TTSModel string `env:"VOICE_TTS_MODEL" default:"cosyvoice-v3-flash"`
	// TTSVoiceID selects the synthesized timbre. Voices are model-scoped (a v3
	// model needs a _v3 voice). After cloning a custom voice in Bailian
	// (音色复刻) fill its voice_id here.
	TTSVoiceID    string `env:"VOICE_TTS_VOICE_ID" default:"longxiaochun_v3"`
	TTSFormat     string `env:"VOICE_TTS_FORMAT" default:"wav"`
	TTSSampleRate int    `env:"VOICE_TTS_SAMPLE_RATE" default:"24000"`
	// VoiceTTSInstruction is the emotion/tone instruction driving a cloned/designed
	// TTS voice (e.g. "语气开朗俏皮，像女朋友撒娇"). cosyvoice v3 system voices (the
	// long*_v3 family) require a fixed instruction format, so a free-form instruction
	// is never sent for them — only for a cloned/designed voice (声音复刻).
	VoiceTTSInstruction string `env:"VOICE_TTS_INSTRUCTION" default:""`
	// VoiceClonePrefix is the name prefix used when creating a custom voice (声音复刻
	// create_voice). It must be ≤10 alphanumeric chars; the empty value falls back
	// to "cyrene". target_model stays TTSModel (must match the synthesis model).
	VoiceClonePrefix string `env:"VOICE_CLONE_PREFIX" default:"cyrene"`
}

// BackgroundConfig holds the autonomous thinking settings. QuietStartHour and
// QuietEndHour bound the do-not-disturb window in local clock hours; they are the
// single source shared by the backthink service and the channel bridge's
// minimal-disturbance quiet-hours gate (design/12 §6 策略五). Equal non-zero hours
// disable the window. The Push*/Kind fields tune the proactive-push frequency of
// design/11 §6 + design/12 §6 and are mapped into backthink.Config.
type BackgroundConfig struct {
	Enabled               bool `env:"ENABLE_BACKGROUND_THINKING" default:"true"`
	IntervalSec           int  `env:"THINK_INTERVAL_SEC" default:"900"`
	SilenceTimeout        int  `env:"THINK_SILENCE_TIMEOUT_SEC" default:"120"`
	QuietStartHour        int  `env:"QUIET_START_HOUR" default:"23"`
	QuietEndHour          int  `env:"QUIET_END_HOUR" default:"7"`
	PushLimit             int  `env:"THINK_PUSH_LIMIT" default:"1"`
	PushWindowSec         int  `env:"THINK_PUSH_WINDOW_SEC" default:"3600"`
	PushCooldownSec       int  `env:"THINK_PUSH_COOLDOWN_SEC" default:"1800"`
	RecentChatCooldownSec int  `env:"THINK_RECENT_CHAT_COOLDOWN_SEC" default:"600"`
	KindWindowSec         int  `env:"THINK_KIND_WINDOW_SEC" default:"7200"`
}

// WSHubConfig holds the WebSocket hub limits.
type WSHubConfig struct {
	MaxConnections     int `env:"WS_MAX_CONNECTIONS" default:"1000"`
	SessionIdleTimeout int `env:"SESSION_IDLE_TIMEOUT_MIN" default:"30"`
}

// MCPConfig holds Model Context Protocol server settings. The reflection-based
// loader only supports scalars and []string, so servers are supplied as a JSON
// array string (ServersJSON, env MCP_SERVERS_JSON) and parsed at startup by the
// MCP wiring (cmd/cyrene/mcp.go).
type MCPConfig struct {
	ServersJSON string `env:"MCP_SERVERS_JSON"`
}

// CapacityConfig configures the awakened-capacity decision layer
// (DevDocs/design/14-唤醒式能力渐进.md §4). It is opt-in: the loop exposes every
// tool until Enabled is true.
type CapacityConfig struct {
	// Enabled turns on the capacity decider. When false the loop advertises every
	// registered tool (pre-existing behavior). Note: it is wired at the loop
	// periphery, not back-end (Invoke stays unblocked).
	Enabled bool `env:"CAPACITY_ENABLED" default:"false"`
	// MinConfidence is the wake threshold below which the state is forced toward
	// the persona default (设计 §2.3 默认保守). Zero selects the code default 0.6.
	MinConfidence float64 `env:"CAPACITY_MIN_CONFIDENCE" default:"0.6"`
	// ToolSetsJSON, when non-empty, is a JSON map of state→[]tool name, e.g.
	// {"quiet":["terminal","file"],"awake":["terminal","file","web-search"],
	// "full":null}. A missing state selects the built-in default set; null ("full")
	// means unrestricted. Parsed by the capacity wiring at startup (the
	// reflection-based loader only supports scalars and []string), matching the
	// MCP config pattern.
	ToolSetsJSON string `env:"CAPACITY_TOOL_SETS"`
}

// MCPServer describes a single MCP server: a stdio subprocess (Command/Args) or
// an SSE endpoint (URL). Enabled gates activation at startup.
type MCPServer struct {
	Name    string   `json:"name"`
	Command string   `json:"command"`
	Args    []string `json:"args"`
	URL     string   `json:"url"`
	Enabled bool     `json:"enabled"`
}

// ChannelsConfig holds optional multi-account IM channel settings (design/16
// §4.6 §5). The single-account credentials are read from CHANNEL_<NAME>_* env
// vars in cmd/cyrene/channels.go; this block carries the optional multi-account
// registry plus the per-account allow/deny black-or-white lists via a JSON
// string (the reflection-based loader only supports scalars and []string, so a
// structured value is passed as JSON — matching the MCP and Capacity configs).
type ChannelsConfig struct {
	// AccountsJSON is the JSON shape map[platform]→[]ChannelAccount, e.g.
	// {"telegram":[{"id":"conf0","config":{"bot_token":"..."},
	//   "denyUsers":["12345"]}]}. Empty means the single-account env path applies.
	AccountsJSON string `env:"CHANNELS_JSON"`
}

// ChannelAccount is one IM access config (one account) for a platform. It pairs
// an id (used to name the adapter instance and to build the platform:confId
// account key) with the platform-specific credential map and the allow/deny
// black-or-white lists applied at the channelbridge ingress (design/16 §4.6), so
// a blacklisted or non-whitelisted group/user never reaches the LLM.
type ChannelAccount struct {
	// ID is the local config id, unique within its platform. It names the adapter
	// instance and the platform:confId account key (design/16 §4.5 deviceID).
	ID string `json:"id"`
	// Config is the platform-specific credential/feature map decoded into the
	// adapter's typed Config (bot_token, base_url, remote_url, ...). Missing keys
	// fall back to the adapter's own zero values.
	Config map[string]any `json:"config,omitempty"`
	// AllowGroups, when non-empty, is a white list of group ids: a group message
	// is dropped unless its GroupID is in this list.
	AllowGroups []string `json:"allowGroups,omitempty"`
	// DenyGroups is a black list of group ids: a group message whose GroupID is in
	// this list is dropped at ingress.
	DenyGroups []string `json:"denyGroups,omitempty"`
	// AllowUsers, when non-empty, is a white list of user uids: a direct (private)
	// message from a user not in this list is dropped.
	AllowUsers []string `json:"allowUsers,omitempty"`
	// DenyUsers is a black list of user uids: a direct message from a user in this
	// list is dropped at ingress.
	DenyUsers []string `json:"denyUsers,omitempty"`
	// GroupMsgPerMin caps group-chat inbound messages per minute for this account
	// (design/12 §6 策略三). Zero follows the protective default (6); a negative
	// value disables the group rate limit; a positive value pins it.
	GroupMsgPerMin int `json:"groupMsgPerMin,omitempty"`
	// MsgPerMin caps total inbound non-owner messages per minute for this account.
	// Zero or negative disables it.
	MsgPerMin int `json:"msgPerMin,omitempty"`
	// MaxMessages is the total pending/active message cap for this account
	// (design/12 §6 策略四). Zero follows the protective default (50); a negative
	// value disables the cap; a positive value pins it. When the pending set nears
	// this cap the review model prunes low-value messages.
	MaxMessages int `json:"maxMessages,omitempty"`
}

// Load reads configuration, merging an optional .env file (development only)
// with process environment variables. Process environment variables always take
// precedence over .env values, which take precedence over code defaults.
//
// Passing an empty envPath skips .env entirely. A missing .env file is not an
// error; only missing required values are.
func Load(envPath string) (*Config, error) {
	fromEnv := map[string]string{}
	if envPath != "" {
		m, err := parseDotEnv(envPath)
		if err != nil {
			return nil, fmt.Errorf("config: read %s: %w", envPath, err)
		}
		fromEnv = m
	}

	cfg := &Config{}
	if err := decodeValue(reflect.ValueOf(cfg), fromEnv); err != nil {
		return nil, err
	}

	if errs := validateStruct(reflect.ValueOf(cfg).Elem()); len(errs) > 0 {
		parts := make([]string, 0, len(errs))
		for _, e := range errs {
			parts = append(parts, e.Error())
		}
		return nil, fmt.Errorf("config: validation failed: %s", strings.Join(parts, "; "))
	}

	return cfg, nil
}

// MustLoad is Load that panics on error. It is intended for startup, matching
// the "schema validation fails at boot" rule.
func MustLoad(envPath string) *Config {
	cfg, err := Load(envPath)
	if err != nil {
		panic(err)
	}
	return cfg
}