From 14da4793f754d6a96232a3fb70d1307507eda49c Mon Sep 17 00:00:00 2001
From: Ashwin Bhat <ashwin@anthropic.com>
Date: Thu, 10 Oct 2024 14:33:07 -0700
Subject: [PATCH] extract transport logic

---
 client/src/components/ResourcesTab.tsx |  7 ++--
 server/src/index.ts                    | 49 ++++++++++++++++----------
 2 files changed, 34 insertions(+), 22 deletions(-)

diff --git a/client/src/components/ResourcesTab.tsx b/client/src/components/ResourcesTab.tsx
index 5fa08ca..0fca352 100644
--- a/client/src/components/ResourcesTab.tsx
+++ b/client/src/components/ResourcesTab.tsx
@@ -6,6 +6,7 @@ import ListPane from "./ListPane";
 
 export type Resource = {
   uri: string;
+  name: string;
 };
 
 const ResourcesTab = ({
@@ -37,7 +38,7 @@ const ResourcesTab = ({
         <div className="flex items-center w-full">
           <FileText className="w-4 h-4 mr-2 flex-shrink-0 text-gray-500" />
           <span className="flex-1 truncate" title={resource.uri}>
-            {resource.uri}
+            {resource.name}
           </span>
           <ChevronRight className="w-4 h-4 flex-shrink-0 text-gray-400" />
         </div>
@@ -48,8 +49,8 @@ const ResourcesTab = ({
 
     <div className="bg-white rounded-lg shadow">
       <div className="p-4 border-b border-gray-200 flex justify-between items-center">
-        <h3 className="font-semibold truncate" title={selectedResource?.uri}>
-          {selectedResource ? selectedResource.uri : "Select a resource"}
+        <h3 className="font-semibold truncate" title={selectedResource?.name}>
+          {selectedResource ? selectedResource.name : "Select a resource"}
         </h3>
         {selectedResource && (
           <Button
diff --git a/server/src/index.ts b/server/src/index.ts
index b4c345a..2ac006c 100644
--- a/server/src/index.ts
+++ b/server/src/index.ts
@@ -16,33 +16,44 @@ app.use(cors());
 
 let webAppTransports: SSEServerTransport[] = [];
 
+const createTransport = async (query: express.Request["query"]) => {
+  console.log("Query parameters:", query);
+
+  const transportType = query.transportType as string;
+
+  if (transportType === "stdio") {
+    const command = decodeURIComponent(query.command as string);
+    const args = decodeURIComponent(query.args as string).split(",");
+    console.log(`Stdio transport: command=${command}, args=${args}`);
+    const transport = new StdioClientTransport();
+    await transport.spawn({ command, args });
+    console.log("Spawned stdio transport");
+    return transport;
+  } else if (transportType === "sse") {
+    const url = decodeURIComponent(query.url as string);
+    console.log(`SSE transport: url=${url}`);
+    const transport = new SSEClientTransport();
+    await transport.connect(new URL(url));
+    console.log("Connected to SSE transport");
+    return transport;
+  } else {
+    console.error(`Invalid transport type: ${transportType}`);
+    throw new Error("Invalid transport type specified");
+  }
+};
+
 app.get("/sse", async (req, res) => {
   console.log("New SSE connection");
   const transportType = req.query.transportType as string;
   console.log(`Transport type: ${transportType}`);
 
-  let backingServerTransport;
-  console.log("Query parameters:", req.query);
+  const backingServerTransport = await createTransport(req.query);
 
-  if (transportType === "stdio") {
-    const command = decodeURIComponent(req.query.command as string);
-    const args = decodeURIComponent(req.query.args as string).split(",");
-    console.log(`Stdio transport: command=${command}, args=${args}`);
-    backingServerTransport = new StdioClientTransport();
-    await backingServerTransport.spawn({ command, args });
-    console.log("Spawned stdio transport");
-  } else if (transportType === "sse") {
-    const url = decodeURIComponent(req.query.url as string);
-    console.log(`SSE transport: url=${url}`);
-    backingServerTransport = new SSEClientTransport();
-    await backingServerTransport.connect(new URL(url));
-    console.log("Connected to SSE transport");
-  } else {
-    console.error(`Invalid transport type: ${transportType}`);
-    throw new Error("Invalid transport type specified");
-  }
+  console.log("Connected MCP client to backing server transport");
 
   const webAppTransport = new SSEServerTransport("/message");
+  console.log("Created web app transport");
+
   webAppTransports.push(webAppTransport);
   console.log("Created web app transport");