@shun-js/aibaiban-server 1.4.1 → 1.4.3

package/server/service/SEOService.js

@@ -1,11 +1,11 @@
 // path
-const path = require('path');
+const path = require("path");
 
 // mime
-const mime = require('mime-types');
+const mime = require("mime-types");
 
 // qiao
-const { readFile } = require('qiao-file');
+const { readFile } = require("qiao-file");
 
 /**
  * robots
@@ -13,7 +13,7 @@ const { readFile } = require('qiao-file');
  * @param {*} res
  */
 exports.robots = async (req, res) => {
-  const filePath = path.resolve(__dirname, '../../assets/robots.txt');
+  const filePath = path.resolve(__dirname, "../../assets/robots.txt");
   const content = await readFile(filePath);
   res.send(content, mime.lookup(filePath));
 };
@@ -24,7 +24,7 @@ exports.robots = async (req, res) => {
  * @param {*} res
  */
 exports.sitemap = async (req, res) => {
-  const filePath = path.resolve(__dirname, '../../assets/sitemap.xml');
+  const filePath = path.resolve(__dirname, "../../assets/sitemap.xml");
   const content = await readFile(filePath);
   res.send(content, mime.lookup(filePath));
 };
@@ -35,7 +35,7 @@ exports.sitemap = async (req, res) => {
  * @param {*} res
  */
 exports.bingIndexNow = async (req, res) => {
-  const filePath = path.resolve(__dirname, '../../assets/bing.txt');
+  const filePath = path.resolve(__dirname, "../../assets/bing.txt");
   const content = await readFile(filePath);
   res.send(content, mime.lookup(filePath));
 };
@@ -46,7 +46,7 @@ exports.bingIndexNow = async (req, res) => {
  * @param {*} res
  */
 exports.manifestJson = async (req, res) => {
-  const filePath = path.resolve(__dirname, '../../assets/manifest.json');
+  const filePath = path.resolve(__dirname, "../../assets/manifest.json");
   const content = await readFile(filePath);
   res.send(content, mime.lookup(filePath));
 };
@@ -57,7 +57,7 @@ exports.manifestJson = async (req, res) => {
  * @param {*} res
  */
 exports.swJs = async (req, res) => {
-  const filePath = path.resolve(__dirname, '../../assets/sw.js');
+  const filePath = path.resolve(__dirname, "../../assets/sw.js");
   const content = await readFile(filePath);
   res.send(content, mime.lookup(filePath));
 };
@@ -68,7 +68,7 @@ exports.swJs = async (req, res) => {
  * @param {*} res
  */
 exports.icon192Png = async (req, res) => {
-  const filePath = path.resolve(__dirname, '../../assets/icon-192.png');
+  const filePath = path.resolve(__dirname, "../../assets/icon-192.png");
   const content = await readFile(filePath, { encoding: null });
   res.send(content, mime.lookup(filePath));
 };
@@ -79,7 +79,7 @@ exports.icon192Png = async (req, res) => {
  * @param {*} res
  */
 exports.icon512Png = async (req, res) => {
-  const filePath = path.resolve(__dirname, '../../assets/icon-512.png');
+  const filePath = path.resolve(__dirname, "../../assets/icon-512.png");
   const content = await readFile(filePath, { encoding: null });
   res.send(content, mime.lookup(filePath));
 };

package/server/util/check.js

@@ -1,5 +1,5 @@
 // qiao
-const { userCheck } = require('qiao-z-service');
+const { userCheck } = require("qiao-z-service");
 
 /**
  * checkUserAuth
@@ -16,7 +16,7 @@ exports.checkUserAuth = async function (req, res) {
   });
 
   // pass
-  if (userCheckRes && userCheckRes.type === 'success') return true;
+  if (userCheckRes && userCheckRes.type === "success") return true;
 
   // r
   res.json(userCheckRes);

package/server/util/feishu.js

@@ -1,12 +1,12 @@
 // services
-const { feishuBot } = require('@shun-js/shun-service');
+const { feishuBot } = require("@shun-js/shun-service");
 
 /**
  * feishuMsg
  * @param {*} msg
  */
 exports.feishuMsg = (msg) => {
-  if (global.QZ_CONFIG.env !== 'production') return;
+  if (global.QZ_CONFIG.env !== "production") return;
 
   feishuBot({
     url: global.QZ_CONFIG.feishu.url,
@@ -19,9 +19,14 @@ exports.feishuMsg = (msg) => {
 function isBot(req) {
   const ua = req.useragent;
   const hasOSName = ua && ua.os && ua.os.name;
-  const isGoogleBot = ua && ua.browser && ua.browser.name === 'Googlebot';
-  const isMetaBot = ua && ua.browser && ua.browser.name === 'meta-externalagent';
-  const isSafariBot = ua && ua.engine && ua.engine.name === 'WebKit' && ua.engine.version === '605.1.15';
+  const isGoogleBot = ua && ua.browser && ua.browser.name === "Googlebot";
+  const isMetaBot =
+    ua && ua.browser && ua.browser.name === "meta-externalagent";
+  const isSafariBot =
+    ua &&
+    ua.engine &&
+    ua.engine.name === "WebKit" &&
+    ua.engine.version === "605.1.15";
   return !hasOSName || isGoogleBot || isMetaBot || isSafariBot;
 }
 

package/server/util/prompt-agent.js

@@ -145,167 +145,211 @@ flowchart TD
    */
   CANVAS_TOOLS: [
     {
-      type: 'function',
+      type: "function",
       function: {
-        name: 'draw_shape',
-        description: '在白板上绘制基础形状（矩形、椭圆、菱形），可带文字标签。需要被箭头连接时必须指定 id。',
+        name: "draw_shape",
+        description:
+          "在白板上绘制基础形状（矩形、椭圆、菱形），可带文字标签。需要被箭头连接时必须指定 id。",
         parameters: {
-          type: 'object',
+          type: "object",
           properties: {
-            id: { type: 'string', description: '元素ID，箭头绑定时需要引用' },
-            shape: { type: 'string', enum: ['rectangle', 'ellipse', 'diamond'] },
-            x: { type: 'number', description: '左上角X坐标' },
-            y: { type: 'number', description: '左上角Y坐标' },
-            width: { type: 'number', description: '宽度，默认100' },
-            height: { type: 'number', description: '高度，默认100' },
-            label: { type: 'string', description: '形状内的文字标签' },
-            strokeColor: { type: 'string', description: '边框颜色，默认#1e1e1e' },
-            backgroundColor: { type: 'string', description: '填充颜色，默认transparent' },
-            fillStyle: { type: 'string', enum: ['hachure', 'cross-hatch', 'solid', 'zigzag'] },
-            strokeWidth: { type: 'number', enum: [1, 2, 4], description: '1=细,2=中,4=粗' },
-            strokeStyle: { type: 'string', enum: ['solid', 'dashed', 'dotted'] },
-            roughness: { type: 'number', enum: [0, 1, 2], description: '0=精确,1=手绘,2=夸张' },
-            roundness: { type: 'boolean', description: '是否圆角，默认true' },
-            opacity: { type: 'number', description: '不透明度0-100，默认100' },
+            id: { type: "string", description: "元素ID，箭头绑定时需要引用" },
+            shape: {
+              type: "string",
+              enum: ["rectangle", "ellipse", "diamond"],
+            },
+            x: { type: "number", description: "左上角X坐标" },
+            y: { type: "number", description: "左上角Y坐标" },
+            width: { type: "number", description: "宽度，默认100" },
+            height: { type: "number", description: "高度，默认100" },
+            label: { type: "string", description: "形状内的文字标签" },
+            strokeColor: {
+              type: "string",
+              description: "边框颜色，默认#1e1e1e",
+            },
+            backgroundColor: {
+              type: "string",
+              description: "填充颜色，默认transparent",
+            },
+            fillStyle: {
+              type: "string",
+              enum: ["hachure", "cross-hatch", "solid", "zigzag"],
+            },
+            strokeWidth: {
+              type: "number",
+              enum: [1, 2, 4],
+              description: "1=细,2=中,4=粗",
+            },
+            strokeStyle: {
+              type: "string",
+              enum: ["solid", "dashed", "dotted"],
+            },
+            roughness: {
+              type: "number",
+              enum: [0, 1, 2],
+              description: "0=精确,1=手绘,2=夸张",
+            },
+            roundness: { type: "boolean", description: "是否圆角，默认true" },
+            opacity: { type: "number", description: "不透明度0-100，默认100" },
           },
-          required: ['shape', 'x', 'y'],
+          required: ["shape", "x", "y"],
         },
       },
     },
     {
-      type: 'function',
+      type: "function",
       function: {
-        name: 'draw_arrow',
-        description: '绘制箭头。可通过 startId/endId 绑定到已有形状（箭头自动吸附），也可直接指定坐标点。',
+        name: "draw_arrow",
+        description:
+          "绘制箭头。可通过 startId/endId 绑定到已有形状（箭头自动吸附），也可直接指定坐标点。",
         parameters: {
-          type: 'object',
+          type: "object",
           properties: {
-            startX: { type: 'number', description: '起点X（无startId时必填）' },
-            startY: { type: 'number', description: '起点Y（无startId时必填）' },
-            endX: { type: 'number', description: '终点X（无endId时必填）' },
-            endY: { type: 'number', description: '终点Y（无endId时必填）' },
-            startId: { type: 'string', description: '起点绑定的形状ID' },
-            endId: { type: 'string', description: '终点绑定的形状ID' },
-            label: { type: 'string', description: '箭头上的文字标签' },
+            startX: { type: "number", description: "起点X（无startId时必填）" },
+            startY: { type: "number", description: "起点Y（无startId时必填）" },
+            endX: { type: "number", description: "终点X（无endId时必填）" },
+            endY: { type: "number", description: "终点Y（无endId时必填）" },
+            startId: { type: "string", description: "起点绑定的形状ID" },
+            endId: { type: "string", description: "终点绑定的形状ID" },
+            label: { type: "string", description: "箭头上的文字标签" },
             startArrowhead: {
-              type: 'string',
-              enum: ['arrow', 'bar', 'dot', 'triangle', 'diamond'],
-              description: '起点箭头样式，默认无',
+              type: "string",
+              enum: ["arrow", "bar", "dot", "triangle", "diamond"],
+              description: "起点箭头样式，默认无",
             },
             endArrowhead: {
-              type: 'string',
-              enum: ['arrow', 'bar', 'dot', 'triangle', 'diamond'],
-              description: '终点箭头样式，默认arrow',
+              type: "string",
+              enum: ["arrow", "bar", "dot", "triangle", "diamond"],
+              description: "终点箭头样式，默认arrow",
+            },
+            strokeColor: { type: "string" },
+            strokeWidth: { type: "number", enum: [1, 2, 4] },
+            strokeStyle: {
+              type: "string",
+              enum: ["solid", "dashed", "dotted"],
+            },
+            elbowed: {
+              type: "boolean",
+              description: "是否使用直角折线，默认false",
             },
-            strokeColor: { type: 'string' },
-            strokeWidth: { type: 'number', enum: [1, 2, 4] },
-            strokeStyle: { type: 'string', enum: ['solid', 'dashed', 'dotted'] },
-            elbowed: { type: 'boolean', description: '是否使用直角折线，默认false' },
           },
           required: [],
         },
       },
     },
     {
-      type: 'function',
+      type: "function",
       function: {
-        name: 'draw_line',
-        description: '绘制线条（无箭头），支持多个折点',
+        name: "draw_line",
+        description: "绘制线条（无箭头），支持多个折点",
         parameters: {
-          type: 'object',
+          type: "object",
           properties: {
             points: {
-              type: 'array',
-              items: { type: 'array', items: { type: 'number' } },
-              description: '折点坐标数组，如 [[0,0],[100,50],[200,0]]',
+              type: "array",
+              items: { type: "array", items: { type: "number" } },
+              description: "折点坐标数组，如 [[0,0],[100,50],[200,0]]",
+            },
+            strokeColor: { type: "string" },
+            strokeWidth: { type: "number", enum: [1, 2, 4] },
+            strokeStyle: {
+              type: "string",
+              enum: ["solid", "dashed", "dotted"],
             },
-            strokeColor: { type: 'string' },
-            strokeWidth: { type: 'number', enum: [1, 2, 4] },
-            strokeStyle: { type: 'string', enum: ['solid', 'dashed', 'dotted'] },
           },
-          required: ['points'],
+          required: ["points"],
         },
       },
     },
     {
-      type: 'function',
+      type: "function",
       function: {
-        name: 'draw_text',
-        description: '在白板上添加独立文本',
+        name: "draw_text",
+        description: "在白板上添加独立文本",
         parameters: {
-          type: 'object',
+          type: "object",
           properties: {
-            id: { type: 'string' },
-            x: { type: 'number' },
-            y: { type: 'number' },
-            text: { type: 'string' },
-            fontSize: { type: 'number', description: '字号，默认20' },
+            id: { type: "string" },
+            x: { type: "number" },
+            y: { type: "number" },
+            text: { type: "string" },
+            fontSize: { type: "number", description: "字号，默认20" },
             fontFamily: {
-              type: 'number',
+              type: "number",
               enum: [1, 2, 3, 5],
-              description: '1=手写,2=常规,3=等宽,5=圆体',
+              description: "1=手写,2=常规,3=等宽,5=圆体",
             },
-            textAlign: { type: 'string', enum: ['left', 'center', 'right'] },
-            strokeColor: { type: 'string' },
+            textAlign: { type: "string", enum: ["left", "center", "right"] },
+            strokeColor: { type: "string" },
           },
-          required: ['x', 'y', 'text'],
+          required: ["x", "y", "text"],
         },
       },
     },
     {
-      type: 'function',
+      type: "function",
       function: {
-        name: 'draw_group',
-        description: '批量创建多个元素并自动编组（移动时一起移动）。适合画一组相关图形。',
+        name: "draw_group",
+        description:
+          "批量创建多个元素并自动编组（移动时一起移动）。适合画一组相关图形。",
         parameters: {
-          type: 'object',
+          type: "object",
           properties: {
             elements: {
-              type: 'array',
-              description: '元素数组，每个元素格式同其他 draw 工具的参数',
+              type: "array",
+              description: "元素数组，每个元素格式同其他 draw 工具的参数",
               items: {
-                type: 'object',
+                type: "object",
                 properties: {
-                  tool: { type: 'string', enum: ['shape', 'arrow', 'line', 'text'] },
+                  tool: {
+                    type: "string",
+                    enum: ["shape", "arrow", "line", "text"],
+                  },
                   args: {
-                    type: 'object',
-                    description: '对应 draw_shape/draw_arrow/draw_line/draw_text 的参数',
+                    type: "object",
+                    description:
+                      "对应 draw_shape/draw_arrow/draw_line/draw_text 的参数",
                   },
                 },
-                required: ['tool', 'args'],
+                required: ["tool", "args"],
               },
             },
           },
-          required: ['elements'],
+          required: ["elements"],
         },
       },
     },
     {
-      type: 'function',
+      type: "function",
       function: {
-        name: 'draw_frame',
-        description: '创建一个 Frame 容器，包含指定的子元素（子元素会被框在一起）',
+        name: "draw_frame",
+        description:
+          "创建一个 Frame 容器，包含指定的子元素（子元素会被框在一起）",
         parameters: {
-          type: 'object',
+          type: "object",
           properties: {
-            name: { type: 'string', description: 'Frame 名称' },
-            childIds: { type: 'array', items: { type: 'string' }, description: '子元素ID数组' },
-            x: { type: 'number' },
-            y: { type: 'number' },
-            width: { type: 'number' },
-            height: { type: 'number' },
+            name: { type: "string", description: "Frame 名称" },
+            childIds: {
+              type: "array",
+              items: { type: "string" },
+              description: "子元素ID数组",
+            },
+            x: { type: "number" },
+            y: { type: "number" },
+            width: { type: "number" },
+            height: { type: "number" },
           },
-          required: ['childIds'],
+          required: ["childIds"],
         },
       },
     },
     {
-      type: 'function',
+      type: "function",
       function: {
-        name: 'clear_canvas',
-        description: '清空白板上的所有内容',
+        name: "clear_canvas",
+        description: "清空白板上的所有内容",
         parameters: {
-          type: 'object',
+          type: "object",
           properties: {},
           required: [],
         },

package/server/util/relay-client.js

@@ -0,0 +1,70 @@
+const WebSocket = require("ws");
+
+/**
+ * Send a prompt through the WS relay and return the result
+ * @param {object} options
+ * @param {string} options.wsUrl - relay client URL (wss://ws.aibaiban.com/ws/client)
+ * @param {string} options.authSecret - RELAY_AUTH_SECRET
+ * @param {string} options.userId - user ID for relay auth
+ * @param {string} options.prompt - prompt text
+ * @param {number} [options.timeout=300000] - timeout in ms
+ * @returns {Promise<string>} result text
+ */
+exports.sendPromptViaRelay = function (options) {
+  const { wsUrl, authSecret, userId, prompt, timeout = 300000 } = options;
+
+  return new Promise((resolve, reject) => {
+    const ws = new WebSocket(wsUrl);
+    let settled = false;
+
+    const timer = setTimeout(() => {
+      if (!settled) {
+        settled = true;
+        ws.close();
+        reject(new Error("Relay timeout"));
+      }
+    }, timeout);
+
+    function done(err, result) {
+      if (settled) return;
+      settled = true;
+      clearTimeout(timer);
+      ws.close();
+      if (err) reject(err);
+      else resolve(result);
+    }
+
+    ws.on("open", () => {
+      ws.send(JSON.stringify({ type: "auth", userId, token: authSecret }));
+    });
+
+    ws.on("message", (data) => {
+      let msg;
+      try {
+        msg = JSON.parse(data.toString());
+      } catch {
+        return;
+      }
+
+      if (msg.type === "auth-ok") {
+        ws.send(JSON.stringify({ type: "prompt", prompt }));
+      } else if (msg.type === "auth-fail") {
+        done(new Error(`Auth failed: ${msg.reason}`));
+      } else if (msg.type === "response") {
+        done(null, msg.result);
+      } else if (msg.type === "error") {
+        done(new Error(msg.message || "Relay error"));
+      } else if (msg.type === "agent-offline") {
+        done(new Error("AI agent is offline"));
+      }
+    });
+
+    ws.on("error", (err) => {
+      done(new Error(`WebSocket error: ${err.message}`));
+    });
+
+    ws.on("close", () => {
+      done(new Error("WebSocket closed unexpectedly"));
+    });
+  });
+};

package/views/index.html

@@ -6,7 +6,10 @@
 
     <!-- Primary Meta Tags -->
     <title>AI白板 - 在线协作白板工具 | AI生成流程图时序图类图ER图</title>
-    <meta name="title" content="AI白板 - 在线协作白板工具 | AI生成流程图时序图类图ER图" />
+    <meta
+      name="title"
+      content="AI白板 - 在线协作白板工具 | AI生成流程图时序图类图ER图"
+    />
     <meta
       name="description"
       content="AI白板是专业的在线协作白板软件，AI驱动自动生成流程图、时序图、类图、ER图。支持在线协作、远程办公、团队协同。Miro、FigJam、boardmix优质替代方案，一句话完成专业图表绘制。"
@@ -26,34 +29,60 @@
     <!-- Open Graph / Facebook -->
     <meta property="og:type" content="website" />
     <meta property="og:url" content="https://aibaiban.com/" />
-    <meta property="og:title" content="AI白板 - 在线协作白板工具 | AI生成流程图时序图类图ER图" />
+    <meta
+      property="og:title"
+      content="AI白板 - 在线协作白板工具 | AI生成流程图时序图类图ER图"
+    />
     <meta
       property="og:description"
       content="专业的在线协作白板软件，AI自动生成流程图、时序图、类图、ER图。支持在线协作、远程办公。Miro/FigJam优质替代方案。"
     />
-    <meta property="og:image" content="https://static-small.vincentqiao.com/aibaiban/static/og-image.png" />
+    <meta
+      property="og:image"
+      content="https://static-small.vincentqiao.com/aibaiban/static/og-image.png"
+    />
     <meta property="og:site_name" content="AI白板" />
     <meta property="og:locale" content="zh_CN" />
 
     <!-- Twitter Card -->
     <meta name="twitter:card" content="summary_large_image" />
     <meta name="twitter:url" content="https://aibaiban.com/" />
-    <meta name="twitter:title" content="AI白板 - 在线协作白板工具 | AI生成流程图时序图类图ER图" />
+    <meta
+      name="twitter:title"
+      content="AI白板 - 在线协作白板工具 | AI生成流程图时序图类图ER图"
+    />
     <meta
       name="twitter:description"
       content="专业的在线协作白板软件，AI自动生成流程图、时序图、类图、ER图。支持在线协作、远程办公。"
     />
-    <meta name="twitter:image" content="https://static-small.vincentqiao.com/aibaiban/static/og-image.png" />
+    <meta
+      name="twitter:image"
+      content="https://static-small.vincentqiao.com/aibaiban/static/og-image.png"
+    />
 
     <!-- Apple Mobile Web App -->
     <meta name="mobile-web-app-capable" content="yes" />
-    <meta name="apple-mobile-web-app-status-bar-style" content="black-translucent" />
+    <meta
+      name="apple-mobile-web-app-status-bar-style"
+      content="black-translucent"
+    />
     <meta name="apple-mobile-web-app-title" content="AI白板" />
 
     <!-- Favicon -->
-    <link rel="icon" type="image/x-icon" href="https://static-small.vincentqiao.com/aibaiban/static/favicon.ico" />
-    <link rel="icon" type="image/svg+xml" href="https://static-small.vincentqiao.com/aibaiban/static/logo.svg" />
-    <link rel="apple-touch-icon" href="https://static-small.vincentqiao.com/aibaiban/static/apple-touch-icon.png" />
+    <link
+      rel="icon"
+      type="image/x-icon"
+      href="https://static-small.vincentqiao.com/aibaiban/static/favicon.ico"
+    />
+    <link
+      rel="icon"
+      type="image/svg+xml"
+      href="https://static-small.vincentqiao.com/aibaiban/static/logo.svg"
+    />
+    <link
+      rel="apple-touch-icon"
+      href="https://static-small.vincentqiao.com/aibaiban/static/apple-touch-icon.png"
+    />
 
     <!-- Manifest -->
     <link rel="manifest" href="https://aibaiban.com/manifest.json" />
@@ -81,13 +110,20 @@
           "description": "专业的在线协作白板软件，AI驱动自动生成流程图、时序图、类图、ER图。支持在线协作、远程办公、团队协同。",
           "url": "https://aibaiban.com",
           "image": "https://static-small.vincentqiao.com/aibaiban/static/og-image.png",
-          "featureList": ["AI生成流程图", "AI生成时序图", "AI生成类图", "AI生成ER图", "在线协作白板", "远程办公支持"]
+          "featureList": [
+            "AI生成流程图",
+            "AI生成时序图",
+            "AI生成类图",
+            "AI生成ER图",
+            "在线协作白板",
+            "远程办公支持"
+          ]
         }
       ]
     </script>
     <!-- Microsoft Clarity (production only) -->
     <script type="text/javascript">
-      if (location.hostname.includes('aibaiban.com')) {
+      if (location.hostname.includes("aibaiban.com")) {
         (function (c, l, a, r, i, t, y) {
           c[a] =
             c[a] ||
@@ -96,10 +132,10 @@
             };
           t = l.createElement(r);
           t.async = 1;
-          t.src = 'https://www.clarity.ms/tag/' + i;
+          t.src = "https://www.clarity.ms/tag/" + i;
           y = l.getElementsByTagName(r)[0];
           y.parentNode.insertBefore(t, y);
-        })(window, document, 'clarity', 'script', 't5b230u2zp');
+        })(window, document, "clarity", "script", "t5b230u2zp");
       }
     </script>
     <script
@@ -141,8 +177,8 @@
   <body>
     <div id="root"></div>
     <script>
-      if ('serviceWorker' in navigator) {
-        navigator.serviceWorker.register('/sw.js');
+      if ("serviceWorker" in navigator) {
+        navigator.serviceWorker.register("/sw.js");
       }
     </script>
   </body>