mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-21 17:38:10 +00:00
367 lines
12 KiB
Go
367 lines
12 KiB
Go
package cfgmgr
|
||
|
||
import (
|
||
"encoding/json"
|
||
"log"
|
||
"sort"
|
||
|
||
"gitcode.com/JianFeeeee/HomeAgent/internal/plugin"
|
||
sdk "gitcode.com/JianFeeeee/HomeAgent/internal/sdk"
|
||
)
|
||
|
||
func init() {
|
||
plugin.RegisterPluginMeta("cfgmgr", "配置管理", "Config Manager")
|
||
plugin.RegisterFactory("cfgmgr", func(name string, config map[string]interface{}) (sdk.Plugin, error) {
|
||
return New(name), nil
|
||
})
|
||
}
|
||
|
||
type Plugin struct {
|
||
name string
|
||
}
|
||
|
||
func New(name string) *Plugin {
|
||
return &Plugin{name: name}
|
||
}
|
||
|
||
func (p *Plugin) Name() string { return p.name }
|
||
|
||
func (p *Plugin) Start(s *sdk.PluginSDK) error {
|
||
s.SetAutoRestart(true)
|
||
|
||
s.RegisterTool("config_list_plugins", sdk.ToolDef{
|
||
Name: "config_list_plugins",
|
||
Description: "列出所有插件的配置命名空间，返回每个插件的名称和配置项数量",
|
||
Parameters: map[string]interface{}{
|
||
"type": "object",
|
||
"properties": map[string]interface{}{},
|
||
},
|
||
}, p.handleListPlugins(s))
|
||
|
||
s.RegisterTool("config_get", sdk.ToolDef{
|
||
Name: "config_get",
|
||
Description: "获取指定插件或核心的配置项值。scope=core 读取核心配置，scope=插件名 读取该插件的配置",
|
||
Parameters: map[string]interface{}{
|
||
"type": "object",
|
||
"properties": map[string]interface{}{
|
||
"scope": map[string]interface{}{"type": "string", "description": "配置作用域：core 表示核心配置，插件名表示该插件的配置"},
|
||
"key": map[string]interface{}{"type": "string", "description": "配置项键名"},
|
||
},
|
||
"required": []string{"scope", "key"},
|
||
},
|
||
}, p.handleGet(s))
|
||
|
||
s.RegisterTool("config_set", sdk.ToolDef{
|
||
Name: "config_set",
|
||
Description: "设置指定插件或核心的配置项值。scope=core 写入核心配置，scope=插件名 写入该插件的配置。修改后立即生效，无需重启",
|
||
Parameters: map[string]interface{}{
|
||
"type": "object",
|
||
"properties": map[string]interface{}{
|
||
"scope": map[string]interface{}{"type": "string", "description": "配置作用域：core 表示核心配置，插件名表示该插件的配置"},
|
||
"key": map[string]interface{}{"type": "string", "description": "配置项键名"},
|
||
"value": map[string]interface{}{"type": "string", "description": "配置项值（字符串形式，布尔用 true/false，数字用字面量）"},
|
||
},
|
||
"required": []string{"scope", "key", "value"},
|
||
},
|
||
}, p.handleSet(s))
|
||
|
||
s.RegisterTool("config_list_keys", sdk.ToolDef{
|
||
Name: "config_list_keys",
|
||
Description: "列出指定插件或核心的所有配置项键名和当前值。scope=core 列出核心配置，scope=插件名 列出该插件配置",
|
||
Parameters: map[string]interface{}{
|
||
"type": "object",
|
||
"properties": map[string]interface{}{
|
||
"scope": map[string]interface{}{"type": "string", "description": "配置作用域：core 或插件名"},
|
||
"prefix": map[string]interface{}{"type": "string", "description": "键名前缀过滤（可选，默认空表示全部）"},
|
||
},
|
||
"required": []string{"scope"},
|
||
},
|
||
}, p.handleListKeys(s))
|
||
|
||
s.RegisterTool("config_get_defs", sdk.ToolDef{
|
||
Name: "config_get_defs",
|
||
Description: "获取指定插件或核心的配置项定义（类型、描述、默认值、选项等），用于了解每个配置项的含义和合法值",
|
||
Parameters: map[string]interface{}{
|
||
"type": "object",
|
||
"properties": map[string]interface{}{
|
||
"scope": map[string]interface{}{"type": "string", "description": "配置作用域：core 或插件名"},
|
||
"prefix": map[string]interface{}{"type": "string", "description": "键名前缀过滤（可选）"},
|
||
},
|
||
"required": []string{"scope"},
|
||
},
|
||
}, p.handleGetDefs(s))
|
||
|
||
s.RegisterTool("config_dump", sdk.ToolDef{
|
||
Name: "config_dump",
|
||
Description: "导出所有配置（核心+所有插件）的完整快照，用于全局审查或备份",
|
||
Parameters: map[string]interface{}{
|
||
"type": "object",
|
||
"properties": map[string]interface{}{},
|
||
},
|
||
}, p.handleDump(s))
|
||
|
||
s.RegisterTool("config_batch_set", sdk.ToolDef{
|
||
Name: "config_batch_set",
|
||
Description: "批量设置配置项，一次修改多个键值对。items 为 JSON 数组，每项包含 scope/key/value",
|
||
Parameters: map[string]interface{}{
|
||
"type": "object",
|
||
"properties": map[string]interface{}{
|
||
"items": map[string]interface{}{
|
||
"type": "array",
|
||
"description": "配置项数组，每项: {\"scope\":\"core|插件名\", \"key\":\"键名\", \"value\":\"值\"}",
|
||
"items": map[string]interface{}{
|
||
"type": "object",
|
||
"properties": map[string]interface{}{
|
||
"scope": map[string]interface{}{"type": "string", "description": "配置作用域"},
|
||
"key": map[string]interface{}{"type": "string", "description": "配置项键名"},
|
||
"value": map[string]interface{}{"type": "string", "description": "配置项值"},
|
||
},
|
||
"required": []string{"scope", "key", "value"},
|
||
},
|
||
},
|
||
},
|
||
"required": []string{"items"},
|
||
},
|
||
}, p.handleBatchSet(s))
|
||
|
||
log.Printf("[cfgmgr] started")
|
||
return nil
|
||
}
|
||
|
||
func (p *Plugin) Stop() error {
|
||
return nil
|
||
}
|
||
|
||
func (p *Plugin) handleListPlugins(s *sdk.PluginSDK) sdk.ToolHandler {
|
||
return func(args map[string]interface{}) (interface{}, error) {
|
||
sett := s.Settings()
|
||
if sett == nil {
|
||
return map[string]interface{}{"error": "settings API not available"}, nil
|
||
}
|
||
plugins := sett.Plugins()
|
||
result := make([]map[string]interface{}, 0, len(plugins))
|
||
for _, name := range plugins {
|
||
entry := map[string]interface{}{"name": name}
|
||
if name != "core" {
|
||
keys, err := sett.ListPlugin(name, "")
|
||
if err == nil {
|
||
entry["key_count"] = len(keys)
|
||
}
|
||
} else {
|
||
keys, err := sett.ListCore("")
|
||
if err == nil {
|
||
entry["key_count"] = len(keys)
|
||
}
|
||
}
|
||
result = append(result, entry)
|
||
}
|
||
return map[string]interface{}{"plugins": result}, nil
|
||
}
|
||
}
|
||
|
||
func (p *Plugin) handleGet(s *sdk.PluginSDK) sdk.ToolHandler {
|
||
return func(args map[string]interface{}) (interface{}, error) {
|
||
sett := s.Settings()
|
||
if sett == nil {
|
||
return map[string]interface{}{"error": "settings API not available"}, nil
|
||
}
|
||
scope, _ := args["scope"].(string)
|
||
key, _ := args["key"].(string)
|
||
if scope == "" || key == "" {
|
||
return map[string]interface{}{"error": "scope and key are required"}, nil
|
||
}
|
||
|
||
var val interface{}
|
||
var err error
|
||
if scope == "core" {
|
||
val, err = sett.GetCore(key)
|
||
} else {
|
||
val, err = sett.GetPlugin(scope, key)
|
||
}
|
||
if err != nil {
|
||
return map[string]interface{}{"error": err.Error(), "scope": scope, "key": key}, nil
|
||
}
|
||
return map[string]interface{}{"scope": scope, "key": key, "value": val}, nil
|
||
}
|
||
}
|
||
|
||
func (p *Plugin) handleSet(s *sdk.PluginSDK) sdk.ToolHandler {
|
||
return func(args map[string]interface{}) (interface{}, error) {
|
||
sett := s.Settings()
|
||
if sett == nil {
|
||
return map[string]interface{}{"error": "settings API not available"}, nil
|
||
}
|
||
scope, _ := args["scope"].(string)
|
||
key, _ := args["key"].(string)
|
||
value, _ := args["value"].(string)
|
||
if scope == "" || key == "" {
|
||
return map[string]interface{}{"error": "scope and key are required"}, nil
|
||
}
|
||
|
||
var err error
|
||
if scope == "core" {
|
||
err = sett.SetCore(key, value)
|
||
} else {
|
||
err = sett.SetPlugin(scope, key, value)
|
||
}
|
||
if err != nil {
|
||
return map[string]interface{}{"error": err.Error()}, nil
|
||
}
|
||
log.Printf("[cfgmgr] set %s.%s = %s", scope, key, value)
|
||
return map[string]interface{}{"status": "ok", "scope": scope, "key": key, "value": value}, nil
|
||
}
|
||
}
|
||
|
||
func (p *Plugin) handleListKeys(s *sdk.PluginSDK) sdk.ToolHandler {
|
||
return func(args map[string]interface{}) (interface{}, error) {
|
||
sett := s.Settings()
|
||
if sett == nil {
|
||
return map[string]interface{}{"error": "settings API not available"}, nil
|
||
}
|
||
scope, _ := args["scope"].(string)
|
||
prefix, _ := args["prefix"].(string)
|
||
if scope == "" {
|
||
return map[string]interface{}{"error": "scope is required"}, nil
|
||
}
|
||
|
||
var keys []string
|
||
var err error
|
||
if scope == "core" {
|
||
keys, err = sett.ListCore(prefix)
|
||
} else {
|
||
keys, err = sett.ListPlugin(scope, prefix)
|
||
}
|
||
if err != nil {
|
||
return map[string]interface{}{"error": err.Error()}, nil
|
||
}
|
||
sort.Strings(keys)
|
||
|
||
items := make([]map[string]interface{}, 0, len(keys))
|
||
for _, k := range keys {
|
||
var val interface{}
|
||
if scope == "core" {
|
||
val, _ = sett.GetCore(k)
|
||
} else {
|
||
val, _ = sett.GetPlugin(scope, k)
|
||
}
|
||
items = append(items, map[string]interface{}{"key": k, "value": val})
|
||
}
|
||
return map[string]interface{}{"scope": scope, "count": len(items), "items": items}, nil
|
||
}
|
||
}
|
||
|
||
func (p *Plugin) handleGetDefs(s *sdk.PluginSDK) sdk.ToolHandler {
|
||
return func(args map[string]interface{}) (interface{}, error) {
|
||
sett := s.Settings()
|
||
if sett == nil {
|
||
return map[string]interface{}{"error": "settings API not available"}, nil
|
||
}
|
||
scope, _ := args["scope"].(string)
|
||
prefix, _ := args["prefix"].(string)
|
||
if scope == "" {
|
||
return map[string]interface{}{"error": "scope is required"}, nil
|
||
}
|
||
|
||
defPrefix := buildDefPrefix(scope, prefix)
|
||
defs := sett.Defs(defPrefix)
|
||
result := make([]map[string]interface{}, 0, len(defs))
|
||
for _, d := range defs {
|
||
if d == nil {
|
||
continue
|
||
}
|
||
entry := map[string]interface{}{
|
||
"key": d.Key,
|
||
"type": d.Type,
|
||
"display_name": d.DisplayName,
|
||
"description": d.Description,
|
||
}
|
||
if d.Default != nil {
|
||
entry["default"] = d.Default
|
||
}
|
||
if len(d.Options) > 0 {
|
||
entry["options"] = d.Options
|
||
}
|
||
if d.Category != "" {
|
||
entry["category"] = d.Category
|
||
}
|
||
if d.Secret {
|
||
entry["secret"] = true
|
||
}
|
||
result = append(result, entry)
|
||
}
|
||
return map[string]interface{}{"scope": scope, "count": len(result), "defs": result}, nil
|
||
}
|
||
}
|
||
|
||
func (p *Plugin) handleDump(s *sdk.PluginSDK) sdk.ToolHandler {
|
||
return func(args map[string]interface{}) (interface{}, error) {
|
||
sett := s.Settings()
|
||
if sett == nil {
|
||
return map[string]interface{}{"error": "settings API not available"}, nil
|
||
}
|
||
dump := sett.Dump()
|
||
return map[string]interface{}{"config": dump}, nil
|
||
}
|
||
}
|
||
|
||
func (p *Plugin) handleBatchSet(s *sdk.PluginSDK) sdk.ToolHandler {
|
||
return func(args map[string]interface{}) (interface{}, error) {
|
||
sett := s.Settings()
|
||
if sett == nil {
|
||
return map[string]interface{}{"error": "settings API not available"}, nil
|
||
}
|
||
rawItems, ok := args["items"].([]interface{})
|
||
if !ok || len(rawItems) == 0 {
|
||
return map[string]interface{}{"error": "items must be a non-empty array"}, nil
|
||
}
|
||
|
||
var results []map[string]interface{}
|
||
applied := 0
|
||
for i, raw := range rawItems {
|
||
item, ok := raw.(map[string]interface{})
|
||
if !ok {
|
||
results = append(results, map[string]interface{}{"index": i, "error": "invalid item format"})
|
||
continue
|
||
}
|
||
scope, _ := item["scope"].(string)
|
||
key, _ := item["key"].(string)
|
||
value, _ := item["value"].(string)
|
||
if scope == "" || key == "" {
|
||
results = append(results, map[string]interface{}{"index": i, "error": "scope and key required"})
|
||
continue
|
||
}
|
||
|
||
var err error
|
||
if scope == "core" {
|
||
err = sett.SetCore(key, value)
|
||
} else {
|
||
err = sett.SetPlugin(scope, key, value)
|
||
}
|
||
if err != nil {
|
||
results = append(results, map[string]interface{}{"index": i, "scope": scope, "key": key, "error": err.Error()})
|
||
continue
|
||
}
|
||
applied++
|
||
results = append(results, map[string]interface{}{"index": i, "scope": scope, "key": key, "value": value, "status": "ok"})
|
||
}
|
||
log.Printf("[cfgmgr] batch set: %d/%d applied", applied, len(rawItems))
|
||
return map[string]interface{}{"applied": applied, "total": len(rawItems), "results": results}, nil
|
||
}
|
||
}
|
||
|
||
func buildDefPrefix(scope, prefix string) string {
|
||
if scope == "core" {
|
||
if prefix == "" {
|
||
return "core."
|
||
}
|
||
return "core." + prefix
|
||
}
|
||
p := "plugin." + scope + "."
|
||
if prefix != "" {
|
||
p += prefix
|
||
}
|
||
return p
|
||
}
|
||
|
||
var _ = json.Marshal
|