import { McpAgent } from "agents/mcp";
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { closeDb } from "./database/connection";
import { registerAllTools } from "./tools/register-tools";
// Simple MCP server for speedrunethereum.com build discovery
export class SpeedrunMCP extends McpAgent<Env, Record<string, never>, Record<string, never>> {
server = new McpServer({
name: "SpeedRunEthereum Build Discovery",
version: "1.0.0",
});
/**
* Cleanup database connections when Durable Object is shutting down
*/
async cleanup(): Promise<void> {
try {
await closeDb();
console.log("Database connections closed successfully");
} catch (error) {
console.error("Error during database cleanup:", error);
}
}
/**
* Durable Objects alarm handler - used for cleanup
*/
async alarm(): Promise<void> {
await this.cleanup();
}
async init() {
registerAllTools(this.server, this.env);
}
}
export default {
fetch(request: Request, env: Env, ctx: ExecutionContext) {
const url = new URL(request.url);
if (url.pathname === "/sse" || url.pathname === "/sse/message") {
return SpeedrunMCP.serveSSE("/sse").fetch(request, env, ctx);
}
if (url.pathname === "/mcp") {
return SpeedrunMCP.serve("/mcp").fetch(request, env, ctx);
}
return new Response("SpeedRunEthereum MCP Server - Use /mcp or /sse endpoints", { status: 200 });
},
};