feat: add cli tool

Author: QuantGeekDev

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "mcp-framework",
-  "version": "0.1.8",
+  "version": "0.1.9",
   "description": "Framework for building Model Context Protocol (MCP) servers in Typescript",
   "type": "module",
   "author": "Alex Andru <[email protected]>",
@@ -16,7 +16,8 @@
     "dist"
   ],
   "bin": {
-    "mcp-build": "./dist/cli/build.js"
+    "mcp": "./dist/cli/index.js",
+    "mcp-build": "./dist/cli/framework/build.js"
   },
   "scripts": {
     "build": "tsc",
@@ -36,6 +37,9 @@
     "@modelcontextprotocol/sdk": "^0.6.0"
   },
   "dependencies": {
+    "@types/prompts": "^2.4.9",
+    "commander": "^12.1.0",
+    "prompts": "^2.4.2",
     "zod": "^3.23.8"
   },
   "devDependencies": {

package.json

@@ -3,6 +3,12 @@ import { spawnSync } from "child_process";
 import { readFileSync, writeFileSync } from "fs";
 import { join } from "path";
 
+export function buildFramework() {
+  runTsc();
+  addShebang();
+  console.log("MCP Build complete");
+}
+
 function runTsc() {
   const tscPath = join(process.cwd(), "node_modules", ".bin", "tsc");
   const tsc = spawnSync(tscPath, [], {
@@ -30,9 +36,3 @@ function addShebang() {
     process.exit(1);
   }
 }
-
-runTsc();
-
-addShebang();
-
-console.log("MCP Build complete");

src/cli/build.ts src/cli/framework/build.ts

@@ -0,0 +1,49 @@
+#!/usr/bin/env node
+import { Command } from "commander";
+import { createProject } from "./project/create.js";
+import { addTool } from "./project/add-tool.js";
+import { addPrompt } from "./project/add-prompt.js";
+import { addResource } from "./project/add-resource.js";
+import { buildFramework } from "./framework/build.js";
+
+const program = new Command();
+
+program
+  .name("mcp")
+  .description("CLI for managing MCP server projects")
+  .version("0.1.8");
+
+program
+  .command("build-framework", { hidden: true })
+  .description("Build the MCP framework")
+  .action(buildFramework);
+
+program
+  .command("create")
+  .description("Create a new MCP server project")
+  .argument("[name]", "project name")
+  .action(createProject);
+
+program
+  .command("add")
+  .description("Add a new component to your MCP server")
+  .addCommand(
+    new Command("tool")
+      .description("Add a new tool")
+      .argument("[name]", "tool name")
+      .action(addTool)
+  )
+  .addCommand(
+    new Command("prompt")
+      .description("Add a new prompt")
+      .argument("[name]", "prompt name")
+      .action(addPrompt)
+  )
+  .addCommand(
+    new Command("resource")
+      .description("Add a new resource")
+      .argument("[name]", "resource name")
+      .action(addResource)
+  );
+
+program.parse();

src/cli/index.ts

@@ -0,0 +1,86 @@
+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?: string) {
+  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: string) =>
+          /^[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);
+  }
+}

src/cli/project/add-prompt.ts

@@ -0,0 +1,73 @@
+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 addResource(name?: string) {
+  await validateMCPProject();
+
+  let resourceName = name;
+  if (!resourceName) {
+    const response = await prompts([
+      {
+        type: "text",
+        name: "name",
+        message: "What is the name of your resource?",
+        validate: (value: string) =>
+          /^[a-z0-9-]+$/.test(value)
+            ? true
+            : "Resource name can only contain lowercase letters, numbers, and hyphens",
+      },
+    ]);
+
+    if (!response.name) {
+      console.log("Resource creation cancelled");
+      process.exit(1);
+    }
+
+    resourceName = response.name;
+  }
+
+  if (!resourceName) {
+    throw new Error("Resource name is required");
+  }
+
+  const className = toPascalCase(resourceName);
+  const fileName = `${className}Resource.ts`;
+  const resourcesDir = join(process.cwd(), "src/resources");
+
+  try {
+    await mkdir(resourcesDir, { recursive: true });
+
+    const resourceContent = `import { MCPResource, ResourceContent } from "mcp-framework";
+
+class ${className}Resource extends MCPResource {
+  uri = "resource://${resourceName}";
+  name = "${className}";
+  description = "${className} resource description";
+  mimeType = "application/json";
+
+  async read(): Promise<ResourceContent[]> {
+    return [
+      {
+        uri: this.uri,
+        mimeType: this.mimeType,
+        text: JSON.stringify({ message: "Hello from ${className} resource" }),
+      },
+    ];
+  }
+}
+
+export default ${className}Resource;`;
+
+    await writeFile(join(resourcesDir, fileName), resourceContent);
+
+    console.log(
+      `Resource ${resourceName} created successfully at src/resources/${fileName}`
+    );
+  } catch (error) {
+    console.error("Error creating resource:", error);
+    process.exit(1);
+  }
+}