@cniot/mdd-editor 0.3.1 → 0.3.4

package/src/ai/bridgeClient.js

@@ -2,9 +2,19 @@ export const DEFAULT_BRIDGE_URL = 'http://127.0.0.1:17678';
 
 const trimEndSlash = (value = '') => value.replace(/\/+$/, '');
 const RELAY_CHANNEL = 'mdd-ai-bridge';
+const relayCacheMap = new Map();
 
 const isLoopbackURL = (value = '') => /^http:\/\/(127(?:\.\d{1,3}){3}|localhost)(?::\d+)?/i.test(value);
 
+export class BridgeRequestError extends Error {
+  constructor(message, options = {}) {
+    super(message);
+    this.name = 'BridgeRequestError';
+    this.status = options.status || 0;
+    this.data = options.data || null;
+  }
+}
+
 export function shouldUseBridgeRelay(baseURL) {
   return typeof window !== 'undefined' && !window.isSecureContext && isLoopbackURL(baseURL);
 }
@@ -35,14 +45,22 @@ async function request(baseURL, path, options = {}) {
   }
 
   if (!res.ok) {
-    throw new Error(data?.message || `请求本地 AI Bridge 失败: ${res.status}`);
+    throw new BridgeRequestError(data?.message || `请求本地 AI Bridge 失败: ${res.status}`, {
+      status: res.status,
+      data,
+    });
   }
   return data;
 }
 
-function requestViaRelay(baseURL, path, options = {}) {
-  const relayURL = `${trimEndSlash(baseURL)}/relay`;
-  const requestId = createRequestId();
+function getRelayCache(baseURL) {
+  const relayOrigin = trimEndSlash(baseURL);
+  const relayURL = `${relayOrigin}/relay`;
+  const cached = relayCacheMap.get(relayOrigin);
+  if (cached?.window && !cached.window.closed) {
+    return cached;
+  }
+
   const relayWindow = window.open(
     relayURL,
     'mdd-ai-bridge-relay',
@@ -53,9 +71,23 @@ function requestViaRelay(baseURL, path, options = {}) {
     throw new Error('浏览器拦截了本地 Bridge 窗口，请允许弹窗后重试');
   }
 
+  const next = {
+    origin: relayOrigin,
+    window: relayWindow,
+    ready: false,
+  };
+  relayCacheMap.set(relayOrigin, next);
+  return next;
+}
+
+function requestViaRelay(baseURL, path, options = {}) {
+  const relayCache = getRelayCache(baseURL);
+  const requestId = createRequestId();
+
   return new Promise((resolve, reject) => {
     let settled = false;
-    let ready = false;
+    let fallbackPosted = false;
+    let readyPosted = false;
 
     const cleanup = () => {
       window.removeEventListener('message', handleMessage);
@@ -63,8 +95,19 @@ function requestViaRelay(baseURL, path, options = {}) {
       clearTimeout(fallbackTimer);
     };
 
-    const postRequest = () => {
-      relayWindow.postMessage(
+    const postRequest = (source = 'ready') => {
+      if (source === 'ready') {
+        if (readyPosted) return;
+        readyPosted = true;
+      } else {
+        if (fallbackPosted) return;
+        fallbackPosted = true;
+      }
+      if (!relayCache.window || relayCache.window.closed) {
+        finish(reject, new Error('本地 Bridge relay 窗口已关闭，请重试'));
+        return;
+      }
+      relayCache.window.postMessage(
         {
           channel: RELAY_CHANNEL,
           type: 'request',
@@ -72,7 +115,7 @@ function requestViaRelay(baseURL, path, options = {}) {
           path,
           options,
         },
-        trimEndSlash(baseURL),
+        relayCache.origin,
       );
     };
 
@@ -84,14 +127,14 @@ function requestViaRelay(baseURL, path, options = {}) {
     };
 
     const handleMessage = (event) => {
-      if (event.origin !== trimEndSlash(baseURL)) return;
-      if (event.source !== relayWindow) return;
+      if (event.origin !== relayCache.origin) return;
+      if (event.source !== relayCache.window) return;
       const message = event.data || {};
       if (message.channel !== RELAY_CHANNEL) return;
 
-      if (message.type === 'ready' && !ready) {
-        ready = true;
-        postRequest();
+      if (message.type === 'ready') {
+        relayCache.ready = true;
+        postRequest('ready');
         return;
       }
 
@@ -99,7 +142,13 @@ function requestViaRelay(baseURL, path, options = {}) {
       if (!message.ok) {
         finish(
           reject,
-          new Error(message.data?.message || `请求本地 AI Bridge 失败: ${message.status || 0}`),
+          new BridgeRequestError(
+            message.data?.message || `请求本地 AI Bridge 失败: ${message.status || 0}`,
+            {
+              status: message.status || 0,
+              data: message.data,
+            },
+          ),
         );
         return;
       }
@@ -110,10 +159,13 @@ function requestViaRelay(baseURL, path, options = {}) {
       finish(reject, new Error('本地 AI Bridge relay 响应超时，请确认服务已启动'));
     }, 30000);
     const fallbackTimer = setTimeout(() => {
-      if (!ready) postRequest();
+      if (!relayCache.ready) postRequest('fallback');
     }, 500);
 
     window.addEventListener('message', handleMessage);
+    if (relayCache.ready) {
+      setTimeout(() => postRequest('ready'), 0);
+    }
   });
 }