chore(mcp): convert computercontroller server to use the rust sdk (#4772)

crates/goose-cli/src/cli.rs

@@ -7,7 +7,6 @@ use crate::commands::acp::run_acp_agent;
 use crate::commands::bench::agent_generator;
 use crate::commands::configure::handle_configure;
 use crate::commands::info::handle_info;
-use crate::commands::mcp::run_server;
 use crate::commands::project::{handle_project_default, handle_projects_interactive};
 use crate::commands::recipe::{handle_deeplink, handle_list, handle_validate};
 // Import the new handlers from commands::schedule
@@ -743,7 +742,8 @@ pub async fn cli() -> Result<()> {
             return Ok(());
         }
         Some(Command::Mcp { name }) => {
-            let _ = run_server(&name).await;
+            crate::logging::setup_logging(Some(&format!("mcp-{name}")), None)?;
+            let _ = goose_mcp::mcp_server_runner::run_mcp_server(&name).await;
         }
         Some(Command::Acp {}) => {
             let _ = run_acp_agent().await;

crates/goose-cli/src/commands/mcp.rs

@@ -1,123 +0,0 @@
-use anyhow::{anyhow, Result};
-use goose_mcp::{
-    AutoVisualiserRouter, ComputerControllerRouter, DeveloperServer, MemoryServer, TutorialServer,
-};
-use mcp_server::router::RouterService;
-use mcp_server::{BoundedService, ByteTransport, Server};
-use rmcp::{transport::stdio, ServiceExt};
-use tokio::io::{stdin, stdout};
-
-use std::sync::Arc;
-use tokio::sync::Notify;
-
-#[cfg(unix)]
-use nix::sys::signal::{kill, Signal};
-#[cfg(unix)]
-use nix::unistd::getpgrp;
-#[cfg(unix)]
-use nix::unistd::Pid;
-
-pub async fn run_server(name: &str) -> Result<()> {
-    crate::logging::setup_logging(Some(&format!("mcp-{name}")), None)?;
-
-    if name == "googledrive" || name == "google_drive" {
-        return Err(anyhow!(
-            "the built-in Google Drive extension has been removed"
-        ));
-    }
-
-    tracing::info!("Starting MCP server");
-
-    // Handle RMCP-based servers
-    if name == "developer" {
-        let service = DeveloperServer::new()
-            .serve(stdio())
-            .await
-            .inspect_err(|e| {
-                tracing::error!("serving error: {:?}", e);
-            })?;
-
-        service.waiting().await?;
-        return Ok(());
-    }
-
-    if name == "autovisualiser" {
-        let service = AutoVisualiserRouter::new()
-            .serve(stdio())
-            .await
-            .inspect_err(|e| {
-                tracing::error!("serving error: {:?}", e);
-            })?;
-
-        service.waiting().await?;
-        return Ok(());
-    }
-
-    if name == "tutorial" {
-        let service = TutorialServer::new()
-            .serve(stdio())
-            .await
-            .inspect_err(|e| {
-                tracing::error!("serving error: {:?}", e);
-            })?;
-
-        service.waiting().await?;
-        return Ok(());
-    }
-
-    if name == "memory" {
-        let service = MemoryServer::new().serve(stdio()).await.inspect_err(|e| {
-            tracing::error!("serving error: {:?}", e);
-        })?;
-
-        service.waiting().await?;
-        return Ok(());
-    }
-
-    // Handle old MCP-based servers
-    if name == "memory" {
-        let service = MemoryServer::new().serve(stdio()).await.inspect_err(|e| {
-            tracing::error!("serving error: {:?}", e);
-        })?;
-        service.waiting().await?;
-        return Ok(());
-    }
-
-    // Handle old MCP-based servers
-    let router: Option<Box<dyn BoundedService>> = match name {
-        "computercontroller" => Some(Box::new(RouterService(ComputerControllerRouter::new()))),
-        _ => None,
-    };
-
-    let shutdown = Arc::new(Notify::new());
-    let shutdown_clone = shutdown.clone();
-
-    tokio::spawn(async move {
-        crate::signal::shutdown_signal().await;
-        shutdown_clone.notify_one();
-    });
-
-    let server = Server::new(router.unwrap_or_else(|| panic!("Unknown server requested {}", name)));
-    let transport = ByteTransport::new(stdin(), stdout());
-
-    tracing::info!("Server initialized and ready to handle requests");
-
-    tokio::select! {
-        result = server.run(transport) => {
-            Ok(result?)
-        }
-        _ = shutdown.notified() => {
-            // On Unix systems, kill the entire process group
-            #[cfg(unix)]
-            {
-                fn terminate_process_group() {
-                    let pgid = getpgrp();
-                    kill(Pid::from_raw(-pgid.as_raw()), Signal::SIGTERM)
-                        .expect("Failed to send SIGTERM to process group");
-                }
-                terminate_process_group();
-            }
-            Ok(())
-        }
-    }
-}

crates/goose-cli/src/commands/mod.rs

@@ -2,7 +2,6 @@ pub mod acp;
 pub mod bench;
 pub mod configure;
 pub mod info;
-pub mod mcp;
 pub mod project;
 pub mod recipe;
 pub mod schedule;