feat: add Cobra CLI, LSP/MCP registries, workflow engine, and enriched dashboard
Major changes: - Refactor CLI entry point to Cobra commands (root, setup, scan, doctor, install, update, lsp, mcp, skills, config, version) - Add LSP registry with health checks, auto-install, and editor config generation - Add MCP registry with editor detection, status tracking, and per-editor configuration - Add workflow engine with planner and step execution for automated task chains - Add conversation search, export (Markdown/JSON), and detailed token counting - Add streaming shell chat handler with tool call/result events - Add skill validation, dry-run testing, and export endpoints - Enrich dashboard with Tools/Activity/Status tabs and tool cards grid - Add PRD documentation - Complete i18n for both EN and FR 💘 Generated with Crush Assisted-by: GLM-5.1 via Crush <crush@charm.land>
This commit is contained in:
Augustin
@@ -6,17 +6,22 @@ import (
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/muyue/muyue/internal/config"
|
||||
)
|
||||
|
||||
type MCPServer struct {
|
||||
Name string `json:"name"`
|
||||
Command string `json:"command"`
|
||||
Args []string `json:"args"`
|
||||
Env map[string]string `json:"env,omitempty"`
|
||||
Installed bool `json:"installed"`
|
||||
Category string `json:"category"`
|
||||
Name string `json:"name"`
|
||||
Command string `json:"command"`
|
||||
Args []string `json:"args"`
|
||||
Env map[string]string `json:"env,omitempty"`
|
||||
Installed bool `json:"installed"`
|
||||
Category string `json:"category"`
|
||||
Description string `json:"description,omitempty"`
|
||||
Version string `json:"version,omitempty"`
|
||||
Status string `json:"status,omitempty"`
|
||||
}
|
||||
|
||||
type mcpEntry struct {
|
||||
@@ -47,10 +52,52 @@ func ScanServers() []MCPServer {
|
||||
servers[i] = s
|
||||
_, err := exec.LookPath(s.Command)
|
||||
servers[i].Installed = err == nil
|
||||
servers[i].Version = GetInstalledVersion(s.Name)
|
||||
}
|
||||
|
||||
regServers, err := scanRegistryServers()
|
||||
if err == nil {
|
||||
servers = append(servers, regServers...)
|
||||
}
|
||||
|
||||
return servers
|
||||
}
|
||||
|
||||
func scanRegistryServers() ([]MCPServer, error) {
|
||||
reg, err := LoadRegistry()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
knownNames := map[string]bool{}
|
||||
for _, s := range knownMCPServers {
|
||||
knownNames[s.Name] = true
|
||||
}
|
||||
|
||||
var servers []MCPServer
|
||||
for _, rs := range reg.Servers {
|
||||
if knownNames[rs.Name] {
|
||||
continue
|
||||
}
|
||||
servers = append(servers, MCPServer{
|
||||
Name: rs.Name,
|
||||
Command: rs.Command,
|
||||
Args: rs.Args,
|
||||
Env: rs.Env,
|
||||
Category: rs.Category,
|
||||
Description: rs.Description,
|
||||
Installed: isCommandAvailable(rs.Command),
|
||||
Version: GetInstalledVersion(rs.Name),
|
||||
})
|
||||
}
|
||||
return servers, nil
|
||||
}
|
||||
|
||||
func isCommandAvailable(cmd string) bool {
|
||||
_, err := exec.LookPath(cmd)
|
||||
return err == nil
|
||||
}
|
||||
|
||||
func getCoreEntries(homeDir string) []mcpEntry {
|
||||
return []mcpEntry{
|
||||
{"filesystem", "npx", []string{"-y", "@modelcontextprotocol/server-filesystem", filepath.Join(homeDir, "projects")}, nil},
|
||||
@@ -98,7 +145,8 @@ func writeMCPConfig(configPath string, mcpKey string, entries []mcpEntry) error
|
||||
"args": e.args,
|
||||
}
|
||||
if len(e.env) > 0 {
|
||||
entry["env"] = e.env
|
||||
resolved := ResolveEnv(e.env, nil)
|
||||
entry["env"] = resolved
|
||||
}
|
||||
mcpMap[e.name] = entry
|
||||
}
|
||||
@@ -110,7 +158,49 @@ func writeMCPConfig(configPath string, mcpKey string, entries []mcpEntry) error
|
||||
return err
|
||||
}
|
||||
|
||||
return os.WriteFile(configPath, out, 0600)
|
||||
if err := os.WriteFile(configPath, out, 0600); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return ValidateConfig(configPath)
|
||||
}
|
||||
|
||||
func writeMCPConfigForEditor(editor EditorConfig, entries []mcpEntry) error {
|
||||
configDir := filepath.Dir(editor.ConfigPath)
|
||||
if err := os.MkdirAll(configDir, 0700); err != nil {
|
||||
return fmt.Errorf("create config dir %s: %w", editor.Name, err)
|
||||
}
|
||||
|
||||
existing := map[string]interface{}{}
|
||||
data, err := os.ReadFile(editor.ConfigPath)
|
||||
if err == nil {
|
||||
_ = json.Unmarshal(data, &existing)
|
||||
}
|
||||
|
||||
mcpMap := map[string]interface{}{}
|
||||
for _, e := range entries {
|
||||
if editor.TransformCommand != nil {
|
||||
mcpMap[e.name] = editor.TransformCommand(e)
|
||||
} else {
|
||||
entry := map[string]interface{}{
|
||||
"command": e.cmd,
|
||||
"args": e.args,
|
||||
}
|
||||
if len(e.env) > 0 {
|
||||
entry["env"] = e.env
|
||||
}
|
||||
mcpMap[e.name] = entry
|
||||
}
|
||||
}
|
||||
|
||||
existing[editor.ConfigKey] = mcpMap
|
||||
|
||||
out, err := json.MarshalIndent(existing, "", " ")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return os.WriteFile(editor.ConfigPath, out, 0600)
|
||||
}
|
||||
|
||||
func GenerateCrushMCPConfig(cfg *config.MuyueConfig, homeDir string) error {
|
||||
@@ -140,19 +230,154 @@ func GenerateClaudeMCPConfig(cfg *config.MuyueConfig, homeDir string) error {
|
||||
return writeMCPConfig(configPath, "mcpServers", entries)
|
||||
}
|
||||
|
||||
func GenerateCursorMCPConfig(cfg *config.MuyueConfig, homeDir string) error {
|
||||
if homeDir == "" {
|
||||
home, _ := os.UserHomeDir()
|
||||
homeDir = home
|
||||
}
|
||||
|
||||
core := getCoreEntries(homeDir)
|
||||
entries := withProviderEntries(core, cfg, nil)
|
||||
editor := EditorConfig{
|
||||
Name: "cursor",
|
||||
ConfigPath: filepath.Join(homeDir, ".cursor", "mcp.json"),
|
||||
ConfigKey: "mcpServers",
|
||||
Format: "json",
|
||||
TransformCommand: func(e mcpEntry) interface{} {
|
||||
m := map[string]interface{}{
|
||||
"type": "stdio",
|
||||
"command": e.cmd,
|
||||
"args": e.args,
|
||||
}
|
||||
if len(e.env) > 0 {
|
||||
m["env"] = e.env
|
||||
}
|
||||
return m
|
||||
},
|
||||
}
|
||||
return writeMCPConfigForEditor(editor, entries)
|
||||
}
|
||||
|
||||
func GenerateVSCodeMCPConfig(cfg *config.MuyueConfig, homeDir string) error {
|
||||
if homeDir == "" {
|
||||
home, _ := os.UserHomeDir()
|
||||
homeDir = home
|
||||
}
|
||||
|
||||
core := getCoreEntries(homeDir)
|
||||
entries := withProviderEntries(core, cfg, nil)
|
||||
editor := EditorConfig{
|
||||
Name: "vscode",
|
||||
ConfigPath: filepath.Join(homeDir, ".vscode", "mcp.json"),
|
||||
ConfigKey: "servers",
|
||||
Format: "json",
|
||||
}
|
||||
return writeMCPConfigForEditor(editor, entries)
|
||||
}
|
||||
|
||||
func GenerateWindsurfMCPConfig(cfg *config.MuyueConfig, homeDir string) error {
|
||||
if homeDir == "" {
|
||||
home, _ := os.UserHomeDir()
|
||||
homeDir = home
|
||||
}
|
||||
|
||||
core := getCoreEntries(homeDir)
|
||||
entries := withProviderEntries(core, cfg, nil)
|
||||
editor := EditorConfig{
|
||||
Name: "windsurf",
|
||||
ConfigPath: filepath.Join(homeDir, ".windsurf", "mcp.json"),
|
||||
ConfigKey: "mcpServers",
|
||||
Format: "json",
|
||||
}
|
||||
return writeMCPConfigForEditor(editor, entries)
|
||||
}
|
||||
|
||||
func ConfigureAll(cfg *config.MuyueConfig) error {
|
||||
home, err := os.UserHomeDir()
|
||||
if err != nil {
|
||||
return fmt.Errorf("get home dir: %w", err)
|
||||
}
|
||||
|
||||
if err := GenerateCrushMCPConfig(cfg, home); err != nil {
|
||||
return fmt.Errorf("crush MCP config: %w", err)
|
||||
editors := []struct {
|
||||
name string
|
||||
fn func(*config.MuyueConfig, string) error
|
||||
}{
|
||||
{"crush", GenerateCrushMCPConfig},
|
||||
{"claude", GenerateClaudeMCPConfig},
|
||||
{"cursor", GenerateCursorMCPConfig},
|
||||
{"vscode", GenerateVSCodeMCPConfig},
|
||||
{"windsurf", GenerateWindsurfMCPConfig},
|
||||
}
|
||||
|
||||
if err := GenerateClaudeMCPConfig(cfg, home); err != nil {
|
||||
return fmt.Errorf("claude MCP config: %w", err)
|
||||
var errs []string
|
||||
for _, e := range editors {
|
||||
if err := e.fn(cfg, home); err != nil {
|
||||
errs = append(errs, fmt.Sprintf("%s: %s", e.name, err))
|
||||
}
|
||||
}
|
||||
|
||||
SaveReceipt("all", time.Now().Format("2006-01-02"))
|
||||
|
||||
if len(errs) > 0 {
|
||||
return fmt.Errorf("MCP config errors: %s", strings.Join(errs, "; "))
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func ConfigureForEditor(cfg *config.MuyueConfig, editorName string) error {
|
||||
home, err := os.UserHomeDir()
|
||||
if err != nil {
|
||||
return fmt.Errorf("get home dir: %w", err)
|
||||
}
|
||||
|
||||
switch editorName {
|
||||
case "crush":
|
||||
return GenerateCrushMCPConfig(cfg, home)
|
||||
case "claude", "claude-code":
|
||||
return GenerateClaudeMCPConfig(cfg, home)
|
||||
case "cursor":
|
||||
return GenerateCursorMCPConfig(cfg, home)
|
||||
case "vscode", "code":
|
||||
return GenerateVSCodeMCPConfig(cfg, home)
|
||||
case "windsurf":
|
||||
return GenerateWindsurfMCPConfig(cfg, home)
|
||||
default:
|
||||
return fmt.Errorf("unknown editor: %s (supported: crush, claude-code, cursor, vscode, windsurf)", editorName)
|
||||
}
|
||||
}
|
||||
|
||||
func DetectInstalledEditors(homeDir string) []string {
|
||||
if homeDir == "" {
|
||||
home, _ := os.UserHomeDir()
|
||||
homeDir = home
|
||||
}
|
||||
|
||||
editors := []struct {
|
||||
name string
|
||||
path string
|
||||
}{
|
||||
{"crush", filepath.Join(homeDir, ".config", "crush", "crush.json")},
|
||||
{"claude-code", filepath.Join(homeDir, ".claude.json")},
|
||||
{"cursor", filepath.Join(homeDir, ".cursor")},
|
||||
{"vscode", filepath.Join(homeDir, ".vscode")},
|
||||
{"windsurf", filepath.Join(homeDir, ".windsurf")},
|
||||
}
|
||||
|
||||
var detected []string
|
||||
for _, e := range editors {
|
||||
if _, err := os.Stat(e.path); err == nil {
|
||||
detected = append(detected, e.name)
|
||||
}
|
||||
}
|
||||
return detected
|
||||
}
|
||||
|
||||
func GetAllStatuses() []MCPStatus {
|
||||
servers := ScanServers()
|
||||
statuses := make([]MCPStatus, len(servers))
|
||||
for i, s := range servers {
|
||||
statuses[i] = CheckServerStatus(s.Name)
|
||||
}
|
||||
return statuses
|
||||
}
|
||||
|
||||
520
internal/mcp/registry.go
Normal file
520
internal/mcp/registry.go
Normal file
@@ -0,0 +1,520 @@
|
||||
package mcp
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"gopkg.in/yaml.v3"
|
||||
)
|
||||
|
||||
type RegistryServer struct {
|
||||
Name string `yaml:"name" json:"name"`
|
||||
Description string `yaml:"description" json:"description"`
|
||||
Category string `yaml:"category" json:"category"`
|
||||
Package string `yaml:"package" json:"package"`
|
||||
Command string `yaml:"command" json:"command"`
|
||||
Args []string `yaml:"args" json:"args"`
|
||||
Env map[string]string `yaml:"env,omitempty" json:"env,omitempty"`
|
||||
RequiredEnv []string `yaml:"required_env,omitempty" json:"required_env,omitempty"`
|
||||
HomePage string `yaml:"homepage,omitempty" json:"homepage,omitempty"`
|
||||
Tags []string `yaml:"tags,omitempty" json:"tags,omitempty"`
|
||||
Version string `yaml:"version,omitempty" json:"version,omitempty"`
|
||||
InstallType string `yaml:"install_type" json:"install_type"`
|
||||
}
|
||||
|
||||
type Registry struct {
|
||||
SchemaVersion string `yaml:"schema_version"`
|
||||
UpdatedAt time.Time `yaml:"updated_at"`
|
||||
Servers []RegistryServer `yaml:"servers"`
|
||||
}
|
||||
|
||||
type MCPStatus struct {
|
||||
Name string `json:"name"`
|
||||
Installed bool `json:"installed"`
|
||||
Running bool `json:"running"`
|
||||
Healthy bool `json:"healthy"`
|
||||
Version string `json:"version"`
|
||||
Error string `json:"error,omitempty"`
|
||||
}
|
||||
|
||||
type EditorConfig struct {
|
||||
Name string
|
||||
ConfigPath string
|
||||
ConfigKey string
|
||||
LocalConfigPath string
|
||||
Format string
|
||||
TransformCommand func(entry mcpEntry) interface{}
|
||||
}
|
||||
|
||||
var (
|
||||
registryMu sync.RWMutex
|
||||
registryCache *Registry
|
||||
registryPath string
|
||||
)
|
||||
|
||||
func init() {
|
||||
home, _ := os.UserHomeDir()
|
||||
if home != "" {
|
||||
registryPath = filepath.Join(home, ".muyue", "mcp-registry.yaml")
|
||||
}
|
||||
}
|
||||
|
||||
func SetRegistryPath(p string) {
|
||||
registryMu.Lock()
|
||||
defer registryMu.Unlock()
|
||||
registryPath = p
|
||||
registryCache = nil
|
||||
}
|
||||
|
||||
func DefaultRegistry() *Registry {
|
||||
return &Registry{
|
||||
SchemaVersion: "v1",
|
||||
UpdatedAt: time.Now(),
|
||||
Servers: []RegistryServer{
|
||||
{
|
||||
Name: "filesystem", Description: "File system operations for AI tools",
|
||||
Category: "core", Package: "@modelcontextprotocol/server-filesystem",
|
||||
Command: "npx", Args: []string{"-y", "@modelcontextprotocol/server-filesystem"},
|
||||
InstallType: "npm", Tags: []string{"files", "core"},
|
||||
},
|
||||
{
|
||||
Name: "github", Description: "GitHub API integration",
|
||||
Category: "vcs", Package: "@modelcontextprotocol/server-github",
|
||||
Command: "npx", Args: []string{"-y", "@modelcontextprotocol/server-github"},
|
||||
Env: map[string]string{"GITHUB_PERSONAL_ACCESS_TOKEN": ""},
|
||||
RequiredEnv: []string{"GITHUB_PERSONAL_ACCESS_TOKEN"},
|
||||
InstallType: "npm", Tags: []string{"github", "git"},
|
||||
},
|
||||
{
|
||||
Name: "git", Description: "Git repository operations",
|
||||
Category: "vcs", Package: "@modelcontextprotocol/server-git",
|
||||
Command: "npx", Args: []string{"-y", "@modelcontextprotocol/server-git"},
|
||||
InstallType: "npm", Tags: []string{"git"},
|
||||
},
|
||||
{
|
||||
Name: "fetch", Description: "Web fetching and HTTP requests",
|
||||
Category: "web", Package: "@modelcontextprotocol/server-fetch",
|
||||
Command: "npx", Args: []string{"-y", "@modelcontextprotocol/server-fetch"},
|
||||
InstallType: "npm", Tags: []string{"web", "http"},
|
||||
},
|
||||
{
|
||||
Name: "memory", Description: "Persistent memory/knowledge graph",
|
||||
Category: "core", Package: "@modelcontextprotocol/server-memory",
|
||||
Command: "npx", Args: []string{"-y", "@modelcontextprotocol/server-memory"},
|
||||
InstallType: "npm", Tags: []string{"memory", "core"},
|
||||
},
|
||||
{
|
||||
Name: "sequential-thinking", Description: "Structured reasoning and chain-of-thought",
|
||||
Category: "ai", Package: "@modelcontextprotocol/server-sequential-thinking",
|
||||
Command: "npx", Args: []string{"-y", "@modelcontextprotocol/server-sequential-thinking"},
|
||||
InstallType: "npm", Tags: []string{"ai", "reasoning"},
|
||||
},
|
||||
{
|
||||
Name: "brave-search", Description: "Web search via Brave Search API",
|
||||
Category: "web", Package: "@modelcontextprotocol/server-brave-search",
|
||||
Command: "npx", Args: []string{"-y", "@modelcontextprotocol/server-brave-search"},
|
||||
Env: map[string]string{"BRAVE_API_KEY": ""},
|
||||
RequiredEnv: []string{"BRAVE_API_KEY"},
|
||||
InstallType: "npm", Tags: []string{"search", "web"},
|
||||
},
|
||||
{
|
||||
Name: "sqlite", Description: "SQLite database operations",
|
||||
Category: "database", Package: "@modelcontextprotocol/server-sqlite",
|
||||
Command: "npx", Args: []string{"-y", "@modelcontextprotocol/server-sqlite"},
|
||||
InstallType: "npm", Tags: []string{"database", "sqlite"},
|
||||
},
|
||||
{
|
||||
Name: "postgres", Description: "PostgreSQL database operations",
|
||||
Category: "database", Package: "@modelcontextprotocol/server-postgres",
|
||||
Command: "npx", Args: []string{"-y", "@modelcontextprotocol/server-postgres"},
|
||||
InstallType: "npm", Tags: []string{"database", "postgres"},
|
||||
},
|
||||
{
|
||||
Name: "docker", Description: "Docker container management",
|
||||
Category: "devops", Package: "@modelcontextprotocol/server-docker",
|
||||
Command: "npx", Args: []string{"-y", "@modelcontextprotocol/server-docker"},
|
||||
InstallType: "npm", Tags: []string{"docker", "devops"},
|
||||
},
|
||||
{
|
||||
Name: "minimax-web-search", Description: "Web search via MiniMax API",
|
||||
Category: "ai", Package: "@minimax/mcp-web-search",
|
||||
Command: "npx", Args: []string{"-y", "@minimax/mcp-web-search"},
|
||||
Env: map[string]string{"MINIMAX_API_KEY": ""},
|
||||
RequiredEnv: []string{"MINIMAX_API_KEY"},
|
||||
InstallType: "npm", Tags: []string{"ai", "search"},
|
||||
},
|
||||
{
|
||||
Name: "minimax-image", Description: "Image understanding via MiniMax API",
|
||||
Category: "ai", Package: "@minimax/mcp-image-understanding",
|
||||
Command: "npx", Args: []string{"-y", "@minimax/mcp-image-understanding"},
|
||||
Env: map[string]string{"MINIMAX_API_KEY": ""},
|
||||
RequiredEnv: []string{"MINIMAX_API_KEY"},
|
||||
InstallType: "npm", Tags: []string{"ai", "image"},
|
||||
},
|
||||
{
|
||||
Name: "puppeteer", Description: "Browser automation with Puppeteer",
|
||||
Category: "web", Package: "@modelcontextprotocol/server-puppeteer",
|
||||
Command: "npx", Args: []string{"-y", "@modelcontextprotocol/server-puppeteer"},
|
||||
InstallType: "npm", Tags: []string{"browser", "automation"},
|
||||
},
|
||||
{
|
||||
Name: "everything", Description: "Test/debug MCP server with all features",
|
||||
Category: "testing", Package: "@modelcontextprotocol/server-everything",
|
||||
Command: "npx", Args: []string{"-y", "@modelcontextprotocol/server-everything"},
|
||||
InstallType: "npm", Tags: []string{"testing", "debug"},
|
||||
},
|
||||
{
|
||||
Name: "slack", Description: "Slack workspace integration",
|
||||
Category: "communication", Package: "@modelcontextprotocol/server-slack",
|
||||
Command: "npx", Args: []string{"-y", "@modelcontextprotocol/server-slack"},
|
||||
Env: map[string]string{"SLACK_BOT_TOKEN": ""},
|
||||
RequiredEnv: []string{"SLACK_BOT_TOKEN"},
|
||||
InstallType: "npm", Tags: []string{"slack", "communication"},
|
||||
},
|
||||
{
|
||||
Name: "google-maps", Description: "Google Maps integration",
|
||||
Category: "web", Package: "@modelcontextprotocol/server-google-maps",
|
||||
Command: "npx", Args: []string{"-y", "@modelcontextprotocol/server-google-maps"},
|
||||
Env: map[string]string{"GOOGLE_MAPS_API_KEY": ""},
|
||||
RequiredEnv: []string{"GOOGLE_MAPS_API_KEY"},
|
||||
InstallType: "npm", Tags: []string{"maps", "location"},
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func LoadRegistry() (*Registry, error) {
|
||||
registryMu.RLock()
|
||||
if registryCache != nil {
|
||||
defer registryMu.RUnlock()
|
||||
return registryCache, nil
|
||||
}
|
||||
registryMu.RUnlock()
|
||||
|
||||
reg, err := loadRegistryFromDisk()
|
||||
if err != nil {
|
||||
defaultReg := DefaultRegistry()
|
||||
registryMu.Lock()
|
||||
registryCache = defaultReg
|
||||
registryMu.Unlock()
|
||||
return defaultReg, nil
|
||||
}
|
||||
|
||||
registryMu.Lock()
|
||||
registryCache = reg
|
||||
registryMu.Unlock()
|
||||
return reg, nil
|
||||
}
|
||||
|
||||
func loadRegistryFromDisk() (*Registry, error) {
|
||||
if registryPath == "" {
|
||||
return nil, fmt.Errorf("registry path not set")
|
||||
}
|
||||
|
||||
data, err := os.ReadFile(registryPath)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var reg Registry
|
||||
if err := yaml.Unmarshal(data, ®); err != nil {
|
||||
return nil, fmt.Errorf("parse registry: %w", err)
|
||||
}
|
||||
|
||||
return ®, nil
|
||||
}
|
||||
|
||||
func SaveRegistry(reg *Registry) error {
|
||||
if registryPath == "" {
|
||||
return fmt.Errorf("registry path not set")
|
||||
}
|
||||
|
||||
reg.UpdatedAt = time.Now()
|
||||
data, err := yaml.Marshal(reg)
|
||||
if err != nil {
|
||||
return fmt.Errorf("marshal registry: %w", err)
|
||||
}
|
||||
|
||||
if err := os.MkdirAll(filepath.Dir(registryPath), 0755); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err := os.WriteFile(registryPath, data, 0644); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
registryMu.Lock()
|
||||
registryCache = reg
|
||||
registryMu.Unlock()
|
||||
return nil
|
||||
}
|
||||
|
||||
func AddToRegistry(server RegistryServer) error {
|
||||
reg, err := LoadRegistry()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, s := range reg.Servers {
|
||||
if s.Name == server.Name {
|
||||
return fmt.Errorf("server %q already exists in registry", server.Name)
|
||||
}
|
||||
}
|
||||
|
||||
reg.Servers = append(reg.Servers, server)
|
||||
return SaveRegistry(reg)
|
||||
}
|
||||
|
||||
func RemoveFromRegistry(name string) error {
|
||||
reg, err := LoadRegistry()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for i, s := range reg.Servers {
|
||||
if s.Name == name {
|
||||
reg.Servers = append(reg.Servers[:i], reg.Servers[i+1:]...)
|
||||
return SaveRegistry(reg)
|
||||
}
|
||||
}
|
||||
|
||||
return fmt.Errorf("server %q not found in registry", name)
|
||||
}
|
||||
|
||||
func InitRegistry() error {
|
||||
if _, err := os.Stat(registryPath); err == nil {
|
||||
return nil
|
||||
}
|
||||
return SaveRegistry(DefaultRegistry())
|
||||
}
|
||||
|
||||
func ResolveEnv(env map[string]string, providerKeys map[string]string) map[string]string {
|
||||
resolved := make(map[string]string)
|
||||
for k, v := range env {
|
||||
if v != "" {
|
||||
resolved[k] = v
|
||||
continue
|
||||
}
|
||||
|
||||
if providerKeys != nil {
|
||||
for providerKey, apiKey := range providerKeys {
|
||||
if strings.EqualFold(k, providerKey) || strings.Contains(strings.ToUpper(k), strings.ToUpper(providerKey)) {
|
||||
if apiKey != "" {
|
||||
resolved[k] = apiKey
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if resolved[k] == "" {
|
||||
if envVal := os.Getenv(k); envVal != "" {
|
||||
resolved[k] = envVal
|
||||
}
|
||||
}
|
||||
}
|
||||
return resolved
|
||||
}
|
||||
|
||||
func ValidateConfig(configPath string) error {
|
||||
data, err := os.ReadFile(configPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("read config: %w", err)
|
||||
}
|
||||
|
||||
var cfg map[string]interface{}
|
||||
if err := json.Unmarshal(data, &cfg); err != nil {
|
||||
return fmt.Errorf("parse config: %w", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func DiscoverNpmServers() ([]RegistryServer, error) {
|
||||
var servers []RegistryServer
|
||||
|
||||
packages := []struct {
|
||||
pkg string
|
||||
name string
|
||||
desc string
|
||||
cat string
|
||||
args []string
|
||||
}{
|
||||
{"@modelcontextprotocol/server-filesystem", "filesystem", "File system operations", "core", []string{"-y", "@modelcontextprotocol/server-filesystem"}},
|
||||
{"@modelcontextprotocol/server-github", "github", "GitHub API integration", "vcs", []string{"-y", "@modelcontextprotocol/server-github"}},
|
||||
{"@modelcontextprotocol/server-fetch", "fetch", "Web fetching", "web", []string{"-y", "@modelcontextprotocol/server-fetch"}},
|
||||
{"@modelcontextprotocol/server-memory", "memory", "Persistent memory", "core", []string{"-y", "@modelcontextprotocol/server-memory"}},
|
||||
}
|
||||
|
||||
for _, p := range packages {
|
||||
servers = append(servers, RegistryServer{
|
||||
Name: p.name,
|
||||
Description: p.desc,
|
||||
Category: p.cat,
|
||||
Package: p.pkg,
|
||||
Command: "npx",
|
||||
Args: p.args,
|
||||
InstallType: "npm",
|
||||
})
|
||||
}
|
||||
|
||||
return servers, nil
|
||||
}
|
||||
|
||||
func GetInstalledVersion(name string) string {
|
||||
home, _ := os.UserHomeDir()
|
||||
if home == "" {
|
||||
return ""
|
||||
}
|
||||
receiptPath := filepath.Join(home, ".muyue", "receipts", "mcp", name+".json")
|
||||
data, err := os.ReadFile(receiptPath)
|
||||
if err != nil {
|
||||
return ""
|
||||
}
|
||||
var receipt struct {
|
||||
Version string `json:"version"`
|
||||
}
|
||||
if json.Unmarshal(data, &receipt) == nil {
|
||||
return receipt.Version
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func SaveReceipt(name, version string) error {
|
||||
home, _ := os.UserHomeDir()
|
||||
if home == "" {
|
||||
return nil
|
||||
}
|
||||
receiptDir := filepath.Join(home, ".muyue", "receipts", "mcp")
|
||||
os.MkdirAll(receiptDir, 0755)
|
||||
|
||||
receipt := struct {
|
||||
Name string `json:"name"`
|
||||
Version string `json:"version"`
|
||||
UpdatedAt string `json:"updated_at"`
|
||||
}{
|
||||
Name: name,
|
||||
Version: version,
|
||||
UpdatedAt: time.Now().Format(time.RFC3339),
|
||||
}
|
||||
|
||||
data, _ := json.MarshalIndent(receipt, "", " ")
|
||||
return os.WriteFile(filepath.Join(receiptDir, name+".json"), data, 0644)
|
||||
}
|
||||
|
||||
func BuildProviderKeyMap(cfg interface{ GetAPIKeys() map[string]string }) map[string]string {
|
||||
if cfg == nil {
|
||||
return nil
|
||||
}
|
||||
return cfg.GetAPIKeys()
|
||||
}
|
||||
|
||||
func EditorConfigs(homeDir string) []EditorConfig {
|
||||
if homeDir == "" {
|
||||
home, _ := os.UserHomeDir()
|
||||
homeDir = home
|
||||
}
|
||||
|
||||
transformStdio := func(e mcpEntry) interface{} {
|
||||
m := map[string]interface{}{
|
||||
"command": e.cmd,
|
||||
"args": e.args,
|
||||
}
|
||||
if len(e.env) > 0 {
|
||||
m["env"] = e.env
|
||||
}
|
||||
return m
|
||||
}
|
||||
|
||||
transformCursor := func(e mcpEntry) interface{} {
|
||||
m := map[string]interface{}{
|
||||
"type": "stdio",
|
||||
"command": e.cmd,
|
||||
"args": e.args,
|
||||
}
|
||||
if len(e.env) > 0 {
|
||||
m["env"] = e.env
|
||||
}
|
||||
return m
|
||||
}
|
||||
|
||||
return []EditorConfig{
|
||||
{
|
||||
Name: "crush", ConfigPath: filepath.Join(homeDir, ".config", "crush", "crush.json"),
|
||||
ConfigKey: "mcps", Format: "json", TransformCommand: transformStdio,
|
||||
},
|
||||
{
|
||||
Name: "claude-code", ConfigPath: filepath.Join(homeDir, ".claude.json"),
|
||||
ConfigKey: "mcpServers", Format: "json", TransformCommand: transformStdio,
|
||||
},
|
||||
{
|
||||
Name: "cursor", ConfigPath: filepath.Join(homeDir, ".cursor", "mcp.json"),
|
||||
LocalConfigPath: ".cursor/mcp.json", ConfigKey: "mcpServers",
|
||||
Format: "json", TransformCommand: transformCursor,
|
||||
},
|
||||
{
|
||||
Name: "vscode", ConfigPath: filepath.Join(homeDir, ".vscode", "mcp.json"),
|
||||
LocalConfigPath: ".vscode/mcp.json", ConfigKey: "servers",
|
||||
Format: "json", TransformCommand: transformStdio,
|
||||
},
|
||||
{
|
||||
Name: "windsurf", ConfigPath: filepath.Join(homeDir, ".windsurf", "mcp.json"),
|
||||
ConfigKey: "mcpServers", Format: "json", TransformCommand: transformStdio,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func CheckServerStatus(name string) MCPStatus {
|
||||
status := MCPStatus{Name: name}
|
||||
|
||||
reg, err := LoadRegistry()
|
||||
if err != nil {
|
||||
status.Error = "registry unavailable"
|
||||
return status
|
||||
}
|
||||
|
||||
var server *RegistryServer
|
||||
for i := range reg.Servers {
|
||||
if reg.Servers[i].Name == name {
|
||||
server = ®.Servers[i]
|
||||
break
|
||||
}
|
||||
}
|
||||
if server == nil {
|
||||
status.Error = "not in registry"
|
||||
return status
|
||||
}
|
||||
|
||||
_, err = exec.LookPath(server.Command)
|
||||
if err != nil {
|
||||
status.Error = fmt.Sprintf("command %q not found", server.Command)
|
||||
return status
|
||||
}
|
||||
status.Installed = true
|
||||
|
||||
status.Version = GetInstalledVersion(name)
|
||||
|
||||
home, _ := os.UserHomeDir()
|
||||
if home != "" {
|
||||
crushingPath := filepath.Join(home, ".config", "crush", "crush.json")
|
||||
data, err := os.ReadFile(crushingPath)
|
||||
if err == nil {
|
||||
var cfg map[string]interface{}
|
||||
if json.Unmarshal(data, &cfg) == nil {
|
||||
if mcps, ok := cfg["mcps"].(map[string]interface{}); ok {
|
||||
if _, exists := mcps[name]; exists {
|
||||
status.Running = true
|
||||
status.Healthy = true
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return status
|
||||
}
|
||||
228
internal/mcp/registry_test.go
Normal file
228
internal/mcp/registry_test.go
Normal file
@@ -0,0 +1,228 @@
|
||||
package mcp
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestDefaultRegistry(t *testing.T) {
|
||||
reg := DefaultRegistry()
|
||||
if reg.SchemaVersion != "v1" {
|
||||
t.Errorf("Expected v1, got %s", reg.SchemaVersion)
|
||||
}
|
||||
if len(reg.Servers) == 0 {
|
||||
t.Error("Default registry should have servers")
|
||||
}
|
||||
|
||||
names := map[string]bool{}
|
||||
for _, s := range reg.Servers {
|
||||
if names[s.Name] {
|
||||
t.Errorf("Duplicate server name: %s", s.Name)
|
||||
}
|
||||
names[s.Name] = true
|
||||
if s.Command == "" {
|
||||
t.Errorf("Server %s missing command", s.Name)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestSaveAndLoadRegistry(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
registryPath := filepath.Join(tmpDir, "mcp-registry.yaml")
|
||||
SetRegistryPath(registryPath)
|
||||
|
||||
reg := DefaultRegistry()
|
||||
if err := SaveRegistry(reg); err != nil {
|
||||
t.Fatalf("SaveRegistry failed: %v", err)
|
||||
}
|
||||
|
||||
if _, err := os.Stat(registryPath); os.IsNotExist(err) {
|
||||
t.Error("Registry file should exist")
|
||||
}
|
||||
|
||||
loaded, err := LoadRegistry()
|
||||
if err != nil {
|
||||
t.Fatalf("LoadRegistry failed: %v", err)
|
||||
}
|
||||
if len(loaded.Servers) != len(reg.Servers) {
|
||||
t.Errorf("Expected %d servers, got %d", len(reg.Servers), len(loaded.Servers))
|
||||
}
|
||||
}
|
||||
|
||||
func TestAddAndRemoveFromRegistry(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
SetRegistryPath(filepath.Join(tmpDir, "mcp-registry.yaml"))
|
||||
SaveRegistry(DefaultRegistry())
|
||||
|
||||
newServer := RegistryServer{
|
||||
Name: "test-server",
|
||||
Description: "Test server",
|
||||
Category: "test",
|
||||
Command: "npx",
|
||||
Args: []string{"-y", "test-pkg"},
|
||||
InstallType: "npm",
|
||||
}
|
||||
|
||||
if err := AddToRegistry(newServer); err != nil {
|
||||
t.Fatalf("AddToRegistry failed: %v", err)
|
||||
}
|
||||
|
||||
reg, _ := LoadRegistry()
|
||||
found := false
|
||||
for _, s := range reg.Servers {
|
||||
if s.Name == "test-server" {
|
||||
found = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !found {
|
||||
t.Error("test-server should be in registry")
|
||||
}
|
||||
|
||||
if err := RemoveFromRegistry("test-server"); err != nil {
|
||||
t.Fatalf("RemoveFromRegistry failed: %v", err)
|
||||
}
|
||||
|
||||
reg, _ = LoadRegistry()
|
||||
for _, s := range reg.Servers {
|
||||
if s.Name == "test-server" {
|
||||
t.Error("test-server should have been removed")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestResolveEnv(t *testing.T) {
|
||||
env := map[string]string{
|
||||
"API_KEY": "",
|
||||
"HOST": "localhost",
|
||||
}
|
||||
|
||||
os.Setenv("API_KEY", "from-env")
|
||||
defer os.Unsetenv("API_KEY")
|
||||
|
||||
resolved := ResolveEnv(env, nil)
|
||||
if resolved["API_KEY"] != "from-env" {
|
||||
t.Errorf("Expected from-env, got %s", resolved["API_KEY"])
|
||||
}
|
||||
if resolved["HOST"] != "localhost" {
|
||||
t.Errorf("Expected localhost, got %s", resolved["HOST"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestValidateConfig(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
configPath := filepath.Join(tmpDir, "test-config.json")
|
||||
os.WriteFile(configPath, []byte(`{"mcps":{}}`), 0644)
|
||||
|
||||
if err := ValidateConfig(configPath); err != nil {
|
||||
t.Errorf("Valid config should pass: %v", err)
|
||||
}
|
||||
|
||||
badPath := filepath.Join(tmpDir, "nonexistent.json")
|
||||
if err := ValidateConfig(badPath); err == nil {
|
||||
t.Error("Nonexistent config should fail")
|
||||
}
|
||||
}
|
||||
|
||||
func TestEditorConfigs(t *testing.T) {
|
||||
configs := EditorConfigs("/tmp")
|
||||
if len(configs) < 3 {
|
||||
t.Errorf("Expected at least 3 editor configs, got %d", len(configs))
|
||||
}
|
||||
|
||||
names := map[string]bool{}
|
||||
for _, c := range configs {
|
||||
if names[c.Name] {
|
||||
t.Errorf("Duplicate editor: %s", c.Name)
|
||||
}
|
||||
names[c.Name] = true
|
||||
if c.ConfigPath == "" {
|
||||
t.Errorf("Editor %s missing config path", c.Name)
|
||||
}
|
||||
if c.ConfigKey == "" {
|
||||
t.Errorf("Editor %s missing config key", c.Name)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestDiscoverNpmServers(t *testing.T) {
|
||||
servers, err := DiscoverNpmServers()
|
||||
if err != nil {
|
||||
t.Fatalf("DiscoverNpmServers failed: %v", err)
|
||||
}
|
||||
if len(servers) == 0 {
|
||||
t.Error("Should discover some npm servers")
|
||||
}
|
||||
}
|
||||
|
||||
func TestReceiptRoundTrip(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
os.Setenv("HOME", tmpDir)
|
||||
defer os.Unsetenv("HOME")
|
||||
|
||||
SetRegistryPath(filepath.Join(tmpDir, "reg.yaml"))
|
||||
|
||||
if err := SaveReceipt("test-server", "1.2.3"); err != nil {
|
||||
t.Fatalf("SaveReceipt failed: %v", err)
|
||||
}
|
||||
|
||||
version := GetInstalledVersion("test-server")
|
||||
if version != "1.2.3" {
|
||||
t.Errorf("Expected 1.2.3, got %s", version)
|
||||
}
|
||||
}
|
||||
|
||||
func TestInitRegistry(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
SetRegistryPath(filepath.Join(tmpDir, "init-reg.yaml"))
|
||||
|
||||
if err := InitRegistry(); err != nil {
|
||||
t.Fatalf("InitRegistry failed: %v", err)
|
||||
}
|
||||
|
||||
if _, err := os.Stat(filepath.Join(tmpDir, "init-reg.yaml")); os.IsNotExist(err) {
|
||||
t.Error("Registry file should be created")
|
||||
}
|
||||
|
||||
if err := InitRegistry(); err != nil {
|
||||
t.Fatalf("Second InitRegistry should not fail: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestDetectInstalledEditors(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
os.MkdirAll(filepath.Join(tmpDir, ".config", "crush"), 0755)
|
||||
os.WriteFile(filepath.Join(tmpDir, ".config", "crush", "crush.json"), []byte(`{}`), 0644)
|
||||
os.MkdirAll(filepath.Join(tmpDir, ".cursor"), 0755)
|
||||
|
||||
editors := DetectInstalledEditors(tmpDir)
|
||||
if len(editors) < 2 {
|
||||
t.Errorf("Expected at least 2 editors, got %d", len(editors))
|
||||
}
|
||||
|
||||
found := map[string]bool{}
|
||||
for _, e := range editors {
|
||||
found[e] = true
|
||||
}
|
||||
if !found["crush"] {
|
||||
t.Error("Should detect crush")
|
||||
}
|
||||
if !found["cursor"] {
|
||||
t.Error("Should detect cursor")
|
||||
}
|
||||
}
|
||||
|
||||
func TestCheckServerStatus(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
os.Setenv("HOME", tmpDir)
|
||||
defer os.Unsetenv("HOME")
|
||||
|
||||
SetRegistryPath(filepath.Join(tmpDir, "reg.yaml"))
|
||||
SaveRegistry(DefaultRegistry())
|
||||
|
||||
status := CheckServerStatus("nonexistent")
|
||||
if status.Error == "" {
|
||||
t.Error("Should have error for nonexistent server")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user