mcp

cli/package.json

@@ -52,6 +52,7 @@
     "pub:beta": "pnpm build && npm publish --tag beta --access public",
     "pub:next": "pnpm build && npm publish --tag next --access public",
     "pub:release": "pnpm build && npm publish --access public",
+    "mcp-inspector": "CLIENT_PORT=8080 SERVER_PORT=9000 pnpx @modelcontextprotocol/inspector node ./dist/index.js mcp",
     "test": "vitest run",
     "test:watch": "vitest",
     "test:coverage": "vitest run --coverage"
@@ -60,6 +61,7 @@
     "@clack/core": "^0.3.4",
     "@clack/prompts": "^0.7.0",
     "@ianvs/prettier-plugin-sort-imports": "^4.2.1",
+    "@modelcontextprotocol/sdk": "^1.7.0",
     "@proofgeist/fmdapi": "^4.1.5",
     "@types/glob": "^8.1.0",
     "axios": "^1.7.3",
@@ -99,6 +101,7 @@
     "@types/randomstring": "^1.3.0",
     "@types/react": "^18.3.11",
     "@types/semver": "^7.5.8",
+    "@vitest/coverage-v8": "^1.4.0",
     "drizzle-kit": "^0.21.4",
     "drizzle-orm": "^0.30.10",
     "mysql2": "^3.9.7",
@@ -116,7 +119,6 @@
     "type-fest": "^3.13.1",
     "typescript": "^5.6.3",
     "vitest": "^1.4.0",
-    "@vitest/coverage-v8": "^1.4.0",
-    "zod": "^3.23.8"
+    "zod": "^3.24.2"
   }
 }

cli/src/index.ts

@@ -15,6 +15,7 @@ import { makeUpgradeCommand } from "./cli/update/makeUpgradeCommand.js";
 import { UserAbortedError } from "./cli/utils.js";
 import { npmName } from "./consts.js";
 import { ciOption } from "./globalOptions.js";
+import { makeMcpCommand } from "./mcp/command.js";
 import { initProgramState, state } from "./state.js";
 import { getVersion } from "./utils/getProofKitVersion.js";
 import { getSettings, type Settings } from "./utils/parseSettings.js";
@@ -67,6 +68,7 @@ const main = async () => {
   program.addCommand(makeTypegenCommand());
   program.addCommand(makeDeployCommand());
   program.addCommand(makeUpgradeCommand());
+  program.addCommand(makeMcpCommand());
 
   await program.parseAsync(process.argv);
   process.exit(0);

cli/src/mcp/command.ts

@@ -0,0 +1,15 @@
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { program } from "commander";
+
+import { makeServer } from "./server.js";
+
+export function makeMcpCommand() {
+  return program.command("mcp").action(async () => {
+    const server = makeServer();
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+
+    // eslint-disable-next-line @typescript-eslint/no-empty-function
+    await new Promise(() => {}); // keep the process running
+  });
+}

cli/src/mcp/server.ts

@@ -0,0 +1,50 @@
+import {
+  McpServer,
+  ResourceTemplate,
+} from "@modelcontextprotocol/sdk/server/mcp.js";
+import { z } from "zod";
+
+import { getVersion } from "~/utils/getProofKitVersion.js";
+
+interface ScaffoldPageParams {
+  name: string;
+  path: string;
+  techStack?: string;
+  template?: string;
+}
+
+interface ScaffoldProjectParams {
+  name: string;
+  path: string;
+  techStack?: string;
+  template?: string;
+}
+
+export function makeServer() {
+  const server = new McpServer({
+    name: "proofkit",
+    version: getVersion(),
+  });
+
+  server.tool(
+    "init",
+    "description",
+    { name: z.string(), type: z.enum(["browser", "wv"]) },
+    ({ name, type }) => {
+      console.log("create new project", name, type);
+      return { content: [{ type: "text", text: "Project created" }] };
+    }
+  );
+
+  server.resource(
+    "project-info",
+    new ResourceTemplate("proofkit://info?path={path}", { list: undefined }),
+    async (uri, { path }) => {
+      return {
+        contents: [{ uri: uri.href, text: `Project info for ${path}` }],
+      };
+    }
+  );
+
+  return server;
+}