import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { z } from "zod";
import { createCodeExecutor, createSearchExecutor } from "./executor";
import { truncateResponse } from "./truncate";
import { PRODUCTS } from "./data/products";
const CLOUDFLARE_TYPES = `
interface CloudflareRequestOptions {
method: "GET" | "POST" | "PUT" | "PATCH" | "DELETE";
path: string;
query?: Record<string, string | number | boolean | undefined>;
body?: unknown;
contentType?: string; // Custom Content-Type header (defaults to application/json if body is present)
rawBody?: boolean; // If true, sends body as-is without JSON.stringify
}
interface CloudflareResponse<T = unknown> {
success: boolean;
result: T;
errors: Array<{ code: number; message: string }>;
messages: Array<{ code: number; message: string }>;
result_info?: {
page: number;
per_page: number;
total_pages: number;
count: number;
total_count: number;
};
}
declare const cloudflare: {
request<T = unknown>(options: CloudflareRequestOptions): Promise<CloudflareResponse<T>>;
};
declare const accountId: string;
`;
const SPEC_TYPES = `
interface OperationInfo {
summary?: string;
description?: string;
tags?: string[];
parameters?: Array<{ name: string; in: string; required?: boolean; schema?: unknown; description?: string }>;
requestBody?: { required?: boolean; content?: Record<string, { schema?: unknown }> };
responses?: Record<string, { description?: string; content?: Record<string, { schema?: unknown }> }>;
}
interface PathItem {
get?: OperationInfo;
post?: OperationInfo;
put?: OperationInfo;
patch?: OperationInfo;
delete?: OperationInfo;
}
declare const spec: {
paths: Record<string, PathItem>;
};
`;
export function createServer(env: Env, apiToken: string): McpServer {
const server = new McpServer({
name: "cloudflare-api",
version: "0.1.0",
});
const executeCode = createCodeExecutor(env);
const executeSearch = createSearchExecutor(env);
server.registerTool(
"search",
{
description: `Search the Cloudflare OpenAPI spec. All $refs are pre-resolved inline.
Products: ${PRODUCTS.slice(0, 30).join(", ")}... (${PRODUCTS.length} total)
Types:
${SPEC_TYPES}
Examples:
// Find endpoints by product
async () => {
const results = [];
for (const [path, methods] of Object.entries(spec.paths)) {
for (const [method, op] of Object.entries(methods)) {
if (op.tags?.some(t => t.toLowerCase() === 'workers')) {
results.push({ method: method.toUpperCase(), path, summary: op.summary });
}
}
}
return results;
}
// Get endpoint with requestBody schema (refs are resolved)
async () => {
const op = spec.paths['/accounts/{account_id}/d1/database']?.post;
return { summary: op?.summary, requestBody: op?.requestBody };
}
// Get endpoint parameters
async () => {
const op = spec.paths['/accounts/{account_id}/workers/scripts']?.get;
return op?.parameters;
}`,
inputSchema: {
code: z
.string()
.describe(
"JavaScript async arrow function to search the OpenAPI spec"
),
},
},
async ({ code }) => {
try {
const result = await executeSearch(code);
return {
content: [{ type: "text", text: truncateResponse(result) }],
};
} catch (error) {
const errorMessage =
error instanceof Error ? error.message : String(error);
return {
content: [{ type: "text", text: `Error: ${errorMessage}` }],
isError: true,
};
}
}
);
server.registerTool(
"execute",
{
description: `Execute JavaScript code against the Cloudflare API. First use the 'search' tool to find the right endpoints, then write code using the cloudflare.request() function.
Available in your code:
${CLOUDFLARE_TYPES}
Your code must be an async arrow function that returns the result.
Example:
async () => {
const response = await cloudflare.request({
method: "GET",
path: \`/accounts/\${accountId}/workers/scripts\`
});
return response.result;
}
// Upload a Worker script (service worker syntax)
async () => {
const workerCode = \`addEventListener('fetch', event => {
event.respondWith(new Response('Hello World!'));
});\`;
return await cloudflare.request({
method: "PUT",
path: \`/accounts/\${accountId}/workers/scripts/my-worker\`,
body: workerCode,
contentType: "application/javascript",
rawBody: true
});
}`,
inputSchema: {
code: z.string().describe("JavaScript async arrow function to execute"),
account_id: z
.string()
.describe(
"Your Cloudflare account ID (run `npx wrangler whoami` to see available accounts)"
),
},
},
async ({ code, account_id }) => {
try {
const result = await executeCode(code, account_id, apiToken);
return {
content: [{ type: "text", text: truncateResponse(result) }],
};
} catch (error) {
const errorMessage =
error instanceof Error ? error.message : String(error);
return {
content: [{ type: "text", text: `Error: ${errorMessage}` }],
isError: true,
};
}
}
);
return server;
}
