AskaEth
70c9fba529
- capability/plugin/manager.go: Manager(Install/Uninstall/List/Restore/Close); Install=连接+发现+mcp:name注册工具,失败不落盘; holder持有*mcp.Registry; boot Restore重装
- config/runtimeconfig/plugins.go: PluginConfig(Normalize/Validate), LoadPlugins/SavePlugins(0600+tmp+rename)/Upsert/Delete/Masked(apiKey+敏感env掩码)
- mcp: MCPConfig加APIKey/Env, http/sse Bearer应用, stdio子进程env合并; registry.ToolNames()
- gateway: adminPluginInstall/Uninstall/ListInstalledPlugins; Deps.Plugins注入
- cmd/cyrene/plugins.go: wirePlugins(boot Restore+close)
- 契约: body{kind:mcp,name,config{transport, endpoint?/command?,args?,apiKey?,env?}} → {installed,name,kind,transport,toolCount}; 失败502不落盘/非法400/未注入501; 卸载连带unregister+删配置
- 测试(manager/plugins/admin/mcp_auth/组合根), build/vet/test绿+3平台交叉编译
- Docs/api-client v0.28 + api-reference install=501改真实; .gitignore加plugins.json
- 前端api.ts仍发旧body(无config)会400, 面板安装表单待后续
48 lines
1.3 KiB
Go
48 lines
1.3 KiB
Go
package mcp
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"testing"
|
|
)
|
|
|
|
func TestEnvWith_OverridesAndPreserves(t *testing.T) {
|
|
t.Parallel()
|
|
base := []string{"PATH=/usr/bin", "HOME=/root"}
|
|
got := envWith(base, map[string]string{"HOME": "/tmp", "MCP_TOKEN": "abc"})
|
|
if len(got) != 3 {
|
|
t.Fatalf("len = %d, want 3 (%v)", len(got), got)
|
|
}
|
|
seen := map[string]bool{}
|
|
for _, kv := range got {
|
|
seen[kv] = true
|
|
}
|
|
if !seen["PATH=/usr/bin"] || !seen["HOME=/tmp"] || !seen["MCP_TOKEN=abc"] {
|
|
t.Fatalf("merged env = %v", got)
|
|
}
|
|
if len(envWith(base, nil)) != len(base) {
|
|
t.Fatal("empty extra changed the base environment")
|
|
}
|
|
}
|
|
|
|
func TestHTTPTransport_SendsAPIKey(t *testing.T) {
|
|
t.Parallel()
|
|
got := make(chan string, 1)
|
|
srv := httpTestServer(t, func(w http.ResponseWriter, r *http.Request) {
|
|
got <- r.Header.Get("Authorization")
|
|
httpJSONResponse(w, r)
|
|
})
|
|
defer srv.Close()
|
|
tr := newHTTPTransport(MCPConfig{Name: "s", URL: srv.URL, APIKey: "sk-1234"}, defaultLogger())
|
|
defer tr.Close()
|
|
if err := tr.Start(context.Background()); err != nil {
|
|
t.Fatalf("Start: %v", err)
|
|
}
|
|
if err := tr.Call(context.Background(), "tools/list", nil, &listToolsResult{}); err != nil {
|
|
t.Fatalf("Call: %v", err)
|
|
}
|
|
if auth := <-got; auth != "Bearer sk-1234" {
|
|
t.Fatalf("Authorization = %q, want bearer key", auth)
|
|
}
|
|
}
|