gemini-proxy-client 1.0.21 → 1.0.22

package/dist/browser/manager.js

@@ -191,36 +191,80 @@ export class BrowserManager {
     }
     /**
      * 检查是否已登录 Google
+     * 使用多重检测策略，避免误判
      */
     async checkGoogleLogin() {
         if (!this.page)
             return false;
         try {
-            // 检查页面 URL 是否在 AI Studio 域名下且没有登录重定向
             const url = this.page.url();
-            // 如果在 accounts.google.com，说明需要登录
-            if (url.includes('accounts.google.com')) {
+            // 明确的未登录状态：在 Google 登录页面
+            if (url.includes('accounts.google.com/v3/signin') ||
+                url.includes('accounts.google.com/ServiceLogin')) {
+                console.log(chalk.gray(`[${new Date().toLocaleTimeString()}] 检测到登录页面: ${url.substring(0, 60)}...`));
                 return false;
             }
-            // 如果在 AI Studio，检查是否有登录相关的元素
+            // 如果在 AI Studio 域名下，进一步检查
             if (url.includes(GOOGLE_LOGGED_IN_INDICATOR)) {
-                // 检查是否有用户头像或登录状态指示器
-                const isLoggedIn = await this.page.evaluate(() => {
-                    // 检查是否有登录提示或错误
-                    const loginPrompt = document.querySelector('[data-login-required]');
-                    if (loginPrompt)
-                        return false;
-                    // 检查是否有用户相关元素
-                    const userAvatar = document.querySelector('img[alt*="Account"], img[alt*="Profile"]');
-                    return !!userAvatar || !document.body.textContent?.includes('Sign in');
+                const loginStatus = await this.page.evaluate(() => {
+                    // 策略1: 检查是否有明确的登录按钮/链接
+                    const signInButtons = document.querySelectorAll('a[href*="accounts.google.com"], button');
+                    for (const el of signInButtons) {
+                        const text = el.textContent?.toLowerCase() || '';
+                        // 只有当按钮文字明确是 "sign in" 或 "登录" 时才认为未登录
+                        if (text === 'sign in' || text === '登录' || text === 'log in') {
+                            return { loggedIn: false, reason: 'found sign-in button' };
+                        }
+                    }
+                    // 策略2: 检查页面是否有重定向到登录的 meta 标签
+                    const metaRefresh = document.querySelector('meta[http-equiv="refresh"]');
+                    if (metaRefresh) {
+                        const content = metaRefresh.getAttribute('content') || '';
+                        if (content.includes('accounts.google.com')) {
+                            return { loggedIn: false, reason: 'meta refresh to login' };
+                        }
+                    }
+                    // 策略3: 检查是否有用户菜单/头像（正向确认已登录）
+                    const userIndicators = [
+                        'img[alt*="Account"]',
+                        'img[alt*="Profile"]',
+                        'img[alt*="Google Account"]',
+                        '[aria-label*="Account"]',
+                        '[aria-label*="Google Account"]',
+                        '[data-ogsr-up]', // Google 用户面板
+                    ];
+                    for (const selector of userIndicators) {
+                        if (document.querySelector(selector)) {
+                            return { loggedIn: true, reason: `found: ${selector}` };
+                        }
+                    }
+                    // 策略4: 如果页面正常加载了 AI Studio 内容，默认认为已登录
+                    // 检查是否有 AI Studio 特有的元素
+                    const studioIndicators = [
+                        'iframe[src*="scf.usercontent.goog"]',
+                        'iframe[src*="blob:"]',
+                        '[class*="studio"]',
+                        '[class*="gemini"]',
+                    ];
+                    for (const selector of studioIndicators) {
+                        if (document.querySelector(selector)) {
+                            return { loggedIn: true, reason: `studio content: ${selector}` };
+                        }
+                    }
+                    // 默认：如果在 AI Studio 域名下且没有明确的登录提示，认为已登录
+                    return { loggedIn: true, reason: 'default (no sign-in prompt)' };
                 });
-                return isLoggedIn;
+                console.log(chalk.gray(`[${new Date().toLocaleTimeString()}] 登录检测: ${loginStatus.loggedIn ? '已登录' : '未登录'} (${loginStatus.reason})`));
+                return loginStatus.loggedIn;
             }
+            // 不在 AI Studio 域名下，可能是其他页面
+            console.log(chalk.gray(`[${new Date().toLocaleTimeString()}] 当前页面不在 AI Studio: ${url.substring(0, 60)}...`));
             return false;
         }
         catch (error) {
             console.error('检查登录状态失败:', error);
-            return false;
+            // 出错时默认认为已登录，避免误触发重新登录
+            return true;
         }
     }
     /**
@@ -601,11 +645,32 @@ export class BrowserManager {
     }
     /**
      * 处理登录过期
+     * 先以无头模式重启检查，如果确实需要登录再切换到有界面模式
      */
     async handleLoginExpired() {
+        const originalHeadless = this.options.headless;
         // 关闭当前浏览器
         await this.close();
-        // 以有界面模式重新启动
+        // 先以无头模式重新启动，检查是否真的需要登录
+        console.log(chalk.yellow(`[${new Date().toLocaleTimeString()}] 检测到可能的登录问题，正在验证...`));
+        this.options.headless = true;
+        await this.launch();
+        await this.openBuildApp();
+        // 等待页面加载
+        await sleep(3000);
+        // 再次检查登录状态
+        const isLoggedIn = await this.checkGoogleLogin();
+        if (isLoggedIn) {
+            // 实际上已登录，是误判，继续无头模式运行
+            console.log(chalk.green(`[${new Date().toLocaleTimeString()}] ✅ 验证通过，登录状态正常，继续无头模式运行`));
+            await this.saveCookies();
+            await this.clickConnectButton();
+            await this.startKeepAlive();
+            return;
+        }
+        // 确实需要登录，切换到有界面模式
+        console.log(chalk.yellow(`[${new Date().toLocaleTimeString()}] 确认需要重新登录，切换到有界面模式...`));
+        await this.close();
         this.options.headless = false;
         await this.launch();
         await this.openBuildApp();
@@ -615,8 +680,14 @@ export class BrowserManager {
         console.log(chalk.green('✅ 重新登录成功！'));
         // 重新连接
         await this.clickConnectButton();
-        // 切回无头模式 (下次启动时)
+        // 登录成功后，重新以无头模式启动
+        console.log(chalk.gray(`[${new Date().toLocaleTimeString()}] 登录完成，切换回无头模式...`));
+        await this.close();
         this.options.headless = true;
+        await this.launch();
+        await this.openBuildApp();
+        await this.clickConnectButton();
+        await this.startKeepAlive();
     }
     /**
      * 检查 WebSocket 连接状态

package/package.json

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