gemini-proxy-client 1.0.2 → 1.0.4

package/dist/browser/manager.js

@@ -129,27 +129,36 @@ export class BrowserManager {
         if (!this.page)
             throw new Error('Browser not launched');
         // 等待页面完全加载
-        await sleep(2000);
-        // 查找并修改 WebSocket 连接地址 (如果需要)
-        // 这取决于 Build App 的实现方式
-        // 尝试多种方式找到连接按钮
+        await sleep(3000);
+        // 第一步：点击 "Continue to the app" 按钮（如果存在）
+        try {
+            const continueButton = await this.page.waitForSelector('button:has-text("Continue to the app")', { timeout: 5000 });
+            if (continueButton) {
+                await continueButton.click();
+                console.log(chalk.green('✅ 点击了 "Continue to the app" 按钮'));
+                await sleep(3000); // 等待页面切换
+            }
+        }
+        catch (e) {
+            // 按钮不存在，可能已经在 app 页面了
+        }
+        // 第二步：设置服务器地址（如果有输入框）
+        await this.setServerAddress();
+        // 第三步：点击 Connect WS 按钮
         const buttonSelectors = [
-            'button:has-text("Connect WS")', // 实际按钮文字
-            'button[aria-label="Connect WebSocket Proxy"]', // aria-label
+            'button[aria-label="Connect WebSocket Proxy"]', // aria-label 最准确
             'button[title="Connect WebSocket Proxy"]', // title 属性
-            'button:has-text("ws connect")',
-            'button:has-text("connect")',
+            'button:has-text("Connect WS")', // 按钮文字
             'button.bg-green-500', // 绿色按钮
         ];
         let clicked = false;
         for (const selector of buttonSelectors) {
             try {
-                const button = await this.page.$(selector);
+                const button = await this.page.waitForSelector(selector, { timeout: 5000 });
                 if (button) {
-                    // 先尝试设置服务器地址 (如果有输入框)
-                    await this.setServerAddress();
                     await button.click();
                     clicked = true;
+                    console.log(chalk.green(`✅ 点击了连接按钮`));
                     break;
                 }
             }
@@ -158,15 +167,6 @@ export class BrowserManager {
             }
         }
         if (!clicked) {
-            // 如果找不到按钮，可能需要通过其他方式触发连接
-            // 尝试在页面中执行 JavaScript
-            await this.page.evaluate(({ serverUrl, token }) => {
-                // 尝试找到全局的连接函数或 WebSocket 实例
-                const wsUrl = `${serverUrl}?auth_token=${token}`;
-                console.log('Attempting to connect to:', wsUrl);
-                // 这里的实现取决于 Build App 的具体代码结构
-                // 可能需要根据实际情况调整
-            }, { serverUrl: this.options.serverUrl, token: this.options.token });
             console.log(chalk.yellow('⚠️ 未找到连接按钮，请手动点击连接'));
         }
         // 等待连接建立

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "gemini-proxy-client",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "Gemini Proxy Build App 客户端 - 使用 Camoufox 自动保持连接",
   "main": "dist/index.js",
   "bin": {