kova/youchat_proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add manual login

Browse Source
This commit is contained in:
被遗忘的记忆
2024-08-25 15:17:15 +08:00
committed by GitHub
parent f0e9d8e24b
commit 5dc7bfc1b6
1 changed files with 496 additions and 415 deletions
Download Patch File Download Diff File Expand all files Collapse all files
provider.mjs
+112 -31
View File
@@ -1,14 +1,14 @@
import { EventEmitter } from "events";
import { connect } from "puppeteer-real-browser";
import { v4 as uuidv4 } from "uuid";
import {EventEmitter} from "events";
import {connect} from "puppeteer-real-browser";
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 { execSync } from 'child_process';
import {fileURLToPath} from "url";
import {createDirectoryIfNotExists, createDocx, extractCookie, getSessionCookie, sleep} from "./utils.mjs";
import {execSync} from 'child_process';
import os from 'os';
import './proxyAgent.mjs';
import { formatMessages } from './formatMessages.mjs';
import {formatMessages} from './formatMessages.mjs';
const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);
@@ -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 = {
@@ -265,7 +339,7 @@ class YouProvider {
throw new Error(`用户 ${username} 的会话无效`);
}
const { page, browser } = session;
const {page, browser} = session;
const emitter = new EventEmitter();
// 处理模式轮换逻辑
if (this.isCustomModeEnabled && this.isRotationEnabled) {
@@ -288,15 +362,15 @@ class YouProvider {
if (!isLoaded) {
console.log('页面尚未加载完成，等待加载...');
await page.waitForNavigation({ waitUntil: 'domcontentloaded', timeout: 10000 }).catch(() => {
await page.waitForNavigation({waitUntil: 'domcontentloaded', timeout: 10000}).catch(() => {
console.log('页面加载超时，继续执行');
});
}
await page.goto("https://you.com/?chatMode=default", { waitUntil: "domcontentloaded" });
await page.goto("https://you.com/?chatMode=default", {waitUntil: "domcontentloaded"});
// 计算用户消息长度
let userMessage = [{ question: "", answer: "" }];
let userMessage = [{question: "", answer: ""}];
let userQuery = "";
let lastUpdate = true;
@@ -307,7 +381,7 @@ class YouProvider {
} else if (userMessage[userMessage.length - 1].question == "") {
userMessage[userMessage.length - 1].question += msg.content + "\n";
} else {
userMessage.push({ question: msg.content + "\n", answer: "" });
userMessage.push({question: msg.content + "\n", answer: ""});
}
lastUpdate = true;
} else if (msg.role == "assistant") {
@@ -316,7 +390,7 @@ class YouProvider {
} else if (userMessage[userMessage.length - 1].answer == "") {
userMessage[userMessage.length - 1].answer += msg.content + "\n";
} else {
userMessage.push({ question: "", answer: msg.content + "\n" });
userMessage.push({question: "", answer: msg.content + "\n"});
}
lastUpdate = false;
}
@@ -342,7 +416,7 @@ class YouProvider {
hideInstructions: true,
includeFollowUps: false,
instructions: "Please review the attached prompt",
instructionsSummary:"",
instructionsSummary: "",
isUserOwned: true,
name: proxyModelName,
visibility: "private",
@@ -355,12 +429,12 @@ class YouProvider {
proxyModel,
uuidv4().substring(0, 4)
);
if(userChatMode.chat_mode_id){
if (userChatMode.chat_mode_id) {
this.config.sessions[session.configIndex].user_chat_mode_id[proxyModel] = userChatMode.chat_mode_id;
// 写回config
fs.writeFileSync("./config.mjs", "export const config = " + JSON.stringify(this.config, null, 4));
}else{
if(userChatMode.error) console.log(userChatMode.error)
} else {
if (userChatMode.error) console.log(userChatMode.error)
console.log("Failed to create user chat mode, will use default mode instead.");
}
}
@@ -483,13 +557,17 @@ 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("正在发送请求");
emitter.emit("start", traceId);
try {
await page.goto(`https://you.com/search?q=&fromSearchBar=true&tbm=youchat&chatMode=custom`, { waitUntil: "domcontentloaded" });
await page.goto(`https://you.com/search?q=&fromSearchBar=true&tbm=youchat&chatMode=custom`, {waitUntil: "domcontentloaded"});
await page.evaluate(
async (url, traceId) => {
var evtSource = new EventSource(url);
@@ -515,7 +593,7 @@ class YouProvider {
headers: {
"content-type": "application/json",
},
body: JSON.stringify({ chatId: traceId }),
body: JSON.stringify({chatId: traceId}),
method: "DELETE",
});
},
@@ -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 = () => {
@@ -548,7 +629,7 @@ class YouProvider {
}, traceId).catch(console.error);
};
return { completion: emitter, cancel };
return {completion: emitter, cancel};
}
}