add mcp server to CLI

Author: andrelandgraf

atw-cli/bun.lock

@@ -2,7 +2,7 @@
   "name": "atw-cli",
   "module": "index.ts",
   "type": "module",
-  "version": "1.2.0",
+  "version": "1.3.0",
   "private": true,
   "scripts": {
     "prettier:check": "prettier --check .",
@@ -18,13 +18,15 @@
   },
   "dependencies": {
     "@inkjs/ui": "^2.0.0",
+    "@modelcontextprotocol/sdk": "^1.7.0",
     "@rocicorp/zero": "^0.16.2025031000",
     "commander": "^13.1.0",
     "ink": "^5.2.0",
     "ink-link": "^4.1.0",
     "picocolors": "^1.1.1",
     "qrcode-terminal": "^0.12.0",
     "react": "^18.3.1",
-    "react-devtools-core": "^6.1.1"
+    "react-devtools-core": "^6.1.1",
+    "zod": "^3.24.2"
   }
 }

atw-cli/package.json

@@ -1,5 +1,11 @@
 import { createJarvis } from "../core/create-jarvis";
 import { createRegisterCommand } from "./register";
+import { createMCPServeCommand } from "./mcp";
+import { createZero } from "../core/create-zero";
 
 const jarvis = createJarvis("Plopix");
-export const commands = [createRegisterCommand({ tBot: jarvis })];
+const zero = createZero();
+export const commands = [
+  createRegisterCommand({ tBot: jarvis, zero }),
+  createMCPServeCommand({ zero }),
+];

atw-cli/src/commands/index.ts

@@ -0,0 +1,82 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import packageJson from "../../package.json";
+import { Command } from "commander";
+import { z } from "zod";
+import { registerAction, type Event } from "../actions/register";
+import type { AtwZero } from "../core/create-zero";
+import { sleep } from "bun";
+
+type Deps = {
+  zero: AtwZero;
+};
+
+export const createMCPServeCommand = ({ zero }: Deps): Command => {
+  const command = new Command("mcp-serve")
+    .description("Start the MCP server.")
+    .action(async () => {
+      const mcpServer = new McpServer({
+        name: "All Things Web meetup events",
+        version: packageJson.version,
+      });
+
+      mcpServer.tool(
+        "get-events",
+        "Get all upcoming published All Things Web events",
+        {},
+        async () => {
+            const getEvents = async (): Promise<Event[]> => {
+                let count = 0;
+              return new Promise(async (resolve) => {
+                for(let i = 0; i < 10; i++) {
+                  const events = zero.query.events
+                    //.where("startDate", ">", new Date().getTime())
+                    .orderBy("startDate", "desc")
+                    .limit(10)
+                    .run();
+                  if(!events.length) {
+                    count += 1;
+                    await sleep(200);
+                  } else {
+                    resolve(events);
+                  }
+                }
+              });
+            };
+            const events = await getEvents();
+          return { content: events.map((event) => ({
+            type: "text",
+            text: `
+             name: ${event.name}
+             start: ${event.startDate}
+             location: ${event.shortLocation}
+             tagline: ${event.tagline}
+             eventId: ${event.id}`
+          })) };
+        },
+      );
+
+      mcpServer.tool(
+        "register",
+        "Register to an All Things Web event",
+        {
+          eventId: z.string().describe("The event id to register to."),
+          email: z.string().email().describe("The email to register with."),
+        },
+        async ({ eventId, email }) => {
+          const results = await registerAction(email, eventId);
+          let text = "Registered!";
+          if (results.success === false) {
+            text = `Oh no! Something went wrong! ${results.error}`;
+          }
+          return {
+            content: [{ type: "text", text }, {type: "text", text: `eventId: ${eventId}`}],
+          };
+        },
+      );
+
+      const stdioServer = new StdioServerTransport();
+      await mcpServer.connect(stdioServer);
+    });
+  return command;
+};

atw-cli/src/commands/mcp.ts

@@ -5,9 +5,11 @@ import pc from "picocolors";
 import { logo } from "..";
 import { AppLayout } from "../ui/app-layout";
 import type { TalkativeBot } from "../contracts/talkative-bot-interface";
+import type { AtwZero } from "../core/create-zero";
 
 type Deps = {
   tBot: TalkativeBot;
+  zero: AtwZero;
 };
 export const createRegisterCommand = (deps: Deps): Command => {
   const command = new Command("register")
@@ -18,7 +20,7 @@ export const createRegisterCommand = (deps: Deps): Command => {
         "We can't wait to see you soon! Select one of the events below to register.",
       );
       const { waitUntilExit, unmount } = render(
-        <AppLayout title="Register to an event!">
+        <AppLayout zero={deps.zero} title="Register to an event!">
           <RegisterJourney unmount={() => unmount()} deps={deps} />
         </AppLayout>,
         {

atw-cli/src/commands/register.tsx

@@ -0,0 +1,14 @@
+import { Zero } from "@rocicorp/zero";
+import { schema } from "@lib/zero-sync/schema";
+
+export type AtwZero = ReturnType<typeof createZero>;
+
+export const createZero = () => {
+  const z = new Zero({
+    userID: "anon",
+    server: "https://allthingsweb-sync.fly.dev",
+    schema,
+    kvStore: "mem",
+  });
+  return z;
+};

atw-cli/src/core/create-zero.ts

@@ -47,4 +47,3 @@ try {
   console.error(exception);
   process.exit(1);
 }
-process.exit(0);

atw-cli/src/index.ts

@@ -1,30 +1,25 @@
 import { Box } from "ink";
 import { ZeroProvider } from "@rocicorp/zero/react";
-import { Zero } from "@rocicorp/zero";
-import { schema } from "@lib/zero-sync/schema";
 import { ThemeProvider } from "@inkjs/ui";
 import type React from "react";
 import { customTheme } from "./theme";
 import { Header } from "./components/header";
 import { Footer } from "./footer";
+import type { AtwZero } from "../core/create-zero";
 
 type AppLayoutProps = {
   children: React.ReactNode;
   title: string;
+  zero: AtwZero;
 };
 
 export const AppLayout = ({
   children,
   title,
+  zero,
 }: AppLayoutProps): React.ReactNode => {
-  const z = new Zero({
-    userID: "anon",
-    server: "https://allthingsweb-sync.fly.dev",
-    schema,
-    kvStore: "mem",
-  });
   return (
-    <ZeroProvider zero={z}>
+    <ZeroProvider zero={zero}>
       <ThemeProvider theme={customTheme}>
         <Header level={1}>{title}</Header>
         <Box flexDirection="column" padding={1}>

atw-cli/src/ui/app-layout.tsx

@@ -2,10 +2,9 @@ import { eq } from "drizzle-orm";
 import { buildContainer } from "~/modules/container.server";
 import { eventsTable, InsertEvent } from "@lib/db/schema.server";
 
-const slug = "2025-03-12-all-things-web-hack-evening";
+const slug = "2025-04-26-hackathon-at-sentry";
 const eventData = {
-  startDate: new Date("2025-03-12T17:30:00.000Z"),
-  endDate: new Date("2025-03-12T20:30:00.000Z"),
+  isDraft: false,
 } satisfies Partial<InsertEvent>;
 
 async function main() {