feat: add mcp server

packages/mcp-server/README.md

@@ -0,0 +1,13 @@
+# mcp-server
+
+## Usage:
+
+```sh
+UPLOADTHING_TOKEN=your-token npx @uploadthing/mcp-server
+```
+
+or
+
+```sh
+npx @uploadthing/mcp-server your-token
+```

packages/mcp-server/package.json

@@ -0,0 +1,25 @@
+{
+  "name": "@uploadthing/mcp-server",
+  "type": "module",
+  "version": "0.0.1",
+  "bin": "./dist/bin/index.js",
+  "scripts": {
+    "build": "bunchee --tsconfig tsconfig.build.json --no-external",
+    "dev": "bunchee -w --tsconfig tsconfig.build.json --no-clean --no-external"
+  },
+  "devDependencies": {
+    "@uploadthing/eslint-config": "workspace:*",
+    "@uploadthing/tsconfig": "workspace:*",
+    "bunchee": "^6.2.0",
+    "eslint": "9.17.0",
+    "typescript": "5.7.2"
+  },
+  "dependencies": {
+    "@juliusmarminge/trpc-mcp": "^0.2.0",
+    "@modelcontextprotocol/sdk": "^1.6.0",
+    "@trpc/server": "11.0.0-rc.772",
+    "effect": "3.12.0",
+    "uploadthing": "workspace:*",
+    "zod": "^3.24.1"
+  }
+}

packages/mcp-server/src/bin/index.ts

@@ -0,0 +1,28 @@
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+
+import { createServer } from "../server";
+
+const tokenstr =
+  process.argv.find((arg) => arg.startsWith("eyJhc")) ??
+  process.env.UPLOADTHING_TOKEN;
+
+if (!tokenstr) {
+  console.error(
+    `
+No uploadthing token provided.
+Either set it as an environment variable or pass it as an argument.
+Usage:
+    UPLOADTHING_TOKEN=your-token npx @uploadthing/mcp-server
+    npx @uploadthing/mcp-server your-token
+    `.trim(),
+  );
+  process.exit(1);
+}
+
+const transport = new StdioServerTransport();
+await createServer(tokenstr)
+  .connect(transport)
+  .catch((error) => {
+    console.error("Fatal error in main():", error);
+    process.exit(1);
+  });

packages/mcp-server/src/server.ts

@@ -0,0 +1,90 @@
+import { createMcpServer, type McpMeta } from "@juliusmarminge/trpc-mcp";
+import {
+  ListResourcesRequestSchema,
+  ReadResourceRequestSchema,
+} from "@modelcontextprotocol/sdk/types.js";
+import { initTRPC } from "@trpc/server";
+import { Schema } from "effect";
+import { z } from "zod";
+
+import { UTApi } from "uploadthing/server";
+import { UploadThingToken } from "uploadthing/types";
+
+import { version } from "../package.json";
+
+export function createServer(tokenstr: string) {
+  const utapi = new UTApi({ token: tokenstr });
+  const token = Schema.decodeUnknownSync(UploadThingToken)(tokenstr);
+
+  const t = initTRPC.meta<McpMeta>().create();
+
+  const toolRouter = t.router({
+    listFiles: t.procedure
+      .meta({
+        mcp: {
+          enabled: true,
+          name: "listFiles",
+          description: "List all files in the UploadThing app",
+        },
+      })
+      .input(
+        z.object({
+          cursor: z.string().optional(),
+        }),
+      )
+      .mutation(async ({ input }) => {
+        const offset = input.cursor ? Number.parseInt(input.cursor) : 0;
+        const files = await utapi.listFiles({
+          offset,
+        });
+
+        return files.files.map((file) => ({
+          uri: `https://${token.appId}.ufs.sh/f/${file.key}`,
+          ...file,
+        }));
+      }),
+  });
+
+  const server = createMcpServer({ name: "uploadthing", version }, toolRouter);
+
+  // Haven't added resource capabilities to trpc-mcp yet
+  server.setRequestHandler(ListResourcesRequestSchema, async (req) => {
+    const offset = req.params?.cursor ? Number.parseInt(req.params.cursor) : 0;
+    const files = await utapi.listFiles({
+      offset,
+    });
+
+    const nextCursor = files.hasMore
+      ? String(offset + files.files.length)
+      : null;
+
+    return {
+      nextCursor,
+      resources: await Promise.all(
+        files.files.map((file) => {
+          return {
+            uri: `https://${token.appId}.utfs.io/f/${file.key}`,
+            name: file.name,
+            // mimeType: file.type
+          };
+        }),
+      ),
+    };
+  });
+
+  server.setRequestHandler(ReadResourceRequestSchema, async (req) => {
+    const key = new URL(req.params.uri).pathname.slice(1);
+    const { ufsUrl } = await utapi.generateSignedURL(key);
+    const response = await fetch(ufsUrl);
+
+    return {
+      content: {
+        uri: req.params.uri,
+        mimeType: response.headers.get("Content-Type"),
+        blob: response.blob(),
+      },
+    };
+  });
+
+  return server;
+}

packages/mcp-server/tsconfig.build.json

@@ -0,0 +1,4 @@
+{
+  "extends": "@uploadthing/tsconfig/base.build",
+  "include": ["src/**/*", "*.ts", "test"]
+}

packages/mcp-server/tsconfig.json

@@ -0,0 +1,4 @@
+{
+  "extends": "@uploadthing/tsconfig/base",
+  "include": ["src/**/*", "*.ts", "test"]
+}