import { McpAgent } from "agents/mcp";
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { z } from "zod";
import { todo } from "node:test";
import { title } from "process";
import { JSONRPCErrorSchema } from "@modelcontextprotocol/sdk/types.js";
import { toASCII } from "punycode";
interface Project {
id: string;
name: string;
description: string;
createdAt: string;
updatedAt: string;
}
interface Todo {
id: string;
projectId: string;
title: string;
description: string;
status: "pending" | "in_progress" | "completed";
priority: "low" | "medium" | "high";
createdAt: string;
updatedAt: string;
}
// Define our MCP agent with tools
export class MyMCP extends McpAgent {
server = new McpServer({
name: "Project Planner",
version: "1.0.0",
});
private get kv(): KVNamespace {
return (this.env as Env).PROJECT_PLANNER;
}
private async getProjectList(): Promise<string[]> {
const listData = await this.kv.get("project:list");
return listData ? JSON.parse(listData) : [];
}
private async getTodoList(projectId: string): Promise<string[]> {
const listData = await this.kv.get(`project:${projectId}:todos`);
return listData ? JSON.parse(listData) : [];
}
private async getTodosByProject(projectId: string): Promise<Todo[]> {
const todoList = await this.getTodoList(projectId);
const todos: Todo[] = [];
for (const todoId of todoList) {
const todoData = await this.kv.get(`todo:${todoId}`);
if (todoData) {
todos.push(JSON.parse(todoData));
}
}
return todos;
}
async init() {
this.server.tool(
"create_project",
"Create a new Project",
{
name: z.string().describe("Project name"),
description: z.string().optional().describe("Project description"),
},
async ({ name, description }) => {
const projectId = crypto.randomUUID();
const project: Project = {
id: projectId,
name,
description: description || "",
createdAt: new Date().toISOString(),
updatedAt: new Date().toISOString(),
};
await this.kv.put(`project:${projectId}`, JSON.stringify(project));
const projectList = await this.getProjectList();
projectList.push(projectId);
await this.kv.put("project:list", JSON.stringify(projectList));
return {
content: [
{
type: "text",
text: JSON.stringify(project, null, 2),
},
],
};
}
);
this.server.tool("list_projects", "Get all the Projects", {}, async () => {
const projectList = await this.getProjectList();
const projects: Project[] = [];
for (const projectId of projectList) {
const projectData = await this.kv.get(`project:${projectId}`);
if (projectData) {
projects.push(JSON.parse(projectData));
}
}
return {
content: [
{
type: "text",
text: JSON.stringify(projects, null, 2),
},
],
};
});
this.server.tool(
"get_projects",
"Get a specific project by Id",
{ project_id: z.string().describe("Project Id") },
async ({ project_id }) => {
const projectData = await this.kv.get(`project:${project_id}`);
if (!projectData) {
throw new Error(`Project not founf with id ${project_id} `);
}
const project: Project = JSON.parse(projectData);
const todos = await this.getTodosByProject(project_id);
return {
content: [
{
type: "text",
text: JSON.stringify({ project, todos }, null, 2),
},
],
};
}
);
this.server.tool(
"delete_project",
"Delete a specific project and all its todos",
{ project_id: z.string().describe("Project Id") },
async ({ project_id }) => {
const projectData = await this.kv.get(`project:${project_id}`);
if (!projectData) {
throw new Error(`Project not found with id ${project_id} `);
}
const todos = await this.getTodosByProject(project_id);
for (const todo of todos) {
await this.kv.delete(`todo:${todo.id}`);
}
await this.kv.delete(`project:${project_id}:todos`);
await this.kv.delete(`project:${project_id}`);
const projectList = await this.getProjectList();
const updatedList = projectList.filter((id) => id !== project_id);
await this.kv.put("project:list", JSON.stringify(updatedList));
return {
content: [
{
type: "text",
text: `Project with ${project_id} and its corresponding todos are deleted`,
},
],
};
}
);
this.server.tool(
"create_todo",
"Create a new todo in a project",
{
project_id: z.string().describe("Project ID"),
title: z.string().describe("Todo title"),
description: z.string().optional().describe("Todo description"),
priority: z
.enum(["low", "medium", "high"])
.optional()
.describe("Todo Priority"),
},
async ({ project_id, title, description, priority }) => {
const projectData = await this.kv.get(`project:${project_id}`);
if (!projectData) {
throw new Error(`Project not found with id:${project_id}`);
}
const todoId = crypto.randomUUID();
const todo: Todo = {
id: todoId,
projectId: project_id,
title,
description: description || "",
status: "pending",
priority: priority || "medium",
createdAt: new Date().toISOString(),
updatedAt: new Date().toISOString(),
};
await this.kv.put(`todo:${todoId}`, JSON.stringify(todo));
const todoList = await this.getTodoList(project_id);
todoList.push(todoId);
await this.kv.put(
`project:${project_id}:todos`,
JSON.stringify(todoList)
);
return {
content: [
{
type: "text",
text: JSON.stringify(todo, null, 2),
},
],
};
}
);
this.server.tool(
"update_todo",
"update a todo's properties in a project",
{
todo_id: z.string().describe("Todo ID"),
title: z.string().describe("New Todo title"),
description: z.string().optional().describe("New Todo description"),
status: z
.enum(["pending", "in_progress", "completed"])
.optional()
.describe("New Todo Status"),
priority: z
.enum(["low", "medium", "high"])
.optional()
.describe("New Todo Priority"),
},
async ({ todo_id, title, description, status, priority }) => {
const todoData = await this.kv.get(`todo:${todo_id}`);
if (!todoData) {
throw new Error(`Todo with ID: ${todo_id} not found `);
}
const todo: Todo = JSON.parse(todoData);
if (title !== undefined) todo.title = title;
if (description !== undefined) todo.description = description;
if (status !== undefined) todo.status = status;
if (priority !== undefined) todo.priority = priority;
todo.updatedAt = new Date().toISOString();
await this.kv.put(`todo:${todo_id}`, JSON.stringify(todo));
return {
content: [
{
type: "text",
text: JSON.stringify(todo, null, 2),
},
],
};
}
);
this.server.tool(
"delete_todo",
"delete a todo from a project",
{
todo_id: z.string().describe("Todo ID"),
},
async ({ todo_id }) => {
const todoData = await this.kv.get(`todo:${todo_id}`);
if (!todoData) {
throw new Error(`Todo with ID: ${todo_id} not found `);
}
const todo: Todo = JSON.parse(todoData);
const todoList = await this.getTodoList(todo.projectId);
const updatedList = todoList.filter((id) => id !== todo_id);
await this.kv.put(
`project:${todo.projectId}:todos`,
JSON.stringify(updatedList)
);
await this.kv.delete(`todo:${todo_id}`);
return {
content: [
{
type: "text",
text: `Todo ${todo_id} has been deleted`,
},
],
};
}
);
this.server.tool(
"get_todo",
"Get a specific todo by ID",
{
todo_id: z.string().describe("Todo ID"),
},
async ({ todo_id }) => {
const todoData = await this.kv.get(`todo:${todo_id}`);
if (!todoData) {
throw new Error(`Todo with ID: ${todo_id} not found `);
}
const todo: Todo = JSON.parse(todoData);
return {
content: [
{
type: "text",
text: JSON.stringify(todo, null, 2),
},
],
};
}
);
this.server.tool(
"list_todos",
"List all todos project",
{
project_id: z.string().describe("Project ID"),
status: z
.enum(["pending", "in_progress", "completed", "all"])
.optional()
.describe("Filter by status"),
},
async ({ project_id, status }) => {
const projectData = await this.kv.get(`project:${project_id}`);
if (!projectData) {
throw new Error(`Todo with ID: ${project_id} not found `);
}
let todos = await this.getTodosByProject(project_id);
if (status && status !== "all") {
todos = todos.filter((todo) => todo.status === status);
}
return {
content: [
{
type: "text",
text: JSON.stringify(todos, null, 2),
},
],
};
}
);
}
}
export default {
fetch(request: Request, env: Env, ctx: ExecutionContext) {
const url = new URL(request.url);
if (url.pathname === "/sse" || url.pathname === "/sse/message") {
return MyMCP.serveSSE("/sse").fetch(request, env, ctx);
}
if (url.pathname === "/mcp") {
return MyMCP.serve("/mcp").fetch(request, env, ctx);
}
return new Response("Not found", { status: 404 });
},
};
