mcp-framework/dist/cli/project/add-prompt.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 { toPascalCase } from "../utils/string-utils.js";
import { validateMCPProject } from "../utils/validate-project.js";
export async function addPrompt(name) {
    await validateMCPProject();
    let promptName = name;
    if (!promptName) {
        const response = await prompts([
            {
                type: "text",
                name: "name",
                message: "What is the name of your prompt?",
                validate: (value) => /^[a-z0-9-]+$/.test(value)
                    ? true
                    : "Prompt name can only contain lowercase letters, numbers, and hyphens",
            },
        ]);
        if (!response.name) {
            console.log("Prompt creation cancelled");
            process.exit(1);
        }
        promptName = response.name;
    }
    if (!promptName) {
        throw new Error("Prompt name is required");
    }
    const className = toPascalCase(promptName);
    const fileName = `${className}Prompt.ts`;
    const promptsDir = join(process.cwd(), "src/prompts");
    try {
        await mkdir(promptsDir, { recursive: true });
        const promptContent = `import { MCPPrompt } from "mcp-framework";
import { z } from "zod";

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

class ${className}Prompt extends MCPPrompt<${className}Input> {
  name = "${promptName}";
  description = "${className} prompt description";

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

  async generateMessages({ message }: ${className}Input) {
    return [
      {
        role: "user",
        content: {
          type: "text",
          text: message,
        },
      },
    ];
  }
}

export default ${className}Prompt;`;
        await writeFile(join(promptsDir, fileName), promptContent);
        console.log(`Prompt ${promptName} created successfully at src/prompts/${fileName}`);
    }
    catch (error) {
        console.error("Error creating prompt:", error);
        process.exit(1);
    }
}