mcp-framework/dist/cli/project/add-tool.js

Framework for building Model Context Protocol (MCP) servers in Typescript

import { writeFile, mkdir } from "fs/promises";
import { join } from "path";
import prompts from "prompts";
import { validateMCPProject } from "../utils/validate-project.js";
import { toPascalCase } from "../utils/string-utils.js";
export async function addTool(name) {
    await validateMCPProject();
    let toolName = name;
    if (!toolName) {
        const response = await prompts([
            {
                type: "text",
                name: "name",
                message: "What is the name of your tool?",
                validate: (value) => /^[a-z0-9-]+$/.test(value)
                    ? true
                    : "Tool name can only contain lowercase letters, numbers, and hyphens",
            },
        ]);
        if (!response.name) {
            console.log("Tool creation cancelled");
            process.exit(1);
        }
        toolName = response.name;
    }
    if (!toolName) {
        throw new Error("Tool name is required");
    }
    const className = toPascalCase(toolName);
    const fileName = `${className}Tool.ts`;
    const toolsDir = join(process.cwd(), "src/tools");
    try {
        await mkdir(toolsDir, { recursive: true });
        const toolContent = `import { MCPTool } from "mcp-framework";
import { z } from "zod";

interface ${className}Input {
  message: string;
}

class ${className}Tool extends MCPTool<${className}Input> {
  name = "${toolName}";
  description = "${className} tool description";

  schema = {
    message: {
      type: z.string(),
      description: "Message to process",
    },
  };

  async execute(input: ${className}Input) {
    return \`Processed: \${input.message}\`;
  }
}

export default ${className}Tool;`;
        await writeFile(join(toolsDir, fileName), toolContent);
        console.log(`Tool ${toolName} created successfully at src/tools/${fileName}`);
    }
    catch (error) {
        console.error("Error creating tool:", error);
        process.exit(1);
    }
}