@atlisp/mcp 1.0.10 → 1.0.11

package/src/index.js

@@ -1,379 +0,0 @@
-import express from 'express';
-import { Server } from '@modelcontextprotocol/sdk/server/index.js';
-import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
-import {
-  ListToolsRequestSchema,
-  CallToolRequestSchema,
-} from '@modelcontextprotocol/sdk/types.js';
-import fs from 'fs';
-import path from 'path';
-import os from 'os';
-import crypto from 'crypto';
-import { cad } from './cad.js';
-
-const DEBUG_FILE = path.join(os.tmpdir(), 'mcp-server-debug.log');
-function log(msg) {
-  fs.writeFileSync(DEBUG_FILE, new Date().toISOString() + ' ' + msg + '\n', { flag: 'a' });
-}
-log('Server starting, cad loaded: ' + typeof cad);
-
-const PORT = process.env.PORT || 8110;
-const HOST = process.env.HOST || '0.0.0.0';
-const TRANSPORT = process.env.TRANSPORT || 'http';
-const API_KEY = process.env.MCP_API_KEY || '';
-if (API_KEY) {
-  log('API Key authentication enabled');
-}
-
-export const CAD_PLATFORMS = ['AutoCAD', 'ZWCAD', 'GStarCAD', 'BricsCAD'];
-export const FILE_EXTENSIONS = {
-  'AutoCAD': ['.fas', '.vlx'],
-  'ZWCAD': ['.zelx', '.vls'],
-  'BricsCAD': ['.des']
-};
-
-export const MOCK_PACKAGES = [
-  { name: 'base', description: '基础函数库', version: '1.0.0' },
-  { name: 'at-pm', description: '工程项目管理', version: '2.1.0' },
-  { name: 'network', description: '网络功能模块', version: '1.5.0' },
-  { name: 'userman', description: '用户管理', version: '1.2.0' },
-  { name: 'pkgman', description: '包管理器', version: '2.0.0' },
-  { name: 'sidebar', description: '侧边栏工具', version: '1.0.0' },
-  { name: 'aibot', description: 'AI 助手', version: '0.9.0' },
-  { name: 'tips', description: '每日提示', version: '1.0.0' },
-  { name: 'function', description: '函数库', version: '3.0.0' }
-];
-
-export const tools = [
-  {
-    name: 'connect_cad',
-    description: '连接到 CAD (AutoCAD/ZWCAD/GStarCAD/BricsCAD)',
-    inputSchema: { type: 'object', properties: {} }
-  },
-  {
-    name: 'eval_lisp',
-    description: '在 CAD 中执行 AutoLISP 代码（不返回结果）',
-    inputSchema: {
-      type: 'object',
-      properties: { code: { type: 'string', description: '要执行的 LISP 代码' } },
-      required: ['code']
-    }
-  },
-  {
-    name: 'eval_lisp_with_result',
-    description: '在 CAD 中执行 AutoLISP 代码并返回结果',
-    inputSchema: {
-      type: 'object',
-      properties: { code: { type: 'string', description: '要执行的 LISP 代码' } },
-      required: ['code']
-    }
-  },
-  {
-    name: 'get_cad_info',
-    description: '获取当前 CAD 信息',
-    inputSchema: { type: 'object', properties: {} }
-  },
-  {
-    name: 'list_packages',
-    description: '列出已安装的 @lisp 包',
-    inputSchema: { type: 'object', properties: {} }
-  },
-  {
-    name: 'search_packages',
-    description: '搜索 @lisp 包',
-    inputSchema: {
-      type: 'object',
-      properties: { query: { type: 'string', description: '搜索关键词' } },
-      required: ['query']
-    }
-  },
-  {
-    name: 'install_package',
-    description: '安装 @lisp 包到 CAD',
-    inputSchema: {
-      type: 'object',
-      properties: { packageName: { type: 'string', description: '包名称' } },
-      required: ['packageName']
-    }
-  },
-  {
-    name: 'get_platform_info',
-    description: '获取 CAD 平台信息',
-    inputSchema: { type: 'object', properties: {} }
-  },
-  {
-    name: 'get_file_extensions',
-    description: '获取指定平台的文件扩展名',
-    inputSchema: {
-      type: 'object',
-      properties: { platform: { type: 'string', enum: CAD_PLATFORMS, description: 'CAD 平台' } },
-      required: ['platform']
-    }
-  },
-  {
-    name: 'get_install_code',
-    description: '获取 @lisp 安装代码',
-    inputSchema: { type: 'object', properties: {} }
-  },
-  {
-    name: 'at_command',
-    description: '执行 @lisp 命令',
-    inputSchema: {
-      type: 'object',
-      properties: { command: { type: 'string', description: '@lisp 命令' } },
-      required: ['command']
-    }
-  },
-  {
-    name: 'new_document',
-    description: '在 CAD 中新建空白文档',
-    inputSchema: { type: 'object', properties: {} }
-  },
-  {
-    name: 'install_atlisp',
-    description: '在 CAD 中安装 @lisp',
-    inputSchema: { type: 'object', properties: {} }
-  }
-];
-
-export async function handleToolCall(name, args) {
-  switch (name) {
-    case 'connect_cad':
-      log('connect_cad called, cad.connected before: ' + cad.connected);
-      try {
-        log('calling cad.connect()...');
-        const connected = await cad.connect();
-        log('cad.connect() returned: ' + connected + ', cad.connected: ' + cad.connected);
-        if (connected) {
-          return { content: [{ type: 'text', text: `已连接到 ${cad.getPlatform()} ${cad.getVersion()}` }] };
-        }
-      } catch (e) {
-        log('connect_cad error: ' + e.message);
-      }
-      return { content: [{ type: 'text', text: '未找到运行中的 CAD，请确保 CAD 已启动并启用 COM 接口' }], isError: true };
-
-    case 'eval_lisp':
-      if (!cad.connected) { await cad.connect(); }
-      if (!cad.connected) return { content: [{ type: 'text', text: '未连接 CAD' }], isError: true };
-      await cad.sendCommand(args.code + '\n');
-      return { content: [{ type: 'text', text: `已发送命令: ${args.code}` }] };
-
-    case 'eval_lisp_with_result':
-      if (!cad.connected) { await cad.connect(); }
-      if (!cad.connected) return { content: [{ type: 'text', text: '未连接 CAD' }], isError: true };
-      const result = await cad.sendCommandWithResult(args.code);
-      if (result !== null) {
-        return { content: [{ type: 'text', text: result }] };
-      }
-      return { content: [{ type: 'text', text: '执行失败或无返回值' }], isError: true };
-
-    case 'get_cad_info':
-      if (!cad.connected) { await cad.connect(); }
-      if (!cad.connected) return { content: [{ type: 'text', text: 'CAD 未连接' }] };
-      const info = await cad.getInfo();
-      return { content: [{ type: 'text', text: info }] };
-
-    case 'list_packages':
-      const packages = MOCK_PACKAGES.map(p => `${p.name} v${p.version} - ${p.description}`);
-      return { content: [{ type: 'text', text: `已安装的 @lisp 包:\n${packages.join('\n')}` }] };
-
-    case 'search_packages':
-      const results = MOCK_PACKAGES.filter(p => p.name.includes(args.query) || p.description.includes(args.query));
-      if (results.length === 0) return { content: [{ type: 'text', text: `未找到包含 "${args.query}" 的包` }] };
-      return { content: [{ type: 'text', text: `搜索结果:\n${results.map(p => `${p.name} v${p.version} - ${p.description}`).join('\n')}` }] };
-
-    case 'install_package':
-      const pkg = MOCK_PACKAGES.find(p => p.name === args.packageName);
-      if (!pkg) return { content: [{ type: 'text', text: `错误: 包 "${args.packageName}" 不存在` }], isError: true };
-      if (cad.connected) await cad.sendCommand(`(@::load-module '${args.packageName})\n`);
-      return { content: [{ type: 'text', text: `包 "${args.packageName}" 安装命令已发送！` }] };
-
-    case 'get_platform_info':
-      return { content: [{ type: 'text', text: `支持的 CAD 平台:\n${CAD_PLATFORMS.join('\n')}\n当前连接: ${cad.connected ? '已连接' : '未连接'}` }] };
-
-    case 'get_file_extensions':
-      const exts = FILE_EXTENSIONS[args.platform];
-      if (!exts) return { content: [{ type: 'text', text: `错误: 不支持的平台 "${args.platform}"` }], isError: true };
-      return { content: [{ type: 'text', text: `${args.platform} 文件扩展名: ${exts.join(', ')}` }] };
-
-    case 'get_install_code':
-      const code = '(progn(vl-load-com)(setq s strcat h "http" o(vlax-create-object (s"win"h".win"h"request.5.1"))v vlax-invoke e eval r read)(v o\'open "get" (s h"://atlisp.""cn/@"):vlax-true)(v o\'send)(v o\'WaitforResponse 1000)(e(r(vlax-get-property o\'ResponseText))))';
-      return { content: [{ type: 'text', text: `复制以下代码到 CAD 命令行安装 @lisp:\n\n${code}` }] };
-
-    case 'at_command':
-      if (!cad.connected) return { content: [{ type: 'text', text: '请先连接 CAD' }], isError: true };
-      await cad.sendCommand(`${args.command}\n`);
-      return { content: [{ type: 'text', text: `已发送命令: ${args.command}` }] };
-
-    case 'new_document':
-      if (!cad.connected) { await cad.connect(); }
-      if (!cad.connected) return { content: [{ type: 'text', text: '未连接 CAD' }], isError: true };
-      const docResult = await cad.newDoc();
-      if (docResult) {
-        return { content: [{ type: 'text', text: '已在 CAD 中新建空白文档' }] };
-      }
-      return { content: [{ type: 'text', text: '新建文档失败' }], isError: true };
-
-    case 'install_atlisp':
-      if (!cad.connected) { await cad.connect(); }
-      if (!cad.connected) return { content: [{ type: 'text', text: '未连接 CAD' }], isError: true };
-      const installCode ='(progn(vl-load-com)(setq s strcat h "http" o(vlax-create-object (s"win"h".win"h"request.5.1"))v vlax-invoke e eval r read)(v o\'open "get" (s h"://atlisp.""cn/@"):vlax-true)(v o\'send)(v o\'WaitforResponse 1000)(e(r(vlax-get-property o\'ResponseText))))';
-      await cad.sendCommand(installCode + '\n');
-      return { content: [{ type: 'text', text: '已发送 @lisp 安装代码到 CAD' }] };
-
-    default:
-      return { content: [{ type: 'text', text: `未知工具: ${name}` }], isError: true };
-  }
-}
-
-async function initCadConnection() {
-  log('Attempting to connect to CAD on startup...');
-  try {
-    const connected = await cad.connect();
-    if (connected) {
-      log('Auto-connected to CAD: ' + cad.getPlatform() + ' ' + cad.getVersion());
-      console.error('已自动连接到 CAD: ' + cad.getPlatform() + ' ' + cad.getVersion());
-    } else {
-      log('No CAD found on startup');
-    }
-  } catch (e) {
-    log('Auto-connect failed: ' + e.message);
-  }
-}
-
-export function createMcpServer() {
-  const server = new Server(
-    { name: 'atlisp-mcp-server', version: '1.0.10' },
-    { capabilities: { tools: {} } }
-  );
-
-  server.setRequestHandler(ListToolsRequestSchema, async () => ({ tools }));
-
-  server.setRequestHandler(CallToolRequestSchema, async (request) => {
-    const { name, arguments: args } = request.params;
-    return await handleToolCall(name, args || {});
-  });
-
-  return server;
-}
-
-export async function startServer() {
-  if (TRANSPORT === 'stdio') {
-    await initCadConnection();
-    const mcpServer = createMcpServer();
-    const transport = new StdioServerTransport();
-    await mcpServer.connect(transport);
-    console.error('@lisp MCP Server 运行在 stdio 模式');
-  } else {
-    const app = express();
-    app.use(express.json({ type: ['application/json', 'application/json-rpc'] }));
-
-    if (API_KEY) {
-      app.use((req, res, next) => {
-        if (req.path === '/health') return next();
-        const auth = req.get('Authorization');
-        if (auth === `Bearer ${API_KEY}`) return next();
-        res.status(401).json({ error: 'Unauthorized' });
-      });
-    }
-
-    app.get('/health', (req, res) => {
-      res.json({ status: 'ok', cad: cad.connected ? 'connected' : 'disconnected' });
-    });
-
-    let mcpServer = null;
-    const sessionId = crypto.randomUUID();
-
-    app.post('/mcp', async (req, res) => {
-      if (!mcpServer) {
-        mcpServer = createMcpServer();
-      }
-
-      const body = req.body || {};
-      const method = body.method;
-      const params = body.params;
-      const id = body.id;
-      const accept = req.get('Accept') || '';
-      log('accept: "' + accept + '", wantsSSE: ' + accept.includes('text/event-stream'));
-
-      res.setHeader('mcp-session-id', sessionId);
-
-      const sendResponse = (data) => {
-        const json = JSON.stringify(data);
-        if (accept.includes('text/event-stream')) {
-          res.setHeader('Content-Type', 'text/event-stream');
-          return res.end('data: ' + json + '\n\n');
-        } else {
-          res.setHeader('Content-Type', 'application/json');
-          return res.end(json);
-        }
-      };
-
-      if (!id) {
-        return res.status(204).end();
-      }
-
-      try {
-        if (method === 'initialize') {
-          sendResponse({
-            jsonrpc: '2.0',
-            id,
-            result: {
-              protocolVersion: '2024-11-05',
-              capabilities: { tools: {} },
-              serverInfo: { name: 'atlisp-mcp-server', version: '1.0.10' }
-            }
-          });
-          return;
-        }
-
-        if (method === 'tools/list') {
-          sendResponse({
-            jsonrpc: '2.0',
-            id,
-            result: { tools }
-          });
-          return;
-        }
-
-        if (method === 'tools/call') {
-          const toolName = params?.name;
-          if (!toolName) {
-            sendResponse({
-              jsonrpc: '2.0',
-              id,
-              error: { code: -32602, message: 'Invalid params: missing tool name' }
-            });
-            return;
-          }
-          const toolArgs = params?.arguments || {};
-          const result = await handleToolCall(toolName, toolArgs);
-          sendResponse({ jsonrpc: '2.0', id, result });
-          return;
-        }
-
-        sendResponse({
-          jsonrpc: '2.0',
-          id,
-          error: { code: -32601, message: 'Method not found' }
-        });
-      } catch (e) {
-        log('MCP error: ' + e.message);
-        return res.status(500).json({
-          jsonrpc: '2.0',
-          id,
-          error: { code: -32603, message: e.message }
-        });
-      }
-    });
-
-    app.listen(PORT, HOST, async () => {
-      await initCadConnection();
-      console.error(`@lisp MCP Server 启动 - http://${HOST}:${PORT} - Streamable HTTP`);
-    });
-  }
-}
-
-if (!process.env.VITEST) {
-  await startServer();
-}