add manual login

provider.mjs

@@ -4,7 +4,7 @@ import { v4 as uuidv4 } from "uuid";
 import path from "path";
 import fs from "fs";
 import {fileURLToPath} from "url";
-import { createDirectoryIfNotExists, sleep, extractCookie, getSessionCookie, createDocx } from "./utils.mjs";
+import {createDirectoryIfNotExists, createDocx, extractCookie, getSessionCookie, sleep} from "./utils.mjs";
 import {execSync} from 'child_process';
 import os from 'os';
 import './proxyAgent.mjs';
@@ -54,9 +54,20 @@ class YouProvider {
         // 检测Chrome和Edge浏览器
         const browserPath = this.detectBrowser();
 
-        // extract essential jwt session and token from cookie
+        this.sessions = {};
+        const timeout = 120000; // 120 秒超时
+
+        if (process.env.USE_MANUAL_LOGIN === "true") {
+            this.sessions['manual_login'] = {
+                configIndex: 0,
+                valid: false,
+            };
+            console.log("当前使用手动登录模式，跳过config.mjs文件中的 cookie 验证");
+        } else {
+            // 使用配置文件中的 cookie
             for (let index = 0; index < config.sessions.length; index++) {
-			let session = config.sessions[index], {jwtSession, jwtToken, ds, dsr} = extractCookie(session.cookie);
+                let session = config.sessions[index];
+                let {jwtSession, jwtToken, ds, dsr} = extractCookie(session.cookie);
                 if (jwtSession && jwtToken) {
                     // 旧版cookie处理
                     try {
@@ -94,12 +105,15 @@ class YouProvider {
                 }
             }
             console.log(`已添加 ${Object.keys(this.sessions).length} 个 cookie，开始验证有效性`);
+        }
 
         for (let username of Object.keys(this.sessions)) {
             let session = this.sessions[username];
             createDirectoryIfNotExists(path.join(__dirname, "browser_profiles", username));
+            const isWindows = os.platform() === 'win32';
+
             await connect({
-                headless: "auto",
+                headless: isWindows ? false : 'auto',
                 turnstile: true,
                 customConfig: {
                     userDataDir: path.join(__dirname, "browser_profiles", username),
@@ -107,27 +121,53 @@ class YouProvider {
                 },
             })
                 .then(async (response) => {
-                    const { page, browser, setTarget } = response;
+                    const { page, browser } = response;
+                    if (process.env.USE_MANUAL_LOGIN === "true") {
+                        console.log(`正在为 session #${session.configIndex} 进行手动登录...`);
+                        await page.goto("https://you.com", { timeout: timeout });
+                        // 等待页面加载完毕
+                        await sleep(5000);
+                        console.log(`请在打开的浏览器窗口中手动登录 You.com (session #${session.configIndex})`);
+                        await this.waitForManualLogin(page);
+                        const cookies = await page.cookies();
+                        const sessionCookie = this.extractSessionCookie(cookies);
+                        if (sessionCookie) {
+                            this.sessions[sessionCookie.email] = {
+                                ...session,
+                                ...sessionCookie,
+                            };
+                            delete this.sessions[username];
+                            username = sessionCookie.email;
+                            session = this.sessions[username];
+                            console.log(`成功获取 ${sessionCookie.email} 登录的 cookie`);
+                        } else {
+                            console.error(`未能获取到 session #${session.configIndex} 有效登录的 cookie`);
+                            await browser.close();
+                            return;
+                        }
+                    } else {
                         await page.setCookie(...getSessionCookie(
                             session.jwtSession,
                             session.jwtToken,
                             session.ds,
                             session.dsr
                         ));
+                        await page.goto("https://you.com", { timeout: timeout });
+                    }
 
-                    page.goto("https://you.com", { timeout: 60000 });
                     await sleep(5000); // 等待加载完毕
+
                     // 如果遇到盾了就多等一段时间
                     let pageContent = await page.content();
                     if (pageContent.indexOf("https://challenges.cloudflare.com") > -1) {
-                        console.log(`请在30秒内完成人机验证`);
+                        console.log(`请在30秒内完成人机验证 (${username})`);
-                        page.evaluate(() => {
+                        await page.evaluate(() => {
                             alert("请在30秒内完成人机验证");
                         });
                         await sleep(30000);
                     }
 
-                    // get page content and try parse JSON
+                    // 验证 cookie 有效性
                     try {
                         let content = await page.evaluate(() => {
                             return fetch("https://you.com/api/user/getYouProState").then((res) => res.text());
@@ -150,13 +190,47 @@ class YouProvider {
                     }
                 })
                 .catch((e) => {
-                    console.error(`初始化浏览器失败`);
+                    console.error(`初始化浏览器失败 (${username})`);
                     console.error(e);
                 });
         }
         console.log(`验证完毕，有效cookie数量 ${Object.keys(this.sessions).filter((username) => this.sessions[username].valid).length}`);
     }
 
+    async waitForManualLogin(page) {
+        return new Promise((resolve) => {
+            const checkLoginStatus = async () => {
+                const isLoggedIn = await page.evaluate(() => {
+                    return !!document.querySelector('button[aria-label="User menu"]');
+                });
+                if (isLoggedIn) {
+                    console.log("检测到登录成功");
+                    resolve();
+                } else {
+                    setTimeout(checkLoginStatus, 1000);
+                }
+            };
+
+            page.on('request', request => {
+                if (request.url().includes('https://you.com/api/instrumentation')) {
+                    resolve();
+                }
+            });
+
+            checkLoginStatus();
+        });
+    }
+
+    extractSessionCookie(cookies) {
+        const ds = cookies.find(c => c.name === 'DS')?.value;
+        const dsr = cookies.find(c => c.name === 'DSR')?.value;
+        if (ds) {
+            const jwt = JSON.parse(Buffer.from(ds.split(".")[1], "base64").toString());
+            return {ds, dsr, email: jwt.email};
+        }
+        return null;
+    }
+
     detectBrowser() {
         const platform = os.platform();
         let browsers = {
@@ -483,7 +557,11 @@ class YouProvider {
         req_param.append("domain", "youchat");
         req_param.append("mkt", "ja-JP");
         if (uploadedFile)
-			req_param.append("userFiles", JSON.stringify([{ user_filename: randomFileName, filename: uploadedFile.filename, size: messageBuffer.length }]));
+            req_param.append("userFiles", JSON.stringify([{
+                user_filename: randomFileName,
+                filename: uploadedFile.filename,
+                size: messageBuffer.length
+            }]));
         req_param.append("chat", JSON.stringify(userMessage));
         var url = "https://you.com/api/streamingSearch?" + req_param.toString();
         console.log("正在发送请求");
@@ -539,7 +617,10 @@ class YouProvider {
         } catch (error) {
             console.error("评估过程中出错:", error);
             emitter.emit("error", error);
-			return { completion: emitter, cancel: () => {} };
+            return {
+                completion: emitter, cancel: () => {
+                }
+            };
         }
 
         const cancel = () => {