initial commit

commands/mcp.ts

@@ -0,0 +1,74 @@
+import { Command } from 'commander';
+import { readConfig } from "../config.js";
+import { addCreatePubkeyCommand } from "../logic/create-pubkey.js";
+import { addFindSnippetsCommand } from "../logic/find_snippets.js";
+import { addFindUserCommand } from "../logic/find_user.js";
+import { addListUsernamesCommand } from "../logic/list_usernames.js";
+import { addPublishCodeSnippetCommand } from "../logic/publish-code-snippet.js";
+import { addPublishCommand } from "../logic/publish.js";
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+
+// Define type for command functions
+type CommandFunction = (server: McpServer) => void;
+
+// Map of command names to their registration functions
+const commandMap: Record<string, CommandFunction> = {
+    publish: addPublishCommand,
+    "publish-snippet": addPublishCodeSnippetCommand,
+    "create-pubkey": addCreatePubkeyCommand,
+    "find-user": addFindUserCommand,
+    "find-snippets": addFindSnippetsCommand,
+    "list-usernames": addListUsernamesCommand,
+};
+
+// Global server instance
+let mcpServer: McpServer | null = null;
+
+export function registerMcpCommand(program: Command): void {
+    program
+        .command('mcp')
+        .description('Start the MCP server')
+        .action(async () => {
+            try {
+                // Create the MCP server
+                mcpServer = new McpServer({
+                    name: "Nostr Publisher",
+                    version: "1.0.0",
+                });
+
+                // Register all MCP commands
+                registerMcpCommands(mcpServer);
+
+                // Connect the server to the transport
+                const transport = new StdioServerTransport();
+                await mcpServer.connect(transport);
+            } catch (error) {
+                console.error("Error starting MCP server:", error);
+                process.exit(1);
+            }
+        });
+}
+
+// Register all MCP commands, filtered by config if specified
+export function registerMcpCommands(server: McpServer) {
+    const config = readConfig();
+    const enabledCommands = config.mcpCommands;
+
+    // If mcpCommands is specified in config, only register those commands
+    if (enabledCommands && enabledCommands.length > 0) {
+        for (const cmd of enabledCommands) {
+            if (commandMap[cmd]) {
+                commandMap[cmd](server);
+            }
+        }
+    } else {
+        // Otherwise register all commands
+        addPublishCommand(server);
+        addPublishCodeSnippetCommand(server);
+        addCreatePubkeyCommand(server);
+        addFindUserCommand(server);
+        addFindSnippetsCommand(server);
+        addListUsernamesCommand(server);
+    }
+}