@atlisp/mcp 1.3.0 → 1.3.1

package/src/cad-worker.js

@@ -1,498 +0,0 @@
-#!/usr/bin/env node
-import edge from 'edge-js';
-import process from 'process';
-import fs from 'fs';
-import path from 'path';
-
-const BUSY_RETRIES = parseInt(process.env.BUSY_RETRIES || '10', 10);
-const BUSY_DELAY = parseInt(process.env.BUSY_DELAY || '500', 10);
-
-const DEBUG_FILE = process.env.DEBUG_FILE || path.join(process.env.TEMP || '/tmp', 'cad-worker-debug.log');
-
-function log(msg) {
-    const entry = `${new Date().toISOString()} ${msg}\n`;
-    if (process.env.DEBUG) {
-        fs.appendFile(DEBUG_FILE, entry, () => {});
-    }
-}
-
-const cadConnect = edge.func(function() {/*
-  async (input) => {
-      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 (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 (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 (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 (Exception ex) {
-          System.Diagnostics.Debug.WriteLine("BricsCAD connect error: " + ex.Message);
-      }
-      return new { success = false, error = "No CAD found" };
-  }
-*/});
-
-const cadLaunch = edge.func(function() {/*
-  async (input) => {
-      string[] platforms = { "AutoCAD", "ZWCAD", "GStarCAD", "BricsCAD" };
-      string[] progIds = { "AutoCAD.Application", "ZWCAD.Application", "GStarCAD.Application", "BricsCAD.Application" };
-
-      for (int i = 0; i < platforms.Length; i++) {
-          try {
-              dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject(progIds[i]);
-              if (cad != null) {
-                  string ver = cad.Version.ToString();
-                  return new { success = true, platform = platforms[i], version = ver, launched = false };
-              }
-          } catch {}
-      }
-
-      for (int i = 0; i < platforms.Length; i++) {
-          try {
-              var type = System.Type.GetTypeFromProgID(progIds[i]);
-              if (type != null) {
-                  dynamic cad = System.Activator.CreateInstance(type);
-                  if (cad != null) {
-                      for (int retry = 0; retry < 60; retry++) {
-                          await System.Threading.Tasks.Task.Delay(500);
-                          try {
-                              string ver = cad.Version.ToString();
-                              return new { success = true, platform = platforms[i], version = ver, launched = true };
-                          } catch {}
-                      }
-                  }
-              }
-          } catch (Exception ex) {
-              System.Diagnostics.Debug.WriteLine("Create COM " + platforms[i] + " error: " + ex.Message);
-          }
-      }
-
-      return new { success = false, error = "No CAD found" };
-  }
-*/});
-
-const cadSend = edge.func(function() {/*
-  async (input) => {
-      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;
-              if (doc != null) {
-                  doc.SendCommand(code);
-                  return new { success = true };
-              }
-          }
-      } catch (Exception ex) {
-          System.Diagnostics.Debug.WriteLine("cadSend error: " + ex.Message);
-      }
-      return new { success = false };
-  }
- */});
-
-const cadSendWithResult = edge.func(function() {/*
-  async (input) => {
-      string tempFile = null;
-      string lispFile = null;
-      try {
-          dynamic d = input;
-          string code = d.code.ToString();
-          string platform = d.platform.ToString();
-          string encoding = d.encoding != null ? d.encoding.ToString() : "";
-          string progId = platform + ".Application";
-          string tempDir = System.IO.Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.UserProfile), "@lisp");
-          System.IO.Directory.CreateDirectory(tempDir);
-          
-          tempFile = System.IO.Path.Combine(tempDir, "atlisp_result_" + System.Guid.NewGuid().ToString("N") + ".txt");
-          lispFile = System.IO.Path.Combine(tempDir, "atlisp_code_" + System.Guid.NewGuid().ToString("N") + ".lsp");
-          
-          // 生成 LISP 代码：使用 vl-catch-all-apply 捕获结果和错误
-          // 返回 (success . result) 或 (nil . error-msg)
-          string lispCode = "(progn"
-              + "(setq f(open \"" + tempFile.Replace("\\", "\\\\") + "\" \"w\"))"
-              + "(setq r(vl-catch-all-apply '(lambda ()(progn " + code + "))))"
-              + "(if (vl-catch-all-error-p r)"
-              + "(princ(strcat \"ERROR:\"(vl-catch-all-error-message r)) f)"
-              + "(princ r f))"
-              + "(close f)"
-              + "(princ))";
-          
-            // 使用系统 ANSI 编码以兼容 GstarCAD / ZWCAD
-            System.IO.File.WriteAllText(lispFile, lispCode, System.Text.Encoding.GetEncoding(System.Globalization.CultureInfo.CurrentCulture.TextInfo.ANSICodePage));
-          
-          dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject(progId);
-          if (cad != null) {
-              dynamic doc = cad.ActiveDocument;
-              if (doc != null) {
-                  // 加载 .lsp 文件执行
-                  doc.SendCommand("(load \"" + lispFile.Replace("\\", "\\\\") + "\")\n");
-                  
-                  int retries = 10;
-                  string result = "";
-                  while (retries > 0 && !System.IO.File.Exists(tempFile)) {
-                      System.Threading.Thread.Sleep(200);
-                      retries--;
-                  }
-                  if (System.IO.File.Exists(tempFile)) {
-                      retries = 5;
-                      while (retries > 0) {
-                          try {
-                              byte[] bytes = System.IO.File.ReadAllBytes(tempFile);
-                              System.Text.Encoding enc;
-                              
-                              // 优先使用指定的编码
-                              if (!string.IsNullOrEmpty(encoding)) {
-                                  enc = System.Text.Encoding.GetEncoding(encoding);
-                              } else {
-                                  // 自动检测编码：先尝试 UTF-8，如果失败则使用 GBK
-                                  try {
-                                      string utf8Result = System.Text.Encoding.UTF8.GetString(bytes);
-                                      if (!utf8Result.Contains("\ufffd")) {
-                                          enc = System.Text.Encoding.UTF8;
-                                      } else {
-                                          enc = System.Text.Encoding.GetEncoding("gbk");
-                                      }
-                                  } catch {
-                                      try {
-                                          enc = System.Text.Encoding.GetEncoding("gbk");
-                                      } catch {
-                                          enc = System.Text.Encoding.Default;
-                                      }
-                                  }
-                              }
-                              
-                              result = enc.GetString(bytes).Trim();
-                              
-                              // 检查是否有错误前缀
-                              if (result.StartsWith("ERROR:")) {
-                                  return new { success = false, error = result.Substring(6).Trim() };
-                              }
-                              
-                              try { System.IO.File.Delete(tempFile); } catch {}
-                              try { System.IO.File.Delete(lispFile); } catch {}
-                              break;
-                          } catch (System.IO.IOException) {
-                              System.Threading.Thread.Sleep(200);
-                              retries--;
-                          }
-                      }
-                      if (!string.IsNullOrEmpty(result)) {
-                          return new { success = true, result = result };
-                      }
-                      return new { success = true, result = "" };
-                  }
-                  return new { success = true, result = "" };
-              }
-          }
-      } catch (Exception ex) {
-          System.Diagnostics.Debug.WriteLine("cadSendWithResult error: " + ex.Message);
-          if (tempFile != null) {
-              try { if (System.IO.File.Exists(tempFile)) System.IO.File.Delete(tempFile); } catch {}
-          }
-          if (lispFile != null) {
-              try { if (System.IO.File.Exists(lispFile)) System.IO.File.Delete(lispFile); } catch {}
-          }
-          return new { success = false, error = ex.Message };
-      }
-      if (tempFile != null) {
-          try { if (System.IO.File.Exists(tempFile)) System.IO.File.Delete(tempFile); } catch {}
-      }
-      if (lispFile != null) {
-          try { if (System.IO.File.Exists(lispFile)) System.IO.File.Delete(lispFile); } catch {}
-      }
-      return new { success = false };
-  }
-  */});
-
-process.stdin.setEncoding('utf8');
-
-let stdinBuffer = '';
-process.stdin.on('data', (data) => {
-  stdinBuffer += data;
-  const lines = stdinBuffer.split('\n');
-  stdinBuffer = lines.pop() || '';
-  for (const line of lines) {
-    if (!line.trim()) continue;
-    try {
-      const msg = JSON.parse(line);
-      const requestId = msg.requestId;
-      handleMessage(msg).then(result => {
-        process.stdout.write(JSON.stringify({ ...result, requestId }) + '\n');
-      }).catch(err => {
-        process.stdout.write(JSON.stringify({ error: err.message, requestId }) + '\n');
-      });
-    } catch (e) {
-      process.stdout.write(JSON.stringify({ error: 'invalid message' }) + '\n');
-    }
-  }
-});
-
-const cadIsBusy = edge.func(function() {/*
-  async (input) => {
-      try {
-          string platform = input.ToString();
-          string progId = platform + ".Application";
-          dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject(progId);
-          if (cad != null) {
-              bool isBusy = false;
-              if (platform == "ZWCAD") {
-                  isBusy = !(cad.GetZcadState().IsQuiescent);
-              } else {
-                  isBusy = !(cad.GetAcadState().IsQuiescent);
-              }
-              return new { isBusy = isBusy };
-          }
-      } catch (Exception ex) {
-          System.Diagnostics.Debug.WriteLine("cadIsBusy error: " + ex.Message);
-      }
-      return new { isBusy = false };
-  }
- */});
-
-const cadHasDoc = edge.func(function() {/*
-  async (input) => {
-      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 (Exception ex) {
-          System.Diagnostics.Debug.WriteLine("cadHasDoc error: " + ex.Message);
-      }
-      return new { hasDoc = false, docCount = 0 };
-  }
- */});
-
-const cadNewDoc = edge.func(function() {/*
-  async (input) => {
-      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" };
-  }
- */});
-
-const cadBringToFront = edge.func(function() {/*
-  async (input) => {
-      try {
-          string platform = input.ToString();
-          string progId = platform + ".Application";
-          dynamic cad = System.Runtime.InteropServices.Marshal.GetActiveObject(progId);
-          if (cad != null) {
-              cad.Visible = true;
-              return new { success = true };
-          }
-      } catch (Exception ex) {
-          System.Diagnostics.Debug.WriteLine("cadBringToFront error: " + ex.Message);
-          return new { success = false, error = ex.Message };
-      }
-      return new { success = false, error = "CAD not found" };
-  }
- */});
-
-async function waitForIdle(platform) {
-  let retries = BUSY_RETRIES;
-  while (retries > 0) {
-    const checkResult = await new Promise((resolve, reject) => {
-      cadIsBusy(platform || 'AutoCAD', (e, r) => {
-        if (e) reject(e);
-        else resolve(r);
-      });
-    });
-    if (!checkResult.isBusy) return true;
-    retries--;
-    if (retries > 0) await new Promise(resolve => setTimeout(resolve, BUSY_DELAY));
-  }
-  return false;
-}
-
-export function checkParens(code) {
-  let depth = 0;
-  let inString = false;
-  for (let i = 0; i < code.length; i++) {
-    const ch = code[i];
-    if (inString) {
-      if (ch === '\\' && i + 1 < code.length) {
-        i++;
-        continue;
-      }
-      if (ch === '"') {
-        inString = false;
-      }
-      continue;
-    }
-    if (ch === '"') {
-      inString = true;
-      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') {
-    try {
-      const result = await new Promise((resolve, reject) => {
-        cadConnect({}, (e, r) => {
-          if (e) reject(e);
-          else resolve(r);
-        });
-      });
-      if (result && result.success) return result;
-    } catch (e) {
-      log('cadConnect failed: ' + e.message);
-    }
-    return new Promise((resolve, reject) => {
-      cadLaunch({}, (e, r) => {
-        if (e) reject(e);
-        else resolve(r);
-      });
-    });
-  }
-  if (msg.type === 'hasdoc') {
-    return new Promise((resolve, reject) => {
-      cadHasDoc(msg.platform || 'AutoCAD', (e, r) => {
-        if (e) reject(e);
-        else resolve(r);
-      });
-    });
-  }
-  if (msg.type === 'newdoc') {
-    return new Promise((resolve, reject) => {
-      cadNewDoc(msg.platform || 'AutoCAD', (e, r) => {
-        if (e) reject(e);
-        else resolve(r);
-      });
-    });
-  }
-  if (msg.type === 'bringToFront') {
-    return new Promise((resolve, reject) => {
-      cadBringToFront(msg.platform || 'AutoCAD', (e, r) => {
-        if (e) reject(e);
-        else resolve(r);
-      });
-    });
-  }
-  if (msg.type === 'send') {
-    const docCheck = await new Promise((resolve, reject) => {
-      cadHasDoc(msg.platform || 'AutoCAD', (e, r) => {
-        if (e) reject(e);
-        else resolve(r);
-      });
-    });
-    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 || 'AutoCAD'))) {
-      return { success: false, error: 'CAD is busy, timeout' };
-    }
-    log(`send: ${msg.code}`);
-    return new Promise((resolve, reject) => {
-      cadSend({ code: msg.code, platform: msg.platform }, (e, r) => {
-        if (e) {
-          log(`send error: ${e.message}`);
-          reject(e);
-        } else {
-          log(`send result: ${JSON.stringify(r)}`);
-          resolve(r);
-        }
-      });
-    });
-  }
-  if (msg.type === 'sendResult') {
-    const docCheck = await new Promise((resolve, reject) => {
-      cadHasDoc(msg.platform || 'AutoCAD', (e, r) => {
-        if (e) reject(e);
-        else resolve(r);
-      });
-    });
-    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' };
-    }
-    log(`sendResult: ${msg.code}`);
-    return new Promise((resolve, reject) => {
-      cadSendWithResult({ code: msg.code, platform: msg.platform, encoding: msg.encoding || '' }, (e, r) => {
-        if (e) {
-          log(`sendResult error: ${e.message}`);
-          reject(e);
-        } else {
-          log(`sendResult result: ${JSON.stringify(r)}`);
-          resolve(r);
-        }
-      });
-    });
-  }
-  if (msg.type === 'isBusy') {
-    return new Promise((resolve, reject) => {
-      cadIsBusy(msg.platform || 'AutoCAD', (e, r) => {
-        if (e) reject(e);
-        else resolve(r);
-      });
-    });
-  }
-  return { error: 'unknown message type' };
-}