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

This commit is contained in:
Alex Hancock
2025-09-24 10:03:00 -04:00
committed by GitHub
parent 0d25cea181
commit dba0f4eaef
9 changed files with 983 additions and 1192 deletions
-77
View File
@@ -1,77 +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};
pub async fn run(name: &str) -> Result<()> {
crate::logging::setup_logging(Some(&format!("mcp-{name}")))?;
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
let router: Option<Box<dyn BoundedService>> = match name {
"computercontroller" => Some(Box::new(RouterService(ComputerControllerRouter::new()))),
_ => None,
};
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");
Ok(server.run(transport).await?)
}
-1
View File
@@ -1,2 +1 @@
pub mod agent;
pub mod mcp;
+2 -1
View File
@@ -36,7 +36,8 @@ async fn main() -> anyhow::Result<()> {
commands::agent::run().await?;
}
Commands::Mcp { name } => {
commands::mcp::run(name).await?;
logging::setup_logging(Some(&format!("mcp-{name}")))?;
goose_mcp::mcp_server_runner::run_mcp_server(name).await?;
}
}