add manual login
This commit is contained in:
被遗忘的记忆
+93
-12
@@ -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秒内完成人机验证`);
|
||||
page.evaluate(() => {
|
||||
console.log(`请在30秒内完成人机验证 (${username})`);
|
||||
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 = () => {
|
||||
|
||||
Reference in New Issue
Block a user