lark-mcp-server 0.0.1 → 0.0.3

package/index.js

@@ -268,27 +268,138 @@ async function sendRequest(request) {
   }
 }
 
-// 读取 stdin
+// 修复 JSON 字符串中的控制字符
+// JSON 规范不允许字符串内有实际的换行符，必须转义为 \n
+function fixJsonControlChars(str) {
+  let result = '';
+  let inString = false;
+  let escapeNext = false;
+
+  for (let i = 0; i < str.length; i++) {
+    const char = str[i];
+    const code = char.charCodeAt(0);
+
+    if (escapeNext) {
+      result += char;
+      escapeNext = false;
+      continue;
+    }
+
+    if (char === '\\' && inString) {
+      result += char;
+      escapeNext = true;
+      continue;
+    }
+
+    if (char === '"') {
+      inString = !inString;
+      result += char;
+      continue;
+    }
+
+    // 如果在字符串内且遇到控制字符，转义它们
+    if (inString && code < 32) {
+      if (code === 10) {
+        result += '\\n';
+      } else if (code === 13) {
+        result += '\\r';
+      } else if (code === 9) {
+        result += '\\t';
+      } else {
+        // 其他控制字符用 Unicode 转义
+        result += '\\u' + code.toString(16).padStart(4, '0');
+      }
+      continue;
+    }
+
+    result += char;
+  }
+
+  return result;
+}
+
+// 读取 stdin - 使用更智能的 JSON 解析
 let buffer = '';
 process.stdin.setEncoding('utf8');
 process.stdin.on('data', (chunk) => {
   buffer += chunk;
 
   // 尝试解析完整的 JSON 消息
-  let newlineIndex;
-  while ((newlineIndex = buffer.indexOf('\n')) !== -1) {
-    const line = buffer.slice(0, newlineIndex);
-    buffer = buffer.slice(newlineIndex + 1);
-
-    if (line.trim()) {
-      try {
-        const request = JSON.parse(line);
-        log(`Received: ${request.method}, id=${request.id}`);
-        sendRequest(request);
-      } catch (e) {
-        log('Parse error:', e.message);
+  // 使用大括号配对来检测完整的 JSON 对象
+  let startIndex = 0;
+
+  while (startIndex < buffer.length) {
+    // 找到第一个 {
+    const jsonStart = buffer.indexOf('{', startIndex);
+    if (jsonStart === -1) {
+      // 没有找到 JSON 开始，清理 buffer 中 startIndex 之前的内容
+      buffer = buffer.slice(startIndex);
+      break;
+    }
+
+    // 尝试从这个位置解析 JSON
+    let braceCount = 0;
+    let inString = false;
+    let escapeNext = false;
+    let jsonEnd = -1;
+
+    for (let i = jsonStart; i < buffer.length; i++) {
+      const char = buffer[i];
+
+      if (escapeNext) {
+        escapeNext = false;
+        continue;
+      }
+
+      if (char === '\\' && inString) {
+        escapeNext = true;
+        continue;
+      }
+
+      if (char === '"' && !escapeNext) {
+        inString = !inString;
+        continue;
       }
+
+      if (!inString) {
+        if (char === '{') {
+          braceCount++;
+        } else if (char === '}') {
+          braceCount--;
+          if (braceCount === 0) {
+            jsonEnd = i + 1;
+            break;
+          }
+        }
+      }
+    }
+
+    if (jsonEnd === -1) {
+      // JSON 不完整，等待更多数据
+      buffer = buffer.slice(startIndex);
+      break;
     }
+
+    // 提取完整的 JSON
+    let jsonStr = buffer.slice(jsonStart, jsonEnd);
+
+    // 修复 JSON 字符串内的控制字符
+    jsonStr = fixJsonControlChars(jsonStr);
+
+    try {
+      const request = JSON.parse(jsonStr);
+      log(`Received: ${request.method}, id=${request.id}`);
+      sendRequest(request);
+    } catch (e) {
+      log('Parse error:', e.message, 'JSON preview:', jsonStr.slice(0, 100));
+    }
+
+    startIndex = jsonEnd;
+  }
+
+  // 清理已处理的部分
+  if (startIndex > 0 && startIndex <= buffer.length) {
+    buffer = buffer.slice(startIndex);
   }
 });
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "lark-mcp-server",
-  "version": "0.0.1",
+  "version": "0.0.3",
   "description": "飞书 MCP 本地代理 - 让 AI 编程工具读取飞书文档，Token 本地安全存储",
   "type": "module",
   "bin": {