部署 zai2api 到 HF Space

captcha-provider/server.js

@@ -150,8 +150,16 @@ async function acquireToken() {
       Object.defineProperty(navigator, 'languages', { get: () => ['zh-CN', 'zh', 'en-US', 'en'] });
       window.chrome = { runtime: {}, loadTimes: () => ({}) };
     });
-    const freshPage = await localCtx.newPage();
-    await freshPage.goto(ZAI_URL, { waitUntil: 'domcontentloaded', timeout: 30000 });
+    const freshPage = await localCtx.newPage();
+    freshPage.on('console', (msg) => {
+      if (msg.type() === 'warning' || msg.type() === 'error') {
+        console.log(`[provider:page:${msg.type()}] ${msg.text()}`);
+      }
+    });
+    freshPage.on('pageerror', (err) => {
+      console.error(`[provider:page:error] ${err.message}`);
+    });
+    await freshPage.goto(ZAI_URL, { waitUntil: 'domcontentloaded', timeout: 30000 });
     await freshPage.waitForTimeout(3000);
 
     // 注入阿里云 captcha SDK
@@ -169,42 +177,64 @@ async function acquireToken() {
     await freshPage.waitForTimeout(1000);
 
     const token = await freshPage.evaluate(async (sceneId) => {
-      if (typeof window.initAliyunCaptcha !== 'function') {
-        throw new Error('initAliyunCaptcha not available after injection');
-      }
-      const id = 'c-' + Date.now() + '-' + Math.random().toString(36).slice(2, 6);
-      const triggerId = 't-' + id;
-      const container = document.createElement('div');
-      container.id = id;
-      container.style.cssText = 'position:fixed;top:-9999px;left:-9999px;width:1px;height:1px;overflow:hidden;';
-      document.body.appendChild(container);
-      const trigger = document.createElement('button');
-      trigger.id = triggerId;
-      trigger.style.cssText = 'display:none;';
-      container.appendChild(trigger);
-
-      try {
-        return await new Promise((resolve, reject) => {
+      if (typeof window.initAliyunCaptcha !== 'function') {
+        throw new Error('initAliyunCaptcha not available after injection');
+      }
+      const id = 'chat-captcha-element';
+      const triggerId = 'chat-captcha-trigger';
+      document.getElementById(id)?.remove();
+      document.getElementById(triggerId)?.remove();
+      const container = document.createElement('div');
+      container.id = id;
+      container.style.cssText = 'position:absolute;left:-99999px;top:-99999px;width:0;height:0;overflow:hidden;pointer-events:none;';
+      document.body.appendChild(container);
+      const trigger = document.createElement('button');
+      trigger.id = triggerId;
+      trigger.type = 'button';
+      trigger.setAttribute('aria-hidden', 'true');
+      trigger.tabIndex = -1;
+      trigger.style.cssText = 'position:absolute;left:-99999px;top:-99999px;width:1px;height:1px;opacity:0;';
+      document.body.appendChild(trigger);
+
+      try {
+        return await new Promise((resolve, reject) => {
           const timeout = setTimeout(() => reject(new Error('captcha timeout')), 60000);
-          window.initAliyunCaptcha({
-            SceneId: sceneId,
-            mode: 'popup',
-            element: `#${id}`,
-            button: `#${triggerId}`,
-            language: 'cn',
+          let clickCount = 0;
+          const clickTrigger = () => {
+            if (clickCount >= 3) return;
+            clickCount++;
+            setTimeout(() => trigger.click(), 500);
+          };
+          window.initAliyunCaptcha({
+            SceneId: sceneId,
+            mode: 'popup',
+            element: `#${id}`,
+            button: `#${triggerId}`,
+            captchaLogoImg: 'https://z-cdn.chatglm.cn/z-ai/static/logo.svg',
+            language: 'cn',
             timeout: 30000,
-            delayBeforeSuccess: false,
-            success: (token) => { clearTimeout(timeout); resolve(token); },
-            fail: () => {},
-            onError: (err) => { clearTimeout(timeout); reject(new Error('captcha error: ' + JSON.stringify(err))); },
-            onClose: () => {},
-          });
-          setTimeout(() => trigger.click(), 500);
-        });
-      } finally {
-        try { container.remove(); } catch {}
-      }
-    }, SCENE_ID);
+            delayBeforeSuccess: false,
+            success: (token) => { clearTimeout(timeout); resolve(token); },
+            fail: (err) => {
+              console.warn('chat captcha fail', err);
+              if (clickCount < 3) {
+                clickTrigger();
+              } else {
+                clearTimeout(timeout);
+                reject(new Error('captcha fail: ' + JSON.stringify(err)));
+              }
+            },
+            onError: (err) => { clearTimeout(timeout); reject(new Error('captcha error: ' + JSON.stringify(err))); },
+            onClose: () => { clearTimeout(timeout); reject(new Error('captcha closed')); },
+            getInstance: () => { clickTrigger(); },
+          });
+          setTimeout(clickTrigger, 1500);
+        });
+      } finally {
+        try { container.remove(); } catch {}
+        try { trigger.remove(); } catch {}
+      }
+    }, SCENE_ID);
 
     return token;
   } finally {