@znt/mcp 1.1.1 → 2.0.1

package/index.js

@@ -1,129 +1,28 @@
 #!/usr/bin/env node
-import { Client } from '@modelcontextprotocol/sdk/client/index.js';
-import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js';
-import { Server } from '@modelcontextprotocol/sdk/server/index.js';
-import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
-import {
-  CallToolRequestSchema,
-  ListToolsRequestSchema,
-} from '@modelcontextprotocol/sdk/types.js';
 
-// ─── Configuration ───────────────────────────────────────────────
-const ZNT_API_URL = (process.env.ZNT_API_URL || 'http://localhost:8080').replace(/\/$/, '');
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 
-let clientInstance = null;
-let clientPromise = null;
+import { createZntMcpServer } from "./src/server.js";
 
-/**
- * Establishes an SSE connection to the native Znt Go MCP Server
- */
-async function connectClient() {
-  const sseUrl = new URL(`${ZNT_API_URL}/mcp/sse`);
-  const transport = new SSEClientTransport(sseUrl);
-  const client = new Client(
-    { name: 'znt-mcp-adapter', version: '1.2.0' },
-    { capabilities: {} }
-  );
+const application = createZntMcpServer();
 
-  client.onerror = () => {
-    clientInstance = null;
-    clientPromise = null;
-  };
-
-  client.onclose = () => {
-    clientInstance = null;
-    clientPromise = null;
-  };
-
-  await client.connect(transport);
-  clientInstance = client;
-  return client;
+try {
+  await application.initialize();
+} catch (error) {
+  // Keep the MCP endpoint available. Tool calls retry the complete
+  // install-and-start sequence after a transient download or startup failure.
+  console.error(`znt-core is not ready: ${error.message}`);
 }
 
-/**
- * Returns active connected client or initiates a new connection
- */
-async function getZntClient() {
-  if (clientInstance) {
-    return clientInstance;
-  }
-  if (clientPromise) {
-    return clientPromise;
-  }
-
-  clientPromise = connectClient().catch((err) => {
-    clientInstance = null;
-    clientPromise = null;
-    throw new Error(`Failed to connect to Znt Go server at ${ZNT_API_URL}/mcp. Make sure Znt backend is running (go run main.go). Detail: ${err.message}`);
-  });
-
-  return clientPromise;
-}
-
-// ─── MCP Server Setup (STDIO) ────────────────────────────────────
-const server = new Server(
-  { name: 'znt-mcp-adapter', version: '1.2.0' },
-  { capabilities: { tools: {} } }
-);
-
-// ─── Dynamic Tools Listing ───────────────────────────────────────
-server.setRequestHandler(ListToolsRequestSchema, async () => {
-  try {
-    const client = await getZntClient();
-    const res = await client.listTools();
-    return { tools: res.tools || [] };
-  } catch (err) {
-    // If backend is offline, return descriptive fallback status tool
-    return {
-      tools: [
-        {
-          name: 'znatok_server_offline',
-          description: `Znt backend is offline (${err.message}). Start server with 'go run main.go' to enable tools.`,
-          inputSchema: { type: 'object', properties: {} }
-        }
-      ]
-    };
-  }
-});
-
-// ─── Dynamic Tools Proxy Execution ──────────────────────────────
-server.setRequestHandler(CallToolRequestSchema, async (request) => {
-  const toolName = request.params.name;
-  if (toolName === 'znatok_server_offline') {
-    return {
-      content: [{
-        type: 'text',
-        text: `Znt backend server is not running on ${ZNT_API_URL}. Please start Znt backend by running 'go run main.go' in workspace.`
-      }],
-      isError: true
-    };
-  }
-
-  try {
-    const client = await getZntClient();
-    const res = await client.callTool({
-      name: toolName,
-      arguments: request.params.arguments || {}
-    });
-    return res;
-  } catch (err) {
-    return {
-      content: [{
-        type: 'text',
-        text: `Error executing tool ${toolName} via Znt Go backend: ${err.message}`
-      }],
-      isError: true
-    };
-  }
-});
+await application.server.connect(new StdioServerTransport());
 
-// ─── Run Server over STDIO ───────────────────────────────────────
-async function main() {
-  const transport = new StdioServerTransport();
-  await server.connect(transport);
+let closing = false;
+async function close() {
+  if (closing) return;
+  closing = true;
+  application.disconnect();
+  await application.server.close();
 }
 
-main().catch((error) => {
-  console.error('Fatal error running Znt MCP adapter:', error);
-  process.exit(1);
-});
+process.once("SIGINT", close);
+process.once("SIGTERM", close);

package/package.json

@@ -1,9 +1,34 @@
 {
   "name": "@znt/mcp",
-  "version": "1.1.1",
-  "description": "Model Context Protocol adapter for Znt",
-  "main": "index.js",
+  "version": "2.0.1",
+  "description": "Model Context Protocol server for znt-core through @znt/sdk-nodejs",
   "type": "module",
+  "main": "./index.js",
+  "bin": {
+    "znt-mcp": "./index.js"
+  },
+  "files": [
+    "index.js",
+    "src",
+    "README.md",
+    "AGENTS_EXAMPLE.md"
+  ],
+  "scripts": {
+    "start": "node index.js",
+    "test": "node --test",
+    "pack:check": "npm pack --dry-run"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.30.0",
+    "@znt/sdk-nodejs": "^1.0.0",
+    "yaml": "^2.9.0"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "engines": {
+    "node": ">=18"
+  },
   "keywords": [
     "mcp",
     "mcp-server",
@@ -12,23 +37,7 @@
     "znatok",
     "semantic-search",
     "code-analysis",
-    "call-graph",
-    "ast",
-    "ai-agent",
-    "antigravity",
-    "claude-desktop"
+    "call-graph"
   ],
-  "bin": {
-    "znt-mcp": "./index.js"
-  },
-  "files": [
-    "index.js"
-  ],
-  "scripts": {
-    "start": "node index.js"
-  },
-  "dependencies": {
-    "@modelcontextprotocol/sdk": "^1.0.1",
-    "ws": "^8.21.1"
-  }
+  "license": "UNLICENSED"
 }