AskaEth
86aef7b454
主人 2026-10-24：「能否将每个 MCP 服务器所承载的插件和工具按上游(来源)服务器分类
并与其他插件相对隔离？」
归属信息此前后端根本没有（GET /admin/plugins 每条只有 name/kind/description/两个
布尔），而 `mcp:` 是每台 server 一条的 KindMCP 条目的注册键，**不是**工具的命名空间
（工具名由 server 原样给出，连前缀都没有）⇒ 只能靠注册时的元数据，不能解析名字。
后端（唯一汇点盖章）：
- 新增 pkg/plugins/sdk/origin.go：Origin{Kind,Server,Transport} + 可选接口
OriginAware（没实现 = 宿主内置，老工具零改动）+ OriginOf（零值视为无来源）。
- registry.Capability 加 Origin 字段；RegisterTool 盖章（覆盖热装与 .env
MCP_SERVERS_JSON 两条接入路径的合流处），ReplaceTool **同样重新盖章**（它逐字段
重建条目，漏了就是「热替换把来源抹掉」）。
- mcp.mcpTool 实现 Origin()（server 名与传输都在 cfg 上），transport 用新的
transportNameOf 报**实际会用的**形态（stdio/sse/http），与 newTransport 同一推断。
- 网关 PluginSummary 加可选 origin（omitempty）：零值 ⇒ 键缺席，不是空对象。
- 卸载守卫：对单把 MCP 工具或 `mcp:<server>` 登记条目直接 400
（plugin_uninstall_scope）并指向「MCP 服务器」页 —— 原来会绕过插件管理器裸删，
server 的 names 账本不同步、重连后工具复活。判据是来源章，不是名字前缀。
前端（呈现分组）：
- admin/modules/pluginOrigin.ts：纯函数按 origin.server 分节（按 server 名排序），
transport 未报告时标题省略括号，缺 server 名不冒充内置。
- Plugins.tsx：工具段两级（`来自 fs（stdio）· N 把工具` + 宿主内置），MCP 工具行去掉
「卸载」改为指向 MCP 页；`kind:"mcp"` 登记条目与未知 kind（iot）各有段如实显示，
不再静默丢弃（design/09 §5.2.1 红线③）。
零影响（有断言）：tool_policies 表/策略写路径未改；ToolSchemas 不发来源（测试断言
模型面既无 origin 键也无 server 名）；停用 server ⇒ 条目连来源一起消失，无孤儿。
测试：registry 盖章/保章（两条均已实测变异变红）/内置零值/改策略不清章；mcp 注册
路径盖章 + transport 推断表；网关读面缺席与存在、热装整链、卸载守卫；前端 esbuild+cjs
逻辑测试 12 项。验收：go build/vet/test 全绿，frontend `npm run build` 通过。
112 lines
4.2 KiB
Go
112 lines
4.2 KiB
Go
package registry
|
||
|
||
import (
|
||
"context"
|
||
"testing"
|
||
|
||
"git.yeij.top/AskaEth/Cyrene_v2/pkg/plugins/sdk"
|
||
)
|
||
|
||
// 本文件钉住「来源章」（DevDocs/audit/MCP工具按上游分组-2026-10-24.md §2 方案①）：
|
||
// 归属只能来自注册时的元数据，因此盖章必须发生在**注册汇点**上，且热替换不能把它抹掉。
|
||
//
|
||
// 三条判据各自**能变红**：
|
||
// - 去掉 RegisterTool 里的盖章 → 第一条红；
|
||
// - 去掉 ReplaceTool 里的盖章（它逐字段重建条目）→ 第二条红；
|
||
// - 给所有工具都编一个章（比如按名字猜）→ 第三条红。
|
||
|
||
// originTool 是自报来源的最小工具（宿主内置工具不实现 sdk.OriginAware）。
|
||
type originTool struct {
|
||
name string
|
||
origin sdk.Origin
|
||
}
|
||
|
||
func (t originTool) Name() string { return t.name }
|
||
func (t originTool) Description() string { return "来自 MCP" }
|
||
func (t originTool) Parameters() (map[string]any, error) {
|
||
return map[string]any{"type": "object"}, nil
|
||
}
|
||
func (t originTool) Execute(_ context.Context, _ any) (any, error) {
|
||
return nil, nil
|
||
}
|
||
func (t originTool) Origin() sdk.Origin { return t.origin }
|
||
|
||
var _ sdk.OriginAware = originTool{}
|
||
|
||
// TestRegisterToolStampsOrigin：注册汇点把工具自报的来源盖在能力条目上。
|
||
func TestRegisterToolStampsOrigin(t *testing.T) {
|
||
r := New()
|
||
want := sdk.Origin{Kind: sdk.OriginKindMCP, Server: "fs", Transport: "stdio"}
|
||
if err := r.RegisterTool(originTool{name: "read_file_mcp", origin: want}); err != nil {
|
||
t.Fatalf("RegisterTool: %v", err)
|
||
}
|
||
cap, err := r.Get("read_file_mcp")
|
||
if err != nil {
|
||
t.Fatalf("Get: %v", err)
|
||
}
|
||
if cap.Origin != want {
|
||
t.Fatalf("条目上的来源章 = %+v, want %+v（RegisterTool 是唯一汇点，必须在这里盖章）", cap.Origin, want)
|
||
}
|
||
}
|
||
|
||
// TestReplaceToolKeepsOrigin：热替换必须重新盖章 —— 它逐字段重建 Capability，
|
||
// 漏一位就等于「面板保存一次，这把工具从它的 server 分节里掉进内置」。
|
||
func TestReplaceToolKeepsOrigin(t *testing.T) {
|
||
r := New()
|
||
origin := sdk.Origin{Kind: sdk.OriginKindMCP, Server: "fs", Transport: "http"}
|
||
if err := r.RegisterTool(originTool{name: "read_file_mcp", origin: origin}); err != nil {
|
||
t.Fatalf("RegisterTool: %v", err)
|
||
}
|
||
// 热替换成同名的新实现，来源相同（换实现不换 server）。
|
||
if err := r.ReplaceTool(originTool{name: "read_file_mcp", origin: origin}); err != nil {
|
||
t.Fatalf("ReplaceTool: %v", err)
|
||
}
|
||
cap, err := r.Get("read_file_mcp")
|
||
if err != nil {
|
||
t.Fatalf("Get: %v", err)
|
||
}
|
||
if cap.Origin != origin {
|
||
t.Fatalf("热替换后来源章 = %+v, want %+v（ReplaceTool 重建条目，必须重新盖章）", cap.Origin, origin)
|
||
}
|
||
}
|
||
|
||
// TestRegisterToolLeavesBuiltinOriginZero：宿主内置工具（没实现 OriginAware）
|
||
// 的条目**没有**来源章。它是「键缺席」的根据 —— 读面据此省略 origin，而不是编一个。
|
||
func TestRegisterToolLeavesBuiltinOriginZero(t *testing.T) {
|
||
r := New()
|
||
if err := r.RegisterTool(&stubTool{name: "web_search", desc: "内置"}); err != nil {
|
||
t.Fatalf("RegisterTool: %v", err)
|
||
}
|
||
cap, err := r.Get("web_search")
|
||
if err != nil {
|
||
t.Fatalf("Get: %v", err)
|
||
}
|
||
if !cap.Origin.IsZero() {
|
||
t.Fatalf("内置工具却带来源章 %+v：没有 OriginAware 就不得编一个来源", cap.Origin)
|
||
}
|
||
}
|
||
|
||
// TestSetPolicyKeepsOrigin：策略写路径不受来源章影响，也不许把它覆盖掉。
|
||
// 这是「tool_policies / 权限闸零影响」的断言 —— 策略仍按**名字**存，
|
||
// 来源章只在条目上，两者互不干涉。
|
||
func TestSetPolicyKeepsOrigin(t *testing.T) {
|
||
r := New()
|
||
origin := sdk.Origin{Kind: sdk.OriginKindMCP, Server: "fs", Transport: "stdio"}
|
||
if err := r.RegisterTool(originTool{name: "read_file_mcp", origin: origin}); err != nil {
|
||
t.Fatalf("RegisterTool: %v", err)
|
||
}
|
||
if err := r.SetAllowInGroup("read_file_mcp", true); err != nil {
|
||
t.Fatalf("SetAllowInGroup: %v", err)
|
||
}
|
||
cap, err := r.Get("read_file_mcp")
|
||
if err != nil {
|
||
t.Fatalf("Get: %v", err)
|
||
}
|
||
if !cap.Policy.AllowInGroup || cap.Policy.AllowInStranger {
|
||
t.Fatalf("策略 = %+v, want 只有 AllowInGroup=true（按名字写的策略行）", cap.Policy)
|
||
}
|
||
if cap.Origin != origin {
|
||
t.Fatalf("改策略把来源章冲掉了：%+v, want %+v", cap.Origin, origin)
|
||
}
|
||
}
|