@shun-js/aibaiban-server 1.4.1 → 1.4.3

package/app.js

@@ -1,12 +1,12 @@
 // config
-const { parseServerConfig } = require('@shun-js/shun-config');
+const { parseServerConfig } = require("@shun-js/shun-config");
 
 // init
 (async () => {
   // config
   const config = await parseServerConfig(process.argv);
   if (!config) {
-    console.log('read server config fail');
+    console.log("read server config fail");
     return;
   }
 
@@ -20,23 +20,23 @@ const { parseServerConfig } = require('@shun-js/shun-config');
   options.config = config;
 
   // options.redis
-  options.redis = require('qiao-redis');
+  options.redis = require("qiao-redis");
   options.redisOptions = config.redisOptions;
 
   // options log
-  options.log = require('qiao-log');
-  options.logOptions = require('./server/log-options.js')();
+  options.log = require("qiao-log");
+  options.logOptions = require("./server/log-options.js")();
 
   // options rate limit
-  options.rateLimitLib = require('qiao-rate-limit');
+  options.rateLimitLib = require("qiao-rate-limit");
   options.rateLimitOptions = config.rateLimitOptions;
 
   // options checks
-  options.checks = [require('./server/util/check.js').checkUserAuth];
+  options.checks = [require("./server/util/check.js").checkUserAuth];
 
   // options modules
-  options.modules = [require('qiao-z-sms').init, require('qiao-z-nuser').init];
+  options.modules = [require("qiao-z-sms").init, require("qiao-z-nuser").init];
 
-  const app = await require('qiao-z')(options);
+  const app = await require("qiao-z")(options);
   app.listen(config.port);
 })();

package/assets/sw.js

@@ -1,3 +1,3 @@
 // Service Worker - 仅用于满足 PWA 安装条件
-self.addEventListener('install', () => self.skipWaiting());
-self.addEventListener('activate', (e) => e.waitUntil(self.clients.claim()));
+self.addEventListener("install", () => self.skipWaiting());
+self.addEventListener("activate", (e) => e.waitUntil(self.clients.claim()));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shun-js/aibaiban-server",
-  "version": "1.4.1",
+  "version": "1.4.3",
   "description": "aibaiban.com server",
   "keywords": [
     "ai aibaiban"
@@ -38,6 +38,7 @@
     "qiao-z-service": "^6.0.0",
     "qiao-z-sms": "^6.0.0",
     "viho-llm": "^1.1.0",
+    "ws": "^8.20.0",
     "zod": "^4.3.6",
     "zod-to-json-schema": "^3.25.1"
   },
@@ -45,5 +46,5 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "gitHead": "31816288306545bca83a6378d28bcb10109b5365"
+  "gitHead": "a07951954c905f106f4bbec45d23b133a73d15dd"
 }

package/server/controller/IndexController.js

@@ -1,12 +1,12 @@
 // service
-const service = require('../service/IndexService.js');
+const service = require("../service/IndexService.js");
 
 /**
  * controller
  */
 module.exports = (app) => {
   // index
-  app.get('/', (req, res) => {
+  app.get("/", (req, res) => {
     service.index(req, res);
   });
 };

package/server/controller/LLMController.js

@@ -1,12 +1,18 @@
 // service
-const service = require('../service/LLMService.js');
+const service = require("../service/LLMService.js");
+const ccService = require("../service/CCService.js");
 
 /**
  * controller
  */
 module.exports = (app) => {
   // draw agent
-  app.post('/draw-agent', (req, res) => {
+  app.post("/draw-agent", (req, res) => {
     service.drawAgent(req, res);
   });
+
+  // draw cc (Claude Code via Docker)
+  app.post("/draw-cc", (req, res) => {
+    ccService.drawCC(req, res);
+  });
 };

package/server/controller/SEOController.js

@@ -1,32 +1,32 @@
 // service
-const service = require('../service/SEOService.js');
+const service = require("../service/SEOService.js");
 
 /**
  * controller
  */
 module.exports = (app) => {
   // seo
-  app.get('/robots.txt', (req, res) => {
+  app.get("/robots.txt", (req, res) => {
     service.robots(req, res);
   });
-  app.get('/sitemap.xml', (req, res) => {
+  app.get("/sitemap.xml", (req, res) => {
     service.sitemap(req, res);
   });
-  app.get('/4cb288d7aef5469c92616c0f5b5aeb89.txt', (req, res) => {
+  app.get("/4cb288d7aef5469c92616c0f5b5aeb89.txt", (req, res) => {
     service.bingIndexNow(req, res);
   });
 
   // pwa
-  app.get('/manifest.json', (req, res) => {
+  app.get("/manifest.json", (req, res) => {
     service.manifestJson(req, res);
   });
-  app.get('/sw.js', (req, res) => {
+  app.get("/sw.js", (req, res) => {
     service.swJs(req, res);
   });
-  app.get('/icon-192.png', (req, res) => {
+  app.get("/icon-192.png", (req, res) => {
     service.icon192Png(req, res);
   });
-  app.get('/icon-512.png', (req, res) => {
+  app.get("/icon-512.png", (req, res) => {
     service.icon512Png(req, res);
   });
 };

package/server/log-options.js

@@ -4,19 +4,19 @@
  */
 module.exports = () => {
   // log options
-  const logLevel = 'debug';
-  const logPattern = 'yyyy-MM-dd-hh';
-  const logPath = require('path').resolve(__dirname, '../logs/qiao-z.log');
+  const logLevel = "debug";
+  const logPattern = "yyyy-MM-dd-hh";
+  const logPath = require("path").resolve(__dirname, "../logs/qiao-z.log");
 
   return {
     pm2: true,
-    pm2InstanceVar: 'INSTANCE_ID',
+    pm2InstanceVar: "INSTANCE_ID",
     appenders: {
       stdout: {
-        type: 'stdout',
+        type: "stdout",
       },
       datefile: {
-        type: 'dateFile',
+        type: "dateFile",
         pattern: logPattern,
         filename: logPath,
         keepFileExt: true,
@@ -27,7 +27,7 @@ module.exports = () => {
     categories: {
       default: {
         level: logLevel,
-        appenders: ['stdout', 'datefile'],
+        appenders: ["stdout", "datefile"],
       },
     },
   };

package/server/service/CCService.js

@@ -0,0 +1,111 @@
+const { chatFeishuMsg, errorFeishuMsg } = require("../util/feishu.js");
+const { sendPromptViaRelay } = require("../util/relay-client.js");
+
+/**
+ * drawCC - Claude Code via Docker
+ * 所有请求直接通过 WS relay 发送给 Docker 内的 Claude Code 处理
+ */
+exports.drawCC = async (req, res) => {
+  const methodName = "drawCC";
+  const messages = req.body.messages;
+
+  if (!messages?.length) {
+    const msg = "need messages";
+    req.logger.error(methodName, msg);
+    res.jsonFail(msg);
+    return;
+  }
+
+  const relayConfig = global.QZ_CONFIG.relay;
+  if (!relayConfig?.wsUrl) {
+    req.logger.error(methodName, "relay config missing");
+    res.jsonFail("relay not configured");
+    return;
+  }
+
+  res.streamingStart();
+
+  const lastUserMsg =
+    messages.filter((m) => m.role === "user").pop()?.content || "";
+  req.logger.info(methodName, "lastUserMsg", lastUserMsg);
+  chatFeishuMsg(req, `cc-${lastUserMsg}`);
+
+  try {
+    const startTime = Date.now();
+
+    res.streaming(
+      `data: ${JSON.stringify({ type: "status", step: "thinking" })}\n\n`,
+    );
+
+    const keepalive = setInterval(() => {
+      res.streaming(": keepalive\n\n");
+    }, 15000);
+
+    const result = await sendPromptViaRelay({
+      wsUrl: relayConfig.wsUrl,
+      authSecret: relayConfig.authSecret,
+      userId: req.user?.id || "guest",
+      prompt: lastUserMsg,
+      timeout: relayConfig.timeout || 600000,
+    });
+
+    clearInterval(keepalive);
+    const duration = Date.now() - startTime;
+
+    // 解析 Claude Code 返回结果
+    // Claude Code 可能返回：纯 JSON、markdown 包裹的 JSON、或普通文本
+    let parsed = null;
+    try {
+      let raw = result;
+      // 剥离 markdown 代码块: ```json\n...\n```
+      const fenceMatch = raw.match(/```(?:json)?\s*\n([\s\S]*?)\n```/);
+      if (fenceMatch) {
+        raw = fenceMatch[1].trim();
+      }
+      // 处理可能的双重 JSON 编码
+      let decoded = raw;
+      if (typeof decoded === "string") {
+        decoded = JSON.parse(decoded);
+      }
+      if (typeof decoded === "string") {
+        decoded = JSON.parse(decoded);
+      }
+      if (decoded && decoded.type === "excalidraw" && decoded.elements) {
+        parsed = decoded;
+      }
+    } catch {
+      // 非 excalidraw JSON
+    }
+
+    if (parsed) {
+      req.logger.info(
+        methodName,
+        "excalidraw elements",
+        parsed.elements.length,
+        `${duration}ms`,
+      );
+      chatFeishuMsg(req, `cc-excalidraw-${parsed.elements.length}-elements`);
+      res.streaming(`data: ${JSON.stringify({ type: "clear" })}\n\n`);
+      res.streaming(
+        `data: ${JSON.stringify({ type: "draw", elements: parsed.elements, duration })}\n\n`,
+      );
+    } else {
+      res.streaming(
+        `data: ${JSON.stringify({ type: "message", content: result })}\n\n`,
+      );
+    }
+
+    res.streamingEnd();
+  } catch (error) {
+    req.logger.error(methodName, "error", error);
+    errorFeishuMsg(req, error.message);
+    const userMessage =
+      error.message === "AI agent is offline"
+        ? "AI 助手离线中，请稍后重试"
+        : "服务暂时出错，请稍后重试";
+    res.streaming(
+      `data: ${JSON.stringify({ type: "error", message: userMessage })}\n\n`,
+    );
+    res.streamingEnd();
+  }
+};

package/server/service/IndexService.js

@@ -5,7 +5,7 @@
  */
 exports.index = async (req, res) => {
   // const
-  const pagePath = './views/index.html';
+  const pagePath = "./views/index.html";
 
   // is static
   const isStatic = await res.staticRender(pagePath);