Adding Cursor support

Author: Shchvova

src/install.rs

@@ -1,19 +1,25 @@
 use color_eyre::eyre::{eyre, Result, WrapErr};
+use color_eyre::Help;
 use roblox_install::RobloxStudio;
 use serde_json::{json, Value};
 use std::fs::File;
 use std::io::BufReader;
 use std::io::Write;
 use std::path::Path;
 use std::path::PathBuf;
+use std::vec;
 use std::{env, fs, io};
 
-const DISCLAIMER: &str = "Roblox Studio MCP is ready to go.
-        
-Please restart Studio and Claude to apply the changes.
+fn get_message(successes: String) -> String {
+    format!("Roblox Studio MCP is ready to go.
+Please restart Studio and MCP clients to apply the changes.
+
+MCP Clients set up:
+{successes}
 
 Note: connecting a third-party LLM to Roblox Studio via an MCP server will share your data with that external service provider. Please review their privacy practices carefully before proceeding.
-To uninstall, delete the MCPStudioPlugin.rbxm from your Plugins directory.";
+To uninstall, delete the MCPStudioPlugin.rbxm from your Plugins directory.")
+}
 
 // returns OS dependant claude_desktop_config.json path
 fn get_claude_config() -> Result<PathBuf> {
@@ -35,6 +41,13 @@ fn get_claude_config() -> Result<PathBuf> {
     Ok(config_path)
 }
 
+fn get_cursor_config() -> Result<PathBuf> {
+    let home_dir = env::var_os("HOME")
+        .or_else(|| env::var_os("USERPROFILE"))
+        .unwrap();
+    Ok(Path::new(&home_dir).join(".cursor").join("mcp.json"))
+}
+
 #[cfg(target_os = "macos")]
 fn get_exe_path() -> Result<PathBuf> {
     use core_foundation::url::CFURL;
@@ -52,7 +65,46 @@ fn get_exe_path() -> io::Result<PathBuf> {
     env::current_exe()
 }
 
-async fn install_internal() -> Result<()> {
+pub fn install_to_config<'a>(
+    config_path: Result<PathBuf>,
+    exe_path: &Path,
+    name: &'a str,
+) -> Result<&'a str> {
+    let config_path = config_path?;
+    let mut config: serde_json::Map<String, Value> = {
+        if !config_path.exists() {
+            let mut file = File::create(&config_path).map_err(|e| {
+                eyre!("Could not create {name} config file at {config_path:?}: {e:#?}")
+            })?;
+            file.write_all(serde_json::to_string(&serde_json::Map::new())?.as_bytes())?;
+        }
+        let config_file = File::open(&config_path)
+            .map_err(|error| eyre!("Could not read or create {name} config file: {error:#?}"))?;
+        let reader = BufReader::new(config_file);
+        serde_json::from_reader(reader)?
+    };
+
+    if !matches!(config.get("mcpServers"), Some(Value::Object(_))) {
+        config.insert("mcpServers".to_string(), json!({}));
+    }
+
+    config["mcpServers"]["Roblox Studio"] = json!({
+      "command": &exe_path,
+      "args": [
+        "--stdio"
+      ]
+    });
+
+    let mut file = File::create(&config_path)?;
+    file.write_all(serde_json::to_string_pretty(&config)?.as_bytes())
+        .map_err(|e| eyre!("Could not write to {name} config file at {config_path:?}: {e:#?}"))?;
+
+    println!("Installed MCP Studio plugin to {name} config {config_path:?}");
+
+    Ok(name)
+}
+
+async fn install_internal() -> Result<String> {
     let plugin_bytes = include_bytes!(concat!(env!("OUT_DIR"), "/MCPStudioPlugin.rbxm"));
     let studio = RobloxStudio::locate()?;
     let plugins = studio.plugins_path();
@@ -76,43 +128,32 @@ async fn install_internal() -> Result<()> {
         output_plugin.display()
     );
 
-    let claude_config_path = get_claude_config()?;
-
-    let mut config: serde_json::Map<String, Value> = {
-        if !claude_config_path.exists() {
-            let mut file = File::create(&claude_config_path).map_err(|e| {
-                eyre!("Could not create Claude config file at {claude_config_path:?}: {e:#?}")
-            })?;
-            file.write_all(serde_json::to_string(&serde_json::Map::new())?.as_bytes())?;
-        }
-        let claude_config_file = File::open(&claude_config_path)
-            .map_err(|error| eyre!("Could not read or create claude config file: {error:#?}"))?;
-        let reader = BufReader::new(claude_config_file);
-        serde_json::from_reader(reader)?
-    };
-
-    if !matches!(config.get("mcpServers"), Some(Value::Object(_))) {
-        config.insert("mcpServers".to_string(), json!({}));
-    }
-
     let this_exe = get_exe_path()?;
-    config["mcpServers"]["Roblox Studio"] = json!({
-      "command": &this_exe,
-      "args": [
-        "--stdio"
-      ]
-    });
 
-    let mut file = File::create(&claude_config_path)?;
-    file.write_all(serde_json::to_string_pretty(&config)?.as_bytes())
-        .map_err(|e| {
-            eyre!("Could not write to Claude config file at {claude_config_path:?}: {e:#?}")
-        })?;
+    let mut errors = vec![];
+    let claude_config_path = get_claude_config();
+    let results = vec![
+        install_to_config(claude_config_path, &this_exe, "Claude"),
+        install_to_config(get_cursor_config(), &this_exe, "Cursor"),
+    ];
+
+    let successes: Vec<_> = results
+        .into_iter()
+        .filter_map(|r| r.map_err(|e| errors.push(e)).ok())
+        .collect();
+
+    if successes.is_empty() {
+        let error = errors.into_iter().fold(
+            eyre!("Failed to install to either Claude or Cursor"),
+            |report, e| report.note(e),
+        );
+        return Err(error);
+    }
 
-    println!("Installed MCP Studio plugin to Claude config {claude_config_path:?}");
     println!();
-    println!("{DISCLAIMER}");
-    Ok(())
+    let msg = get_message(successes.join("\n"));
+    println!("{}", msg);
+    Ok(msg)
 }
 
 #[cfg(target_os = "windows")]
@@ -128,14 +169,13 @@ pub async fn install() -> Result<()> {
 #[cfg(target_os = "macos")]
 pub async fn install() -> Result<()> {
     use native_dialog::{DialogBuilder, MessageLevel};
-    let alert_builder = if let Err(e) = install_internal().await {
-        DialogBuilder::message()
+    let alert_builder = match install_internal().await {
+        Err(e) => DialogBuilder::message()
             .set_level(MessageLevel::Error)
-            .set_text(format!("Errors occurred: {:#}", e))
-    } else {
-        DialogBuilder::message()
+            .set_text(format!("Errors occurred: {:#}", e)),
+        Ok(msg) => DialogBuilder::message()
             .set_level(MessageLevel::Info)
-            .set_text(DISCLAIMER)
+            .set_text(msg),
     };
     let _ = alert_builder.set_title("Roblox Studio MCP").alert().show();
     Ok(())