Agent-System-Backend/internal/settings/settings.go

// Package settings 实现分层设置（4.9）：system 默认 → user → project → conversation，
// schema 驱动设置页（生效时机 hot / new-run / restart 显式标注）。
package settings

import (
	"sort"
	"strings"

	"agent-system-backend/internal/store"
	core "agent-system-core"
)

// Scope 层级键：user 与 project:<id> 落库；conversation 经会话画像选择实现。
const (
	ScopeUser          = "user"
	ScopeProjectPrefix = "project:"
)

// SystemDefaults 是系统级默认（代码内置，settings.system.json 语义）。
func SystemDefaults() map[string]any {
	return map[string]any{
		"models": map[string]any{
			"chat": map[string]any{
				"baseUrl":   "", // 例：https://api.openai.com/v1 （OpenAI 兼容）
				"apiKey":    "",
				"model":     "",
				"fallbacks": "", // 逗号分隔；空 = 不回退
			},
			"lightweight": map[string]any{"baseUrl": "", "apiKey": "", "model": ""},
			"embedding":   map[string]any{"baseUrl": "", "apiKey": "", "model": ""},
			"rerank":      map[string]any{"baseUrl": "", "apiKey": "", "model": ""},
		},
		"tools": map[string]any{
			"tavily": map[string]any{"apiKeys": "", "endpoint": ""}, // 网络搜索（逗号分隔密钥池；空 = 不注册搜索工具）
		},
		"runtime": map[string]any{
			"mock":              false,    // true = 全部会话走 mock runner（演示/测试）
			"squad":             true,     // 小队编排（Agent-Squad 引擎）可用
			"entry":             "direct", // direct = 现有对话循环；squad = 小队分诊入口（灰度，达标后再改默认）
			"rateLimitPerMin":   0,        // 租户限流（次/分；0 = 不限）
			"redisAddr":         "",       // 限流共享后端（原生 Redis addr；空 = 进程内）
			"verify":            true,     // 终轮确定性检查与独立验证
			"llmRetries":        2,        // 直连模型在尚未输出时的额外重试次数
			"codeOrchestration": false,    // 模型用脚本批量调用工具；默认关
		},
		"storage": map[string]any{
			"dataDir": "./data", // 启动注入实际值（--data）
		},
		"context": map[string]any{
			"maxHistoryMessages": 40,
			"maxToolResultChars": 20000,
			"userInstructions":   "",
			"contextWindow":      32000,
			"messageShape":       "merge",
		},
		"crossSession": map[string]any{
			"read": false, // 跨会话读取项目闸门（提案 2026-09-30-cross-session-reading）：关 = 全项目禁读
		},
		"retrieval": map[string]any{"topK": 5},
		"ui":        map[string]any{"theme": "dark", "cardflow": false},
		"deliverables": map[string]any{
			"style":              "modern", // modern / formal / minimal
			"accentColor":        "",       // 空 = 主题默认主色；#RRGGBB 或 RRGGBB
			"templateArtifactId": "",       // 非空 = 该工件作为默认模板，落到 inputs/_default-template*
		},
		"plugins": map[string]any{"dir": "./plugins"},
	}
}

// Schema 返回设置描述清单（设置页渲染依据；插件设置项由 pluginhost 追加）。
func Schema() []core.SettingDesc {
	return []core.SettingDesc{
		{Key: "models.chat.baseUrl", Group: "models", Label: "主模型 Base URL", Type: core.SettingString, Default: "", Effect: core.EffectHot, Description: "OpenAI 兼容端点，如 https://api.openai.com/v1"},
		{Key: "models.chat.apiKey", Group: "models", Label: "主模型 API Key", Type: core.SettingSecret, Default: "", Effect: core.EffectHot},
		{Key: "models.chat.model", Group: "models", Label: "主模型名称", Type: core.SettingString, Default: "", Effect: core.EffectNewRun},
		{Key: "models.chat.fallbacks", Group: "models", Label: "主模型回退链", Type: core.SettingString, Default: "", Effect: core.EffectNewRun, Description: "逗号分隔的模型名。只在尚未输出时，前一个模型返回 5xx 才改用下一个。空表示不回退"},
		{Key: "models.chat.maxOutputTokens", Group: "models", Label: "主模型输出上限", Type: core.SettingNumber, Default: 0, Effect: core.EffectNewRun, Description: "单次回复的 max_tokens。0 = 不限制，由模型服务端决定"},
		{Key: "models.lightweight.baseUrl", Group: "models", Label: "轻量模型 Base URL", Type: core.SettingString, Default: "", Effect: core.EffectHot, Description: "标题生成/澄清等轻任务；空则回落主模型"},
		{Key: "models.lightweight.apiKey", Group: "models", Label: "轻量模型 API Key", Type: core.SettingSecret, Default: "", Effect: core.EffectHot},
		{Key: "models.lightweight.model", Group: "models", Label: "轻量模型名称", Type: core.SettingString, Default: "", Effect: core.EffectNewRun},
		{Key: "models.lightweight.maxOutputTokens", Group: "models", Label: "轻量模型输出上限", Type: core.SettingNumber, Default: 0, Effect: core.EffectNewRun, Description: "单次回复的 max_tokens。0 = 不限制，由模型服务端决定"},
		{Key: "models.embedding.baseUrl", Group: "models", Label: "Embedding Base URL", Type: core.SettingString, Default: "", Effect: core.EffectHot, Description: "RAG/记忆检索用（向量管线远期启用）"},
		{Key: "models.embedding.apiKey", Group: "models", Label: "Embedding API Key", Type: core.SettingSecret, Default: "", Effect: core.EffectHot},
		{Key: "models.embedding.model", Group: "models", Label: "Embedding 模型", Type: core.SettingString, Default: "", Effect: core.EffectHot},
		{Key: "models.rerank.baseUrl", Group: "models", Label: "Reranker Base URL", Type: core.SettingString, Default: "", Effect: core.EffectHot},
		{Key: "models.rerank.apiKey", Group: "models", Label: "Reranker API Key", Type: core.SettingSecret, Default: "", Effect: core.EffectHot},
		{Key: "models.rerank.model", Group: "models", Label: "Reranker 模型", Type: core.SettingString, Default: "", Effect: core.EffectHot},
		{Key: "runtime.mock", Group: "runtime", Label: "演示模式（mock runner）", Type: core.SettingBool, Default: false, Effect: core.EffectNewRun, Description: "不调用真实模型，按脚本回放事件流"},
		{Key: "runtime.squad", Group: "runtime", Label: "小队编排（Agent-Squad）", Type: core.SettingBool, Default: true, Effect: core.EffectNewRun, Description: "启用 Agent-Squad 引擎编排（#squad 指令与 squad_orchestrate 工具）"},
		{Key: "runtime.entry", Group: "runtime", Label: "对话入口", Type: core.SettingEnum, Default: "direct", Options: []string{"direct", "squad"}, Effect: core.EffectNewRun, Description: "direct 走现有对话循环。squad 由小队先分诊：简单任务单步完成，复杂任务由 leader 编排。灰度期默认 direct"},
		{Key: "runtime.rateLimitPerMin", Group: "runtime", Label: "租户限流（次/分）", Type: core.SettingNumber, Default: 0, Effect: core.EffectHot, Description: "0 = 不限；多用户建议 300–1200"},
		{Key: "runtime.redisAddr", Group: "runtime", Label: "Redis 地址（限流共享）", Type: core.SettingString, Default: "", Effect: core.EffectRestart, Description: "原生 Redis（如 127.0.0.1:6379）；空 = 进程内限流（单实例）"},
		{Key: "runtime.verify", Group: "runtime", Label: "终轮结果检查", Type: core.SettingBool, Default: true, Effect: core.EffectNewRun, Description: "终轮确定性检查与独立验证"},
		{Key: "runtime.llmRetries", Group: "runtime", Label: "模型重试次数", Type: core.SettingNumber, Default: 2, Effect: core.EffectNewRun, Description: "直连模型在还没有任何输出时，遇到超时、限流或服务端错误的额外重试次数。0 = 不重试。已经开始输出后不再重试"},
		{Key: "runtime.codeOrchestration", Group: "runtime", Label: "代码编排", Type: core.SettingBool, Default: false, Effect: core.EffectNewRun, Description: "允许模型用 run_tool_script 写脚本批量调用工具。默认关闭"},
		{Key: "runtime.composeFinal", Group: "runtime", Label: "终轮整理回复", Type: core.SettingBool, Default: true, Effect: core.EffectNewRun, Description: "用过工具的回合结束后，再用主模型写一段直接回答用户的最终回复"},
		{Key: "context.maxHistoryMessages", Group: "context", Label: "历史窗口条数", Type: core.SettingNumber, Default: 40, Effect: core.EffectNewRun},
		{Key: "context.maxToolResultChars", Group: "context", Label: "工具结果上限（字符）", Type: core.SettingNumber, Default: 20000, Effect: core.EffectNewRun},
		{Key: "context.userInstructions", Group: "context", Label: "个人指令", Type: core.SettingString, Default: "", Effect: core.EffectNewRun, Description: "每次对话附在上下文末尾，不改系统前缀"},
		{Key: "context.contextWindow", Group: "context", Label: "上下文窗口（估算 token）", Type: core.SettingNumber, Default: 32000, Effect: core.EffectNewRun, Description: "超出时只丢掉最早的历史。估算为 UTF-8 字节 / 3.35，并用上一请求的真实用量校准"},
		{Key: "context.messageShape", Group: "context", Label: "消息整形", Type: core.SettingEnum, Default: "merge", Options: []string{"merge", "strict"}, Effect: core.EffectNewRun, Description: "merge 合并相邻同角色。strict 再把多余的 system 收成 user，满足严格交替"},
		{Key: "crossSession.read", Group: "crossSession", Label: "跨会话读取", Type: core.SettingBool, Default: false, Effect: core.EffectNewRun, Description: "允许本项目会话读取同项目其他会话的对话记录（list_conversations / read_conversation 工具）；会话可单独关闭读取或设为私有"},
		{Key: "retrieval.topK", Group: "retrieval", Label: "检索命中数", Type: core.SettingNumber, Default: 5, Effect: core.EffectHot},
		{Key: "ui.theme", Group: "ui", Label: "主题", Type: core.SettingEnum, Default: "dark", Options: []string{"dark", "light"}, Effect: core.EffectHot},
		{Key: "ui.cardflow", Group: "ui", Label: "长结果进右侧卡片", Type: core.SettingBool, Default: false, Effect: core.EffectNewRun, Description: "开启后报告/纪要/方案等产物型长内容经 submit_card 进收件箱，对话只留结论摘要与锚点"},
		{Key: "tools.tavily.apiKeys", Group: "tools", Label: "Tavily API Keys（池）", Type: core.SettingSecret, Default: "", Effect: core.EffectRestart, Description: "逗号分隔；配置后注册 tavily_search/web_search 工具（429/401/403 自动轮换下一把）"},
		{Key: "tools.tavily.endpoint", Group: "tools", Label: "Tavily 端点", Type: core.SettingString, Default: "", Effect: core.EffectRestart, Description: "空 = 官方 https://api.tavily.com/search；可指向自建代理"},
		{Key: "tools.mcp.overrides", Group: "tools", Label: "MCP 项目覆盖", Type: core.SettingString, Default: "", Effect: core.EffectNewRun, Description: "JSON 对象。键是服务器名，值可含 enabled 与 exposure（direct 或 deferred）"},
		{Key: "deliverables.style", Group: "deliverables", Label: "交付物默认样式", Type: core.SettingEnum, Default: "modern", Options: []string{"modern", "formal", "minimal"}, Effect: core.EffectNewRun, Description: "modern 现代专业 / formal 公文 / minimal 极简。本轮指令、上传模板或参考文档优先于这项"},
		{Key: "deliverables.accentColor", Group: "deliverables", Label: "交付物强调色", Type: core.SettingString, Default: "", Effect: core.EffectNewRun, Description: "6 位十六进制，如 1B3A4B。空 = 使用样式自带主色"},
		{Key: "deliverables.templateArtifactId", Group: "deliverables", Label: "默认模板工件 ID", Type: core.SettingString, Default: "", Effect: core.EffectNewRun, Description: "填写已上传模板的工件 ID。每次运行会复制到会话 inputs/_default-template"},
		{Key: "plugins.dir", Group: "plugins", Label: "插件目录", Type: core.SettingString, Default: "./plugins", Effect: core.EffectRestart},
	}
}

// Service 提供有效设置计算与分层写入。
type Service struct {
	st     *store.Store
	system map[string]any // 启动注入的系统层覆盖（如数据目录），位于 defaults 与 user 之间
}

func NewService(st *store.Store) *Service { return &Service{st: st} }

// SetSystemOverride 注入启动期系统层覆盖（组合根调用；不落库）。
func (s *Service) SetSystemOverride(values map[string]any) { s.system = values }

// Effective 计算覆盖链结果：defaults → system(启动注入) → user(租户) → project。
// owner 是租户身份（认证中间件注入；未启用认证 = store.OwnerDefault）。
func (s *Service) Effective(owner, projectID string) core.EffectiveSettings {
	eff := core.MergeSettings(SystemDefaults(), map[string]any{})
	if s.system != nil {
		eff = core.MergeSettings(eff, s.system)
	}
	if user, err := s.st.GetSettingsScope(owner, ScopeUser); err == nil {
		eff = core.MergeSettings(eff, user)
	}
	if projectID != "" {
		if proj, err := s.st.GetSettingsScope(owner, ScopeProjectPrefix+projectID); err == nil {
			eff = core.MergeSettings(eff, proj)
		}
	}
	return eff
}

func (s *Service) UserScope(owner string) (map[string]any, error) {
	return s.st.GetSettingsScope(owner, ScopeUser)
}

func (s *Service) ProjectScope(owner, projectID string) (map[string]any, error) {
	return s.st.GetSettingsScope(owner, ScopeProjectPrefix+projectID)
}

func (s *Service) MergeUser(owner string, values map[string]any) error {
	return s.mergeScope(owner, ScopeUser, values)
}

func (s *Service) MergeProject(owner, projectID string, values map[string]any) error {
	return s.mergeScope(owner, ScopeProjectPrefix+projectID, values)
}

// mergeScope 先把库里和本次写入的点路径键展开成嵌套，再深合并。
// 掩码占位符视为未修改，不覆盖已存密钥。
func (s *Service) mergeScope(owner, scope string, values map[string]any) error {
	incoming := prepareWrite(values)
	if err := s.st.UpdateSettingsScope(owner, scope, func(cur map[string]any) (map[string]any, error) {
		return core.MergeSettings(normalizeKeys(cur), incoming), nil
	}); err != nil {
		return err
	}
	s.snapshotThrottled(owner, scope)
	return nil
}

// NormalizeStats 是启动迁移的结果。MaskPaths 只含 owner、scope 和键路径，不含值。
type NormalizeStats struct {
	DottedKeys int
	Scopes     int
	MaskPaths  []string
}

// NormalizeStored 把已落库的点路径键展开成嵌套。嵌套对象胜出，已生效的值不变。
// 存量值恰好是掩码占位符时记入 MaskPaths，原值保留（密钥已无法恢复）。
func (s *Service) NormalizeStored() (NormalizeStats, error) {
	var stats NormalizeStats
	refs, err := s.st.ListSettingsScopes()
	if err != nil {
		return stats, err
	}
	for _, ref := range refs {
		var dotted int
		var masks []string
		err := s.st.UpdateSettingsScope(ref.Owner, ref.Scope, func(cur map[string]any) (map[string]any, error) {
			dotted = countDottedKeys(cur)
			next := normalizeKeys(cur)
			for _, p := range maskPlaceholderPaths(next, "") {
				masks = append(masks, ref.Owner+" "+ref.Scope+" "+p)
			}
			return next, nil
		})
		if err != nil {
			return stats, err
		}
		stats.MaskPaths = append(stats.MaskPaths, masks...)
		if dotted > 0 {
			stats.DottedKeys += dotted
			stats.Scopes++
		}
	}
	return stats, nil
}

// MaskPlaceholder 是 GET 对密钥的掩码。写回这个值视为未修改。
const MaskPlaceholder = "••••••"

// Masked 返回脱敏视图。schema 中 Type=secret 或 Secret=true 的路径一律掩码；
// 叶子键名（apiKey/apiKeys 等）再兜底一层，覆盖 schema 未声明的密钥。
// 就地掩码：调用方须传入可丢弃的 map（Effective 每次深合并产出新 map）。
func Masked(values map[string]any, schema []core.SettingDesc) map[string]any {
	if values == nil {
		return map[string]any{}
	}
	maskSecrets(values)
	for _, d := range schema {
		if d.Type == core.SettingSecret || d.Secret {
			maskAtPath(values, d.Key)
		}
	}
	return values
}

func maskSecrets(m map[string]any) {
	for k, v := range m {
		if vm, ok := v.(map[string]any); ok {
			maskSecrets(vm)
			continue
		}
		if isSecretKey(k) {
			if s, ok := v.(string); ok && s != "" {
				m[k] = MaskPlaceholder
			}
		}
	}
}

func isSecretKey(k string) bool {
	switch k {
	case "apiKey", "api_key", "apiKeys", "secret", "token":
		return true
	default:
		return false
	}
}

// prepareWrite 丢弃掩码占位符，再把点路径键展开成嵌套。
func prepareWrite(values map[string]any) map[string]any {
	return normalizeKeys(dropMaskPlaceholders(values))
}

// normalizeKeys 递归展开点路径键。同一层嵌套对象与点路径冲突时嵌套胜出；
// 点路径键按字典序依次合并，后写入的覆盖先写入的。不改入参。
func normalizeKeys(in map[string]any) map[string]any {
	if len(in) == 0 {
		return map[string]any{}
	}
	plain := make(map[string]any, len(in))
	dotted := make([]string, 0)
	for k, v := range in {
		if strings.Contains(k, ".") {
			dotted = append(dotted, k)
			continue
		}
		plain[k] = normalizeValue(v)
	}
	sort.Strings(dotted)
	expanded := map[string]any{}
	for _, k := range dotted {
		expanded = core.MergeSettings(expanded, expandDotted(k, normalizeValue(in[k])))
	}
	return core.MergeSettings(expanded, plain)
}

func normalizeValue(v any) any {
	vm, ok := v.(map[string]any)
	if !ok {
		return v
	}
	return normalizeKeys(vm)
}

// expandDotted 把 "a.b.c" 收成 {a:{b:{c: val}}}。
func expandDotted(key string, val any) map[string]any {
	parts := strings.Split(key, ".")
	built := val
	for i := len(parts) - 1; i >= 0; i-- {
		built = map[string]any{parts[i]: built}
	}
	return built.(map[string]any)
}

func dropMaskPlaceholders(m map[string]any) map[string]any {
	out := make(map[string]any, len(m))
	for k, v := range m {
		if vm, ok := v.(map[string]any); ok {
			out[k] = dropMaskPlaceholders(vm)
			continue
		}
		if s, ok := v.(string); ok && s == MaskPlaceholder {
			continue
		}
		out[k] = v
	}
	return out
}

func countDottedKeys(m map[string]any) int {
	n := 0
	for k, v := range m {
		if strings.Contains(k, ".") {
			n++
		}
		if vm, ok := v.(map[string]any); ok {
			n += countDottedKeys(vm)
		}
	}
	return n
}

func maskPlaceholderPaths(m map[string]any, prefix string) []string {
	var out []string
	for k, v := range m {
		path := k
		if prefix != "" {
			path = prefix + "." + k
		}
		if vm, ok := v.(map[string]any); ok {
			out = append(out, maskPlaceholderPaths(vm, path)...)
			continue
		}
		if s, ok := v.(string); ok && s == MaskPlaceholder {
			out = append(out, path)
		}
	}
	sort.Strings(out)
	return out
}

// maskAtPath 掩码 schema 路径。路径作为单个点键存在时同样掩码。
func maskAtPath(m map[string]any, path string) {
	if s, ok := m[path].(string); ok && s != "" {
		m[path] = MaskPlaceholder
	}
	parts := strings.Split(path, ".")
	cur := m
	for i, p := range parts {
		v, ok := cur[p]
		if !ok {
			return
		}
		if i == len(parts)-1 {
			if s, ok := v.(string); ok && s != "" {
				cur[p] = MaskPlaceholder
			}
			return
		}
		next, ok := v.(map[string]any)
		if !ok {
			return
		}
		cur = next
	}
}