Merge branch 'feat/m30-mcporigin-stamp-tool-origin-and-grou' into dev
This commit is contained in:
AskaEth committed
commit
679c8abf1d
16 files changed
+1241
-48
No files matched your search
@@ -7,12 +7,28 @@ import "context"
|
||||
const backthinkListLimit = 50
|
||||
|
||||
// PluginSummary 是管理列表中一个能力/插件的线上结构。
|
||||
//
|
||||
// Origin（v0.83 契约增量）回答「这条能力来自哪台上游 server」，供面板把 MCP 工具的
|
||||
// 呈现按 server 分节（DevDocs/audit/MCP工具按上游分组-2026-10-24.md §2 方案①）。
|
||||
// 它是**可选键**：宿主内置能力根本不实现来源，因此该键整体**缺席**（不是 `{}` 也不是
|
||||
// `null`）。归属来自注册时的元数据，绝不解析 `name` 推断 —— MCP 工具名里没有任何前缀。
|
||||
// 老客户端忽略未知键即零改动；严格解析器需要容忍/声明该可选字段。
|
||||
type PluginSummary struct {
|
||||
Name string `json:"name"`
|
||||
Kind string `json:"kind"`
|
||||
Description string `json:"description"`
|
||||
AllowInGroup bool `json:"allowInGroup"`
|
||||
AllowInStranger bool `json:"allowInStranger"`
|
||||
Name string `json:"name"`
|
||||
Kind string `json:"kind"`
|
||||
Description string `json:"description"`
|
||||
AllowInGroup bool `json:"allowInGroup"`
|
||||
AllowInStranger bool `json:"allowInStranger"`
|
||||
Origin *PluginOrigin `json:"origin,omitempty"`
|
||||
}
|
||||
|
||||
// PluginOrigin 是一条能力的上游来源。Server 是 server 名（= plugins.json /
|
||||
// MCP_SERVERS_JSON 的 name），Transport 是 stdio / sse / http；两者按 src 的实际
|
||||
// 已知程度省略，Kind 恒有值（目前只有 "mcp"）。
|
||||
type PluginOrigin struct {
|
||||
Kind string `json:"kind"`
|
||||
Server string `json:"server,omitempty"`
|
||||
Transport string `json:"transport,omitempty"`
|
||||
}
|
||||
|
||||
// ChannelSummary 是一个已注册渠道适配器的线上结构。Accounts
|
||||
|
||||
@@ -26,6 +26,7 @@ import (
|
||||
capreg "git.yeij.top/AskaEth/Cyrene_v2/internal/capability/registry"
|
||||
"git.yeij.top/AskaEth/Cyrene_v2/internal/config/runtimeconfig"
|
||||
apperr "git.yeij.top/AskaEth/Cyrene_v2/internal/errors"
|
||||
"git.yeij.top/AskaEth/Cyrene_v2/pkg/plugins/sdk"
|
||||
)
|
||||
|
||||
// pluginInstallTimeout 限定安装请求所等待的 connect+discover 握手
|
||||
@@ -208,6 +209,18 @@ func (s *Server) adminPluginUninstall(w http.ResponseWriter, r *http.Request) {
|
||||
writeError(w, apperr.Wrap(apperr.ErrInternal, "plugins_unavailable", "capability registry is not configured"))
|
||||
return
|
||||
}
|
||||
// 🔴 单把 MCP 工具与 `mcp:<server>` 登记条目**不能**走这条裸删路径
|
||||
// （DevDocs/audit/MCP工具按上游分组-2026-10-24.md §1.5-2）：管理器
|
||||
// Uninstall 只按 server 名/ID 匹配，落到这里就等于绕过插件管理器直接改注册表 ——
|
||||
// 该 server 的 names 账本不同步（下次 refresh 会把工具当「新出现的」再注册回来，
|
||||
// 或让 KindMCP 条目与工具各自消失一半）。判据是**注册时的来源章**，不是名字前缀。
|
||||
if cap, err := reg.Get(name); err == nil {
|
||||
if reason := s.serverScopedUninstall(name, cap); reason != "" {
|
||||
s.auditClaims(r, "plugin.uninstall", "error")
|
||||
writeError(w, apperr.Wrap(apperr.ErrBadRequest, "plugin_uninstall_scope", "%s", reason))
|
||||
return
|
||||
}
|
||||
}
|
||||
if err := reg.Remove(name); err != nil {
|
||||
s.auditClaims(r, "plugin.uninstall", "error")
|
||||
if errors.Is(err, capreg.ErrNotFound) {
|
||||
@@ -221,6 +234,59 @@ func (s *Server) adminPluginUninstall(w http.ResponseWriter, r *http.Request) {
|
||||
writeJSON(w, http.StatusOK, map[string]any{"name": name, "uninstalled": true})
|
||||
}
|
||||
|
||||
// serverScopedUninstall 报告「这个名字必须按整台 server 处理、不能在这里单删」的原因；
|
||||
// 返回空串 = 可以按能力名删除（技能 / 普通工具 / 渠道）。
|
||||
//
|
||||
// 它拦住两类真实事故（同一条根因：绕开插件管理器裸删）：
|
||||
// - 一把 MCP 工具：server 的 names 账本不同步，重连/refresh 后工具复活，用户以为卸了；
|
||||
// - `mcp:<server>` 登记条目：server 条目消失了而它的工具还在，注册表里留下无法归属的孤儿。
|
||||
//
|
||||
// 文案按**接入路径**分流，因为两条路的入口不同（owner 2026-10-24 复审 M30-p2-1）：
|
||||
// - 由插件管理器托管（落在 plugins.json）⇒ 「MCP 服务器」页确实能启停/卸载它；
|
||||
// - 由 `.env` 的 MCP_SERVERS_JSON 在启动期接入（cmd/cyrene/mcp.go，**不经** Manager）
|
||||
// ⇒「MCP 服务器」页只列 manager 托管的那批，**看不到它**，指向那页就是假指引；
|
||||
// 必须让主人去改 .env 并重启。
|
||||
//
|
||||
// 判据是**管理器持久清单里有没有这台 server**，不是名字前缀/传输推断出来的「路径猜测」。
|
||||
func (s *Server) serverScopedUninstall(name string, cap capreg.Capability) string {
|
||||
server := cap.Origin.Server
|
||||
if cap.Kind == capreg.KindMCP {
|
||||
if named, ok := pluginmgr.MCPServerName(name); ok {
|
||||
server = named
|
||||
}
|
||||
return fmt.Sprintf("%q 是一台 MCP server 的登记条目；%s", name, s.uninstallPointer(server))
|
||||
}
|
||||
if cap.Origin.Kind == sdk.OriginKindMCP {
|
||||
return fmt.Sprintf("%q 是 MCP server %q 的工具，单把工具不能单独卸载；%s", name, server, s.uninstallPointer(server))
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
// uninstallPointer 返回「这台 server 该去哪里启停/卸载」的**可操作**话术。
|
||||
func (s *Server) uninstallPointer(server string) string {
|
||||
if s.pluginManagerHasServer(server) {
|
||||
return "请在「MCP 服务器」页启停或卸载该 server"
|
||||
}
|
||||
return "该 server 由 .env 的 MCP_SERVERS_JSON 在启动时接入（不在 plugins.json 里，后台面板不管理它）——" +
|
||||
"要停用或移除，请改 .env 里的 MCP_SERVERS_JSON（把对应的 server 删掉或设 enabled=false），改完需重启生效"
|
||||
}
|
||||
|
||||
// pluginManagerHasServer 报告该 server 名是否由插件管理器托管（= 落在 plugins.json，
|
||||
// 因此「MCP 服务器」页看得见它）。`.env` 的 MCP_SERVERS_JSON 接入的 server 不经过
|
||||
// Manager，这里如实回答 false。没有管理器（测试/未接线）时也回答 false —— 与
|
||||
// 「面板看不见它」的事实一致，不给一个到不了的指引。
|
||||
func (s *Server) pluginManagerHasServer(server string) bool {
|
||||
if s.plugins == nil || server == "" {
|
||||
return false
|
||||
}
|
||||
for _, it := range s.plugins.List() {
|
||||
if it.Config.Name == server {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// adminListInstalledPlugins 列出运行时安装的插件，附其脱敏配置
|
||||
// 与每 server 工具数。它是读端点（owner/admin），
|
||||
// 绝不原样返回 API key 或敏感 env 值。
|
||||
|
||||
@@ -39,6 +39,10 @@ func (s *Server) adminListMemories(w http.ResponseWriter, r *http.Request) {
|
||||
}
|
||||
|
||||
// adminListPlugins 列出所有已注册能力（管理读）。
|
||||
//
|
||||
// 每条能力带上它的上游来源（若有）：来源章的读取面在这里一次性给出，
|
||||
// 前端「插件与能力」页据此把工具按 server 分节。**策略字段与集合完全不变** ——
|
||||
// 只是多了一个可选键，权限判定与模型可见的工具面都不读它。
|
||||
func (s *Server) adminListPlugins(w http.ResponseWriter, r *http.Request) {
|
||||
out := make([]PluginSummary, 0)
|
||||
if s.caps != nil {
|
||||
@@ -49,12 +53,26 @@ func (s *Server) adminListPlugins(w http.ResponseWriter, r *http.Request) {
|
||||
Description: cap.Description,
|
||||
AllowInGroup: cap.Policy.AllowInGroup,
|
||||
AllowInStranger: cap.Policy.AllowInStranger,
|
||||
Origin: pluginOriginOf(cap),
|
||||
})
|
||||
}
|
||||
}
|
||||
writeJSON(w, http.StatusOK, out)
|
||||
}
|
||||
|
||||
// pluginOriginOf 把能力条目上的来源章映射为线上结构；零值（宿主内置 / 来源未知）
|
||||
// 返回 nil，使 JSON 里**没有** origin 键 —— 缺席是「宿主内置」的表达。
|
||||
func pluginOriginOf(cap capreg.Capability) *PluginOrigin {
|
||||
if cap.Origin.IsZero() {
|
||||
return nil
|
||||
}
|
||||
return &PluginOrigin{
|
||||
Kind: cap.Origin.Kind,
|
||||
Server: cap.Origin.Server,
|
||||
Transport: cap.Origin.Transport,
|
||||
}
|
||||
}
|
||||
|
||||
// adminListChannels 列出已注册的渠道适配器（管理读）。当接线了
|
||||
// 渠道配置存储时，它还会附上各平台已配置的账号，并做脱敏，
|
||||
// 确保原始 token 绝不抵达客户端。
|
||||
|
||||
@@ -0,0 +1,331 @@
|
||||
package gateway
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
capreg "git.yeij.top/AskaEth/Cyrene_v2/internal/capability/registry"
|
||||
"git.yeij.top/AskaEth/Cyrene_v2/pkg/plugins/sdk"
|
||||
)
|
||||
|
||||
// 本文件钉住「插件与能力」读面的**来源章**（DevDocs/audit/MCP工具按上游分组-2026-10-24.md
|
||||
// §2 方案① / §6 契约增量），以及与之配套的服务端拒绝：
|
||||
//
|
||||
// - 来自 MCP server 的工具带 `origin:{kind,server,transport}`；
|
||||
// - 宿主内置工具的该键**缺席**（不是 `{}`、不是 `null`）—— 缺席才表示「没有上游」；
|
||||
// - 只改呈现：`allowInGroup`/`allowInStranger` 逐字不变，`kind` 仍是 `tool`；
|
||||
// - 对一把 MCP 工具调 `POST /admin/plugins/uninstall` 必须 400（绕开管理器裸删会让
|
||||
// server 的 names 账本不同步），而普通能力的卸载行为不变。
|
||||
|
||||
// originStubTool 是自报来源的最小工具。
|
||||
type originStubTool struct {
|
||||
name string
|
||||
origin sdk.Origin
|
||||
}
|
||||
|
||||
func (t originStubTool) Name() string { return t.name }
|
||||
func (t originStubTool) Description() string { return "MCP 工具" }
|
||||
func (t originStubTool) Parameters() (map[string]any, error) {
|
||||
return map[string]any{"type": "object"}, nil
|
||||
}
|
||||
func (t originStubTool) Execute(_ context.Context, _ any) (any, error) {
|
||||
return nil, nil
|
||||
}
|
||||
func (t originStubTool) Origin() sdk.Origin { return t.origin }
|
||||
|
||||
// builtinStubTool 是宿主内置工具（不实现 sdk.OriginAware）。
|
||||
type builtinStubTool struct{ name string }
|
||||
|
||||
func (t builtinStubTool) Name() string { return t.name }
|
||||
func (t builtinStubTool) Description() string { return "内置工具" }
|
||||
func (t builtinStubTool) Parameters() (map[string]any, error) {
|
||||
return map[string]any{"type": "object"}, nil
|
||||
}
|
||||
func (t builtinStubTool) Execute(_ context.Context, _ any) (any, error) {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
// listPlugins 取管理读面的能力清单（保留原始 JSON 以便断言键的**缺席**）。
|
||||
func listPlugins(t *testing.T, url, token string) []byte {
|
||||
t.Helper()
|
||||
resp, raw := doReq(t, http.MethodGet, url+"/api/v1/admin/plugins", nil, token)
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
t.Fatalf("plugins list status = %d, body = %s", resp.StatusCode, raw)
|
||||
}
|
||||
return raw
|
||||
}
|
||||
|
||||
// TestAdminListPluginsExposesOrigin 是读面的主判据：MCP 工具带章、内置工具缺席。
|
||||
func TestAdminListPluginsExposesOrigin(t *testing.T) {
|
||||
t.Parallel()
|
||||
env := newPluginTestEnv(t, nil)
|
||||
mcpOrigin := sdk.Origin{Kind: sdk.OriginKindMCP, Server: "fs", Transport: "stdio"}
|
||||
if err := env.caps.RegisterTool(originStubTool{name: "read_file_mcp", origin: mcpOrigin}); err != nil {
|
||||
t.Fatalf("register mcp tool: %v", err)
|
||||
}
|
||||
if err := env.caps.RegisterTool(builtinStubTool{name: "web_search"}); err != nil {
|
||||
t.Fatalf("register builtin tool: %v", err)
|
||||
}
|
||||
|
||||
raw := listPlugins(t, env.ts.URL, env.token)
|
||||
var list struct {
|
||||
Data []PluginSummary `json:"data"`
|
||||
}
|
||||
if err := json.Unmarshal(raw, &list); err != nil {
|
||||
t.Fatalf("decode list: %v (%s)", err, raw)
|
||||
}
|
||||
byName := make(map[string]PluginSummary, len(list.Data))
|
||||
for _, it := range list.Data {
|
||||
byName[it.Name] = it
|
||||
}
|
||||
mcpCap, ok := byName["read_file_mcp"]
|
||||
if !ok {
|
||||
t.Fatalf("MCP 工具不在清单里：%s", raw)
|
||||
}
|
||||
if mcpCap.Origin == nil {
|
||||
t.Fatalf("MCP 工具没有 origin：%+v", mcpCap)
|
||||
}
|
||||
if *mcpCap.Origin != (PluginOrigin{Kind: "mcp", Server: "fs", Transport: "stdio"}) {
|
||||
t.Fatalf("origin = %+v, want {mcp fs stdio}", *mcpCap.Origin)
|
||||
}
|
||||
// 呈现层新增不改既有语义：kind 与两个策略布尔逐字不变。
|
||||
if mcpCap.Kind != "tool" || mcpCap.AllowInGroup || mcpCap.AllowInStranger {
|
||||
t.Fatalf("既有字段被改了：%+v", mcpCap)
|
||||
}
|
||||
builtin, ok := byName["web_search"]
|
||||
if !ok {
|
||||
t.Fatalf("内置工具不在清单里：%s", raw)
|
||||
}
|
||||
if builtin.Origin != nil {
|
||||
t.Fatalf("内置工具带上了 origin：%+v", builtin.Origin)
|
||||
}
|
||||
// 🔴 线上必须是「键缺席」而不是空对象：`"origin":{}` / `"origin":null` 都会被
|
||||
// 前端/严格解析器读成「有来源但内容为空」。
|
||||
entry := rawEntryOf(t, raw, "web_search")
|
||||
if strings.Contains(entry, "origin") {
|
||||
t.Fatalf("内置工具的条目里出现了 origin 键：%s", entry)
|
||||
}
|
||||
}
|
||||
|
||||
// rawEntryOf 从原始响应里截出某个名为 name 的条目的 JSON 片段，用于断言键的缺席
|
||||
// （反序列化后的结构体无法区分「缺键」与「零值」）。
|
||||
func rawEntryOf(t *testing.T, raw []byte, name string) string {
|
||||
t.Helper()
|
||||
var items []json.RawMessage
|
||||
var envelope struct {
|
||||
Data []json.RawMessage `json:"data"`
|
||||
}
|
||||
if err := json.Unmarshal(raw, &envelope); err == nil {
|
||||
items = envelope.Data
|
||||
} else if err := json.Unmarshal(raw, &items); err != nil {
|
||||
t.Fatalf("decode raw list: %v (%s)", err, raw)
|
||||
}
|
||||
for _, item := range items {
|
||||
var probe struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
if err := json.Unmarshal(item, &probe); err != nil {
|
||||
t.Fatalf("decode item: %v (%s)", err, item)
|
||||
}
|
||||
if probe.Name == name {
|
||||
return string(item)
|
||||
}
|
||||
}
|
||||
t.Fatalf("条目 %q 不在清单里：%s", name, raw)
|
||||
return ""
|
||||
}
|
||||
|
||||
// TestAdminPluginUninstallRejectsMCPTool 是那条 high finding 的服务端一半：
|
||||
// 单把 MCP 工具不能绕过插件管理器裸删（消息该指向哪，按接入路径另有专门用例）。
|
||||
func TestAdminPluginUninstallRejectsMCPTool(t *testing.T) {
|
||||
t.Parallel()
|
||||
env := newPluginTestEnv(t, nil)
|
||||
origin := sdk.Origin{Kind: sdk.OriginKindMCP, Server: "fs", Transport: "stdio"}
|
||||
if err := env.caps.RegisterTool(originStubTool{name: "read_file_mcp", origin: origin}); err != nil {
|
||||
t.Fatalf("register mcp tool: %v", err)
|
||||
}
|
||||
|
||||
resp, raw := doReq(t, http.MethodPost, env.ts.URL+"/api/v1/admin/plugins/uninstall",
|
||||
map[string]any{"name": "read_file_mcp"}, env.token)
|
||||
if resp.StatusCode != http.StatusBadRequest {
|
||||
t.Fatalf("status = %d, want 400（单把 MCP 工具不能在这里卸载），body = %s", resp.StatusCode, raw)
|
||||
}
|
||||
if !strings.Contains(string(raw), "plugin_uninstall_scope") {
|
||||
t.Fatalf("错误码不是 plugin_uninstall_scope：%s", raw)
|
||||
}
|
||||
if _, err := env.caps.Get("read_file_mcp"); err != nil {
|
||||
t.Fatalf("被拒的卸载却把工具删了：%v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// installMCPPlugin 通过 install 端点装一台 server（写进 plugins.json）。
|
||||
// 用于「这台 server 由插件管理器托管」的路径判定。
|
||||
func installMCPPlugin(t *testing.T, env *pluginTestEnv, name string) {
|
||||
t.Helper()
|
||||
resp, raw := doReq(t, http.MethodPost, env.ts.URL+"/api/v1/admin/plugins/install", map[string]any{
|
||||
"kind": "mcp",
|
||||
"name": name,
|
||||
"config": map[string]any{"transport": "http", "endpoint": env.mcpURL},
|
||||
}, env.token)
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
t.Fatalf("install %q status = %d, body = %s", name, resp.StatusCode, raw)
|
||||
}
|
||||
}
|
||||
|
||||
// TestAdminPluginUninstallMCPToolPointsToPluginPage：server 在 plugins.json 里
|
||||
// ⇒ 文案指向「MCP 服务器」页（那页确实能卸载它），**不**提 .env。
|
||||
func TestAdminPluginUninstallMCPToolPointsToPluginPage(t *testing.T) {
|
||||
t.Parallel()
|
||||
env := newPluginTestEnv(t, newFakeMCPHandler("", "add"))
|
||||
installMCPPlugin(t, env, "calc")
|
||||
|
||||
resp, raw := doReq(t, http.MethodPost, env.ts.URL+"/api/v1/admin/plugins/uninstall",
|
||||
map[string]any{"name": "add"}, env.token)
|
||||
if resp.StatusCode != http.StatusBadRequest {
|
||||
t.Fatalf("status = %d, want 400, body = %s", resp.StatusCode, raw)
|
||||
}
|
||||
if !strings.Contains(string(raw), "plugin_uninstall_scope") {
|
||||
t.Fatalf("错误码不是 plugin_uninstall_scope：%s", raw)
|
||||
}
|
||||
if !strings.Contains(string(raw), "MCP 服务器") {
|
||||
t.Fatalf("plugins.json 装出来的 server 应指向「MCP 服务器」页：%s", raw)
|
||||
}
|
||||
if strings.Contains(string(raw), "MCP_SERVERS_JSON") {
|
||||
t.Fatalf("plugins.json 路径不该提 .env 的 MCP_SERVERS_JSON：%s", raw)
|
||||
}
|
||||
if _, err := env.caps.Get("add"); err != nil {
|
||||
t.Fatalf("被拒的卸载却把工具删了：%v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// TestAdminPluginUninstallMCPToolPointsToEnvPath：server **不在** plugins.json
|
||||
// ⇒ 它是 `.env` 的 MCP_SERVERS_JSON 启动期接入的，而「MCP 服务器」页只列 manager
|
||||
// 托管的那批（看不见它）⇒ 文案必须让主人去改 .env，且**不得**指向那页。
|
||||
func TestAdminPluginUninstallMCPToolPointsToEnvPath(t *testing.T) {
|
||||
t.Parallel()
|
||||
env := newPluginTestEnv(t, nil)
|
||||
origin := sdk.Origin{Kind: sdk.OriginKindMCP, Server: "envcalc", Transport: "stdio"}
|
||||
if err := env.caps.RegisterTool(originStubTool{name: "env_add", origin: origin}); err != nil {
|
||||
t.Fatalf("register env-path tool: %v", err)
|
||||
}
|
||||
|
||||
resp, raw := doReq(t, http.MethodPost, env.ts.URL+"/api/v1/admin/plugins/uninstall",
|
||||
map[string]any{"name": "env_add"}, env.token)
|
||||
if resp.StatusCode != http.StatusBadRequest {
|
||||
t.Fatalf("status = %d, want 400, body = %s", resp.StatusCode, raw)
|
||||
}
|
||||
if !strings.Contains(string(raw), "plugin_uninstall_scope") {
|
||||
t.Fatalf("错误码不是 plugin_uninstall_scope：%s", raw)
|
||||
}
|
||||
if !strings.Contains(string(raw), "MCP_SERVERS_JSON") {
|
||||
t.Fatalf(".env 接入的 server 应指向 MCP_SERVERS_JSON：%s", raw)
|
||||
}
|
||||
if strings.Contains(string(raw), "MCP 服务器") {
|
||||
t.Fatalf(".env 接入的 server 不得指向「MCP 服务器」页（那页看不到它）：%s", raw)
|
||||
}
|
||||
if _, err := env.caps.Get("env_add"); err != nil {
|
||||
t.Fatalf("被拒的卸载却把工具删了：%v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// TestAdminPluginUninstallMCPEntryPointsToEnvPath：`mcp:<server>` 登记条目同样要按
|
||||
// 接入路径给指引 —— env 路径下也**不得**指向「MCP 服务器」页。
|
||||
func TestAdminPluginUninstallMCPEntryPointsToEnvPath(t *testing.T) {
|
||||
t.Parallel()
|
||||
env := newPluginTestEnv(t, nil)
|
||||
if err := env.caps.Register(capreg.Capability{
|
||||
Name: "mcp:envcalc", Kind: capreg.KindMCP, Description: "MCP server envcalc", Value: struct{}{},
|
||||
}); err != nil {
|
||||
t.Fatalf("register mcp entry: %v", err)
|
||||
}
|
||||
|
||||
resp, raw := doReq(t, http.MethodPost, env.ts.URL+"/api/v1/admin/plugins/uninstall",
|
||||
map[string]any{"name": "mcp:envcalc"}, env.token)
|
||||
if resp.StatusCode != http.StatusBadRequest {
|
||||
t.Fatalf("status = %d, want 400, body = %s", resp.StatusCode, raw)
|
||||
}
|
||||
if !strings.Contains(string(raw), "plugin_uninstall_scope") {
|
||||
t.Fatalf("错误码不是 plugin_uninstall_scope：%s", raw)
|
||||
}
|
||||
if !strings.Contains(string(raw), "MCP_SERVERS_JSON") {
|
||||
t.Fatalf(".env 接入的登记条目应指向 MCP_SERVERS_JSON：%s", raw)
|
||||
}
|
||||
if strings.Contains(string(raw), "MCP 服务器") {
|
||||
t.Fatalf(".env 接入的登记条目不得指向「MCP 服务器」页：%s", raw)
|
||||
}
|
||||
if _, err := env.caps.Get("mcp:envcalc"); err != nil {
|
||||
t.Fatalf("被拒的卸载却把登记条目删了：%v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// TestAdminPluginUninstallStillRemovesPlainCapability：普通能力（技能/内置工具）
|
||||
// 的卸载路径逐字不变 —— 拒绝只针对「有 MCP 来源」的条目。
|
||||
func TestAdminPluginUninstallStillRemovesPlainCapability(t *testing.T) {
|
||||
t.Parallel()
|
||||
env := newPluginTestEnv(t, nil)
|
||||
if err := env.caps.RegisterTool(builtinStubTool{name: "web_search"}); err != nil {
|
||||
t.Fatalf("register builtin tool: %v", err)
|
||||
}
|
||||
resp, raw := doReq(t, http.MethodPost, env.ts.URL+"/api/v1/admin/plugins/uninstall",
|
||||
map[string]any{"name": "web_search"}, env.token)
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
t.Fatalf("status = %d, want 200, body = %s", resp.StatusCode, raw)
|
||||
}
|
||||
if _, err := env.caps.Get("web_search"); err == nil {
|
||||
t.Fatal("普通能力没有被卸载")
|
||||
}
|
||||
}
|
||||
|
||||
// TestInstalledMCPServerToolsCarryOrigin 走**热装整链**（install 端点 → 插件管理器 →
|
||||
// mcp.Registry → RegisterTool）：装完那台 server 的工具在管理读面上带 origin，
|
||||
// 而按 server 名卸载后条目**连 origin 一起消失**，不留孤儿。
|
||||
//
|
||||
// 它与 `internal/capability/mcp` 的盖章用例互补：那条钉的是两条接入路径的合流处
|
||||
// （RegisterTool），这条钉的是运行时这条路径真的走到了那里。
|
||||
func TestInstalledMCPServerToolsCarryOrigin(t *testing.T) {
|
||||
t.Parallel()
|
||||
env := newPluginTestEnv(t, newFakeMCPHandler("", "add", "mul"))
|
||||
resp, raw := doReq(t, http.MethodPost, env.ts.URL+"/api/v1/admin/plugins/install", map[string]any{
|
||||
"kind": "mcp",
|
||||
"name": "calc",
|
||||
"config": map[string]any{"transport": "http", "endpoint": env.mcpURL},
|
||||
}, env.token)
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
t.Fatalf("install status = %d, body = %s", resp.StatusCode, raw)
|
||||
}
|
||||
|
||||
var list struct {
|
||||
Data []PluginSummary `json:"data"`
|
||||
}
|
||||
if err := json.Unmarshal(listPlugins(t, env.ts.URL, env.token), &list); err != nil {
|
||||
t.Fatalf("decode list: %v", err)
|
||||
}
|
||||
want := PluginOrigin{Kind: "mcp", Server: "calc", Transport: "http"}
|
||||
tools := 0
|
||||
for _, it := range list.Data {
|
||||
if it.Kind != "tool" {
|
||||
continue
|
||||
}
|
||||
tools++
|
||||
if it.Origin == nil || *it.Origin != want {
|
||||
t.Fatalf("工具 %q 的 origin = %+v, want %+v", it.Name, it.Origin, want)
|
||||
}
|
||||
}
|
||||
if tools != 2 {
|
||||
t.Fatalf("工具数 = %d, want 2（add/mul）", tools)
|
||||
}
|
||||
|
||||
// 按 server 名卸载（这是**合法**路径，管理器会连工具带账本一起收），
|
||||
// 卸载后工具条目连 origin 一起消失 —— 「停用 server ⇒ 工具消失」语义不变。
|
||||
resp, raw = doReq(t, http.MethodPost, env.ts.URL+"/api/v1/admin/plugins/uninstall",
|
||||
map[string]any{"name": "calc"}, env.token)
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
t.Fatalf("uninstall status = %d, body = %s", resp.StatusCode, raw)
|
||||
}
|
||||
if left := env.caps.List(capreg.KindTool); len(left) != 0 {
|
||||
t.Fatalf("卸载 server 后还剩 %d 把工具（不应留下带 origin 的孤儿）", len(left))
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user