feat: add Cobra CLI, LSP/MCP registries, workflow engine, and enriched dashboard
All checks were successful
Beta Release / beta (push) Successful in 2m24s

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
2026-04-22 22:22:05 +02:00
parent 66b773ff86
commit 2e50366cd8
42 changed files with 6779 additions and 319 deletions

View File

@@ -5,6 +5,7 @@ import (
"fmt"
"os"
"path/filepath"
"strings"
"sync"
"time"
"unicode/utf8"
@@ -36,6 +37,19 @@ type ConversationStore struct {
conv *Conversation
}
type TokenCount struct {
total int
byRole map[string]int
byMessage int
}
type SearchResult struct {
ID string `json:"id"`
Role string `json:"role"`
Content string `json:"content"`
Time string `json:"time"`
}
func NewConversationStore() *ConversationStore {
dir, err := config.ConfigDir()
if err != nil {
@@ -140,19 +154,109 @@ func (cs *ConversationStore) TrimOld(keepCount int) {
}
func (cs *ConversationStore) ApproxTokenCount() int {
return cs.ApproxTokenCountDetailed().total
}
func (cs *ConversationStore) ApproxTokenCountDetailed() TokenCount {
cs.mu.RLock()
defer cs.mu.RUnlock()
total := utf8.RuneCountInString(cs.conv.Summary)
for _, m := range cs.conv.Messages {
total += utf8.RuneCountInString(m.Content)
result := TokenCount{
byRole: make(map[string]int),
}
return total / charsPerToken
for _, m := range cs.conv.Messages {
count := utf8.RuneCountInString(m.Content) / charsPerToken
result.byMessage += count
result.byRole[m.Role] += count
}
if cs.conv.Summary != "" {
result.total = result.byMessage + utf8.RuneCountInString(cs.conv.Summary)/charsPerToken
} else {
result.total = result.byMessage
}
return result
}
func (cs *ConversationStore) NeedsSummarization() bool {
return cs.ApproxTokenCount() > summarizeThreshold
}
func (cs *ConversationStore) Search(query string) []SearchResult {
cs.mu.RLock()
defer cs.mu.RUnlock()
var results []SearchResult
queryLower := strings.ToLower(query)
for _, msg := range cs.conv.Messages {
if strings.Contains(strings.ToLower(msg.Content), queryLower) {
results = append(results, SearchResult{
ID: msg.ID,
Role: msg.Role,
Content: msg.Content,
Time: msg.Time,
})
}
}
return results
}
func (cs *ConversationStore) ExportMarkdown() string {
cs.mu.RLock()
defer cs.mu.RUnlock()
var sb strings.Builder
sb.WriteString("# Conversation Export\n\n")
sb.WriteString(fmt.Sprintf("Exporté le: %s\n\n", time.Now().Format(time.RFC3339)))
if cs.conv.Summary != "" {
sb.WriteString("## Résumé\n\n")
sb.WriteString(cs.conv.Summary)
sb.WriteString("\n\n---\n\n")
}
sb.WriteString("## Messages\n\n")
for i, msg := range cs.conv.Messages {
roleLabel := msg.Role
if roleLabel == "user" {
roleLabel = "👤 Utilisateur"
} else if roleLabel == "assistant" {
roleLabel = "🤖 Assistant"
} else if roleLabel == "system" {
roleLabel = "⚙️ Système"
}
timestamp := ""
if msg.Time != "" {
if t, err := time.Parse(time.RFC3339, msg.Time); err == nil {
timestamp = t.Format("2006-01-02 15:04")
}
}
sb.WriteString(fmt.Sprintf("### [%d] %s (%s)\n\n", i+1, roleLabel, timestamp))
sb.WriteString(msg.Content)
sb.WriteString("\n\n---\n\n")
}
return sb.String()
}
func (cs *ConversationStore) ExportJSON() string {
cs.mu.RLock()
defer cs.mu.RUnlock()
data, err := json.MarshalIndent(cs.conv, "", " ")
if err != nil {
return "{}"
}
return string(data)
}
func generateMsgID() string {
return time.Now().Format("20060102150405.000") + "-" + fmt.Sprintf("%d", time.Now().UnixNano())
}
}

View File

@@ -1,7 +1,9 @@
package api
import (
"encoding/json"
"net/http"
"os"
"github.com/muyue/muyue/internal/lsp"
"github.com/muyue/muyue/internal/mcp"
@@ -95,9 +97,14 @@ func (s *Server) handleLSP(w http.ResponseWriter, r *http.Request) {
func (s *Server) handleMCP(w http.ResponseWriter, r *http.Request) {
servers := mcp.ScanServers()
home, _ := os.UserHomeDir()
editors := mcp.DetectInstalledEditors(home)
statuses := mcp.GetAllStatuses()
writeJSON(w, map[string]interface{}{
"servers": servers,
"configured": true,
"servers": servers,
"configured": true,
"detected_editors": editors,
"statuses": statuses,
})
}
@@ -106,11 +113,297 @@ func (s *Server) handleMCPConfigure(w http.ResponseWriter, r *http.Request) {
writeError(w, "POST only", http.StatusMethodNotAllowed)
return
}
if err := mcp.ConfigureAll(s.config); err != nil {
var body struct {
Editor string `json:"editor,omitempty"`
}
if r.Body != nil {
json.NewDecoder(r.Body).Decode(&body)
}
if body.Editor != "" {
if err := mcp.ConfigureForEditor(s.config, body.Editor); err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
} else {
if err := mcp.ConfigureAll(s.config); err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
}
writeJSON(w, map[string]string{"status": "ok"})
}
func (s *Server) handleMCPStatus(w http.ResponseWriter, r *http.Request) {
statuses := mcp.GetAllStatuses()
writeJSON(w, map[string]interface{}{
"statuses": statuses,
})
}
func (s *Server) handleMCPRegistry(w http.ResponseWriter, r *http.Request) {
reg, err := mcp.LoadRegistry()
if err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
writeJSON(w, map[string]string{"status": "ok"})
writeJSON(w, map[string]interface{}{
"registry": reg,
})
}
func (s *Server) handleLSPHealth(w http.ResponseWriter, r *http.Request) {
servers := lsp.ScanServers()
type healthInfo struct {
Name string `json:"name"`
Language string `json:"language"`
Installed bool `json:"installed"`
Healthy bool `json:"healthy"`
Detail string `json:"detail,omitempty"`
}
var results []healthInfo
for _, srv := range servers {
healthy, detail := lsp.HealthCheck(srv.Name)
results = append(results, healthInfo{
Name: srv.Name,
Language: srv.Language,
Installed: srv.Installed,
Healthy: healthy,
Detail: detail,
})
}
writeJSON(w, map[string]interface{}{
"servers": results,
})
}
func (s *Server) handleLSPAutoInstall(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
writeError(w, "POST only", http.StatusMethodNotAllowed)
return
}
var body struct {
ProjectDir string `json:"project_dir,omitempty"`
}
if r.Body != nil {
json.NewDecoder(r.Body).Decode(&body)
}
if body.ProjectDir == "" {
home, _ := os.UserHomeDir()
body.ProjectDir = home
}
results, err := lsp.AutoInstallForProject(body.ProjectDir)
if err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
writeJSON(w, map[string]interface{}{
"results": results,
})
}
func (s *Server) handleLSPEditorConfig(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
writeError(w, "POST only", http.StatusMethodNotAllowed)
return
}
var body struct {
Editor string `json:"editor"`
Names []string `json:"names,omitempty"`
}
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
writeError(w, err.Error(), http.StatusBadRequest)
return
}
allServers := lsp.ScanServers()
var selected []lsp.LSPServer
if len(body.Names) > 0 {
nameSet := map[string]bool{}
for _, n := range body.Names {
nameSet[n] = true
}
for _, srv := range allServers {
if nameSet[srv.Name] {
selected = append(selected, srv)
}
}
} else {
for _, srv := range allServers {
if srv.Installed {
selected = append(selected, srv)
}
}
}
config, err := lsp.GenerateEditorConfigs(selected, body.Editor, "")
if err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
writeJSON(w, map[string]interface{}{
"editor": body.Editor,
"config": config,
})
}
func (s *Server) handleSkillValidate(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
writeError(w, "POST only", http.StatusMethodNotAllowed)
return
}
var body struct {
Name string `json:"name"`
}
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
writeError(w, err.Error(), http.StatusBadRequest)
return
}
skill, err := skills.Get(body.Name)
if err != nil {
writeError(w, err.Error(), http.StatusNotFound)
return
}
errs := skills.Validate(skill)
writeJSON(w, map[string]interface{}{
"name": body.Name,
"valid": len(errs) == 0,
"errors": errs,
"dependencies": skills.CheckDependencies(skill),
})
}
func (s *Server) handleSkillTest(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
writeError(w, "POST only", http.StatusMethodNotAllowed)
return
}
var body struct {
Name string `json:"name"`
SampleTask string `json:"sample_task,omitempty"`
}
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
writeError(w, err.Error(), http.StatusBadRequest)
return
}
result := skills.DryRun(body.Name, body.SampleTask)
writeJSON(w, result)
}
func (s *Server) handleSkillExport(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
writeError(w, "POST only", http.StatusMethodNotAllowed)
return
}
var body struct {
Name string `json:"name"`
ExportPath string `json:"export_path"`
}
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
writeError(w, err.Error(), http.StatusBadRequest)
return
}
home, _ := os.UserHomeDir()
if body.ExportPath == "" {
body.ExportPath = home + "/.muyue/exports/" + body.Name + ".md"
}
if err := skills.Export(body.Name, body.ExportPath); err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
writeJSON(w, map[string]string{"status": "ok", "path": body.ExportPath})
}
func (s *Server) handleSkillImport(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
writeError(w, "POST only", http.StatusMethodNotAllowed)
return
}
var body struct {
ImportPath string `json:"import_path"`
}
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
writeError(w, err.Error(), http.StatusBadRequest)
return
}
skill, err := skills.Import(body.ImportPath)
if err != nil {
writeError(w, err.Error(), http.StatusBadRequest)
return
}
if err := skills.Create(skill); err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
writeJSON(w, map[string]interface{}{"status": "ok", "skill": skill.Name})
}
func (s *Server) handleDashboardStatus(w http.ResponseWriter, r *http.Request) {
mcpStatuses := mcp.GetAllStatuses()
lspServers := lsp.ScanServers()
skillList, _ := skills.List()
mcpHealthy := 0
mcpTotal := len(mcpStatuses)
for _, st := range mcpStatuses {
if st.Healthy {
mcpHealthy++
}
}
lspInstalled := 0
lspTotal := len(lspServers)
for _, srv := range lspServers {
if srv.Installed {
lspInstalled++
}
}
skillsDeployed := len(skillList)
var skillIssues []string
for _, sk := range skillList {
missing := skills.CheckDependencies(&sk)
if len(missing) > 0 {
for _, dep := range missing {
skillIssues = append(skillIssues, sk.Name+": missing "+dep.Type+" "+dep.Name)
}
}
}
writeJSON(w, map[string]interface{}{
"mcp": map[string]interface{}{
"total": mcpTotal,
"healthy": mcpHealthy,
"servers": mcpStatuses,
},
"lsp": map[string]interface{}{
"total": lspTotal,
"installed": lspInstalled,
"servers": lspServers,
},
"skills": map[string]interface{}{
"total": skillsDeployed,
"issues": skillIssues,
"deployed": skillList,
},
})
}
func (s *Server) handleScan(w http.ResponseWriter, r *http.Request) {

View File

@@ -0,0 +1,269 @@
package api
import (
"encoding/json"
"net/http"
"os"
"path/filepath"
"strings"
"time"
"github.com/muyue/muyue/internal/config"
"github.com/muyue/muyue/internal/lsp"
"github.com/muyue/muyue/internal/skills"
)
type SavedConversation struct {
ID string `json:"id"`
Title string `json:"title"`
Summary string `json:"summary,omitempty"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
Messages []MessageEntry `json:"messages,omitempty"`
}
type MessageEntry struct {
ID string `json:"id"`
Role string `json:"role"`
Content string `json:"content"`
Time string `json:"time"`
}
type conversationsStore struct {
Path string
Items []SavedConversation
}
func conversationsPath() string {
dir, _ := config.ConfigDir()
return filepath.Join(dir, "conversations.json")
}
func listConversations() ([]SavedConversation, error) {
path := conversationsPath()
data, err := os.ReadFile(path)
if err != nil {
if os.IsNotExist(err) {
return []SavedConversation{}, nil
}
return nil, err
}
var store conversationsStore
if err := json.Unmarshal(data, &store); err != nil {
return []SavedConversation{}, nil
}
return store.Items, nil
}
func saveConversations(items []SavedConversation) error {
path := conversationsPath()
dir := filepath.Dir(path)
os.MkdirAll(dir, 0755)
data, err := json.MarshalIndent(struct {
Items []SavedConversation `json:"items"`
}{Items: items}, "", " ")
if err != nil {
return err
}
return os.WriteFile(path, data, 0600)
}
func (s *Server) handleListConversations(w http.ResponseWriter, r *http.Request) {
if r.Method != "GET" {
writeError(w, "GET only", http.StatusMethodNotAllowed)
return
}
convs, err := listConversations()
if err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
conv := s.convStore.Get()
tokenInfo := s.convStore.ApproxTokenCountDetailed()
writeJSON(w, map[string]interface{}{
"conversations": convs,
"current_messages": conv,
"tokens": tokenInfo.total,
"tokens_by_role": tokenInfo.byRole,
"summary": s.convStore.GetSummary(),
})
}
func (s *Server) handleDeleteConversation(w http.ResponseWriter, r *http.Request) {
if r.Method != "DELETE" {
writeError(w, "DELETE only", http.StatusMethodNotAllowed)
return
}
id := strings.TrimPrefix(r.URL.Path, "/api/conversations/")
id = strings.TrimPrefix(id, "/")
if id == "" {
s.convStore.Clear()
writeJSON(w, map[string]string{"status": "cleared"})
return
}
convs, err := listConversations()
if err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
filtered := make([]SavedConversation, 0, len(convs))
found := false
for _, c := range convs {
if c.ID == id {
found = true
continue
}
filtered = append(filtered, c)
}
if !found {
writeError(w, "conversation not found", http.StatusNotFound)
return
}
if err := saveConversations(filtered); err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
writeJSON(w, map[string]string{"status": "deleted"})
}
func (s *Server) handleSearchConversations(w http.ResponseWriter, r *http.Request) {
if r.Method != "GET" {
writeError(w, "GET only", http.StatusMethodNotAllowed)
return
}
query := r.URL.Query().Get("q")
if query == "" {
writeError(w, "query parameter 'q' is required", http.StatusBadRequest)
return
}
results := s.convStore.Search(query)
writeJSON(w, map[string]interface{}{
"query": query,
"results": results,
"count": len(results),
})
}
func (s *Server) handleExportConversation(w http.ResponseWriter, r *http.Request) {
if r.Method != "GET" {
writeError(w, "GET only", http.StatusMethodNotAllowed)
return
}
format := r.URL.Query().Get("format")
if format == "markdown" || format == "md" {
w.Header().Set("Content-Type", "text/markdown; charset=utf-8")
w.Write([]byte(s.convStore.ExportMarkdown()))
return
}
w.Header().Set("Content-Type", "application/json")
w.Write([]byte(s.convStore.ExportJSON()))
}
func (s *Server) handleLSPInstall(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
writeError(w, "POST only", http.StatusMethodNotAllowed)
return
}
var body struct {
Name string `json:"name"`
}
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
writeError(w, err.Error(), http.StatusBadRequest)
return
}
if body.Name == "" {
writeError(w, "name is required", http.StatusBadRequest)
return
}
if err := lsp.InstallServer(body.Name); err != nil {
writeJSON(w, map[string]interface{}{
"success": false,
"error": err.Error(),
})
return
}
writeJSON(w, map[string]interface{}{
"success": true,
"server": body.Name,
})
}
func (s *Server) handleSkillsDeploy(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
writeError(w, "POST only", http.StatusMethodNotAllowed)
return
}
var body struct {
Name string `json:"name"`
}
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
writeError(w, err.Error(), http.StatusBadRequest)
return
}
if body.Name != "" {
skill, err := skills.Get(body.Name)
if err != nil {
writeError(w, err.Error(), http.StatusNotFound)
return
}
if err := skills.Deploy(skill); err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
writeJSON(w, map[string]string{"status": "deployed", "skill": body.Name})
return
}
if err := skills.DeployAll(); err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
writeJSON(w, map[string]string{"status": "all deployed"})
}
func (s *Server) handleSSHConnections(w http.ResponseWriter, r *http.Request) {
if r.Method != "GET" {
writeError(w, "GET only", http.StatusMethodNotAllowed)
return
}
cfg, err := config.Load()
if err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
writeJSON(w, map[string]interface{}{
"connections": cfg.Terminal.SSH,
})
}
func (s *Server) handleSSHTest(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
writeError(w, "POST only", http.StatusMethodNotAllowed)
return
}
var body struct {
Host string `json:"host"`
Port int `json:"port"`
User string `json:"user"`
}
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
writeError(w, err.Error(), http.StatusBadRequest)
return
}
if body.Host == "" || body.User == "" {
writeError(w, "host and user are required", http.StatusBadRequest)
return
}
if body.Port == 0 {
body.Port = 22
}
writeJSON(w, map[string]interface{}{
"success": true,
"message": "SSH connection test not implemented (requires net.DialTimeout)",
})
}

View File

@@ -0,0 +1,298 @@
package api
import (
"context"
"encoding/json"
"net/http"
"strings"
"github.com/muyue/muyue/internal/agent"
"github.com/muyue/muyue/internal/orchestrator"
)
const maxShellToolIterations = 10
type ShellChatRequest struct {
Message string `json:"message"`
Context string `json:"context,omitempty"`
History []string `json:"history,omitempty"`
Cwd string `json:"cwd,omitempty"`
Platform string `json:"platform,omitempty"`
Stream bool `json:"stream"`
}
type ShellChatResponse struct {
Content string `json:"content,omitempty"`
ToolCalls []ToolCallInfo `json:"tool_calls,omitempty"`
Error string `json:"error,omitempty"`
}
type ToolCallInfo struct {
ID string `json:"id"`
Name string `json:"name"`
Args map[string]interface{} `json:"args"`
Result *toolResponseData `json:"result,omitempty"`
Error string `json:"error,omitempty"`
}
func (s *Server) handleShellChat(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
writeError(w, "POST only", http.StatusMethodNotAllowed)
return
}
var req ShellChatRequest
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
writeError(w, err.Error(), http.StatusBadRequest)
return
}
if req.Message == "" {
writeError(w, "message is required", http.StatusBadRequest)
return
}
orb, err := orchestrator.New(s.config)
if err != nil {
writeError(w, err.Error(), http.StatusServiceUnavailable)
return
}
orb.SetSystemPrompt(s.buildShellSystemPrompt(req))
orb.SetTools(s.agentToolsJSON)
if req.Stream {
s.handleShellChatStream(w, orb, req)
} else {
s.handleShellChatNonStream(w, orb, req)
}
}
func (s *Server) buildShellSystemPrompt(req ShellChatRequest) string {
var sb strings.Builder
sb.WriteString(`Tu es l'assistant Shell de Muyue. Tu as accès à un terminal et peux aider l'utilisateur avec:
- Exécuter des commandes shell
- Expliquer des erreurs de commandes
- Suggérer des commandes appropriées pour la tâche demandée
- Lire et explorer des fichiers
- Configurer l'environnement de développement
Tu peux appeler des outils pour exécuter des commandes, lire des fichiers, etc. Sois précis et concis dans tes réponses.
`)
if req.Cwd != "" {
sb.WriteString("Répertoire courant: " + req.Cwd + "\n")
}
if req.Platform != "" {
sb.WriteString("Plateforme: " + req.Platform + "\n")
}
if req.Context != "" {
sb.WriteString("\nContexte du terminal:\n" + req.Context + "\n")
}
if len(req.History) > 0 {
sb.WriteString("\nDernières commandes exécutées:\n")
for _, h := range req.History {
sb.WriteString(" " + h + "\n")
}
}
return sb.String()
}
func (s *Server) handleShellChatStream(w http.ResponseWriter, orb *orchestrator.Orchestrator, req ShellChatRequest) {
w.Header().Set("Content-Type", "text/event-stream")
w.Header().Set("Cache-Control", "no-cache")
w.Header().Set("Connection", "keep-alive")
w.Header().Set("Access-Control-Allow-Origin", "*")
w.WriteHeader(http.StatusOK)
flusher, canFlush := w.(http.Flusher)
writeSSE := func(data map[string]interface{}) {
b, _ := json.Marshal(data)
w.Write([]byte("data: " + string(b) + "\n\n"))
if canFlush {
flusher.Flush()
}
}
ctx := context.Background()
messages := []orchestrator.Message{
{Role: "user", Content: req.Message},
}
var finalContent string
var toolCalls []ToolCallInfo
for i := 0; i < maxShellToolIterations; i++ {
resp, err := orb.SendWithTools(messages)
if err != nil {
writeSSE(map[string]interface{}{"error": err.Error()})
return
}
choice := resp.Choices[0]
content := cleanThinkingTags(choice.Message.Content)
if content != "" {
for _, ch := range strings.Split(content, "") {
writeSSE(map[string]interface{}{"content": ch})
}
finalContent = content
}
if len(choice.Message.ToolCalls) == 0 {
break
}
assistantMsg := orchestrator.Message{
Role: "assistant",
Content: content,
ToolCalls: choice.Message.ToolCalls,
}
messages = append(messages, assistantMsg)
for _, tc := range choice.Message.ToolCalls {
toolCallData := map[string]interface{}{
"tool_call_id": tc.ID,
"name": tc.Function.Name,
"args": tc.Function.Arguments,
}
writeSSE(map[string]interface{}{"tool_call": toolCallData})
argsMap := make(map[string]interface{})
json.Unmarshal([]byte(tc.Function.Arguments), &argsMap)
tcInfo := ToolCallInfo{
ID: tc.ID,
Name: tc.Function.Name,
Args: argsMap,
}
call := agent.ToolCall{
ID: tc.ID,
Name: tc.Function.Name,
Arguments: json.RawMessage(tc.Function.Arguments),
}
result, execErr := s.agentRegistry.Execute(ctx, call)
if execErr != nil {
tcInfo.Error = execErr.Error()
writeSSE(map[string]interface{}{"tool_result": tcInfo})
} else {
tcInfo.Result = &toolResponseData{
Content: result.Content,
IsError: result.IsError,
Meta: result.Meta,
}
writeSSE(map[string]interface{}{"tool_result": tcInfo})
}
toolCalls = append(toolCalls, tcInfo)
messages = append(messages, orchestrator.Message{
Role: "tool",
Content: result.Content,
ToolCallID: tc.ID,
Name: tc.Function.Name,
})
}
finalContent = ""
}
if finalContent == "" && len(toolCalls) > 0 {
finalContent = "(opérations terminées)"
}
writeJSONResp, _ := json.Marshal(ShellChatResponse{
Content: finalContent,
ToolCalls: toolCalls,
})
writeSSE(map[string]interface{}{"done": true, "response": string(writeJSONResp)})
}
func (s *Server) handleShellChatNonStream(w http.ResponseWriter, orb *orchestrator.Orchestrator, req ShellChatRequest) {
ctx := context.Background()
messages := []orchestrator.Message{
{Role: "user", Content: req.Message},
}
var finalContent string
var toolCalls []ToolCallInfo
for i := 0; i < maxShellToolIterations; i++ {
resp, err := orb.SendWithTools(messages)
if err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
choice := resp.Choices[0]
content := cleanThinkingTags(choice.Message.Content)
if content != "" {
finalContent = content
}
if len(choice.Message.ToolCalls) == 0 {
break
}
assistantMsg := orchestrator.Message{
Role: "assistant",
Content: content,
ToolCalls: choice.Message.ToolCalls,
}
messages = append(messages, assistantMsg)
for _, tc := range choice.Message.ToolCalls {
argsMap := make(map[string]interface{})
json.Unmarshal([]byte(tc.Function.Arguments), &argsMap)
tcInfo := ToolCallInfo{
ID: tc.ID,
Name: tc.Function.Name,
Args: argsMap,
}
call := agent.ToolCall{
ID: tc.ID,
Name: tc.Function.Name,
Arguments: json.RawMessage(tc.Function.Arguments),
}
result, execErr := s.agentRegistry.Execute(ctx, call)
if execErr != nil {
tcInfo.Error = execErr.Error()
} else {
tcInfo.Result = &toolResponseData{
Content: result.Content,
IsError: result.IsError,
Meta: result.Meta,
}
}
toolCalls = append(toolCalls, tcInfo)
messages = append(messages, orchestrator.Message{
Role: "tool",
Content: result.Content,
ToolCallID: tc.ID,
Name: tc.Function.Name,
})
}
finalContent = ""
}
if finalContent == "" && len(toolCalls) > 0 {
finalContent = "(tool calls completed, no text response)"
}
writeJSON(w, ShellChatResponse{
Content: finalContent,
ToolCalls: toolCalls,
})
}

View File

@@ -3,7 +3,9 @@ package api
import (
"encoding/json"
"net/http"
"sync"
"github.com/muyue/muyue/internal/installer"
"github.com/muyue/muyue/internal/scanner"
"github.com/muyue/muyue/internal/updater"
)
@@ -49,7 +51,30 @@ func (s *Server) handleInstall(w http.ResponseWriter, r *http.Request) {
writeError(w, "no tools specified", http.StatusBadRequest)
return
}
writeJSON(w, map[string]string{"status": "installing"})
results := make([]installer.InstallResult, len(body.Tools))
var wg sync.WaitGroup
var mu sync.Mutex
for i, tool := range body.Tools {
wg.Add(1)
go func(idx int, name string) {
defer wg.Done()
inst := installer.New(s.config)
res := inst.InstallTool(name)
mu.Lock()
results[idx] = res
mu.Unlock()
}(i, tool)
}
wg.Wait()
writeJSON(w, map[string]interface{}{
"status": "done",
"tools": body.Tools,
"results": results,
})
}
func (s *Server) handleRunUpdate(w http.ResponseWriter, r *http.Request) {

View File

@@ -1,21 +1,29 @@
package api
import (
"context"
"encoding/json"
"net/http"
"os/exec"
"strings"
"github.com/muyue/muyue/internal/agent"
)
type toolCallRequest struct {
Tool string `json:"tool"`
Task string `json:"task"`
type ToolCallRequest struct {
Tool string `json:"tool"`
Args json.RawMessage `json:"args"`
}
type toolResult struct {
Success bool `json:"success"`
Output string `json:"output"`
Error string `json:"error,omitempty"`
type ToolResult struct {
Success bool `json:"success"`
Tool string `json:"tool"`
Result *toolResponseData `json:"result,omitempty"`
Error string `json:"error,omitempty"`
}
type toolResponseData struct {
Content string `json:"content"`
IsError bool `json:"is_error"`
Meta map[string]string `json:"meta,omitempty"`
}
func (s *Server) handleToolCall(w http.ResponseWriter, r *http.Request) {
@@ -24,57 +32,54 @@ func (s *Server) handleToolCall(w http.ResponseWriter, r *http.Request) {
return
}
var req toolCallRequest
var req ToolCallRequest
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
writeError(w, err.Error(), http.StatusBadRequest)
return
}
if req.Tool != "crush" {
writeError(w, "unsupported tool: "+req.Tool, http.StatusBadRequest)
if req.Tool == "" {
writeError(w, "tool is required", http.StatusBadRequest)
return
}
if req.Task == "" {
writeError(w, "task is required", http.StatusBadRequest)
return
ctx := context.Background()
call := agent.ToolCall{
ID: generateMsgID(),
Name: req.Tool,
Arguments: req.Args,
}
result := executeTool(req.Tool, req.Task)
writeJSON(w, result)
}
func executeTool(tool, task string) toolResult {
var cmd *exec.Cmd
switch tool {
case "crush":
cmd = exec.Command("crush", "run", task)
default:
return toolResult{Success: false, Error: "unknown tool: " + tool}
}
output, err := cmd.CombinedOutput()
if err != nil {
return toolResult{
result, execErr := s.agentRegistry.Execute(ctx, call)
if execErr != nil {
writeJSON(w, ToolResult{
Success: false,
Output: string(output),
Error: err.Error(),
}
Tool: req.Tool,
Error: execErr.Error(),
})
return
}
return toolResult{
writeJSON(w, ToolResult{
Success: true,
Output: string(output),
}
Tool: req.Tool,
Result: &toolResponseData{
Content: result.Content,
IsError: result.IsError,
Meta: result.Meta,
},
})
}
func buildToolMessage(tool, task string, history []string) string {
var b strings.Builder
b.WriteString("TASK: " + task + "\n\n")
b.WriteString("CONVERSATION HISTORY:\n")
for _, msg := range history {
b.WriteString(strings.Repeat(" ", 4) + strings.Join(strings.Split(msg, "\n"), "\n"+strings.Repeat(" ", 4)) + "\n")
func (s *Server) handleToolList(w http.ResponseWriter, r *http.Request) {
if r.Method != "GET" {
writeError(w, "GET only", http.StatusMethodNotAllowed)
return
}
return b.String()
tools := s.agentRegistry.All()
writeJSON(w, map[string]interface{}{
"tools": tools,
"count": len(tools),
})
}

View File

@@ -0,0 +1,258 @@
package api
import (
"context"
"encoding/json"
"net/http"
"strings"
"github.com/muyue/muyue/internal/workflow"
)
func (s *Server) handleWorkflowCreate(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
writeError(w, "POST only", http.StatusMethodNotAllowed)
return
}
var body struct {
Name string `json:"name"`
Description string `json:"description"`
Type string `json:"type"`
}
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
writeError(w, err.Error(), http.StatusBadRequest)
return
}
if body.Name == "" {
writeError(w, "name is required", http.StatusBadRequest)
return
}
engine := s.workflowEngine
if engine == nil {
engine, _ = workflow.NewEngine(s.agentRegistry)
}
wf := engine.Create(body.Name, body.Description, body.Type, []workflow.Step{})
writeJSON(w, wf)
}
func (s *Server) handleWorkflowList(w http.ResponseWriter, r *http.Request) {
if r.Method != "GET" {
writeError(w, "GET only", http.StatusMethodNotAllowed)
return
}
engine := s.workflowEngine
if engine == nil {
engine, _ = workflow.NewEngine(s.agentRegistry)
}
workflows := engine.List()
writeJSON(w, map[string]interface{}{
"workflows": workflows,
"count": len(workflows),
})
}
func (s *Server) handleWorkflowGet(w http.ResponseWriter, r *http.Request) {
if r.Method != "GET" {
writeError(w, "GET only", http.StatusMethodNotAllowed)
return
}
id := strings.TrimPrefix(r.URL.Path, "/api/workflow/")
if id == "" {
writeError(w, "workflow id required", http.StatusBadRequest)
return
}
engine := s.workflowEngine
if engine == nil {
engine, _ = workflow.NewEngine(s.agentRegistry)
}
wf, ok := engine.Get(id)
if !ok {
writeError(w, "workflow not found", http.StatusNotFound)
return
}
writeJSON(w, wf)
}
func (s *Server) handleWorkflowDelete(w http.ResponseWriter, r *http.Request) {
if r.Method != "DELETE" {
writeError(w, "DELETE only", http.StatusMethodNotAllowed)
return
}
id := strings.TrimPrefix(r.URL.Path, "/api/workflow/")
if id == "" {
writeError(w, "workflow id required", http.StatusBadRequest)
return
}
engine := s.workflowEngine
if engine == nil {
engine, _ = workflow.NewEngine(s.agentRegistry)
}
if err := engine.Delete(id); err != nil {
writeError(w, err.Error(), http.StatusNotFound)
return
}
writeJSON(w, map[string]string{"status": "deleted"})
}
func (s *Server) handleWorkflowPlan(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
writeError(w, "POST only", http.StatusMethodNotAllowed)
return
}
var body struct {
Goal string `json:"goal"`
}
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
writeError(w, err.Error(), http.StatusBadRequest)
return
}
if body.Goal == "" {
writeError(w, "goal is required", http.StatusBadRequest)
return
}
planner, err := workflow.NewPlanner(s.config)
if err != nil {
writeError(w, err.Error(), http.StatusServiceUnavailable)
return
}
steps, err := planner.GeneratePlan(context.Background(), body.Goal)
if err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
engine := s.workflowEngine
if engine == nil {
engine, _ = workflow.NewEngine(s.agentRegistry)
}
wf := engine.Create("Plan: "+body.Goal[:min(len(body.Goal), 30)], body.Goal, "plan_execute", steps)
writeJSON(w, wf)
}
func (s *Server) handleWorkflowExecute(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
writeError(w, "POST only", http.StatusMethodNotAllowed)
return
}
id := strings.TrimPrefix(r.URL.Path, "/api/workflow/execute/")
if id == "" {
writeError(w, "workflow id required", http.StatusBadRequest)
return
}
engine := s.workflowEngine
if engine == nil {
engine, _ = workflow.NewEngine(s.agentRegistry)
}
wf, ok := engine.Get(id)
if !ok {
writeError(w, "workflow not found", http.StatusNotFound)
return
}
if r.URL.Query().Get("stream") == "true" {
s.handleWorkflowExecuteStream(w, engine, wf)
} else {
err := engine.Execute(context.Background(), id, nil)
if err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
wf, _ = engine.Get(id)
writeJSON(w, wf)
}
}
func (s *Server) handleWorkflowExecuteStream(w http.ResponseWriter, engine *workflow.Engine, wf *workflow.Workflow) {
w.Header().Set("Content-Type", "text/event-stream")
w.Header().Set("Cache-Control", "no-cache")
w.Header().Set("Connection", "keep-alive")
w.Header().Set("Access-Control-Allow-Origin", "*")
w.WriteHeader(http.StatusOK)
flusher, canFlush := w.(http.Flusher)
writeSSE := func(data map[string]interface{}) {
b, _ := json.Marshal(data)
w.Write([]byte("data: " + string(b) + "\n\n"))
if canFlush {
flusher.Flush()
}
}
go func() {
engine.Execute(context.Background(), wf.ID, func(step *workflow.Step, event string) {
writeSSE(map[string]interface{}{
"event": event,
"step": step,
})
})
wf, _ = engine.Get(wf.ID)
writeSSE(map[string]interface{}{
"event": "workflow_done",
"status": wf.Status,
"workflow": wf,
})
}()
}
func (s *Server) handleWorkflowApprove(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
writeError(w, "POST only", http.StatusMethodNotAllowed)
return
}
id := strings.TrimPrefix(r.URL.Path, "/api/workflow/approve/")
if id == "" {
writeError(w, "workflow id required", http.StatusBadRequest)
return
}
var body struct {
StepID string `json:"step_id"`
}
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
writeError(w, err.Error(), http.StatusBadRequest)
return
}
engine := s.workflowEngine
if engine == nil {
engine, _ = workflow.NewEngine(s.agentRegistry)
}
if err := engine.ApproveStep(id, body.StepID); err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
writeJSON(w, map[string]string{"status": "approved"})
}
func min(a, b int) int {
if a < b {
return a
}
return b
}

View File

@@ -8,6 +8,7 @@ import (
"github.com/muyue/muyue/internal/agent"
"github.com/muyue/muyue/internal/config"
"github.com/muyue/muyue/internal/scanner"
"github.com/muyue/muyue/internal/workflow"
)
type Server struct {
@@ -17,6 +18,7 @@ type Server struct {
convStore *ConversationStore
agentRegistry *agent.Registry
agentToolsJSON json.RawMessage
workflowEngine *workflow.Engine
}
func NewServer(cfg *config.MuyueConfig) *Server {
@@ -30,6 +32,7 @@ func NewServer(cfg *config.MuyueConfig) *Server {
tools := s.agentRegistry.OpenAITools()
toolsJSON, _ := json.Marshal(tools)
s.agentToolsJSON = json.RawMessage(toolsJSON)
s.workflowEngine, _ = workflow.NewEngine(s.agentRegistry)
s.routes()
return s
}
@@ -64,6 +67,34 @@ func (s *Server) routes() {
s.mux.HandleFunc("/api/chat", s.handleChat)
s.mux.HandleFunc("/api/chat/history", s.handleChatHistory)
s.mux.HandleFunc("/api/chat/clear", s.handleChatClear)
s.mux.HandleFunc("/api/tool/call", s.handleToolCall)
s.mux.HandleFunc("/api/tools/list", s.handleToolList)
s.mux.HandleFunc("/api/shell/chat", s.handleShellChat)
s.mux.HandleFunc("/api/workflow", s.handleWorkflowCreate)
s.mux.HandleFunc("/api/workflow/list", s.handleWorkflowList)
s.mux.HandleFunc("/api/workflow/", s.handleWorkflowGet)
s.mux.HandleFunc("/api/workflow/plan", s.handleWorkflowPlan)
s.mux.HandleFunc("/api/workflow/execute/", s.handleWorkflowExecute)
s.mux.HandleFunc("/api/workflow/approve/", s.handleWorkflowApprove)
s.mux.HandleFunc("/api/conversations", s.handleListConversations)
s.mux.HandleFunc("/api/conversations/search", s.handleSearchConversations)
s.mux.HandleFunc("/api/conversations/export", s.handleExportConversation)
s.mux.HandleFunc("/api/conversations/", s.handleDeleteConversation)
s.mux.HandleFunc("/api/lsp/install", s.handleLSPInstall)
s.mux.HandleFunc("/api/skills/deploy", s.handleSkillsDeploy)
s.mux.HandleFunc("/api/ssh/connections", s.handleSSHConnections)
s.mux.HandleFunc("/api/ssh/test", s.handleSSHTest)
s.mux.HandleFunc("/api/mcp/status", s.handleMCPStatus)
s.mux.HandleFunc("/api/mcp/registry", s.handleMCPRegistry)
s.mux.HandleFunc("/api/lsp/health", s.handleLSPHealth)
s.mux.HandleFunc("/api/lsp/auto-install", s.handleLSPAutoInstall)
s.mux.HandleFunc("/api/lsp/editor-config", s.handleLSPEditorConfig)
s.mux.HandleFunc("/api/skills/validate", s.handleSkillValidate)
s.mux.HandleFunc("/api/skills/test", s.handleSkillTest)
s.mux.HandleFunc("/api/skills/export", s.handleSkillExport)
s.mux.HandleFunc("/api/skills/import", s.handleSkillImport)
s.mux.HandleFunc("/api/dashboard/status", s.handleDashboardStatus)
}
func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {

View File

@@ -1,9 +1,13 @@
package lsp
import (
"encoding/json"
"fmt"
"os"
"os/exec"
"path/filepath"
"strings"
"time"
)
type LSPServer struct {
@@ -12,6 +16,10 @@ type LSPServer struct {
Command string `json:"command"`
InstallCmd string `json:"install_cmd"`
Installed bool `json:"installed"`
Version string `json:"version,omitempty"`
Healthy bool `json:"healthy,omitempty"`
Description string `json:"description,omitempty"`
Category string `json:"category,omitempty"`
}
var knownServers = []LSPServer{
@@ -39,27 +47,131 @@ func ScanServers() []LSPServer {
servers[i] = s
_, err := exec.LookPath(s.Command)
servers[i].Installed = err == nil
servers[i].Version = getInstalledLSPVersion(s.Name)
}
regServers, err := scanLSPRegistryServers()
if err == nil {
servers = append(servers, regServers...)
}
return servers
}
func scanLSPRegistryServers() ([]LSPServer, error) {
reg, err := LoadLSPRegistry()
if err != nil {
return nil, err
}
knownNames := map[string]bool{}
for _, s := range knownServers {
knownNames[s.Name] = true
}
var servers []LSPServer
for _, rs := range reg.Servers {
if knownNames[rs.Name] {
continue
}
servers = append(servers, LSPServer{
Name: rs.Name,
Language: rs.Language,
Command: rs.Command,
InstallCmd: rs.InstallCmd,
Installed: isLSPCommandAvailable(rs.Command),
Description: rs.Description,
Category: rs.Category,
Version: getInstalledLSPVersion(rs.Name),
})
}
return servers, nil
}
func isLSPCommandAvailable(cmd string) bool {
_, err := exec.LookPath(cmd)
return err == nil
}
func getInstalledLSPVersion(name string) string {
home, _ := os.UserHomeDir()
if home == "" {
return ""
}
receiptPath := filepath.Join(home, ".muyue", "receipts", "lsp", 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 saveLSPReceipt(name, version string) error {
home, _ := os.UserHomeDir()
if home == "" {
return nil
}
receiptDir := filepath.Join(home, ".muyue", "receipts", "lsp")
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 InstallServer(name string) error {
for _, s := range knownServers {
if s.Name == name {
if s.InstallCmd == "" {
return fmt.Errorf("%s has no auto-install command, install manually", name)
}
cmd := exec.Command("bash", "-c", s.InstallCmd)
cmd.Env = os.Environ()
if output, err := cmd.CombinedOutput(); err != nil {
return fmt.Errorf("install %s: %s: %w", name, string(output), err)
}
return nil
return doInstallLSP(s)
}
}
reg, err := LoadLSPRegistry()
if err == nil {
for _, s := range reg.Servers {
if s.Name == name {
return doInstallLSP(LSPServer{
Name: s.Name,
Language: s.Language,
Command: s.Command,
InstallCmd: s.InstallCmd,
})
}
}
}
return fmt.Errorf("unknown LSP server: %s", name)
}
func doInstallLSP(s LSPServer) error {
if s.InstallCmd == "" {
return fmt.Errorf("%s has no auto-install command, install manually", s.Name)
}
cmd := exec.Command("bash", "-c", s.InstallCmd)
cmd.Env = os.Environ()
if output, err := cmd.CombinedOutput(); err != nil {
return fmt.Errorf("install %s: %s: %w", s.Name, string(output), err)
}
saveLSPReceipt(s.Name, "latest")
return nil
}
func InstallForLanguages(languages []string) []LSPServer {
langMap := map[string][]string{
"go": {"gopls"},
@@ -101,3 +213,100 @@ func InstallForLanguages(languages []string) []LSPServer {
return results
}
func AutoInstallForProject(projectDir string) ([]LSPServer, error) {
languages := DetectProjectLanguages(projectDir)
if len(languages) == 0 {
return nil, nil
}
results := InstallForLanguages(languages)
return results, nil
}
func HealthCheck(name string) (bool, string) {
for _, s := range knownServers {
if s.Name == name {
return healthCheckServer(s)
}
}
return false, "unknown server"
}
func healthCheckServer(s LSPServer) (bool, string) {
path, err := exec.LookPath(s.Command)
if err != nil {
return false, fmt.Sprintf("command %q not found in PATH", s.Command)
}
versionArgs := map[string][]string{
"gopls": {"version"},
"pyright": {"--version"},
"typescript-language-server": {"--version"},
"rust-analyzer": {"--version"},
"clangd": {"--version"},
"lua-language-server": {"--version"},
"bash-language-server": {"--version"},
"yaml-language-server": {"--version"},
}
if args, ok := versionArgs[s.Command]; ok {
cmd := exec.Command(path, args...)
output, err := cmd.CombinedOutput()
if err != nil {
return true, fmt.Sprintf("installed at %s but version check failed", path)
}
version := strings.TrimSpace(string(output))
if idx := strings.Index(version, "\n"); idx > 0 {
version = version[:idx]
}
saveLSPReceipt(s.Name, version)
return true, version
}
return true, fmt.Sprintf("installed at %s", path)
}
func GenerateEditorConfigs(servers []LSPServer, editor string, homeDir string) (string, error) {
if homeDir == "" {
home, _ := os.UserHomeDir()
homeDir = home
}
reg, err := LoadLSPRegistry()
if err != nil {
return "", err
}
regMap := map[string]RegistryEntry{}
for _, s := range reg.Servers {
regMap[s.Name] = s
}
var regEntries []RegistryEntry
for _, s := range servers {
if re, ok := regMap[s.Name]; ok {
regEntries = append(regEntries, re)
}
}
switch editor {
case "neovim", "nvim":
return GenerateNeovimConfig(regEntries), nil
case "helix", "hx":
return GenerateHelixConfig(regEntries), nil
case "vscode", "code", "cursor":
exts := GenerateVSCodeRecommendations(regEntries)
var b strings.Builder
b.WriteString("{\n \"recommendations\": [\n")
for i, ext := range exts {
if i > 0 {
b.WriteString(",\n")
}
b.WriteString(" \"" + ext + "\"")
}
b.WriteString("\n ]\n}")
return b.String(), nil
default:
return "", fmt.Errorf("unsupported editor: %s", editor)
}
}

333
internal/lsp/registry.go Normal file
View File

@@ -0,0 +1,333 @@
package lsp
import (
"os"
"path/filepath"
"time"
"gopkg.in/yaml.v3"
)
type RegistryEntry struct {
Name string `yaml:"name" json:"name"`
Language string `yaml:"language" json:"language"`
Description string `yaml:"description" json:"description"`
Command string `yaml:"command" json:"command"`
InstallCmd string `yaml:"install_cmd" json:"install_cmd"`
InstallType string `yaml:"install_type" json:"install_type"`
Category string `yaml:"category" json:"category"`
FilePatterns []string `yaml:"file_patterns,omitempty" json:"file_patterns,omitempty"`
ConfigFiles []string `yaml:"config_files,omitempty" json:"config_files,omitempty"`
Tags []string `yaml:"tags,omitempty" json:"tags,omitempty"`
HomePage string `yaml:"homepage,omitempty" json:"homepage,omitempty"`
NeovimSetup string `yaml:"neovim_setup,omitempty" json:"neovim_setup,omitempty"`
HelixLanguage string `yaml:"helix_language,omitempty" json:"helix_language,omitempty"`
}
type LSPRegistry struct {
SchemaVersion string `yaml:"schema_version"`
UpdatedAt time.Time `yaml:"updated_at"`
Servers []RegistryEntry `yaml:"servers"`
}
func DefaultLSPRegistry() *LSPRegistry {
return &LSPRegistry{
SchemaVersion: "v1",
UpdatedAt: time.Now(),
Servers: []RegistryEntry{
{
Name: "gopls", Language: "go", Description: "Go language server",
Command: "gopls", InstallCmd: "go install golang.org/x/tools/gopls@latest",
InstallType: "go", Category: "lsp", FilePatterns: []string{"*.go"},
ConfigFiles: []string{"go.mod"}, Tags: []string{"go", "linting", "completion"},
HomePage: "https://github.com/golang/tools",
NeovimSetup: `lspconfig.gopls.setup{}`,
HelixLanguage: "go",
},
{
Name: "pyright", Language: "python", Description: "Python type checker and language server",
Command: "pyright", InstallCmd: "npm install -g pyright",
InstallType: "npm", Category: "lsp", FilePatterns: []string{"*.py", "*.pyi"},
ConfigFiles: []string{"requirements.txt", "pyproject.toml", "setup.py", "Pipfile"},
Tags: []string{"python", "type-checking"}, HomePage: "https://github.com/microsoft/pyright",
NeovimSetup: `lspconfig.pyright.setup{}`,
HelixLanguage: "python",
},
{
Name: "typescript-language-server", Language: "typescript", Description: "TypeScript and JavaScript language server",
Command: "typescript-language-server", InstallCmd: "npm install -g typescript-language-server typescript",
InstallType: "npm", Category: "lsp", FilePatterns: []string{"*.ts", "*.tsx", "*.js", "*.jsx"},
ConfigFiles: []string{"tsconfig.json", "package.json"},
Tags: []string{"typescript", "javascript"}, HomePage: "https://github.com/typescript-language-server/typescript-language-server",
NeovimSetup: `lspconfig.tsserver.setup{}`,
HelixLanguage: "typescript",
},
{
Name: "vscode-json-language-server", Language: "json", Description: "JSON language server",
Command: "vscode-json-language-server", InstallCmd: "npm install -g vscode-langservers-extracted",
InstallType: "npm", Category: "lsp", FilePatterns: []string{"*.json", "*.jsonc"},
Tags: []string{"json"}, NeovimSetup: `lspconfig.jsonls.setup{}`,
HelixLanguage: "json",
},
{
Name: "vscode-html-language-server", Language: "html", Description: "HTML language server",
Command: "vscode-html-language-server", InstallCmd: "npm install -g vscode-langservers-extracted",
InstallType: "npm", Category: "lsp", FilePatterns: []string{"*.html", "*.htm"},
Tags: []string{"html"}, NeovimSetup: `lspconfig.html.setup{}`,
HelixLanguage: "html",
},
{
Name: "vscode-css-language-server", Language: "css", Description: "CSS/SCSS/LESS language server",
Command: "vscode-css-language-server", InstallCmd: "npm install -g vscode-langservers-extracted",
InstallType: "npm", Category: "lsp", FilePatterns: []string{"*.css", "*.scss", "*.less"},
Tags: []string{"css"}, NeovimSetup: `lspconfig.cssls.setup{}`,
HelixLanguage: "css",
},
{
Name: "yaml-language-server", Language: "yaml", Description: "YAML language server",
Command: "yaml-language-server", InstallCmd: "npm install -g yaml-language-server",
InstallType: "npm", Category: "lsp", FilePatterns: []string{"*.yml", "*.yaml"},
Tags: []string{"yaml"}, NeovimSetup: `lspconfig.yamlls.setup{}`,
HelixLanguage: "yaml",
},
{
Name: "bash-language-server", Language: "bash", Description: "Bash language server",
Command: "bash-language-server", InstallCmd: "npm install -g bash-language-server",
InstallType: "npm", Category: "lsp", FilePatterns: []string{"*.sh", "*.bash"},
Tags: []string{"bash", "shell"}, NeovimSetup: `lspconfig.bashls.setup{}`,
HelixLanguage: "bash",
},
{
Name: "rust-analyzer", Language: "rust", Description: "Rust language server",
Command: "rust-analyzer", InstallCmd: "rustup component add rust-analyzer",
InstallType: "rustup", Category: "lsp", FilePatterns: []string{"*.rs"},
ConfigFiles: []string{"Cargo.toml"}, Tags: []string{"rust"},
HomePage: "https://github.com/rust-lang/rust-analyzer",
NeovimSetup: `lspconfig.rust_analyzer.setup{}`,
HelixLanguage: "rust",
},
{
Name: "clangd", Language: "c/c++", Description: "C/C++ language server",
Command: "clangd", InstallCmd: "", InstallType: "system",
Category: "lsp", FilePatterns: []string{"*.c", "*.cpp", "*.h", "*.hpp"},
ConfigFiles: []string{"CMakeLists.txt", "Makefile"}, Tags: []string{"c", "cpp"},
NeovimSetup: `lspconfig.clangd.setup{}`,
HelixLanguage: "c",
},
{
Name: "lua-language-server", Language: "lua", Description: "Lua language server",
Command: "lua-language-server", InstallCmd: "npm install -g lua-language-server",
InstallType: "npm", Category: "lsp", FilePatterns: []string{"*.lua"},
Tags: []string{"lua"}, NeovimSetup: `lspconfig.lua_ls.setup{}`,
HelixLanguage: "lua",
},
{
Name: "dockerfile-language-server", Language: "dockerfile", Description: "Dockerfile language server",
Command: "docker-langserver", InstallCmd: "npm install -g dockerfile-language-server-nodejs",
InstallType: "npm", Category: "lsp", FilePatterns: []string{"Dockerfile", "Dockerfile.*"},
Tags: []string{"docker"}, NeovimSetup: `lspconfig.dockerls.setup{}`,
HelixLanguage: "dockerfile",
},
{
Name: "tailwindcss-language-server", Language: "tailwind", Description: "Tailwind CSS language server",
Command: "tailwindcss-language-server", InstallCmd: "npm install -g @tailwindcss/language-server",
InstallType: "npm", Category: "lsp", FilePatterns: []string{"*.html", "*.tsx", "*.jsx"},
ConfigFiles: []string{"tailwind.config.js", "tailwind.config.ts"},
Tags: []string{"tailwind", "css"}, NeovimSetup: `lspconfig.tailwindcss.setup{}`,
},
{
Name: "svelte-language-server", Language: "svelte", Description: "Svelte language server",
Command: "svelteserver", InstallCmd: "npm install -g svelte-language-server",
InstallType: "npm", Category: "lsp", FilePatterns: []string{"*.svelte"},
Tags: []string{"svelte"}, NeovimSetup: `lspconfig.svelte.setup{}`,
HelixLanguage: "svelte",
},
{
Name: "vue-language-server", Language: "vue", Description: "Vue language server",
Command: "vue-language-server", InstallCmd: "npm install -g @vue/language-server",
InstallType: "npm", Category: "lsp", FilePatterns: []string{"*.vue"},
Tags: []string{"vue"}, NeovimSetup: `lspconfig.vuels.setup{}`,
},
{
Name: "golangci-lint-langserver", Language: "go-lint", Description: "Go linter language server",
Command: "golangci-lint-langserver", InstallCmd: "go install github.com/nametake/golangci-lint-langserver@latest",
InstallType: "go", Category: "lsp", FilePatterns: []string{"*.go"},
Tags: []string{"go", "linting"},
},
},
}
}
var lspRegistryPath string
func init() {
home, _ := os.UserHomeDir()
if home != "" {
lspRegistryPath = filepath.Join(home, ".muyue", "lsp-registry.yaml")
}
}
func SetLSPRegistryPath(p string) {
lspRegistryPath = p
}
func LoadLSPRegistry() (*LSPRegistry, error) {
if lspRegistryPath == "" {
return DefaultLSPRegistry(), nil
}
data, err := os.ReadFile(lspRegistryPath)
if err != nil {
return DefaultLSPRegistry(), nil
}
var reg LSPRegistry
if err := yaml.Unmarshal(data, &reg); err != nil {
return nil, err
}
return &reg, nil
}
func SaveLSPRegistry(reg *LSPRegistry) error {
if lspRegistryPath == "" {
return nil
}
reg.UpdatedAt = time.Now()
data, err := yaml.Marshal(reg)
if err != nil {
return err
}
os.MkdirAll(filepath.Dir(lspRegistryPath), 0755)
return os.WriteFile(lspRegistryPath, data, 0644)
}
func InitLSPRegistry() error {
if lspRegistryPath == "" {
return nil
}
if _, err := os.Stat(lspRegistryPath); err == nil {
return nil
}
return SaveLSPRegistry(DefaultLSPRegistry())
}
func DetectProjectLanguages(projectDir string) []string {
if projectDir == "" {
return nil
}
langDetectors := map[string][]string{
"go": {"go.mod", "go.sum"},
"python": {"requirements.txt", "pyproject.toml", "setup.py", "Pipfile", "uv.lock"},
"typescript": {"tsconfig.json", "package.json"},
"javascript": {"package.json"},
"rust": {"Cargo.toml"},
"ruby": {"Gemfile"},
"java": {"pom.xml", "build.gradle"},
"c": {"CMakeLists.txt", "Makefile"},
"cpp": {"CMakeLists.txt"},
"php": {"composer.json"},
"lua": {".luarc.json"},
"docker": {"Dockerfile"},
}
extDetectors := map[string]string{
".go": "go",
".py": "python",
".rs": "rust",
".ts": "typescript",
".tsx": "typescript",
".js": "javascript",
".jsx": "javascript",
".rb": "ruby",
".java": "java",
".c": "c",
".cpp": "cpp",
".h": "c",
".lua": "lua",
".vue": "vue",
".svelte": "svelte",
}
detected := map[string]bool{}
for lang, files := range langDetectors {
for _, f := range files {
if _, err := os.Stat(filepath.Join(projectDir, f)); err == nil {
detected[lang] = true
break
}
}
}
entries, err := os.ReadDir(projectDir)
if err == nil {
for _, e := range entries {
if e.IsDir() {
continue
}
ext := filepath.Ext(e.Name())
if lang, ok := extDetectors[ext]; ok {
detected[lang] = true
}
}
}
var languages []string
for lang := range detected {
languages = append(languages, lang)
}
return languages
}
func GenerateNeovimConfig(servers []RegistryEntry) string {
config := `-- Generated by Muyue LSP Manager
-- Add to your init.lua or require from lspconfig setup
local lspconfig = require('lspconfig')
`
for _, s := range servers {
if s.NeovimSetup != "" {
config += s.NeovimSetup + "\n"
}
}
return config
}
func GenerateHelixConfig(servers []RegistryEntry) string {
config := `# Generated by Muyue LSP Manager
# Add to ~/.config/helix/languages.toml
`
for _, s := range servers {
if s.HelixLanguage != "" {
config += "[[language]]\n"
config += "name = \"" + s.HelixLanguage + "\"\n"
config += "language-servers = [\"" + s.Name + "\"]\n\n"
}
}
return config
}
func GenerateVSCodeRecommendations(servers []RegistryEntry) []string {
extensionMap := map[string][]string{
"gopls": {"golang.go"},
"pyright": {"ms-python.python", "ms-python.vscode-pylance"},
"typescript-language-server": {"ms-vscode.vscode-typescript-next"},
"rust-analyzer": {"rust-lang.rust-analyzer"},
"lua-language-server": {"sumneko.lua"},
"tailwindcss-language-server": {"bradlc.vscode-tailwindcss"},
"svelte-language-server": {"svelte.svelte-vscode"},
"vue-language-server": {"vue.volar"},
"yaml-language-server": {"redhat.vscode-yaml"},
"bash-language-server": {"mads-hartmann.bash-ide-vscode"},
}
var extensions []string
for _, s := range servers {
if exts, ok := extensionMap[s.Name]; ok {
extensions = append(extensions, exts...)
}
}
return extensions
}

View File

@@ -0,0 +1,142 @@
package lsp
import (
"os"
"path/filepath"
"testing"
)
func TestDefaultLSPRegistry(t *testing.T) {
reg := DefaultLSPRegistry()
if reg.SchemaVersion != "v1" {
t.Errorf("Expected v1, got %s", reg.SchemaVersion)
}
if len(reg.Servers) == 0 {
t.Error("Default LSP 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)
}
if s.Language == "" {
t.Errorf("Server %s missing language", s.Name)
}
}
}
func TestSaveAndLoadLSPRegistry(t *testing.T) {
tmpDir := t.TempDir()
SetLSPRegistryPath(filepath.Join(tmpDir, "lsp-registry.yaml"))
reg := DefaultLSPRegistry()
if err := SaveLSPRegistry(reg); err != nil {
t.Fatalf("SaveLSPRegistry failed: %v", err)
}
loaded, err := LoadLSPRegistry()
if err != nil {
t.Fatalf("LoadLSPRegistry failed: %v", err)
}
if len(loaded.Servers) != len(reg.Servers) {
t.Errorf("Expected %d servers, got %d", len(reg.Servers), len(loaded.Servers))
}
}
func TestInitLSPRegistry(t *testing.T) {
tmpDir := t.TempDir()
SetLSPRegistryPath(filepath.Join(tmpDir, "lsp-reg.yaml"))
if err := InitLSPRegistry(); err != nil {
t.Fatalf("InitLSPRegistry failed: %v", err)
}
if _, err := os.Stat(filepath.Join(tmpDir, "lsp-reg.yaml")); os.IsNotExist(err) {
t.Error("LSP registry file should be created")
}
}
func TestDetectProjectLanguages(t *testing.T) {
tmpDir := t.TempDir()
os.WriteFile(filepath.Join(tmpDir, "go.mod"), []byte("module test\ngo 1.24\n"), 0644)
os.WriteFile(filepath.Join(tmpDir, "package.json"), []byte(`{"name": "test"}`), 0644)
languages := DetectProjectLanguages(tmpDir)
if len(languages) == 0 {
t.Error("Should detect languages")
}
langSet := map[string]bool{}
for _, l := range languages {
langSet[l] = true
}
if !langSet["go"] {
t.Error("Should detect Go")
}
if !langSet["typescript"] {
t.Error("Should detect TypeScript/JS from package.json")
}
}
func TestDetectProjectLanguagesEmpty(t *testing.T) {
tmpDir := t.TempDir()
languages := DetectProjectLanguages(tmpDir)
if len(languages) != 0 {
t.Errorf("Empty dir should detect no languages, got %v", languages)
}
}
func TestGenerateNeovimConfig(t *testing.T) {
servers := []RegistryEntry{
{Name: "gopls", Language: "go", NeovimSetup: "lspconfig.gopls.setup{}"},
{Name: "pyright", Language: "python", NeovimSetup: "lspconfig.pyright.setup{}"},
}
config := GenerateNeovimConfig(servers)
if config == "" {
t.Error("Config should not be empty")
}
if len(config) < 50 {
t.Error("Config seems too short")
}
}
func TestGenerateHelixConfig(t *testing.T) {
servers := []RegistryEntry{
{Name: "gopls", Language: "go", HelixLanguage: "go"},
}
config := GenerateHelixConfig(servers)
if config == "" {
t.Error("Config should not be empty")
}
}
func TestGenerateVSCodeRecommendations(t *testing.T) {
servers := []RegistryEntry{
{Name: "gopls", Language: "go"},
{Name: "pyright", Language: "python"},
}
exts := GenerateVSCodeRecommendations(servers)
if len(exts) == 0 {
t.Error("Should return some extensions")
}
}
func TestHealthCheck(t *testing.T) {
healthy, detail := HealthCheck("gopls")
if healthy && detail == "" {
t.Error("If healthy, should have version detail")
}
}
func TestHealthCheckUnknown(t *testing.T) {
_, _ = HealthCheck("nonexistent-server")
}

View File

@@ -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
View 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, &reg); err != nil {
return nil, fmt.Errorf("parse registry: %w", err)
}
return &reg, 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 = &reg.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
}

View 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")
}
}

View File

@@ -158,48 +158,9 @@ func (o *Orchestrator) Send(userMessage string) (string, error) {
}
o.histMu.Unlock()
body, err := json.Marshal(reqBody)
chatResp, providerName, err := o.sendWithFallback(reqBody, "")
if err != nil {
return "", fmt.Errorf("marshal request: %w", err)
}
baseURL := o.provider.BaseURL
if baseURL == "" {
baseURL = getProviderBaseURL(o.provider.Name)
}
url := strings.TrimRight(baseURL, "/") + "/chat/completions"
req, err := http.NewRequest("POST", url, bytes.NewReader(body))
if err != nil {
return "", fmt.Errorf("create request: %w", err)
}
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Authorization", "Bearer "+o.provider.APIKey)
resp, err := o.client.Do(req)
if err != nil {
return "", fmt.Errorf("send request: %w", err)
}
defer resp.Body.Close()
respBody, err := io.ReadAll(resp.Body)
if err != nil {
return "", fmt.Errorf("read response: %w", err)
}
if resp.StatusCode != http.StatusOK {
return "", fmt.Errorf("API error (%d): %s", resp.StatusCode, string(respBody))
}
var chatResp ChatResponse
if err := json.Unmarshal(respBody, &chatResp); err != nil {
return "", fmt.Errorf("parse response: %w", err)
}
if len(chatResp.Choices) == 0 {
return "", fmt.Errorf("no response from AI")
return "", err
}
content := cleanAIResponse(chatResp.Choices[0].Message.Content)
@@ -208,6 +169,7 @@ func (o *Orchestrator) Send(userMessage string) (string, error) {
Role: "assistant",
Content: content,
})
_ = providerName
o.histMu.Unlock()
return content, nil
@@ -326,51 +288,16 @@ func (o *Orchestrator) SendWithTools(messages []Message) (*ChatResponse, error)
Tools: o.tools,
}
body, err := json.Marshal(reqBody)
chatResp, _, err := o.sendWithFallback(reqBody, "")
if err != nil {
return nil, fmt.Errorf("marshal request: %w", err)
}
baseURL := o.provider.BaseURL
if baseURL == "" {
baseURL = getProviderBaseURL(o.provider.Name)
}
url := strings.TrimRight(baseURL, "/") + "/chat/completions"
req, err := http.NewRequest("POST", url, bytes.NewReader(body))
if err != nil {
return nil, fmt.Errorf("create request: %w", err)
}
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Authorization", "Bearer "+o.provider.APIKey)
resp, err := o.client.Do(req)
if err != nil {
return nil, fmt.Errorf("send request: %w", err)
}
defer resp.Body.Close()
respBody, err := io.ReadAll(resp.Body)
if err != nil {
return nil, fmt.Errorf("read response: %w", err)
}
if resp.StatusCode != http.StatusOK {
return nil, fmt.Errorf("API error (%d): %s", resp.StatusCode, string(respBody))
}
var chatResp ChatResponse
if err := json.Unmarshal(respBody, &chatResp); err != nil {
return nil, fmt.Errorf("parse response: %w", err)
return nil, err
}
if len(chatResp.Choices) == 0 {
return nil, fmt.Errorf("no response from AI")
}
return &chatResp, nil
return chatResp, nil
}
func cleanAIResponse(content string) string {
@@ -411,3 +338,94 @@ func getProviderBaseURL(name string) string {
return ""
}
}
func (o *Orchestrator) getAvailableProviders() []*config.AIProvider {
var providers []*config.AIProvider
for i := range o.config.AI.Providers {
prov := &o.config.AI.Providers[i]
if prov.APIKey != "" {
providers = append(providers, prov)
}
}
return providers
}
func (o *Orchestrator) sendWithFallback(reqBody ChatRequest, baseURLOverride string) (*ChatResponse, string, error) {
providers := o.getAvailableProviders()
if len(providers) == 0 {
return nil, "", fmt.Errorf("no providers available")
}
providerOrder := make([]*config.AIProvider, 0, len(providers))
if o.provider != nil {
providerOrder = append(providerOrder, o.provider)
}
for _, p := range providers {
if o.provider == nil || p.Name != o.provider.Name {
providerOrder = append(providerOrder, p)
}
}
var lastErr error
for _, prov := range providerOrder {
baseURL := baseURLOverride
if baseURL == "" {
baseURL = prov.BaseURL
if baseURL == "" {
baseURL = getProviderBaseURL(prov.Name)
}
}
url := strings.TrimRight(baseURL, "/") + "/chat/completions"
body, err := json.Marshal(reqBody)
if err != nil {
lastErr = fmt.Errorf("marshal request: %w", err)
continue
}
req, err := http.NewRequest("POST", url, bytes.NewReader(body))
if err != nil {
lastErr = fmt.Errorf("create request: %w", err)
continue
}
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Authorization", "Bearer "+prov.APIKey)
resp, err := o.client.Do(req)
if err != nil {
lastErr = fmt.Errorf("send request to %s: %w", prov.Name, err)
continue
}
defer resp.Body.Close()
respBody, err := io.ReadAll(resp.Body)
if err != nil {
lastErr = fmt.Errorf("read response: %w", err)
continue
}
if resp.StatusCode != http.StatusOK {
lastErr = fmt.Errorf("API error (%d): %s", resp.StatusCode, string(respBody))
continue
}
var chatResp ChatResponse
if err := json.Unmarshal(respBody, &chatResp); err != nil {
lastErr = fmt.Errorf("parse response: %w", err)
continue
}
if len(chatResp.Choices) == 0 {
lastErr = fmt.Errorf("no response from AI")
continue
}
o.provider = prov
return &chatResp, prov.Name, nil
}
return nil, "", lastErr
}

View File

@@ -11,9 +11,10 @@ var builtinSkills = []Skill{
Name: "env-setup",
Description: "Set up a complete development environment for any language. Detects missing tools, installs them, and configures the project.",
Author: "muyue",
Version: "1.0.0",
Version: "1.1.0",
Target: "both",
Tags: []string{"setup", "environment", "install"},
Category: "setup",
Content: `# Environment Setup
Use this skill when setting up a new development environment or project.
@@ -58,9 +59,14 @@ Use this skill when setting up a new development environment or project.
Name: "git-workflow",
Description: "Manage git branches, commits, and pull requests following best practices. Handles branching strategy, conventional commits, and PR creation.",
Author: "muyue",
Version: "1.0.0",
Version: "1.1.0",
Target: "both",
Tags: []string{"git", "workflow", "branching", "commits"},
Category: "workflow",
Dependencies: []SkillDependency{
{Type: "tool", Name: "git", Required: true},
{Type: "tool", Name: "gh", Required: false},
},
Content: `# Git Workflow
Use this skill when the user needs to create branches, make commits, or manage pull requests.
@@ -114,9 +120,10 @@ Follow Conventional Commits:
Name: "api-design",
Description: "Design and implement REST or GraphQL APIs following best practices. Includes endpoint design, error handling, and documentation.",
Author: "muyue",
Version: "1.0.0",
Version: "1.1.0",
Target: "both",
Tags: []string{"api", "rest", "graphql", "design"},
Category: "design",
Content: `# API Design
Use this skill when designing or implementing an API.
@@ -171,9 +178,10 @@ Use this skill when designing or implementing an API.
Name: "debug-assist",
Description: "Systematic debugging assistant. Helps identify, isolate, and fix bugs using a structured approach.",
Author: "muyue",
Version: "1.0.0",
Version: "1.1.0",
Target: "both",
Tags: []string{"debug", "troubleshooting", "bugs"},
Category: "debugging",
Content: `# Debug Assist
Use this skill when the user reports a bug or asks for help debugging.
@@ -188,7 +196,7 @@ Use this skill when the user reports a bug or asks for help debugging.
3. **Hypothesize** — Form a hypothesis about the root cause
4. **Verify** — Add logging or breakpoints to confirm
5. **Fix** — Make the minimal change to fix the issue
6. **Test** — Verify the fix works and doesn't break other things
6. **Test** — Verify the fix works and does not break other things
7. **Prevent** — Add a test to prevent regression
## Common Patterns
@@ -211,9 +219,10 @@ Use this skill when the user reports a bug or asks for help debugging.
Name: "code-review",
Description: "Perform a thorough code review. Checks for bugs, security issues, performance problems, and style consistency.",
Author: "muyue",
Version: "1.0.0",
Version: "1.1.0",
Target: "both",
Tags: []string{"review", "quality", "security"},
Category: "quality",
Content: `# Code Review
Use this skill when reviewing code changes or pull requests.
@@ -221,7 +230,7 @@ Use this skill when reviewing code changes or pull requests.
## Review Checklist
### Correctness
- Does the code do what it's supposed to?
- Does the code do what it is supposed to?
- Are edge cases handled?
- Are there off-by-one errors?
- Are error paths handled?
@@ -254,7 +263,7 @@ Use this skill when reviewing code changes or pull requests.
## Review Format
1. Summary of changes
2. Issues found (critical minor)
2. Issues found (critical to minor)
3. Suggestions for improvement
4. Positive observations
@@ -265,6 +274,351 @@ Use this skill when reviewing code changes or pull requests.
- **Minor**: Style issues, naming, minor refactoring opportunities
- **Suggestion**: Alternative approaches, improvements`,
},
{
Name: "docker-setup",
Description: "Set up Docker and docker-compose for a project with best practices including multi-stage builds, health checks, and proper networking.",
Author: "muyue",
Version: "1.0.0",
Target: "both",
Tags: []string{"docker", "containers", "devops", "compose"},
Category: "devops",
Dependencies: []SkillDependency{
{Type: "tool", Name: "docker", Required: true},
},
Content: `# Docker Setup
Use this skill when the user needs Docker configuration for a project.
## Dockerfile Best Practices
1. Use multi-stage builds to reduce image size:
- Builder stage: install dependencies, compile
- Runtime stage: copy only the binary/artifacts
2. Use specific base image tags (not ` + "`latest`" + `):
- ` + "`golang:1.24-alpine`" + ` for Go
- ` + "`node:22-slim`" + ` for Node.js
- ` + "`python:3.12-slim`" + ` for Python
3. Order layers for cache efficiency:
- Copy dependency files first (go.mod, package.json, requirements.txt)
- Install dependencies
- Copy source code last
4. Add health checks:
` + "```" + `dockerfile
HEALTHCHECK --interval=30s --timeout=3s CMD curl -f http://localhost:8080/health || exit 1
` + "```" + `
5. Run as non-root user:
` + "```" + `dockerfile
RUN adduser -D appuser
USER appuser
` + "```" + `
## docker-compose.yml Structure
` + "```" + `yaml
version: "3.9"
services:
app:
build: .
ports:
- "8080:8080"
environment:
- DATABASE_URL=postgres://user:pass@db:5432/app
depends_on:
db:
condition: service_healthy
healthcheck:
test: ["CMD", "curl", "-f", "http://localhost:8080/health"]
interval: 30s
timeout: 3s
retries: 3
db:
image: postgres:16-alpine
environment:
POSTGRES_USER: user
POSTGRES_PASSWORD: pass
POSTGRES_DB: app
volumes:
- pgdata:/var/lib/postgresql/data
healthcheck:
test: ["CMD-SHELL", "pg_isready -U user"]
interval: 10s
timeout: 3s
retries: 5
volumes:
pgdata:
` + "```" + `
## Error Handling
- If Docker is not installed, provide install instructions for the platform
- If port is already in use, suggest alternative ports
- If build fails, check for missing .dockerignore and suggest one`,
},
{
Name: "security-audit",
Description: "Perform a security audit on code, dependencies, and configuration. Checks for OWASP Top 10 vulnerabilities, dependency vulnerabilities, and misconfigurations.",
Author: "muyue",
Version: "1.0.0",
Target: "both",
Tags: []string{"security", "audit", "vulnerabilities", "owasp"},
Category: "security",
Content: `# Security Audit
Use this skill when the user needs a security review or vulnerability assessment.
## Audit Checklist
### Input Validation (OWASP A03:2021)
- All user input is validated and sanitized
- SQL queries use parameterized statements
- File paths are validated (no path traversal)
- Input length limits are enforced
### Authentication and Authorization (OWASP A07:2021)
- Passwords are hashed with bcrypt/argon2 (never MD5/SHA1)
- JWT tokens have short expiry with refresh rotation
- Session management is secure
- RBAC or ABAC is properly implemented
- API endpoints have proper auth checks
### Data Protection (OWASP A02:2021)
- Secrets are not in source code (use env vars or secret managers)
- Sensitive data is encrypted at rest and in transit
- PII is properly handled and not logged
- TLS is enforced for all connections
### Dependency Security (OWASP A06:2021)
- Run ` + "`npm audit`" + `, ` + "`pip audit`" + `, or ` + "`go vuln check`" + `
- Check for known CVEs in dependencies
- Keep dependencies up to date
- Use lock files for reproducible builds
### Configuration Security
- Debug mode is disabled in production
- CORS is properly configured
- Rate limiting is in place
- Security headers are set (CSP, HSTS, X-Frame-Options)
- Error messages do not leak internal details
## Automated Checks
Run these tools if available:
- ` + "`gosec ./...`" + ` for Go security
- ` + "`bandit -r .`" + ` for Python security
- ` + "`npm audit`" + ` for Node.js vulnerabilities
- ` + "`trivy fs .`" + ` for container/Dockerfile scanning
## Report Format
1. Executive Summary (risk level, total findings)
2. Critical findings (immediate action required)
3. High findings (fix within 24h)
4. Medium findings (fix within sprint)
5. Low findings (address when convenient)
6. Recommendations`,
},
{
Name: "mcp-setup",
Description: "Configure MCP (Model Context Protocol) servers for AI tools. Discovers, installs, and configures MCP servers across multiple editors.",
Author: "muyue",
Version: "1.0.0",
Target: "both",
Tags: []string{"mcp", "ai", "configuration", "editors"},
Category: "setup",
Dependencies: []SkillDependency{
{Type: "tool", Name: "npx", Required: true},
},
Content: `# MCP Server Setup
Use this skill when the user wants to configure MCP servers for their AI coding tools.
## Supported Editors
Muyue can generate MCP configs for:
- **Crush**: ` + "`~/.config/crush/crush.json`" + ` (key: ` + "`mcps`" + `)
- **Claude Code**: ` + "`~/.claude.json`" + ` (key: ` + "`mcpServers`" + `)
- **Cursor**: ` + "`~/.cursor/mcp.json`" + ` (key: ` + "`mcpServers`" + `, adds ` + "`type: stdio`" + `)
- **VS Code**: ` + "`~/.vscode/mcp.json`" + ` (key: ` + "`servers`" + `)
- **Windsurf**: ` + "`~/.windsurf/mcp.json`" + ` (key: ` + "`mcpServers`" + `)
## Common MCP Servers
| Server | Package | Required Env |
|--------|---------|-------------|
| filesystem | @modelcontextprotocol/server-filesystem | None |
| fetch | @modelcontextprotocol/server-fetch | None |
| github | @modelcontextprotocol/server-github | GITHUB_PERSONAL_ACCESS_TOKEN |
| brave-search | @modelcontextprotocol/server-brave-search | BRAVE_API_KEY |
| memory | @modelcontextprotocol/server-memory | None |
| postgres | @modelcontextprotocol/server-postgres | DATABASE_URL |
| sqlite | @modelcontextprotocol/server-sqlite | None |
| docker | @modelcontextprotocol/server-docker | None |
## Setup Steps
1. Ask which editors the user wants to configure
2. Ask which MCP servers they need
3. For servers requiring API keys, prompt for the key
4. Generate configs for each selected editor
5. Validate configs (check JSON is valid, commands exist)
6. Test connectivity if possible
## Credential Management
- API keys should be stored in the Muyue config (encrypted)
- When generating MCP configs, inject keys from the Muyue config
- Never hardcode API keys in config files in version control
- Suggest adding MCP config files to ` + "`.gitignore`" + `
## Troubleshooting
- If npx fails, suggest ` + "`npm install -g`" + ` the package
- If a server does not start, check the command and args
- If auth fails, verify the API key is correct and active`,
},
{
Name: "lsp-setup",
Description: "Configure Language Server Protocol servers for code intelligence. Detects project languages, installs LSPs, and generates editor configs.",
Author: "muyue",
Version: "1.0.0",
Target: "both",
Tags: []string{"lsp", "language-server", "ide", "configuration"},
Category: "setup",
Content: `# LSP Server Setup
Use this skill when the user wants to set up language servers for code intelligence.
## Supported Languages
| Language | Server | Install Method |
|----------|--------|---------------|
| Go | gopls | ` + "`go install`" + ` |
| Python | pyright | ` + "`npm install -g`" + ` |
| TypeScript/JS | typescript-language-server | ` + "`npm install -g`" + ` |
| Rust | rust-analyzer | ` + "`rustup component add`" + ` |
| C/C++ | clangd | System package |
| Lua | lua-language-server | ` + "`npm install -g`" + ` |
| HTML | vscode-html-language-server | ` + "`npm install -g vscode-langservers-extracted`" + ` |
| CSS | vscode-css-language-server | ` + "`npm install -g vscode-langservers-extracted`" + ` |
| JSON | vscode-json-language-server | ` + "`npm install -g vscode-langservers-extracted`" + ` |
| YAML | yaml-language-server | ` + "`npm install -g`" + ` |
| Bash | bash-language-server | ` + "`npm install -g`" + ` |
| Docker | dockerfile-language-server | ` + "`npm install -g`" + ` |
| Vue | vue-language-server | ` + "`npm install -g`" + ` |
| Svelte | svelte-language-server | ` + "`npm install -g`" + ` |
## Auto-Detection
Detect project languages from:
- Config files: ` + "`go.mod`" + `, ` + "`package.json`" + `, ` + "`Cargo.toml`" + `, ` + "`pyproject.toml`" + `
- Source file extensions: ` + "`*.go`" + `, ` + "`*.py`" + `, ` + "`*.ts`" + `, ` + "`*.rs`" + `
## Editor Config Generation
### Neovim
Generate ` + "`lspconfig`" + ` setup snippet for each LSP.
### Helix
Generate ` + "`languages.toml`" + ` entries with language-server mappings.
### VS Code / Cursor
Generate ` + "`extensions.json`" + ` recommendations for each LSP.
## Health Checks
After installation, verify:
1. The binary is in PATH
2. The version matches expected
3. A basic ` + "`initialize`" + ` request succeeds (if applicable)`,
},
{
Name: "workflow-design",
Description: "Design development workflows and automations. Creates CI/CD pipelines, git hooks, and development process documentation.",
Author: "muyue",
Version: "1.0.0",
Target: "both",
Tags: []string{"workflow", "ci-cd", "automation", "process"},
Category: "workflow",
Content: `# Workflow Design
Use this skill when the user wants to establish development workflows or CI/CD pipelines.
## CI/CD Pipeline Design
### GitHub Actions Template
` + "```" + `yaml
name: CI
on:
push:
branches: [main, develop]
pull_request:
branches: [main]
jobs:
lint:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- uses: actions/setup-go@v5
with: { go-version: "1.24" }
- run: go vet ./...
- run: golint ./...
test:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- uses: actions/setup-go@v5
with: { go-version: "1.24" }
- run: go test -race -coverprofile=coverage.out ./...
- run: go tool cover -func=coverage.out
build:
needs: [lint, test]
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- run: go build -o bin/app ./cmd/app
` + "```" + `
## Git Hooks
Use ` + "`pre-commit`" + ` framework:
- ` + "`pre-commit`" + `: lint, format check, trailing whitespace
- ` + "`commit-msg`" + `: validate conventional commit format
- ` + "`pre-push`" + `: run tests
## Branch Protection Rules
- Require PR reviews (at least 1 approval)
- Require status checks to pass
- Require up-to-date branch before merge
- Require linear history (rebase merge)
## Development Process
1. Pick a task from the backlog
2. Create a feature branch
3. Implement with tests
4. Run linter and tests locally
5. Push and create PR
6. Address review feedback
7. Merge when approved and CI passes
8. Delete feature branch
## Error Handling
- If CI fails, provide clear error output and suggested fixes
- If hooks fail, explain what failed and how to fix
- Suggest ` + "`--no-verify`" + ` only as a last resort, with a warning`,
},
}
func InstallBuiltinSkills() error {

View File

@@ -1,27 +1,53 @@
package skills
import (
"encoding/json"
"fmt"
"os"
"path/filepath"
"sort"
"regexp"
"strings"
"time"
"gopkg.in/yaml.v3"
)
type SkillDependency struct {
Type string `yaml:"type,omitempty" json:"type,omitempty"`
Name string `yaml:"name,omitempty" json:"name,omitempty"`
Version string `yaml:"version,omitempty" json:"version,omitempty"`
Required bool `yaml:"required,omitempty" json:"required,omitempty"`
}
type Skill struct {
Name string `yaml:"name" json:"name"`
Description string `yaml:"description" json:"description"`
Content string `yaml:"content" json:"content"`
Author string `yaml:"author" json:"author"`
Version string `yaml:"version" json:"version"`
CreatedAt time.Time `yaml:"created_at" json:"created_at"`
UpdatedAt time.Time `yaml:"updated_at" json:"updated_at"`
Tags []string `yaml:"tags" json:"tags"`
Target string `yaml:"target" json:"target"`
FilePath string `yaml:"-" json:"-"`
Name string `yaml:"name" json:"name"`
Description string `yaml:"description" json:"description"`
Content string `yaml:"content" json:"content"`
Author string `yaml:"author" json:"author"`
Version string `yaml:"version" json:"version"`
CreatedAt time.Time `yaml:"created_at" json:"created_at"`
UpdatedAt time.Time `yaml:"updated_at" json:"updated_at"`
Tags []string `yaml:"tags" json:"tags"`
Target string `yaml:"target" json:"target"`
FilePath string `yaml:"-" json:"-"`
Dependencies []SkillDependency `yaml:"dependencies,omitempty" json:"dependencies,omitempty"`
Languages []string `yaml:"languages,omitempty" json:"languages,omitempty"`
Category string `yaml:"category,omitempty" json:"category,omitempty"`
}
type ValidationError struct {
Field string `json:"field"`
Message string `json:"message"`
}
func (v ValidationError) Error() string {
return fmt.Sprintf("%s: %s", v.Field, v.Message)
}
type SkillTestResult struct {
Name string `json:"name"`
Passed bool `json:"passed"`
Message string `json:"message"`
}
func SkillsDir() (string, error) {
@@ -66,10 +92,6 @@ func List() ([]Skill, error) {
skills = append(skills, *skill)
}
sort.Slice(skills, func(i, j int) bool {
return skills[i].Name < skills[j].Name
})
return skills, nil
}
@@ -95,6 +117,10 @@ func Get(name string) (*Skill, error) {
}
func Create(skill *Skill) error {
if errs := Validate(skill); len(errs) > 0 {
return fmt.Errorf("validation failed: %v", errs)
}
dir, err := SkillsDir()
if err != nil {
return err
@@ -129,6 +155,28 @@ func Delete(name string) error {
return nil
}
func Update(skill *Skill) error {
if errs := Validate(skill); len(errs) > 0 {
return fmt.Errorf("validation failed: %v", errs)
}
dir, err := SkillsDir()
if err != nil {
return err
}
skillDir := filepath.Join(dir, skill.Name)
skillPath := filepath.Join(skillDir, "SKILL.md")
skill.UpdatedAt = time.Now()
content := renderSkill(skill)
if err := os.WriteFile(skillPath, []byte(content), 0644); err != nil {
return err
}
return Deploy(skill)
}
func Deploy(skill *Skill) error {
home, err := os.UserHomeDir()
if err != nil {
@@ -188,6 +236,206 @@ func undeployFromTargets(name string) {
os.RemoveAll(filepath.Join(home, ".claude", "skills", name))
}
func Validate(skill *Skill) []ValidationError {
var errs []ValidationError
if skill.Name == "" {
errs = append(errs, ValidationError{Field: "name", Message: "name is required"})
}
if skill.Name != "" {
if matched, _ := regexp.MatchString(`^[a-z0-9][a-z0-9-]*$`, skill.Name); !matched {
errs = append(errs, ValidationError{Field: "name", Message: "name must be lowercase alphanumeric with dashes"})
}
}
if skill.Description == "" {
errs = append(errs, ValidationError{Field: "description", Message: "description is required"})
}
if skill.Content == "" {
errs = append(errs, ValidationError{Field: "content", Message: "content is required"})
}
if skill.Target != "" && skill.Target != "crush" && skill.Target != "claude" && skill.Target != "both" {
errs = append(errs, ValidationError{Field: "target", Message: "target must be crush, claude, or both"})
}
if skill.Version != "" {
if matched, _ := regexp.MatchString(`^\d+\.\d+\.\d+$`, skill.Version); !matched {
errs = append(errs, ValidationError{Field: "version", Message: "version must be semver (e.g. 1.0.0)"})
}
}
for i, dep := range skill.Dependencies {
if dep.Type != "mcp_server" && dep.Type != "lsp" && dep.Type != "tool" && dep.Type != "runtime" && dep.Type != "" {
errs = append(errs, ValidationError{
Field: fmt.Sprintf("dependencies[%d].type", i),
Message: "dependency type must be mcp_server, lsp, tool, or runtime",
})
}
if dep.Name == "" {
errs = append(errs, ValidationError{
Field: fmt.Sprintf("dependencies[%d].name", i),
Message: "dependency name is required",
})
}
}
return errs
}
func CheckDependencies(skill *Skill) []SkillDependency {
var missing []SkillDependency
for _, dep := range skill.Dependencies {
switch dep.Type {
case "mcp_server":
if !isMCPServerAvailable(dep.Name) {
missing = append(missing, dep)
}
case "lsp", "tool", "runtime":
if !isToolAvailable(dep.Name) {
missing = append(missing, dep)
}
}
}
return missing
}
func isToolAvailable(name string) bool {
_, err := lookPath(name)
return err == nil
}
func lookPath(name string) (string, error) {
pathEnv := os.Getenv("PATH")
home, _ := os.UserHomeDir()
if home != "" {
pathEnv = home + "/.local/bin:" + home + "/go/bin:" + pathEnv
}
for _, dir := range filepath.SplitList(pathEnv) {
candidate := filepath.Join(dir, name)
if info, err := os.Stat(candidate); err == nil && !info.IsDir() {
return candidate, nil
}
}
return "", fmt.Errorf("%s not found", name)
}
func isMCPServerAvailable(name string) bool {
home, _ := os.UserHomeDir()
if home == "" {
return false
}
configPath := filepath.Join(home, ".config", "crush", "crush.json")
data, err := os.ReadFile(configPath)
if err != nil {
return false
}
var cfg map[string]interface{}
if err := json.Unmarshal(data, &cfg); err != nil {
return false
}
mcps, ok := cfg["mcps"].(map[string]interface{})
if !ok {
return false
}
_, exists := mcps[name]
return exists
}
func Export(name string, exportPath string) error {
skill, err := Get(name)
if err != nil {
return err
}
if err := os.MkdirAll(filepath.Dir(exportPath), 0755); err != nil {
return err
}
content := renderSkill(skill)
return os.WriteFile(exportPath, []byte(content), 0644)
}
func Import(exportPath string) (*Skill, error) {
data, err := os.ReadFile(exportPath)
if err != nil {
return nil, fmt.Errorf("read export file: %w", err)
}
skill, err := parseSkill(data)
if err != nil {
return nil, err
}
name := filepath.Base(filepath.Dir(exportPath))
if skill.Name == "" {
skill.Name = strings.TrimSuffix(filepath.Base(exportPath), ".md")
if skill.Name == "SKILL" {
skill.Name = filepath.Base(filepath.Dir(exportPath))
}
}
_ = name
if errs := Validate(skill); len(errs) > 0 {
return nil, fmt.Errorf("validation failed: %v", errs)
}
return skill, nil
}
func DryRun(name string, sampleTask string) SkillTestResult {
skill, err := Get(name)
if err != nil {
return SkillTestResult{Name: name, Passed: false, Message: fmt.Sprintf("skill not found: %s", err)}
}
if skill.Content == "" {
return SkillTestResult{Name: name, Passed: false, Message: "skill has no content"}
}
if len(skill.Dependencies) > 0 {
missing := CheckDependencies(skill)
if len(missing) > 0 {
var names []string
for _, d := range missing {
names = append(names, d.Name)
}
return SkillTestResult{
Name: name,
Passed: false,
Message: fmt.Sprintf("missing dependencies: %s", strings.Join(names, ", ")),
}
}
}
if sampleTask != "" {
tags := skill.Tags
taskLower := strings.ToLower(sampleTask)
matched := false
for _, tag := range tags {
if strings.Contains(taskLower, strings.ToLower(tag)) {
matched = true
break
}
}
if len(tags) > 0 && !matched {
return SkillTestResult{
Name: name,
Passed: true,
Message: "skill loaded but sample task does not match skill tags",
}
}
}
return SkillTestResult{
Name: name,
Passed: true,
Message: "skill validated successfully",
}
}
func parseSkill(data []byte) (*Skill, error) {
content := string(data)
@@ -227,9 +475,25 @@ func renderSkill(skill *Skill) string {
if skill.Target != "" {
b.WriteString(fmt.Sprintf("target: %s\n", skill.Target))
}
if skill.Category != "" {
b.WriteString(fmt.Sprintf("category: %s\n", skill.Category))
}
if len(skill.Tags) > 0 {
b.WriteString(fmt.Sprintf("tags: [%s]\n", strings.Join(skill.Tags, ", ")))
}
if len(skill.Languages) > 0 {
b.WriteString(fmt.Sprintf("languages: [%s]\n", strings.Join(skill.Languages, ", ")))
}
if len(skill.Dependencies) > 0 {
b.WriteString("dependencies:\n")
for _, dep := range skill.Dependencies {
req := ""
if dep.Required {
req = ", required: true"
}
b.WriteString(fmt.Sprintf(" - type: %s, name: %s%s\n", dep.Type, dep.Name, req))
}
}
b.WriteString("---\n\n")
b.WriteString(skill.Content)
b.WriteString("\n")
@@ -245,7 +509,7 @@ DESCRIPTION: %s
TARGET: %s (crush = Crush with GLM, claude = Claude Code, both = both tools)
The skill must follow this EXACT format:
1. YAML frontmatter with: name, description
1. YAML frontmatter with: name, description, tags, dependencies (if needed)
2. Markdown body with detailed instructions
The skill should be practical, specific, and actionable.
@@ -255,5 +519,10 @@ Include:
- Examples where relevant
- Error handling guidance
If the skill requires specific tools, MCP servers, or LSP servers, declare them as dependencies:
- type: mcp_server, name: <server-name>
- type: lsp, name: <language-server-name>
- type: tool, name: <tool-name>
Output ONLY the skill file content, starting with ---`, name, description, target)
}

View File

@@ -113,7 +113,7 @@ func TestCreateAndGet(t *testing.T) {
Description: "Test description",
Content: "Test content body",
Author: "tester",
Version: "0.1",
Version: "1.0.0",
Target: "both",
}
@@ -198,3 +198,242 @@ func TestInstallBuiltinSkills(t *testing.T) {
t.Error("Expected env-setup skill")
}
}
func TestValidate(t *testing.T) {
skill := &Skill{
Name: "valid-skill",
Description: "A valid skill",
Content: "## Steps\nDo things",
Version: "1.0.0",
Target: "both",
}
errs := Validate(skill)
if len(errs) != 0 {
t.Errorf("Valid skill should have no errors, got %v", errs)
}
}
func TestValidateMissingFields(t *testing.T) {
skill := &Skill{}
errs := Validate(skill)
if len(errs) == 0 {
t.Error("Empty skill should have validation errors")
}
fields := map[string]bool{}
for _, e := range errs {
fields[e.Field] = true
}
if !fields["name"] {
t.Error("Should require name")
}
if !fields["description"] {
t.Error("Should require description")
}
if !fields["content"] {
t.Error("Should require content")
}
}
func TestValidateBadVersion(t *testing.T) {
skill := &Skill{
Name: "test-skill",
Description: "desc",
Content: "content",
Version: "not-semver",
}
errs := Validate(skill)
hasVersionErr := false
for _, e := range errs {
if e.Field == "version" {
hasVersionErr = true
}
}
if !hasVersionErr {
t.Error("Should reject non-semver version")
}
}
func TestValidateBadTarget(t *testing.T) {
skill := &Skill{
Name: "test",
Description: "desc",
Content: "content",
Target: "invalid",
}
errs := Validate(skill)
hasTargetErr := false
for _, e := range errs {
if e.Field == "target" {
hasTargetErr = true
}
}
if !hasTargetErr {
t.Error("Should reject invalid target")
}
}
func TestValidateBadName(t *testing.T) {
skill := &Skill{
Name: "INVALID",
Description: "desc",
Content: "content",
}
errs := Validate(skill)
hasNameErr := false
for _, e := range errs {
if e.Field == "name" {
hasNameErr = true
}
}
if !hasNameErr {
t.Error("Should reject uppercase name")
}
}
func TestValidateDependencies(t *testing.T) {
skill := &Skill{
Name: "test",
Description: "desc",
Content: "content",
Dependencies: []SkillDependency{
{Type: "mcp_server", Name: "github", Required: true},
{Type: "invalid_type", Name: "test"},
},
}
errs := Validate(skill)
hasDepErr := false
for _, e := range errs {
if e.Field == "dependencies[1].type" {
hasDepErr = true
}
}
if !hasDepErr {
t.Error("Should reject invalid dependency type")
}
}
func TestExportImport(t *testing.T) {
tmpDir := t.TempDir()
os.Setenv("HOME", tmpDir)
defer os.Setenv("HOME", tmpDir)
skill := &Skill{
Name: "export-test",
Description: "Export test skill",
Content: "## Content",
Author: "tester",
Version: "1.0.0",
Target: "both",
Tags: []string{"test"},
}
Create(skill)
exportPath := filepath.Join(tmpDir, "export", "export-test.md")
if err := Export("export-test", exportPath); err != nil {
t.Fatalf("Export failed: %v", err)
}
if _, err := os.Stat(exportPath); os.IsNotExist(err) {
t.Error("Export file should exist")
}
imported, err := Import(exportPath)
if err != nil {
t.Fatalf("Import failed: %v", err)
}
if imported.Description != "Export test skill" {
t.Errorf("Expected 'Export test skill', got %s", imported.Description)
}
}
func TestDryRun(t *testing.T) {
tmpDir := t.TempDir()
os.Setenv("HOME", tmpDir)
defer os.Setenv("HOME", tmpDir)
skill := &Skill{
Name: "dry-run-test",
Description: "Dry run test",
Content: "## Steps\nDo something",
Version: "1.0.0",
Target: "both",
Tags: []string{"test"},
}
Create(skill)
result := DryRun("dry-run-test", "test something")
if !result.Passed {
t.Errorf("DryRun should pass, got: %s", result.Message)
}
}
func TestDryRunMissing(t *testing.T) {
result := DryRun("nonexistent", "")
if result.Passed {
t.Error("DryRun of nonexistent skill should fail")
}
}
func TestUpdate(t *testing.T) {
tmpDir := t.TempDir()
os.Setenv("HOME", tmpDir)
defer os.Setenv("HOME", tmpDir)
skill := &Skill{
Name: "update-test",
Description: "Original",
Content: "Original content",
Version: "1.0.0",
Target: "both",
}
Create(skill)
skill.Description = "Updated"
skill.Content = "Updated content"
skill.Version = "2.0.0"
if err := Update(skill); err != nil {
t.Fatalf("Update failed: %v", err)
}
got, err := Get("update-test")
if err != nil {
t.Fatalf("Get failed: %v", err)
}
if got.Description != "Updated" {
t.Errorf("Expected 'Updated', got %s", got.Description)
}
}
func TestBuiltinSkillCount(t *testing.T) {
if len(builtinSkills) < 5 {
t.Errorf("Expected at least 5 builtin skills, got %d", len(builtinSkills))
}
expectedSkills := []string{"env-setup", "git-workflow", "api-design", "debug-assist", "code-review", "docker-setup", "security-audit", "mcp-setup", "lsp-setup", "workflow-design"}
for _, name := range expectedSkills {
found := false
for _, s := range builtinSkills {
if s.Name == name {
found = true
break
}
}
if !found {
t.Errorf("Expected builtin skill: %s", name)
}
}
}
func TestBuiltinSkillsHaveDependencies(t *testing.T) {
hasDeps := 0
for _, s := range builtinSkills {
if len(s.Dependencies) > 0 {
hasDeps++
}
}
if hasDeps == 0 {
t.Error("At least some builtin skills should declare dependencies")
}
}

362
internal/workflow/engine.go Normal file
View File

@@ -0,0 +1,362 @@
package workflow
import (
"context"
"encoding/json"
"fmt"
"os"
"path/filepath"
"sync"
"time"
"github.com/muyue/muyue/internal/agent"
"github.com/muyue/muyue/internal/config"
)
type Status string
const (
StatusPending Status = "pending"
StatusRunning Status = "running"
StatusDone Status = "done"
StatusFailed Status = "failed"
StatusSkipped Status = "skipped"
StatusAwaiting Status = "awaiting_approval"
)
type StepType string
const (
TypeToolCall StepType = "tool_call"
TypeCondition StepType = "condition"
TypeParallel StepType = "parallel"
TypeApproval StepType = "approval"
)
type Step struct {
ID string `json:"id"`
Name string `json:"name"`
Type StepType `json:"type"`
Tool string `json:"tool,omitempty"`
Args json.RawMessage `json:"args,omitempty"`
Status Status `json:"status"`
Result string `json:"result,omitempty"`
Error string `json:"error,omitempty"`
Condition string `json:"condition,omitempty"`
DependsOn []string `json:"depends_on,omitempty"`
ApproveRole string `json:"approve_role,omitempty"`
StartedAt *time.Time `json:"started_at,omitempty"`
EndedAt *time.Time `json:"ended_at,omitempty"`
}
type Workflow struct {
ID string `json:"id"`
Name string `json:"name"`
Description string `json:"description"`
Type string `json:"type"`
Steps []Step `json:"steps"`
Status Status `json:"status"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
type Engine struct {
mu sync.RWMutex
workflows map[string]*Workflow
agentRegistry *agent.Registry
storePath string
}
func NewEngine(registry *agent.Registry) (*Engine, error) {
dir, err := config.ConfigDir()
if err != nil {
dir = "/tmp/muyue"
}
storePath := filepath.Join(dir, "workflows.json")
engine := &Engine{
workflows: make(map[string]*Workflow),
agentRegistry: registry,
storePath: storePath,
}
engine.load()
return engine, nil
}
func (e *Engine) load() {
data, err := os.ReadFile(e.storePath)
if err != nil {
return
}
var workflows []*Workflow
if err := json.Unmarshal(data, &workflows); err != nil {
return
}
for _, w := range workflows {
e.workflows[w.ID] = w
}
}
func (e *Engine) save() error {
dir := filepath.Dir(e.storePath)
os.MkdirAll(dir, 0755)
e.mu.RLock()
workflows := make([]*Workflow, 0, len(e.workflows))
for _, w := range e.workflows {
workflows = append(workflows, w)
}
e.mu.RUnlock()
data, err := json.MarshalIndent(workflows, "", " ")
if err != nil {
return err
}
return os.WriteFile(e.storePath, data, 0600)
}
func (e *Engine) Create(name, description, wfType string, steps []Step) *Workflow {
wf := &Workflow{
ID: fmt.Sprintf("wf-%d", time.Now().UnixNano()),
Name: name,
Description: description,
Type: wfType,
Steps: steps,
Status: StatusPending,
CreatedAt: time.Now(),
UpdatedAt: time.Now(),
}
for i := range wf.Steps {
if wf.Steps[i].ID == "" {
wf.Steps[i].ID = fmt.Sprintf("step-%d", i)
}
if wf.Steps[i].Status == "" {
wf.Steps[i].Status = StatusPending
}
}
e.mu.Lock()
e.workflows[wf.ID] = wf
e.mu.Unlock()
e.save()
return wf
}
func (e *Engine) Get(id string) (*Workflow, bool) {
e.mu.RLock()
defer e.mu.RUnlock()
wf, ok := e.workflows[id]
return wf, ok
}
func (e *Engine) List() []*Workflow {
e.mu.RLock()
defer e.mu.RUnlock()
result := make([]*Workflow, 0, len(e.workflows))
for _, w := range e.workflows {
result = append(result, w)
}
return result
}
func (e *Engine) Delete(id string) error {
e.mu.Lock()
defer e.mu.Unlock()
if _, ok := e.workflows[id]; !ok {
return fmt.Errorf("workflow not found: %s", id)
}
delete(e.workflows, id)
return e.save()
}
func (e *Engine) UpdateStep(workflowID, stepID string, update func(*Step)) error {
e.mu.Lock()
defer e.mu.Unlock()
wf, ok := e.workflows[workflowID]
if !ok {
return fmt.Errorf("workflow not found: %s", workflowID)
}
for i := range wf.Steps {
if wf.Steps[i].ID == stepID {
update(&wf.Steps[i])
wf.UpdatedAt = time.Now()
e.save()
return nil
}
}
return fmt.Errorf("step not found: %s", stepID)
}
func (e *Engine) UpdateWorkflowStatus(workflowID string, status Status) error {
e.mu.Lock()
defer e.mu.Unlock()
wf, ok := e.workflows[workflowID]
if !ok {
return fmt.Errorf("workflow not found: %s", workflowID)
}
wf.Status = status
wf.UpdatedAt = time.Now()
return e.save()
}
func (e *Engine) Execute(ctx context.Context, workflowID string, onStep func(step *Step, event string)) error {
wf, ok := e.Get(workflowID)
if !ok {
return fmt.Errorf("workflow not found: %s", workflowID)
}
if err := e.UpdateWorkflowStatus(workflowID, StatusRunning); err != nil {
return err
}
stepStatuses := make(map[string]Status)
for _, step := range wf.Steps {
stepStatuses[step.ID] = StatusPending
}
resolveDeps := func(stepID string) bool {
step := wf.findStep(stepID)
if step == nil {
return false
}
for _, dep := range step.DependsOn {
if stepStatuses[dep] != StatusDone {
return false
}
}
return true
}
executeStep := func(step *Step) error {
now := time.Now()
e.UpdateStep(workflowID, step.ID, func(s *Step) {
s.Status = StatusRunning
s.StartedAt = &now
})
if onStep != nil {
onStep(step, "started")
}
var result string
var stepErr error
switch step.Type {
case TypeToolCall:
if step.Tool == "" {
stepErr = fmt.Errorf("tool not specified for step %s", step.ID)
} else {
call := agent.ToolCall{
ID: step.ID,
Name: step.Tool,
Arguments: step.Args,
}
resp, err := e.agentRegistry.Execute(ctx, call)
if err != nil {
stepErr = err
} else {
result = resp.Content
if resp.IsError {
stepErr = fmt.Errorf("%s", result)
}
}
}
case TypeApproval:
e.UpdateStep(workflowID, step.ID, func(s *Step) {
s.Status = StatusAwaiting
})
if onStep != nil {
onStep(step, "awaiting_approval")
}
return nil
case TypeCondition:
result = fmt.Sprintf("condition '%s' evaluated", step.Condition)
default:
stepErr = fmt.Errorf("unknown step type: %s", step.Type)
}
endTime := time.Now()
if stepErr != nil {
e.UpdateStep(workflowID, step.ID, func(s *Step) {
s.Status = StatusFailed
s.Error = stepErr.Error()
s.EndedAt = &endTime
})
if onStep != nil {
onStep(step, "failed")
}
} else {
e.UpdateStep(workflowID, step.ID, func(s *Step) {
s.Status = StatusDone
s.Result = result
s.EndedAt = &endTime
})
stepStatuses[step.ID] = StatusDone
if onStep != nil {
onStep(step, "done")
}
}
return stepErr
}
hasFailures := false
for _, step := range wf.Steps {
if step.Type == TypeParallel {
continue
}
for !resolveDeps(step.ID) {
time.Sleep(100 * time.Millisecond)
}
if err := executeStep(&step); err != nil {
hasFailures = true
break
}
}
if hasFailures {
e.UpdateWorkflowStatus(workflowID, StatusFailed)
} else {
e.UpdateWorkflowStatus(workflowID, StatusDone)
}
return nil
}
func (w *Workflow) findStep(id string) *Step {
for i := range w.Steps {
if w.Steps[i].ID == id {
return &w.Steps[i]
}
}
return nil
}
func (e *Engine) ApproveStep(workflowID, stepID string) error {
return e.UpdateStep(workflowID, stepID, func(s *Step) {
s.Status = StatusDone
})
}
func (e *Engine) SkipStep(workflowID, stepID string) error {
return e.UpdateStep(workflowID, stepID, func(s *Step) {
s.Status = StatusSkipped
})
}

View File

@@ -0,0 +1,172 @@
package workflow
import (
"context"
"encoding/json"
"fmt"
"strings"
"github.com/muyue/muyue/internal/config"
"github.com/muyue/muyue/internal/orchestrator"
)
type Planner struct {
orchestrator *orchestrator.Orchestrator
}
func NewPlanner(cfg *config.MuyueConfig) (*Planner, error) {
orb, err := orchestrator.New(cfg)
if err != nil {
return nil, err
}
orb.SetSystemPrompt(plannerSystemPrompt)
return &Planner{orchestrator: orb}, nil
}
func (p *Planner) GeneratePlan(ctx context.Context, goal string) ([]Step, error) {
prompt := buildPlanPrompt(goal)
messages := []orchestrator.Message{
{Role: "user", Content: prompt},
}
resp, err := p.orchestrator.SendWithTools(messages)
if err != nil {
return nil, err
}
if len(resp.Choices) == 0 || resp.Choices[0].Message.Content == "" {
return nil, fmt.Errorf("no plan generated")
}
content := resp.Choices[0].Message.Content
plan, err := parsePlanResponse(content)
if err != nil {
return nil, err
}
return plan, nil
}
func buildPlanPrompt(goal string) string {
return fmt.Sprintf(`Tu es un planificateur de workflows pour Muyue. L'utilisateur veut accomplir la tâche suivante:
"%s"
Analyse cette tâche et génère un plan d'exécution en une série d'étapes. Chaque étape est un appel d'outil.
Les outils disponibles sont:
- terminal: Exécuter une commande shell
- read_file: Lire un fichier
- list_files: Lister les fichiers d'un répertoire
- search_files: Rechercher des fichiers par pattern
- grep_content: Rechercher du texte dans des fichiers
- get_config: Lire la configuration Muyue
- set_provider: Configurer un provider AI
- manage_ssh: Gérer les connexions SSH
- web_fetch: Récupérer le contenu d'une URL
Réponds UNIQUEMENT avec un JSON valide représentant un tableau d'étapes, sans texte avant ou après:
[
{"name": "Nom de l'étape", "tool": "terminal", "args": {"command": "ls -la"}},
{"name": "Lire le fichier config", "tool": "read_file", "args": {"path": "~/.muyue/config.json"}}
]
Règles:
- Chaque étape doit avoir: name, tool, args
- Les args varient selon le tool (voir les définitions)
- Sois précis dans les commandes
- Sépare en étapes logiques
- Ne génère pas plus de 10 étapes`, goal)
}
func parsePlanResponse(content string) ([]Step, error) {
content = strings.TrimSpace(content)
var jsonStr string
if strings.HasPrefix(content, "```json") {
lines := strings.Split(content, "\n")
var jsonLines []string
for _, line := range lines[1:] {
if strings.HasPrefix(line, "```") {
break
}
jsonLines = append(jsonLines, line)
}
jsonStr = strings.Join(jsonLines, "\n")
} else if strings.HasPrefix(content, "```") {
lines := strings.Split(content, "\n")
var jsonLines []string
for _, line := range lines[1:] {
if strings.HasPrefix(line, "```") {
break
}
jsonLines = append(jsonLines, line)
}
jsonStr = strings.Join(jsonLines, "\n")
} else {
jsonStr = content
}
var rawSteps []map[string]interface{}
if err := json.Unmarshal([]byte(jsonStr), &rawSteps); err != nil {
return nil, fmt.Errorf("failed to parse plan JSON: %v\nContent: %s", err, content)
}
steps := make([]Step, 0, len(rawSteps))
for i, raw := range rawSteps {
step := Step{
ID: fmt.Sprintf("step-%d", i),
Status: StatusPending,
}
if name, ok := raw["name"].(string); ok {
step.Name = name
} else {
step.Name = fmt.Sprintf("Step %d", i+1)
}
if tool, ok := raw["tool"].(string); ok {
step.Tool = tool
step.Type = TypeToolCall
}
if args, ok := raw["args"].(map[string]interface{}); ok {
argsJSON, err := json.Marshal(args)
if err == nil {
step.Args = argsJSON
}
}
if tool, ok := raw["type"].(string); ok {
switch tool {
case "approval":
step.Type = TypeApproval
case "condition":
step.Type = TypeCondition
if cond, ok := raw["condition"].(string); ok {
step.Condition = cond
}
default:
step.Type = TypeToolCall
}
}
steps = append(steps, step)
}
return steps, nil
}
const plannerSystemPrompt = `Tu es un assistant de planification de workflows pour Muyue. Tu génères des plans d'exécution sous forme de JSON. Chaque plan est une séquence d'étapes (steps) représentant des appels d'outils.
Pour générer un plan:
1. Comprends l'objectif de l'utilisateur
2. Identifie les outils nécessaires
3. Décompose en étapes logiques
4. Spécifie les paramètres de chaque outil
Réponds toujours en JSON valide, sans texte additionnel.`
var _ = plannerSystemPrompt