switch to fastmcp

mcp/comment.ts

@@ -0,0 +1,66 @@
+import { Octokit } from "@octokit/rest";
+import { type } from "arktype";
+import { resolveRepoContext } from "../utils/repo-context.ts";
+import { tool } from "./shared.ts";
+
+export const Comment = type({
+  issueNumber: type.number.describe("the issue number to comment on"),
+  body: type.string.describe("the comment body content"),
+});
+
+export const CommentTool = tool({
+  name: "create_issue_comment",
+  description: "Create a comment on a GitHub issue",
+  parameters: Comment,
+  execute: async ({ issueNumber, body }: { issueNumber: number; body: string }) => {
+    try {
+      const githubInstallationToken = process.env.GITHUB_INSTALLATION_TOKEN;
+      if (!githubInstallationToken) {
+        throw new Error("GITHUB_INSTALLATION_TOKEN environment variable is required");
+      }
+
+      const repoContext = resolveRepoContext();
+
+      const octokit = new Octokit({
+        auth: githubInstallationToken,
+      });
+
+      const result = await octokit.rest.issues.createComment({
+        owner: repoContext.owner,
+        repo: repoContext.name,
+        issue_number: issueNumber,
+        body: body,
+      });
+
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(
+              {
+                success: true,
+                commentId: result.data.id,
+                url: result.data.html_url,
+                body: result.data.body,
+              },
+              null,
+              2
+            ),
+          },
+        ],
+      };
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      return {
+        content: [
+          {
+            type: "text",
+            text: `Error creating comment: ${errorMessage}`,
+          },
+        ],
+        error: errorMessage,
+        isError: true,
+      };
+    }
+  },
+});

mcp/config.ts

@@ -1,8 +1,6 @@
 /**
  * Simple MCP configuration helper for adding our minimal GitHub comment server
  */
-// const actionPath = process.env.GITHUB_ACTION_PATH || process.cwd();
-
 import { fromHere } from "@ark/fs";
 
 const actionPath = fromHere("..");

mcp/server.ts

@@ -1,92 +1,14 @@
 #!/usr/bin/env node
 // Minimal GitHub Issue Comment MCP Server
-import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import { Octokit } from "@octokit/rest";
-import { type } from "arktype";
-import { z } from "zod";
-import { resolveRepoContext } from "../utils/repo-context.ts";
+import { FastMCP } from "fastmcp";
+import { CommentTool } from "./comment.ts";
+import { addTools } from "./shared.ts";
 
-const server = new McpServer({
+const server = new FastMCP({
   name: "Minimal GitHub Issue Comment Server",
   version: "0.0.1",
 });
 
-// Define the schema for creating issue comments
-const Comment = type({
-  issueNumber: type.number.describe("the issue number to comment on"),
-  body: type.string.describe("the comment body content"),
-});
+addTools(server, [CommentTool]);
 
-server.tool(
-  "create_issue_comment",
-  "Create a comment on a GitHub issue",
-  {
-    issueNumber: z.number().describe("the issue number to comment on"),
-    body: z.string().describe("the comment body content"),
-  },
-  async ({ issueNumber, body }) => {
-    try {
-      Comment.assert({ issueNumber, body });
-
-      const githubInstallationToken = process.env.GITHUB_INSTALLATION_TOKEN;
-      if (!githubInstallationToken) {
-        throw new Error("GITHUB_INSTALLATION_TOKEN environment variable is required");
-      }
-
-      // Resolve repository context from environment
-      const repoContext = resolveRepoContext();
-
-      const octokit = new Octokit({
-        auth: githubInstallationToken,
-      });
-
-      const result = await octokit.rest.issues.createComment({
-        owner: repoContext.owner,
-        repo: repoContext.name,
-        issue_number: issueNumber,
-        body: body,
-      });
-
-      return {
-        content: [
-          {
-            type: "text",
-            text: JSON.stringify(
-              {
-                success: true,
-                commentId: result.data.id,
-                url: result.data.html_url,
-                body: result.data.body,
-              },
-              null,
-              2
-            ),
-          },
-        ],
-      };
-    } catch (error) {
-      const errorMessage = error instanceof Error ? error.message : String(error);
-      return {
-        content: [
-          {
-            type: "text",
-            text: `Error creating comment: ${errorMessage}`,
-          },
-        ],
-        error: errorMessage,
-        isError: true,
-      };
-    }
-  }
-);
-
-async function runServer() {
-  const transport = new StdioServerTransport();
-  await server.connect(transport);
-  process.on("exit", () => {
-    server.close();
-  });
-}
-
-await runServer();
+server.start();

mcp/shared.ts

@@ -0,0 +1,11 @@
+import type { StandardSchemaV1 } from "@standard-schema/spec";
+import type { FastMCP, Tool } from "fastmcp";
+
+export const tool = <const params>(tool: Tool<{}, StandardSchemaV1<params>>) => tool;
+
+export const addTools = (server: FastMCP, tools: Tool<any, any>[]) => {
+  for (const tool of tools) {
+    server.addTool(tool);
+  }
+  return server;
+};