@atlisp/mcp 1.0.10 → 1.0.12

package/README.md

@@ -4,6 +4,10 @@ MCP (Model Context Protocol) Server for @lisp on CAD - 为 CAD 环境提供 @lis
 
 同时为 AI AGENT 操控 cad 提供调用工具和通道。
 
+## 版本
+
+**当前版本**: 1.0.10
+
 ## 功能特性
 
 - 连接 AutoCAD/ZWCAD/GStarCAD/BricsCAD
@@ -126,6 +130,20 @@ atlisp-mcp
 | `MESSAGE_TIMEOUT` | `60000` | 消息超时（毫秒） |
 | `BUSY_RETRIES` | `10` | CAD 忙碌时重试次数 |
 | `BUSY_DELAY` | `500` | CAD 忙碌重试间隔（毫秒） |
+| `MCP_API_KEY` | (无) | API Key 认证（可选） |
+
+### 运行测试
+
+```bash
+npm test
+```
+
+测试覆盖 54 个用例，包括：
+- 工具定义完整性
+- 连接/断开 CAD
+- 执行 LISP 代码
+- 包管理操作
+- 错误处理
 
 ## API
 

package/package.json

@@ -1,21 +1,21 @@
 {
     "name": "@atlisp/mcp",
-    "version": "1.0.10",
+    "version": "1.0.12",
     "description": "MCP Server for @lisp on CAD",
     "type": "module",
     "bin": {
         "atlisp-mcp": "./src/cli.js"
     },
-    "main": "src/index.js",
+    "main": "src/atlisp-mcp.js",
     "exports": {
-        ".": "./src/index.js",
+        ".": "./src/atlisp-mcp.js",
         "./cad": "./src/cad.js"
     },
     "files": [
         "src"
     ],
     "scripts": {
-        "start": "node src/index.js",
+        "start": "node src/atlisp-mcp.js",
         "test": "vitest run",
         "test:watch": "vitest"
     },

package/src/atlisp-mcp.js

@@ -0,0 +1,304 @@
+import express from 'express';
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import {
+    ListToolsRequestSchema,
+    CallToolRequestSchema,
+} from '@modelcontextprotocol/sdk/types.js';
+import { cad } from './cad.js';
+import { CAD_PLATFORMS, FILE_EXTENSIONS, PROTOCOL_VERSION, SERVER_NAME, SERVER_VERSION } from './constants.js';
+import { log } from './logger.js';
+import * as cadHandlers from './handlers/cad-handlers.js';
+import * as packageHandlers from './handlers/package-handlers.js';
+import * as functionHandlers from './handlers/function-handlers.js';
+import { createSseResponseHandler } from './sse-utils.js';
+
+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 { CAD_PLATFORMS, FILE_EXTENSIONS, MOCK_PACKAGES } from './constants.js';
+
+const TOOL_HANDLERS = {
+    connect_cad:           () => cadHandlers.connectCad(),
+    eval_lisp:             (a) => cadHandlers.evalLisp(a.code),
+    eval_lisp_with_result: (a) => cadHandlers.evalLisp(a.code, true),
+    get_cad_info:          () => cadHandlers.getCadInfo(),
+    list_packages:         () => packageHandlers.listPackages(),
+    search_packages:       (a) => packageHandlers.searchPackages(a.query),
+    install_package:       (a) => packageHandlers.installPackage(a.packageName),
+    get_platform_info:     () => ({ content: [{ type: 'text', text: `支持的 CAD 平台:\n${CAD_PLATFORMS.join('\n')}\n当前连接: ${cad.connected ? '已连接' : '未连接'}` }] }),
+    get_file_extensions:   (a) => {
+        const exts = FILE_EXTENSIONS[a.platform];
+        if (!exts) return { content: [{ type: 'text', text: `错误: 不支持的平台 "${a.platform}"` }], isError: true };
+        return { content: [{ type: 'text', text: `${a.platform} 文件扩展名: ${exts.join(', ')}` }] };
+    },
+    get_install_code:      () => ({ content: [{ type: 'text', text: `复制以下代码到 CAD 命令行安装 @lisp:\n\n(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))))` }] }),
+    at_command:            (a) => cadHandlers.atCommand(a.command),
+    new_document:          () => cadHandlers.newDocument(),
+    install_atlisp:        () => cadHandlers.installAtlisp(),
+    get_function_usage:    (a) => functionHandlers.getFunctionUsage(a.name, a.package),
+    list_functions:        (a) => functionHandlers.listFunctions(a.package),
+};
+
+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: {} }
+    },
+    {
+	name: 'get_function_usage',
+	description: '获取 @lisp 函数用法（从本地 JSON 库）',
+	inputSchema: {
+	    type: 'object',
+	    properties: {
+		name: { type: 'string', description: '函数名，如 string:length' },
+		package: { type: 'string', description: '包名，如 string（可选，不传则自动搜索）' }
+	    },
+	    required: ['name']
+	}
+    },
+    {
+	name: 'list_functions',
+	description: '列出 @lisp 函数库中的所有函数',
+	inputSchema: {
+	    type: 'object',
+	    properties: {
+		package: { type: 'string', description: '包名，如 string（可选）' }
+	    }
+	}
+    }
+];
+
+export async function handleToolCall(name, args) {
+    const handler = TOOL_HANDLERS[name];
+    if (!handler) return { content: [{ type: 'text', text: `未知工具: ${name}` }], isError: true };
+    return handler(args);
+}
+
+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);
+    }
+}
+
+let server = null;
+
+export function createMcpServer() {
+    server = new Server(
+	{ name: SERVER_NAME, version: SERVER_VERSION },
+	{ 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(`${SERVER_NAME} 运行在 stdio 模式`);
+    } else {
+        const app = express();
+        app.use(express.json({ type: ['application/json', 'application/json-rpc'] }));
+
+        app.use((req, res, next) => {
+            log(`${req.method} ${req.path} - body: ${JSON.stringify(req.body)}`);
+            next();
+        });
+
+        const PUBLIC_PATHS = ['/health'];
+        if (API_KEY) {
+            app.use((req, res, next) => {
+                if (PUBLIC_PATHS.includes(req.path)) 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' });
+        });
+
+        if (process.env.ENABLE_CORS) {
+            app.use((req, res, next) => {
+                res.setHeader('Access-Control-Allow-Origin', process.env.CORS_ORIGIN || '*');
+                res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS');
+                res.setHeader('Access-Control-Allow-Headers', 'Content-Type, Authorization, Accept, mcp-session-id');
+                if (req.method === 'OPTIONS') return res.status(204).end();
+                next();
+            });
+        }
+
+        createMcpServer();
+
+        const sessionId = crypto.randomUUID();
+
+        ['/sse', '/mcp/sse'].forEach(p => app.all(p, (req, res) => {
+            res.writeHead(200, { 'Content-Type': 'text/event-stream' });
+            const transport = new SSEServerTransport('/mcp', res);
+            server.connect(transport);
+        }));
+
+	app.post('/mcp', async (req, res) => {
+	    const body = req.body || {};
+	    const { method, params, id } = body;
+	    const accept = req.get('Accept') || '';
+	    const wantsSse = accept.includes('text/event-stream');
+	    log('accept: "' + accept + '", wantsSSE: ' + wantsSse);
+
+	    res.setHeader('mcp-session-id', sessionId);
+	    const { send } = createSseResponseHandler(res, wantsSse);
+
+	    if (!id) return res.status(204).end();
+
+	    try {
+		if (method === 'initialize') {
+		    send({ jsonrpc: '2.0', id, result: {
+			protocolVersion: PROTOCOL_VERSION,
+			capabilities: { tools: {} },
+			serverInfo: { name: SERVER_NAME, version: SERVER_VERSION }
+		    } }, true);
+		    return;
+		}
+
+		if (method === 'tools/list') {
+		    send({ jsonrpc: '2.0', id, result: { tools } }, true);
+		    return;
+		}
+
+		if (method === 'tools/call') {
+		    const toolName = params?.name;
+		    if (!toolName) {
+			send({ jsonrpc: '2.0', id, error: { code: -32602, message: 'Invalid params: missing tool name' } }, true);
+			return;
+		    }
+		    const result = await handleToolCall(toolName, params?.arguments || {});
+		    send({ jsonrpc: '2.0', id, result }, true);
+		    return;
+		}
+
+		send({ jsonrpc: '2.0', id, error: { code: -32601, message: 'Method not found' } }, true);
+	    } catch (e) {
+		log('MCP error: ' + e.message);
+		send({ jsonrpc: '2.0', id, error: { code: -32603, message: e.message } }, true);
+	    }
+	});
+
+	app.listen(PORT, HOST, async () => {
+	    await initCadConnection();
+	    console.error(`@lisp MCP Server 启动 - http://${HOST}:${PORT} - Streamable HTTP`);
+	});
+    }
+}
+
+if (!process.env.VITEST) {
+    await startServer();
+}

package/src/cad-worker.js

@@ -7,103 +7,53 @@ const BUSY_DELAY = parseInt(process.env.BUSY_DELAY || '500', 10);
 
 const cadConnect = edge.func(function() {/*
   async (input) => {
-      // Try existing CAD instances
       try {
           dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject("AutoCAD.Application");
           if (cad != null) {
               string ver = cad.Version.ToString();
               return new { success = true, platform = "AutoCAD", version = ver };
           }
-      } catch {}
+      } catch (Exception ex) {
+          System.Diagnostics.Debug.WriteLine("AutoCAD connect error: " + ex.Message);
+      }
       try {
           dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject("ZWCAD.Application");
           if (cad != null) {
               string ver = cad.Version.ToString();
               return new { success = true, platform = "ZWCAD", version = ver };
           }
-      } catch {}
+      } catch (Exception ex) {
+          System.Diagnostics.Debug.WriteLine("ZWCAD connect error: " + ex.Message);
+      }
       try {
           dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject("GStarCAD.Application");
           if (cad != null) {
               string ver = cad.Version.ToString();
               return new { success = true, platform = "GStarCAD", version = ver };
           }
-      } catch {}
+      } catch (Exception ex) {
+          System.Diagnostics.Debug.WriteLine("GStarCAD connect error: " + ex.Message);
+      }
       try {
           dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject("BricsCAD.Application");
           if (cad != null) {
               string ver = cad.Version.ToString();
               return new { success = true, platform = "BricsCAD", version = ver };
           }
-      } catch {}
-      
-      // Try to create new CAD instance
-      string[] cadPaths = new string[] {
-          @"C:\Program Files\Autodesk\AutoCAD 2022\acad.exe",
-          @"C:\Program Files\Autodesk\AutoCAD 2024\acad.exe",
-          @"C:\Program Files\Autodesk\AutoCAD 2023\acad.exe",
-          @"C:\Program Files\Autodesk\AutoCAD 2021\acad.exe",
-          @"C:\Program Files\Autodesk\AutoCAD 2020\acad.exe",
-          @"C:\Program Files\ZWSOFT\ZWCAD 2024\zwcad.exe",
-          @"C:\Program Files\ZWSOFT\ZWCAD 2023\zwcad.exe",
-          @"C:\Program Files\ZWSOFT\ZWCAD 2022\zwcad.exe",
-          @"C:\Program Files\ZWSOFT\ZWCAD 2021\zwcad.exe",
-          @"C:\Program Files\GSTARSOFT\GStarCAD 2024\GCAD.exe",
-          @"C:\Program Files\GSTARSOFT\GStarCAD 2023\GCAD.exe",
-          @"C:\Program Files\Bricsys\BricsCAD\V24\BricsCAD.exe",
-          @"C:\Program Files\Bricsys\BricsCAD\V23\BricsCAD.exe"
-      };
-      
-      foreach (string path in cadPaths) {
-          if (System.IO.File.Exists(path)) {
-              try {
-                  System.Diagnostics.Process.Start(path);
-                  System.Threading.Thread.Sleep(60000);
-                  
-                  // Try to connect to newly started CAD
-                  try {
-                      dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject("AutoCAD.Application");
-                      if (cad != null) {
-                          string ver = cad.Version.ToString();
-                          return new { success = true, platform = "AutoCAD", version = ver };
-                      }
-                  } catch {}
-                  try {
-                      dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject("ZWCAD.Application");
-                      if (cad != null) {
-                          string ver = cad.Version.ToString();
-                          return new { success = true, platform = "ZWCAD", version = ver };
-                      }
-                  } catch {}
-                  try {
-                      dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject("GStarCAD.Application");
-                      if (cad != null) {
-                          string ver = cad.Version.ToString();
-                          return new { success = true, platform = "GStarCAD", version = ver };
-                      }
-                  } catch {}
-                  try {
-                      dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject("BricsCAD.Application");
-                      if (cad != null) {
-                          string ver = cad.Version.ToString();
-                          return new { success = true, platform = "BricsCAD", version = ver };
-                      }
-                  } catch {}
-              } catch {}
-          }
+      } catch (Exception ex) {
+          System.Diagnostics.Debug.WriteLine("BricsCAD connect error: " + ex.Message);
       }
-      
       return new { success = false, error = "No CAD found" };
   }
 */});
 
 const cadSend = edge.func(function() {/*
   async (input) => {
-      dynamic d = input;
-      string code = d.code.ToString();
-      string platform = d.platform.ToString();
-      string progId = platform + ".Application";
       try {
+          dynamic d = input;
+          string code = d.code.ToString();
+          string platform = d.platform.ToString();
+          string progId = platform + ".Application";
           dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject(progId);
           if (cad != null) {
               dynamic doc = cad.ActiveDocument;
@@ -112,49 +62,51 @@ const cadSend = edge.func(function() {/*
                   return new { success = true };
               }
           }
-      } catch {}
+      } catch (Exception ex) {
+          System.Diagnostics.Debug.WriteLine("cadSend error: " + ex.Message);
+      }
       return new { success = false };
   }
  */});
 
 const cadSendWithResult = edge.func(function() {/*
   async (input) => {
-      dynamic d = input;
-      string code = d.code.ToString();
-      string platform = d.platform.ToString();
-      string progId = platform + ".Application";
-      
-      string tempDir = System.IO.Path.GetTempPath();
-      string tempFile = System.IO.Path.Combine(tempDir, "atlisp_result_" + System.Guid.NewGuid().ToString("N") + ".txt");
-      
-      // Use print to write result directly to file
-      string lispCode = "(progn(setq f(open \"" + tempFile.Replace("\\", "\\\\") + "\" \"w\"))(print " + code + " f)(close f))";
-      
+      string tempFile = null;
       try {
+          dynamic d = input;
+          string code = d.code.ToString();
+          string platform = d.platform.ToString();
+          string progId = platform + ".Application";
+          string tempDir = System.IO.Path.GetTempPath();
+          tempFile = System.IO.Path.Combine(tempDir, "atlisp_result_" + System.Guid.NewGuid().ToString("N") + ".txt");
+          string lispCode = "(progn(setq f(open \"" + tempFile.Replace("\\", "\\\\") + "\" \"w\"))(print " + code + " f)(close f))";
           dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject(progId);
           if (cad != null) {
               dynamic doc = cad.ActiveDocument;
               if (doc != null) {
                   doc.SendCommand(lispCode + "\n");
                   System.Threading.Thread.Sleep(500);
-                  
                   if (System.IO.File.Exists(tempFile)) {
-                      string result = System.IO.File.ReadAllText(tempFile).Trim();
-                      try { System.IO.File.Delete(tempFile); } catch {}
+                      string result = System.Text.Encoding.GetEncoding("GBK").GetString(System.IO.File.ReadAllBytes(tempFile)).Trim();
+                      System.IO.File.Delete(tempFile);
                       if (!string.IsNullOrEmpty(result)) {
                           return new { success = true, result = result };
                       }
-                      // Try evaluate result directly if file is empty
                       return new { success = true, result = "" };
                   }
                   return new { success = true, result = "" };
               }
           }
       } catch (Exception ex) {
-          try { if (System.IO.File.Exists(tempFile)) System.IO.File.Delete(tempFile); } catch {}
+          System.Diagnostics.Debug.WriteLine("cadSendWithResult error: " + ex.Message);
+          if (tempFile != null && System.IO.File.Exists(tempFile)) {
+              try { System.IO.File.Delete(tempFile); } catch {}
+          }
           return new { success = false, error = ex.Message };
       }
-      try { if (System.IO.File.Exists(tempFile)) System.IO.File.Delete(tempFile); } catch {}
+      if (tempFile != null && System.IO.File.Exists(tempFile)) {
+          try { System.IO.File.Delete(tempFile); } catch {}
+      }
       return new { success = false };
   }
  */});
@@ -180,9 +132,9 @@ process.stdin.on('data', (data) => {
 
 const cadIsBusy = edge.func(function() {/*
   async (input) => {
-      string platform = input.ToString();
-      string progId = platform + ".Application";
       try {
+          string platform = input.ToString();
+          string progId = platform + ".Application";
           dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject(progId);
           if (cad != null) {
               bool isBusy = false;
@@ -193,37 +145,42 @@ const cadIsBusy = edge.func(function() {/*
               }
               return new { isBusy = isBusy };
           }
-      } catch {}
+      } catch (Exception ex) {
+          System.Diagnostics.Debug.WriteLine("cadIsBusy error: " + ex.Message);
+      }
       return new { isBusy = false };
   }
  */});
 
 const cadHasDoc = edge.func(function() {/*
   async (input) => {
-      string platform = input.ToString();
-      string progId = platform + ".Application";
       try {
+          string platform = input.ToString();
+          string progId = platform + ".Application";
           dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject(progId);
           if (cad != null) {
               int docCount = cad.Documents.Count;
               return new { hasDoc = docCount > 0, docCount = docCount };
           }
-      } catch {}
+      } catch (Exception ex) {
+          System.Diagnostics.Debug.WriteLine("cadHasDoc error: " + ex.Message);
+      }
       return new { hasDoc = false, docCount = 0 };
   }
  */});
 
 const cadNewDoc = edge.func(function() {/*
   async (input) => {
-      string platform = input.ToString();
-      string progId = platform + ".Application";
       try {
+          string platform = input.ToString();
+          string progId = platform + ".Application";
           dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject(progId);
           if (cad != null) {
               cad.Documents.Add();
               return new { success = true };
           }
       } catch (Exception ex) {
+          System.Diagnostics.Debug.WriteLine("cadNewDoc error: " + ex.Message);
           return new { success = false, error = ex.Message };
       }
       return new { success = false, error = "CAD not found" };
@@ -246,7 +203,42 @@ async function waitForIdle(platform) {
   return false;
 }
 
+function checkParens(code) {
+  let depth = 0;
+  let inString = false;
+  let escapeNext = false;
+  for (let i = 0; i < code.length; i++) {
+    const ch = code[i];
+    if (escapeNext) {
+      escapeNext = false;
+      continue;
+    }
+    if (ch === '\\') {
+      escapeNext = true;
+      continue;
+    }
+    if (ch === '"') {
+      inString = !inString;
+      continue;
+    }
+    if (inString) continue;
+    if (ch === ';') {
+      while (i < code.length && code[i] !== '\n') i++;
+      continue;
+    }
+    if (ch === '(') depth++;
+    if (ch === ')') depth--;
+    if (depth < 0) return { valid: false, error: '多余右括号 )', pos: i };
+  }
+  if (depth > 0) return { valid: false, error: `缺少 ${depth} 个右括号 )`, pos: code.length - 1 };
+  if (depth < 0) return { valid: false, error: `多余 ${-depth} 个左括号 (` };
+  return { valid: true };
+}
+
 async function handleMessage(msg) {
+  if (msg.type === 'ping') {
+    return { pong: true };
+  }
   if (msg.type === 'connect') {
     return new Promise((resolve, reject) => {
       cadConnect({}, (e, r) => {
@@ -281,6 +273,10 @@ async function handleMessage(msg) {
     if (!docCheck.hasDoc) {
       return { success: false, error: 'No open document' };
     }
+    const parenCheck = checkParens(msg.code);
+    if (!parenCheck.valid) {
+      return { success: false, error: `括号错误: ${parenCheck.error} (位置 ${parenCheck.pos})` };
+    }
     if (!(await waitForIdle(msg.platform))) {
       return { success: false, error: 'CAD is busy, timeout' };
     }
@@ -301,6 +297,10 @@ async function handleMessage(msg) {
     if (!docCheck.hasDoc) {
       return { success: false, error: 'No open document' };
     }
+    const parenCheck = checkParens(msg.code);
+    if (!parenCheck.valid) {
+      return { success: false, error: `括号错误: ${parenCheck.error} (位置 ${parenCheck.pos})` };
+    }
     if (!(await waitForIdle(msg.platform))) {
       return { success: false, error: 'CAD is busy, timeout' };
     }