add flag to install pscale MCP to zed as well

go.mod

@@ -32,6 +32,7 @@ require (
 	github.com/spf13/pflag v1.0.10
 	github.com/spf13/viper v1.21.0
 	github.com/stretchr/testify v1.11.1
+	github.com/tidwall/jsonc v0.3.2
 	github.com/xelabs/go-mysqlstack v1.0.0
 	go.uber.org/zap v1.27.0
 	go.uber.org/zap/exp v0.3.0

go.sum

@@ -226,6 +226,8 @@ github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu
 github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U=
 github.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8=
 github.com/subosito/gotenv v1.6.0/go.mod h1:Dk4QP5c2W3ibzajGcXpNraDfq2IrhjMIvMSWPKKo0FU=
+github.com/tidwall/jsonc v0.3.2 h1:ZTKrmejRlAJYdn0kcaFqRAKlxxFIC21pYq8vLa4p2Wc=
+github.com/tidwall/jsonc v0.3.2/go.mod h1:dw+3CIxqHi+t8eFSpzzMlcVYxKp08UP5CD8/uSFCyJE=
 github.com/wk8/go-ordered-map/v2 v2.1.8 h1:5h/BUHu93oj4gIdvHHHGsScSTMijfx5PeYkE/fJgbpc=
 github.com/wk8/go-ordered-map/v2 v2.1.8/go.mod h1:5nJHM5DyteebpVlHnWMV0rPz6Zp7+xBAnxjb1X5vnTw=
 github.com/xelabs/go-mysqlstack v1.0.0 h1:go/UqwlxKRNh9df+AQ/pAAgcCCHCaeyv0PYZ/quRbbw=

internal/cmd/mcp/install.go

@@ -9,40 +9,9 @@ import (
 
 	"github.com/planetscale/cli/internal/cmdutil"
 	"github.com/spf13/cobra"
+	"github.com/tidwall/jsonc"
 )
 
-// ClaudeConfig represents the structure of the Claude Desktop config file
-type ClaudeConfig map[string]interface{}
-
-// getClaudeConfigDir returns the path to the Claude Desktop config directory based on the OS
-func getClaudeConfigDir() (string, error) {
-	switch runtime.GOOS {
-	case "darwin":
-		// macOS path: ~/Library/Application Support/Claude/
-		homeDir, err := os.UserHomeDir()
-		if err != nil {
-			return "", fmt.Errorf("could not determine user home directory: %w", err)
-		}
-		return filepath.Join(homeDir, "Library", "Application Support", "Claude"), nil
-	case "windows":
-		// Windows path: %APPDATA%\Claude\
-		return filepath.Join(os.Getenv("APPDATA"), "Claude"), nil
-	default:
-		return "", fmt.Errorf("unsupported operating system: %s", runtime.GOOS)
-	}
-}
-
-// getCursorConfigPath returns the path to the Cursor MCP config file
-func getCursorConfigPath() (string, error) {
-	homeDir, err := os.UserHomeDir()
-	if err != nil {
-		return "", fmt.Errorf("could not determine user home directory: %w", err)
-	}
-
-	// Cursor uses ~/.cursor/mcp.json for its MCP configuration
-	return filepath.Join(homeDir, ".cursor", "mcp.json"), nil
-}
-
 // InstallCmd returns a new cobra.Command for the mcp install command.
 func InstallCmd(ch *cmdutil.Helper) *cobra.Command {
 	var target string
@@ -57,86 +26,176 @@ func InstallCmd(ch *cmdutil.Helper) *cobra.Command {
 
 			switch target {
 			case "claude":
-				configDir, err := getClaudeConfigDir()
+				configPath, err = getClaudeConfigPath()
 				if err != nil {
-					return fmt.Errorf("failed to determine Claude config directory: %w", err)
+					return fmt.Errorf("failed to determine Claude config path: %w", err)
 				}
-
-				// Check if the directory exists
-				if _, err := os.Stat(configDir); os.IsNotExist(err) {
-					return fmt.Errorf("no Claude Desktop installation: path %s not found", configDir)
+				if err := installMCPServer(configPath, target, modifyClaudeConfig); err != nil {
+					return err
 				}
-
-				configPath = filepath.Join(configDir, "claude_desktop_config.json")
 			case "cursor":
 				configPath, err = getCursorConfigPath()
 				if err != nil {
 					return fmt.Errorf("failed to determine Cursor config path: %w", err)
 				}
-
-				// Ensure the .cursor directory exists
-				configDir := filepath.Dir(configPath)
-				if _, err := os.Stat(configDir); os.IsNotExist(err) {
-					if err := os.MkdirAll(configDir, 0755); err != nil {
-						return fmt.Errorf("failed to create Cursor config directory: %w", err)
-					}
+				// Cursor uses the same config structure as Claude
+				if err := installMCPServer(configPath, target, modifyClaudeConfig); err != nil {
+					return err
 				}
-			default:
-				return fmt.Errorf("invalid target vendor: %s (supported values: claude, cursor)", target)
-			}
-
-			config := make(ClaudeConfig) // Same config structure for both tools
-
-			// Check if the file exists
-			if _, err := os.Stat(configPath); err == nil {
-				// File exists, read it
-				configData, err := os.ReadFile(configPath)
+			case "zed":
+				configPath, err = getZedConfigPath()
 				if err != nil {
-					return fmt.Errorf("failed to read %s config file: %w", target, err)
+					return fmt.Errorf("failed to determine Zed config path: %w", err)
 				}
-
-				if err := json.Unmarshal(configData, &config); err != nil {
-					return fmt.Errorf("failed to parse %s config file: %w", target, err)
+				if err := installMCPServer(configPath, target, modifyZedConfig); err != nil {
+					return err
 				}
-			}
-
-			// Get or initialize the mcpServers map
-			var mcpServers map[string]interface{}
-			if existingServers, ok := config["mcpServers"].(map[string]interface{}); ok {
-				mcpServers = existingServers
-			} else {
-				mcpServers = make(map[string]interface{})
-			}
-
-			// Add or update the planetscale server configuration
-			mcpServers["planetscale"] = map[string]interface{}{
-				"command": "pscale",
-				"args":    []string{"mcp", "server"},
-			}
-
-			// Update the config with the new mcpServers
-			config["mcpServers"] = mcpServers
-
-			// Write the updated config back to file
-			configJSON, err := json.MarshalIndent(config, "", "  ")
-			if err != nil {
-				return fmt.Errorf("failed to marshal %s config: %w", target, err)
-			}
-
-			if err := os.WriteFile(configPath, configJSON, 0644); err != nil {
-				return fmt.Errorf("failed to write %s config file: %w", target, err)
+			default:
+				return fmt.Errorf("invalid target vendor: %s (supported values: claude, cursor, zed)", target)
 			}
 
 			fmt.Printf("MCP server successfully configured for %s at %s\n", target, configPath)
 			return nil
 		},
 	}
 
-	cmd.Flags().StringVar(&target, "target", "", "Target vendor for MCP installation (required). Possible values: [claude, cursor]")
+	cmd.Flags().StringVar(&target, "target", "", "Target vendor for MCP installation (required). Possible values: [claude, cursor, zed]")
 	cmd.MarkFlagRequired("target")
 	cmd.RegisterFlagCompletionFunc("target", func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
-		return []string{"claude", "cursor"}, cobra.ShellCompDirectiveDefault
+		return []string{"claude", "cursor", "zed"}, cobra.ShellCompDirectiveDefault
 	})
 
 	return cmd
 }
+
+// installMCPServer handles common file I/O for all editors
+func installMCPServer(configPath string, target string, modifyConfig func(map[string]any) error) error {
+	// Check if config directory exists
+	configDir := filepath.Dir(configPath)
+	if _, err := os.Stat(configDir); os.IsNotExist(err) {
+		return fmt.Errorf("no %s installation: path %s not found", target, configDir)
+	}
+
+	// Read existing config or create empty settings
+	var fullSettings map[string]any
+	if fileData, err := os.ReadFile(configPath); err == nil {
+		cleanJSON := jsonc.ToJSON(fileData)
+		if err := json.Unmarshal(cleanJSON, &fullSettings); err != nil {
+			return fmt.Errorf("failed to parse %s config file: %w", target, err)
+		}
+	} else if !os.IsNotExist(err) {
+		return fmt.Errorf("failed to read %s config file: %w", target, err)
+	} else {
+		fullSettings = make(map[string]any)
+	}
+
+	// Let the editor-specific function modify the config
+	if err := modifyConfig(fullSettings); err != nil {
+		return err
+	}
+
+	// Marshal updated config
+	updatedData, err := json.MarshalIndent(fullSettings, "", "  ")
+	if err != nil {
+		return fmt.Errorf("failed to marshal %s config: %w", target, err)
+	}
+
+	// Backup existing file before writing
+	if _, err := os.Stat(configPath); err == nil {
+		backupPath := configPath + "~"
+		if err := os.Rename(configPath, backupPath); err != nil {
+			return fmt.Errorf("failed to create backup: %w", err)
+		}
+		fmt.Printf("Created backup at %s\n", backupPath)
+	}
+
+	// Write updated config
+	if err := os.WriteFile(configPath, updatedData, 0644); err != nil {
+		return fmt.Errorf("failed to write %s config file: %w", target, err)
+	}
+
+	return nil
+}
+
+// modifyClaudeConfig adds planetscale to mcpServers (Claude and Cursor both use this)
+func modifyClaudeConfig(settings map[string]any) error {
+	var mcpServers map[string]any
+	if existingServers, ok := settings["mcpServers"].(map[string]any); ok {
+		mcpServers = existingServers
+	} else {
+		mcpServers = make(map[string]any)
+	}
+
+	mcpServers["planetscale"] = map[string]any{
+		"command": "pscale",
+		"args":    []string{"mcp", "server"},
+	}
+
+	settings["mcpServers"] = mcpServers
+	return nil
+}
+
+// modifyZedConfig adds planetscale to context_servers (Zed-specific)
+func modifyZedConfig(settings map[string]any) error {
+	var contextServers map[string]any
+	if existingServers, ok := settings["context_servers"].(map[string]any); ok {
+		contextServers = existingServers
+	} else {
+		contextServers = make(map[string]any)
+	}
+
+	contextServers["planetscale"] = map[string]any{
+		"source":  "custom",
+		"command": "pscale",
+		"args":    []string{"mcp", "server"},
+	}
+
+	settings["context_servers"] = contextServers
+	return nil
+}
+
+// getClaudeConfigPath returns the path to the Claude Desktop config file based on the OS
+func getClaudeConfigPath() (string, error) {
+	switch runtime.GOOS {
+	case "darwin":
+		// macOS path: ~/Library/Application Support/Claude/claude_desktop_config.json
+		homeDir, err := os.UserHomeDir()
+		if err != nil {
+			return "", fmt.Errorf("could not determine user home directory: %w", err)
+		}
+		return filepath.Join(homeDir, "Library", "Application Support", "Claude", "claude_desktop_config.json"), nil
+	case "windows":
+		// Windows path: %APPDATA%\Claude\claude_desktop_config.json
+		return filepath.Join(os.Getenv("APPDATA"), "Claude", "claude_desktop_config.json"), nil
+	default:
+		return "", fmt.Errorf("unsupported operating system: %s", runtime.GOOS)
+	}
+}
+
+// getCursorConfigPath returns the path to the Cursor MCP config file
+func getCursorConfigPath() (string, error) {
+	homeDir, err := os.UserHomeDir()
+	if err != nil {
+		return "", fmt.Errorf("could not determine user home directory: %w", err)
+	}
+
+	// Cursor uses ~/.cursor/mcp.json for its MCP configuration
+	return filepath.Join(homeDir, ".cursor", "mcp.json"), nil
+}
+
+// getZedConfigPath returns the path to the Zed config file based on the OS
+func getZedConfigPath() (string, error) {
+	homeDir, err := os.UserHomeDir()
+	if err != nil {
+		return "", fmt.Errorf("could not determine user home directory: %w", err)
+	}
+
+	switch runtime.GOOS {
+	case "darwin", "linux":
+		return filepath.Join(homeDir, ".config", "zed", "settings.json"), nil
+	case "windows":
+		return filepath.Join(os.Getenv("APPDATA"), "Zed", "settings.json"), nil
+	default:
+		return "", fmt.Errorf("unsupported operating system: %s", runtime.GOOS)
+	}
+}