Export MCP register tools functions (#631)

Author: ericglau

.changeset/famous-masks-cheat.md

@@ -0,0 +1,5 @@
+---
+'@openzeppelin/contracts-mcp': minor
+---
+
+Export functions to register MCP tools

.changeset/loud-clowns-divide.md

@@ -0,0 +1,5 @@
+---
+'@openzeppelin/wizard-common': minor
+---
+
+Bump minor version for semantic versioning stability

packages/mcp/package.json

@@ -5,8 +5,9 @@
   "license": "AGPL-3.0-only",
   "repository": "https://github.com/OpenZeppelin/contracts-wizard",
   "main": "dist/index.js",
+  "types": "dist/index.d.ts",
   "bin": {
-    "contracts-mcp": "dist/index.js"
+    "contracts-mcp": "dist/cli.js"
   },
   "ts:main": "src/index.ts",
   "files": [
@@ -37,4 +38,4 @@
     "ava": "^6.0.0",
     "ts-node": "^10.4.0"
   }
-}
+}

packages/mcp/src/cli.ts

@@ -0,0 +1,15 @@
+#!/usr/bin/env node
+
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { createServer } from './server';
+
+async function main() {
+  const transport = new StdioServerTransport();
+  const server = createServer();
+  await server.connect(transport);
+}
+
+main().catch(error => {
+  console.error('Failed to start OpenZeppelin Contracts MCP Server:', error);
+  process.exit(1);
+});

packages/mcp/src/index.ts

@@ -1,11 +1,4 @@
-#!/usr/bin/env node
-
-import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
-import { createServer } from './server';
-
-// Start receiving messages on stdin and sending messages on stdout
-const transport = new StdioServerTransport();
-(async () => {
-  const server = createServer();
-  await server.connect(transport);
-})();
+export { registerSolidityTools } from './solidity/tools';
+export { registerCairoTools } from './cairo/tools';
+export { registerStellarTools } from './stellar/tools';
+export { registerStylusTools } from './stylus/tools';