Refactor tools (#25)

Author: dcramer

src/index.ts

@@ -1,11 +1,11 @@
 import { withSentry } from "@sentry/cloudflare";
 import OAuthProvider from "@cloudflare/workers-oauth-provider";
 import app from "./app";
-import SentryMCP from "./mcp";
+import SentryMCP from "./mcp/server";
 import { SCOPES } from "./routes/auth";
 
 // required for Durable Objects
-export { default as SentryMCP } from "./mcp";
+export { default as SentryMCP } from "./mcp/server";
 
 const oAuthProvider = new OAuthProvider({
   apiRoute: "/sse",

src/mcp.ts

@@ -0,0 +1,15 @@
+import { z } from "zod";
+
+export const ParamOrganizationSlug = z
+  .string()
+  .describe("The organization's slug. This will default to the first org you have access to.");
+
+export const ParamTeamSlug = z
+  .string()
+  .describe("The team's slug. This will default to the first team you have access to.");
+
+export const ParamIssueShortId = z.string().describe("The Issue ID. e.g. `PROJECT-1Z43`");
+
+export const ParamPlatform = z
+  .string()
+  .describe("The platform for the project (e.g., python, javascript, react, etc.)");

src/mcp/schema.ts

@@ -0,0 +1,50 @@
+import { McpAgent } from "agents/mcp";
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import type { Props } from "../types";
+import { logError } from "../lib/logging";
+import { TOOL_DEFINITIONS, TOOL_HANDLERS } from "./tools";
+
+// Context from the auth process, encrypted & stored in the auth token
+// and provided to the DurableMCP as this.props
+export default class SentryMCP extends McpAgent<Env, unknown, Props> {
+  server = new McpServer({
+    name: "Sentry MCP",
+    version: "0.1.0",
+  });
+
+  async init() {
+    for (const tool of TOOL_DEFINITIONS) {
+      const handler = TOOL_HANDLERS[tool.name];
+
+      this.server.tool(tool.name, tool.description, tool.paramsSchema || {}, async (...args) => {
+        try {
+          // TODO(dcramer): I'm too dumb to figure this out
+          // @ts-ignore
+          const output = await handler(this.props, ...args);
+
+          return {
+            content: [
+              {
+                type: "text",
+                text: output,
+              },
+            ],
+          };
+        } catch (error) {
+          logError(error);
+          return {
+            content: [
+              {
+                type: "text",
+                text: `**Error**\n\nIt looks like there was a problem communicating with the Sentry API:\n\n${
+                  error instanceof Error ? error.message : String(error)
+                }`,
+              },
+            ],
+            isError: true,
+          };
+        }
+      });
+    }
+  }
+}