mirror of
https://gitcode.com/JianFeeeee/homeagent-sdk.git
synced 2026-10-31 14:27:30 +00:00
feat(examples): 内置插件外置化 —— cmd/localuse/mcp 迁入 example + files 新增按行号精确编辑
## 背景 主仓按「只有必须使用内部 SDK 能力的插件才内置」的判定标准(见主仓 ARCHITECTURE「内置插件的判定标准」)把纯公开 SDK 插件移出编译期内置清单。 本仓为它们提供外部插件工程(example/ 下,hmapdev 可直接构建)。 ★ 终审翻案:kbtree 与 multimodal **不**在本次外置之列 —— 逐方法对照 公开 SDK 后发现它们依赖内部专有能力(kbtree 要 KnowledgeAPI 的 Subtree/CategoryCounts/SearchIn/Categories 与树视图类型;multimodal 要 LLMAPI.Chat 与多模态消息类型),公开面没有,已加回主仓内置清单。 ## 新增工程(全部对齐当前模板 + 随用随起契约) | 工程 | 工具 | auto_start | 说明 | |---|---|---|---| | cmd | cmd_run | false(按需) | 沙箱执行 shell 命令 | | localuse | screensee/screensue/camerasue/speakeruse/clipboardsee/clipboardsue/computeruse | false(按需) | 本机外设 | | mcp | mcp_add/list/remove/restart_server | false(按需) | MCP 服务器接入(stdio/SSE) | 每工程:plg.json(含 sdk 字段声明)+ skill/<name>-tools/SKILL.md + README。按 auto_start 契约(sdk/autostart.go 的 OnDemandNeedsSkill), **auto_start=false 必须配 skill**,否则工具不出现在初始清单 ⇒ 模型不知道 它存在 ⇒ 永远不会被拉起,等于没装还占地方。 ## 移植要点 - 注册:内置版 init()+RegisterFactory → 外部版导出 NewPluginFactory; RegisterPluginMeta 的显示名进 plg.json/plugin.json。 - mcp:本地 rpcRequest/rpcResponse/rpcError 与 hmapdev 生成的 z_proc_gen.go 同名冲突(构建期生成物与本目录源码同包编译), 本地类型改名 mcpRPC*;internal/safeclose 内联为 safeCloseChan (外部插件不能引 internal 包,close-of-closed-channel 纪律随之带过来)。 - module 名:cmd 的 go.mod 用 homeagent-cmd(避免与标准库 cmd 混淆); name_en 对齐插件名,hmapdev 以 toSnake(name_en) 命名产物 (此前误写 Command/Local,产物成了 command_/local_)。 ## files:新增 files_edit_line(按行号精确编辑) files_edit 要求 old 文本全文件唯一,改「第三个 case 的 break」这类 非唯一文本时只能退化为整文件重写(有损)。files_edit_line 按行号 定位,op=replace/insert/delete,支持: - 多条 edit 从后往前应用(前面行号不漂移) - expect 安全闸:行内容与预期不符则整体拒绝(防行号过期改错位置) - CRLF/末尾换行保真 skill 同步补 edit_line 行。测试 5 条(含 expect 拒绝路径)。 ## 验证 hmapdev build 四工程全过:cmd/localuse/mcp/files 产物齐; go test files 全绿。
This commit is contained in:
428
example/mcp/plugin.go
Normal file
428
example/mcp/plugin.go
Normal file
@ -0,0 +1,428 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"strings"
|
||||
"sync"
|
||||
|
||||
sdk "github.com/JianFeeeee/homeagentsdk/sdk"
|
||||
)
|
||||
|
||||
// MCP 服务器配置(来自 config_mcp 表或 skill.json)
|
||||
type serverConfig struct {
|
||||
Name string `json:"name"`
|
||||
Command string `json:"command,omitempty"` // stdio 模式
|
||||
Args []string `json:"args,omitempty"`
|
||||
Env []string `json:"env,omitempty"`
|
||||
URL string `json:"url,omitempty"` // SSE 模式
|
||||
}
|
||||
|
||||
func NewPluginFactory(name string, config map[string]interface{}) (sdk.Plugin, error) {
|
||||
return New(name), nil
|
||||
}
|
||||
|
||||
type Plugin struct {
|
||||
name string
|
||||
servers []*Server
|
||||
configs []serverConfig
|
||||
sdk *sdk.PluginSDK
|
||||
mu sync.Mutex
|
||||
wg sync.WaitGroup
|
||||
}
|
||||
|
||||
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)
|
||||
p.sdk = s
|
||||
cfgs, err := p.loadConfig(s)
|
||||
if err != nil {
|
||||
return fmt.Errorf("load mcp config: %w", err)
|
||||
}
|
||||
p.configs = cfgs
|
||||
if len(cfgs) == 0 {
|
||||
log.Printf("[mcp] no servers configured, idle")
|
||||
return nil
|
||||
}
|
||||
|
||||
for _, cfg := range cfgs {
|
||||
if err := p.connectAndRegister(cfg); err != nil {
|
||||
log.Printf("[mcp] connect %s: %v", cfg.Name, err)
|
||||
}
|
||||
}
|
||||
|
||||
tDef := sdk.ToolDef{
|
||||
Name: "mcp_restart_server",
|
||||
Description: "重启 MCP 服务器连接。当 MCP 工具返回 pipe/transport closed 错误时,用此工具重启指定的 MCP 服务器。",
|
||||
Parameters: map[string]interface{}{
|
||||
"type": "object",
|
||||
"properties": map[string]interface{}{
|
||||
"name": map[string]interface{}{
|
||||
"type": "string",
|
||||
"description": "MCP 服务器名称(如 email)",
|
||||
},
|
||||
},
|
||||
"required": []string{"name"},
|
||||
},
|
||||
}
|
||||
if err := s.RegisterTool("mcp_restart_server", tDef, p.restartServerHandler); err != nil {
|
||||
log.Printf("[mcp] register restart tool: %v", err)
|
||||
}
|
||||
|
||||
// MCP 服务器动态管理工具
|
||||
addDef := sdk.ToolDef{
|
||||
Name: "mcp_add_server",
|
||||
Description: "动态添加并连接一个新的 MCP 服务器。支持 stdio 模式(指定 command)和 SSE 模式(指定 url)。添加后该服务器的所有工具立即可用。",
|
||||
Parameters: map[string]interface{}{
|
||||
"type": "object",
|
||||
"properties": map[string]interface{}{
|
||||
"name": map[string]interface{}{"type": "string", "description": "服务器名称(唯一标识,如 my-service)"},
|
||||
"command": map[string]interface{}{"type": "string", "description": "stdio 模式:可执行文件路径(如 npx)"},
|
||||
"args": map[string]interface{}{"type": "string", "description": "命令行参数,JSON 字符串数组(如 [\"-y\", \"@modelcontextprotocol/server-everything\"])"},
|
||||
"url": map[string]interface{}{"type": "string", "description": "SSE 模式:服务器 URL(如 https://api.example.com/mcp)"},
|
||||
"env": map[string]interface{}{"type": "string", "description": "环境变量,JSON 字符串对象(如 {\"KEY\": \"value\"})"},
|
||||
},
|
||||
"required": []string{"name"},
|
||||
},
|
||||
}
|
||||
s.RegisterTool("mcp_add_server", addDef, p.addServerHandler)
|
||||
|
||||
removeDef := sdk.ToolDef{
|
||||
Name: "mcp_remove_server",
|
||||
Description: "断开并移除一个已连接的 MCP 服务器。会关闭连接并清理注册的工具(部分清理在重启后完全生效)。",
|
||||
Parameters: map[string]interface{}{
|
||||
"type": "object",
|
||||
"properties": map[string]interface{}{
|
||||
"name": map[string]interface{}{"type": "string", "description": "要移除的 MCP 服务器名称"},
|
||||
},
|
||||
"required": []string{"name"},
|
||||
},
|
||||
}
|
||||
s.RegisterTool("mcp_remove_server", removeDef, p.removeServerHandler)
|
||||
|
||||
listDef := sdk.ToolDef{
|
||||
Name: "mcp_list_servers",
|
||||
Description: "列出所有已连接的 MCP 服务器及其工具。",
|
||||
Parameters: map[string]interface{}{
|
||||
"type": "object",
|
||||
"properties": map[string]interface{}{},
|
||||
},
|
||||
}
|
||||
s.RegisterTool("mcp_list_servers", listDef, p.listServersHandler)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *Plugin) loadConfig(s *sdk.PluginSDK) ([]serverConfig, error) {
|
||||
// 优先从独立服务器配置键读取(servers.<name>.<field>)
|
||||
keys, _ := s.Settings().List("servers.")
|
||||
if len(keys) > 0 {
|
||||
serverNames := make(map[string]bool)
|
||||
for _, k := range keys {
|
||||
parts := strings.SplitN(k, ".", 3)
|
||||
if len(parts) >= 2 {
|
||||
serverNames[parts[1]] = true
|
||||
}
|
||||
}
|
||||
var cfgs []serverConfig
|
||||
for name := range serverNames {
|
||||
cfg := serverConfig{Name: name}
|
||||
if v, _ := s.Settings().Get("servers." + name + ".command"); v != nil {
|
||||
if s, ok := v.(string); ok {
|
||||
cfg.Command = s
|
||||
}
|
||||
}
|
||||
if v, _ := s.Settings().Get("servers." + name + ".url"); v != nil {
|
||||
if s, ok := v.(string); ok {
|
||||
cfg.URL = s
|
||||
}
|
||||
}
|
||||
if v, _ := s.Settings().Get("servers." + name + ".args"); v != nil {
|
||||
if s, ok := v.(string); ok && s != "" {
|
||||
json.Unmarshal([]byte(s), &cfg.Args)
|
||||
}
|
||||
}
|
||||
if v, _ := s.Settings().Get("servers." + name + ".env"); v != nil {
|
||||
if s, ok := v.(string); ok && s != "" {
|
||||
json.Unmarshal([]byte(s), &cfg.Env)
|
||||
}
|
||||
}
|
||||
if cfg.Command != "" || cfg.URL != "" {
|
||||
cfgs = append(cfgs, cfg)
|
||||
}
|
||||
}
|
||||
if len(cfgs) > 0 {
|
||||
return cfgs, nil
|
||||
}
|
||||
}
|
||||
|
||||
// 回退:从旧版 JSON blob 读取
|
||||
raw, err := s.Settings().Get("servers")
|
||||
if err == nil {
|
||||
switch v := raw.(type) {
|
||||
case string:
|
||||
var cfgs []serverConfig
|
||||
if err := json.Unmarshal([]byte(v), &cfgs); err == nil && len(cfgs) > 0 {
|
||||
return cfgs, nil
|
||||
}
|
||||
case []interface{}:
|
||||
data, _ := json.Marshal(v)
|
||||
var cfgs []serverConfig
|
||||
if json.Unmarshal(data, &cfgs) == nil && len(cfgs) > 0 {
|
||||
return cfgs, nil
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
func (p *Plugin) connectServer(cfg serverConfig) (*Server, []MCPTool, error) {
|
||||
var transport Transport
|
||||
|
||||
if cfg.URL != "" {
|
||||
transport = NewSSETransport(cfg.URL)
|
||||
} else if cfg.Command != "" {
|
||||
var err error
|
||||
transport, err = NewStdioTransport(cfg.Command, cfg.Args, cfg.Env)
|
||||
if err != nil {
|
||||
return nil, nil, fmt.Errorf("stdio transport: %w", err)
|
||||
}
|
||||
} else {
|
||||
return nil, nil, fmt.Errorf("neither command nor url specified")
|
||||
}
|
||||
|
||||
server := NewServer(cfg.Name, transport)
|
||||
|
||||
tools, err := server.ListTools()
|
||||
if err != nil {
|
||||
transport.Close()
|
||||
return nil, nil, fmt.Errorf("list tools: %w", err)
|
||||
}
|
||||
|
||||
return server, tools, nil
|
||||
}
|
||||
|
||||
func (p *Plugin) makeHandler(server *Server, toolName string) sdk.ToolHandler {
|
||||
return func(args map[string]interface{}) (interface{}, error) {
|
||||
return server.CallTool(toolName, args)
|
||||
}
|
||||
}
|
||||
|
||||
func (p *Plugin) connectAndRegister(cfg serverConfig) error {
|
||||
server, tools, err := p.connectServer(cfg)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, tool := range tools {
|
||||
toolName := fmt.Sprintf("%s_%s", cfg.Name, tool.Name)
|
||||
tDef := sdk.ToolDef{
|
||||
Name: toolName,
|
||||
Description: fmt.Sprintf("[MCP/%s] %s", cfg.Name, tool.Description),
|
||||
Parameters: tool.InputSchema,
|
||||
}
|
||||
tHandler := p.makeHandler(server, tool.Name)
|
||||
if err := p.sdk.RegisterTool(toolName, tDef, tHandler); err != nil {
|
||||
log.Printf("[mcp] register tool %s: %v", toolName, err)
|
||||
continue
|
||||
}
|
||||
log.Printf("[mcp] registered tool: %s (%s)", toolName, cfg.Name)
|
||||
}
|
||||
|
||||
p.mu.Lock()
|
||||
p.servers = append(p.servers, server)
|
||||
p.mu.Unlock()
|
||||
log.Printf("[mcp] connected server: %s (%d tools)", cfg.Name, len(tools))
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *Plugin) restartServerHandler(args map[string]interface{}) (interface{}, error) {
|
||||
name, _ := args["name"].(string)
|
||||
if name == "" {
|
||||
return "参数 name 不能为空", nil
|
||||
}
|
||||
|
||||
p.mu.Lock()
|
||||
var idx int = -1
|
||||
for i, s := range p.servers {
|
||||
if s.Name() == name {
|
||||
idx = i
|
||||
break
|
||||
}
|
||||
}
|
||||
if idx == -1 {
|
||||
p.mu.Unlock()
|
||||
return fmt.Sprintf("MCP 服务器 [%s] 不存在", name), nil
|
||||
}
|
||||
oldServer := p.servers[idx]
|
||||
p.mu.Unlock()
|
||||
|
||||
var cfg *serverConfig
|
||||
for i := range p.configs {
|
||||
if p.configs[i].Name == name {
|
||||
cfg = &p.configs[i]
|
||||
break
|
||||
}
|
||||
}
|
||||
if cfg == nil {
|
||||
return fmt.Sprintf("MCP 服务器 [%s] 的配置未找到", name), nil
|
||||
}
|
||||
|
||||
transport, err := newTransport(*cfg)
|
||||
if err != nil {
|
||||
return fmt.Sprintf("创建 MCP 服务器 [%s] 传输层失败: %v", name, err), nil
|
||||
}
|
||||
|
||||
oldServer.SetTransport(transport)
|
||||
|
||||
_, err = oldServer.ListTools()
|
||||
if err != nil {
|
||||
return fmt.Sprintf("MCP 服务器 [%s] 重启后通信仍异常: %v", name, err), nil
|
||||
}
|
||||
|
||||
return fmt.Sprintf("MCP 服务器 [%s] 已成功重启", name), nil
|
||||
}
|
||||
|
||||
func (p *Plugin) addServerHandler(args map[string]interface{}) (interface{}, error) {
|
||||
name, _ := args["name"].(string)
|
||||
if name == "" {
|
||||
return "参数 name 不能为空", nil
|
||||
}
|
||||
|
||||
cfg := serverConfig{Name: name}
|
||||
if v, _ := args["command"].(string); v != "" {
|
||||
cfg.Command = v
|
||||
}
|
||||
if v, _ := args["url"].(string); v != "" {
|
||||
cfg.URL = v
|
||||
}
|
||||
if v, _ := args["args"].(string); v != "" {
|
||||
json.Unmarshal([]byte(v), &cfg.Args)
|
||||
}
|
||||
if v, _ := args["env"].(string); v != "" {
|
||||
// 支持 JSON 对象和 JSON 字符串数组两种格式
|
||||
var envObj map[string]string
|
||||
if err := json.Unmarshal([]byte(v), &envObj); err == nil {
|
||||
for k, val := range envObj {
|
||||
cfg.Env = append(cfg.Env, k+"="+val)
|
||||
}
|
||||
} else {
|
||||
json.Unmarshal([]byte(v), &cfg.Env)
|
||||
}
|
||||
}
|
||||
|
||||
if cfg.Command == "" && cfg.URL == "" {
|
||||
return "必须指定 command(stdio 模式)或 url(SSE 模式)", nil
|
||||
}
|
||||
|
||||
// 检查是否已存在同名服务器
|
||||
p.mu.Lock()
|
||||
for _, s := range p.servers {
|
||||
if s.Name() == name {
|
||||
p.mu.Unlock()
|
||||
return fmt.Sprintf("MCP 服务器 [%s] 已存在。如要重启请使用 mcp_restart_server,如要替换请先 mcp_remove_server", name), nil
|
||||
}
|
||||
}
|
||||
p.mu.Unlock()
|
||||
|
||||
if err := p.connectAndRegister(cfg); err != nil {
|
||||
return fmt.Sprintf("连接 MCP 服务器 [%s] 失败: %v", name, err), nil
|
||||
}
|
||||
|
||||
p.mu.Lock()
|
||||
p.configs = append(p.configs, cfg)
|
||||
p.mu.Unlock()
|
||||
|
||||
return fmt.Sprintf("MCP 服务器 [%s] 已成功连接并注册所有工具", name), nil
|
||||
}
|
||||
|
||||
func (p *Plugin) removeServerHandler(args map[string]interface{}) (interface{}, error) {
|
||||
name, _ := args["name"].(string)
|
||||
if name == "" {
|
||||
return "参数 name 不能为空", nil
|
||||
}
|
||||
|
||||
p.mu.Lock()
|
||||
var keptServers []*Server
|
||||
var removed bool
|
||||
for _, s := range p.servers {
|
||||
if s.Name() == name {
|
||||
s.Close()
|
||||
removed = true
|
||||
} else {
|
||||
keptServers = append(keptServers, s)
|
||||
}
|
||||
}
|
||||
p.servers = keptServers
|
||||
|
||||
var keptConfigs []serverConfig
|
||||
for _, c := range p.configs {
|
||||
if c.Name != name {
|
||||
keptConfigs = append(keptConfigs, c)
|
||||
}
|
||||
}
|
||||
p.configs = keptConfigs
|
||||
p.mu.Unlock()
|
||||
|
||||
if !removed {
|
||||
return fmt.Sprintf("MCP 服务器 [%s] 不存在", name), nil
|
||||
}
|
||||
|
||||
return fmt.Sprintf("MCP 服务器 [%s] 已断开连接。工具注册信息将在重启后完全清理。", name), nil
|
||||
}
|
||||
|
||||
func (p *Plugin) listServersHandler(args map[string]interface{}) (interface{}, error) {
|
||||
p.mu.Lock()
|
||||
defer p.mu.Unlock()
|
||||
|
||||
if len(p.servers) == 0 {
|
||||
return "没有已连接的 MCP 服务器。", nil
|
||||
}
|
||||
|
||||
var lines []string
|
||||
lines = append(lines, fmt.Sprintf("已连接的 MCP 服务器 (%d):", len(p.servers)))
|
||||
for _, s := range p.servers {
|
||||
tools, err := s.ListTools()
|
||||
var toolInfo string
|
||||
if err == nil && len(tools) > 0 {
|
||||
var names []string
|
||||
for _, t := range tools {
|
||||
names = append(names, t.Name)
|
||||
}
|
||||
toolInfo = strings.Join(names, ", ")
|
||||
} else if err != nil {
|
||||
toolInfo = fmt.Sprintf("(查询工具失败: %v)", err)
|
||||
} else {
|
||||
toolInfo = "(无工具)"
|
||||
}
|
||||
lines = append(lines, fmt.Sprintf(" %s: %s", s.Name(), toolInfo))
|
||||
}
|
||||
return strings.Join(lines, "\n"), nil
|
||||
}
|
||||
|
||||
func newTransport(cfg serverConfig) (Transport, error) {
|
||||
if cfg.URL != "" {
|
||||
return NewSSETransport(cfg.URL), nil
|
||||
}
|
||||
if cfg.Command != "" {
|
||||
return NewStdioTransport(cfg.Command, cfg.Args, cfg.Env)
|
||||
}
|
||||
return nil, fmt.Errorf("neither command nor url specified")
|
||||
}
|
||||
|
||||
func (p *Plugin) Stop() error {
|
||||
p.mu.Lock()
|
||||
defer p.mu.Unlock()
|
||||
for _, s := range p.servers {
|
||||
s.Close()
|
||||
}
|
||||
p.servers = nil
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user