Kalvin/mcp gdrive (#621)

block · Jan 21, 2025 · 78fc44b · 78fc44b
1 parent e931318
commit 78fc44b
Show file tree

Hide file tree

Showing 5 changed files with 603 additions and 2 deletions.
diff --git a/crates/goose-cli/src/commands/mcp.rs b/crates/goose-cli/src/commands/mcp.rs
@@ -1,6 +1,7 @@
 use anyhow::Result;
 use goose_mcp::{
-    Developer2Router, DeveloperRouter, JetBrainsRouter, MemoryRouter, NonDeveloperRouter,
+    Developer2Router, DeveloperRouter, GoogleDriveRouter, JetBrainsRouter, MemoryRouter,
+    NonDeveloperRouter,
 };
 use mcp_server::router::RouterService;
 use mcp_server::{BoundedService, ByteTransport, Server};
@@ -17,6 +18,10 @@ pub async fn run_server(name: &str) -> Result<()> {
         "developer2" => Some(Box::new(RouterService(Developer2Router::new()))),
         "nondeveloper" => Some(Box::new(RouterService(NonDeveloperRouter::new()))),
         "jetbrains" => Some(Box::new(RouterService(JetBrainsRouter::new()))),
+        "google_drive" => {
+            let router = GoogleDriveRouter::new().await;
+            Some(Box::new(RouterService(router)))
+        }
         "memory" => Some(Box::new(RouterService(MemoryRouter::new()))),
         _ => None,
     };

diff --git a/crates/goose-mcp/Cargo.toml b/crates/goose-mcp/Cargo.toml
@@ -34,6 +34,10 @@ chrono = { version = "0.4.38", features = ["serde"] }
 dirs = "5.0.1"
 tempfile = "3.8"
 include_dir = "0.7.4"
+google-drive3 = "6.0.0"
+webbrowser = "0.8"
+http-body-util = "0.1.2"
+regex = "1.11.1"
 
 [dev-dependencies]
 sysinfo = "0.32.1"