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

fix browser verification

Browse Source
This commit is contained in:
被遗忘的记忆
2025-03-14 05:53:06 +08:00
committed by GitHub
parent 476fa869ba
commit d7ae6d4ded
3 changed files with 1963 additions and 1713 deletions
Download Patch File Download Diff File Expand all files Collapse all files
you_providers/cookieUpdater.mjs
+204
View File
@@ -0,0 +1,204 @@
import fs from "fs";
import path from "path";
import {fileURLToPath} from "url";
import {Mutex} from "async-mutex";
const configMutex = new Mutex(); // 互斥锁
const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);
const CONFIG_FILE_PATH = path.join(__dirname, "../config.mjs");
// 仅在 USE_MANUAL_LOGIN 为 false 且 ENABLE_AUTO_COOKIE_UPDATE 为 true 时生效
const ENABLE_AUTO_COOKIE_UPDATE = process.env.ENABLE_AUTO_COOKIE_UPDATE === "true";
function unifyQuotesForJSON(str) {
// 正则匹配 `` `...` ``
let out = str.replace(/`([^`]*)`/g, (match, p1) => {
const safe = p1.replace(/"/g, '\\"');
return `"${safe}"`;
});
out = out.replace(/'([^']*)'/g, (match, p1) => {
const safe = p1.replace(/"/g, '\\"');
return `"${safe}"`;
});
return out;
}
/**
* cookies 解析出 DS 与 DSR
* @param {Array} cookies 获取到的 cookie 数组
* @returns {{ ds?: string, dsr?: string }}
*/
function parseDSAndDSR(cookies) {
let dsValue, dsrValue;
for (const c of cookies) {
if (c.name === "DS") {
dsValue = c.value;
} else if (c.name === "DSR") {
dsrValue = c.value;
}
}
return {ds: dsValue, dsr: dsrValue};
}
/**
* 从 DS 中解析 email 字段
* @param {string} dsToken DS cookie
* @returns {string|null} 返回 email或null
*/
function decodeEmailFromDs(dsToken) {
try {
const parts = dsToken.split(".");
if (parts.length < 2) return null;
const payload = JSON.parse(Buffer.from(parts[1], "base64").toString("utf-8"));
return payload.email || null;
} catch (err) {
return null;
}
}
/**
* cookie 数组转换 "name=value; name=value"
* @param {Array} cookies
* @returns {string}
*/
function cookiesToStringAll(cookies) {
return cookies.map(c => `${c.name}=${c.value}`).join("; ");
}
/**
* cookie 转换数组
* 每个对象如 { name, value }
* @param {string} cookieStr
* @returns {Array}
*/
function parseCookieString(cookieStr) {
return cookieStr.split("; ").map(entry => {
const [name, value] = entry.split("=", 2);
return {name, value};
});
}
/**
* 本地 configObj.sessions 查找与指定 email 匹配的 session，
* @param {object} configObj 解析后 config
* @param {string} email 匹配的邮箱
* @returns {{ index: number, oldCookie: string, ds: string, dsr: string } | null}
*/
function findSessionByEmail(configObj, email) {
if (!Array.isArray(configObj.sessions)) return null;
for (let i = 0; i < configObj.sessions.length; i++) {
const cookieStr = configObj.sessions[i].cookie || "";
const dsMatch = /DS=([^;\s]+)/.exec(cookieStr);
if (!dsMatch) continue;
const dsValue = dsMatch[1];
const dsEmail = decodeEmailFromDs(dsValue);
if (dsEmail && dsEmail.toLowerCase() === email.toLowerCase()) {
const dsrMatch = /DSR=([^;\s]+)/.exec(cookieStr);
const dsrValue = dsrMatch ? dsrMatch[1] : "";
return {
index: i,
oldCookie: cookieStr,
ds: dsValue,
dsr: dsrValue
};
}
}
return null;
}
/**
* config.mjs 中匹配相同 email 的 session，若 DS 或 DSR 有变化，则更新整个 cookie
* @param {import('puppeteer-core').Page} page
*/
export async function updateLocalConfigCookieByEmail(page) {
if (!ENABLE_AUTO_COOKIE_UPDATE || process.env.USE_MANUAL_LOGIN === "true") {
return;
}
// 尝试从 “https://you.com/api/instrumentation” 获取 cookie
let cookieStringFromInstrumentation = "";
try {
const instrRequest = await page.waitForRequest(
req => req.url().includes("/api/instrumentation"),
{timeout: 5000}
);
if (instrRequest) {
cookieStringFromInstrumentation = instrRequest.headers()["cookie"];
}
} catch (err) {
}
let allCookiesString = "";
if (cookieStringFromInstrumentation) {
allCookiesString = cookieStringFromInstrumentation;
} else {
// 使用 page.cookies() 获取
const cookies = await page.cookies("https://you.com");
allCookiesString = cookiesToStringAll(cookies);
}
const cookieArray = parseCookieString(allCookiesString);
const {ds: newDs, dsr: newDsr} = parseDSAndDSR(cookieArray);
if (!newDs) {
console.log("网页未找到 DS，跳过更新。");
return;
}
const newEmail = decodeEmailFromDs(newDs);
if (!newEmail) {
console.log("[网页无法从 DS 解出 email，跳过更新。");
return;
}
// 互斥区
await configMutex.runExclusive(async () => {
try {
if (!fs.existsSync(CONFIG_FILE_PATH)) {
console.warn(`找不到 config.mjs: ${CONFIG_FILE_PATH}`);
return;
}
const raw = fs.readFileSync(CONFIG_FILE_PATH, "utf8");
// 去掉 export const config =
let jsonString = raw.replace(/^export const config\s*=\s*/, "").trim();
jsonString = unifyQuotesForJSON(jsonString);
const configObj = JSON.parse(jsonString);
const found = findSessionByEmail(configObj, newEmail);
if (!found) {
console.log(`未能在 config 中找到 email=${newEmail} 的 session，跳过更新。`);
return;
}
if (found.ds === newDs && found.dsr === newDsr) {
console.log(`DS/DSR 未变化(email=${newEmail})，不更新。`);
return;
}
configObj.sessions[found.index].cookie = allCookiesString;
const newFileContent = "export const config = " + JSON.stringify(configObj, null, 4);
fs.writeFileSync(CONFIG_FILE_PATH, newFileContent, "utf8");
console.log(`Cookie已更新(email=${newEmail})`);
} catch (err) {
console.warn("Cookie更新过程出错:", err);
}
});
}
/**
* 非阻塞
* @param {import('puppeteer-core').Page} page
*/
export function updateLocalConfigCookieByEmailNonBlocking(page) {
// 保证异步
setImmediate(() => {
updateLocalConfigCookieByEmail(page).catch(err =>
console.error("Cookie update error:", err)
);
});
}
you_providers/youProvider.mjs
+224 -178
View File
@@ -3,7 +3,7 @@ import {v4 as uuidV4} from "uuid";
import path from "path";
import fs from "fs";
import {fileURLToPath} from "url";
import {createDocx, extractCookie, getSessionCookie, sleep} from "../utils.mjs";
import {createDocx, extractCookie, getSessionCookie, sleep} from "../utils/cookieUtils.mjs";
import {exec} from 'child_process';
import '../proxyAgent.mjs';
import {formatMessages} from '../formatMessages.mjs';
@@ -109,14 +109,30 @@ class YouProvider {
}
} else {
// 使用配置文件中的 cookie
// 检查 invalid_accounts 字段
const invalidAccounts = config.invalid_accounts || {};
for (let index = 0; index < config.sessions.length; index++) {
const session = config.sessions[index];
const {jwtSession, jwtToken, ds, dsr, you_subscription, youpro_subscription} = extractCookie(session.cookie);
const {
jwtSession,
jwtToken,
ds,
dsr,
you_subscription,
youpro_subscription
} = extractCookie(session.cookie);
if (jwtSession && jwtToken) {
// 旧版cookie处理
try {
const jwt = JSON.parse(Buffer.from(jwtToken.split(".")[1], "base64").toString());
const username = jwt.user.name;
if (invalidAccounts[username]) {
console.log(`跳过标记失效账号 #${index} ${username} (${invalidAccounts[username]})`);
continue;
}
this.sessions[username] = {
configIndex: index,
jwtSession,
@@ -137,6 +153,12 @@ class YouProvider {
try {
const jwt = JSON.parse(Buffer.from(ds.split(".")[1], "base64").toString());
const username = jwt.email;
if (invalidAccounts[username]) {
console.log(`跳过标记失效账号 #${index} ${username} (${invalidAccounts[username]})`);
continue;
}
this.sessions[username] = {
configIndex: index,
ds,
@@ -371,12 +393,18 @@ class YouProvider {
console.log(`${currentUsername} 无有效订阅`);
console.warn(`警告: ${currentUsername} 可能没有有效的订阅。请检查You是否有有效的Pro或Team订阅。`);
session.valid = false;
// 标记为失效
await markAccountAsInvalid(currentUsername, this.config);
}
} catch (parseErr) {
console.log(`${currentUsername} 已失效 (fetchYouProState 异常)`);
console.warn(`警告: ${currentUsername} 验证失败。请检查cookie是否有效。`);
console.error(parseErr);
session.valid = false;
// 标记为失效
await markAccountAsInvalid(currentUsername, this.config);
}
} catch (errorVisit) {
console.error(`验证账户 ${currentUsername} 时出错:`, errorVisit);
@@ -1135,182 +1163,6 @@ class YouProvider {
let errorCount = 0; // 错误计数器
const ERROR_TIMEOUT = (proxyModel === "openai_o1" || proxyModel === "openai_o1_preview") ? 60000 : 20000; // 错误超时时间
const self = this;
page.exposeFunction("callback" + traceId, async (event, data) => {
if (isEnding) return;
switch (event) {
case "youChatToken": {
data = JSON.parse(data);
let tokenContent = data.youChatToken;
buffer += tokenContent;
if (buffer.endsWith('\\') && !buffer.endsWith('\\\\')) {
// 等待下一个字符
break;
}
let processedContent = unescapeContent(buffer);
buffer = '';
if (!responseStarted) {
responseStarted = true;
startTime = Date.now();
clearTimeout(responseTimeout);
// 自定义终止符延迟触发
customEndMarkerTimer = setTimeout(() => {
customEndMarkerEnabled = true;
}, 20000);
// 停止
if (heartbeatInterval) {
clearInterval(heartbeatInterval);
heartbeatInterval = null;
}
}
// 重置错误计时器
if (errorTimer) {
clearTimeout(errorTimer);
errorTimer = null;
}
// 检测 'unusual query volume'
if (processedContent.includes('unusual query volume')) {
const warningMessage = "您在 you.com 账号的使用已达上限，当前(default/agent)模式已进入冷却期(CD)。请切换模式(default/agent[custom])或耐心等待冷却结束后再继续使用。";
emitter.emit("completion", traceId, warningMessage);
unusualQueryVolumeTriggered = true; // 更新标志位
if (self.isRotationEnabled) {
session.modeStatus[session.currentMode] = false;
self.checkAndSwitchMode();
if (Object.values(session.modeStatus).some(status => status)) {
console.log(`模式达到请求上限，已切换模式 ${session.currentMode}，请重试请求。`);
}
} else {
console.log("检测到请求量异常提示，请求终止。");
}
isEnding = true;
// 终止
setTimeout(async () => {
await cleanup();
emitter.emit("end", traceId);
}, 1000);
self.logger.logRequest({
email: username,
time: requestTime,
mode: session.currentMode,
model: proxyModel,
completed: true,
unusualQueryVolume: true,
});
break;
}
process.stdout.write(processedContent);
accumulatedResponse += processedContent;
if (Date.now() - startTime >= 20000) {
responseAfter20Seconds += processedContent;
}
if (stream) {
emitter.emit("completion", traceId, processedContent);
} else {
finalResponse += processedContent;
}
// 检查自定义结束标记
if (customEndMarkerEnabled && customEndMarker && checkEndMarker(responseAfter20Seconds, customEndMarker)) {
isEnding = true;
console.log("检测到自定义终止，关闭请求");
setTimeout(async () => {
await cleanup();
emitter.emit(stream ? "end" : "completion", traceId, stream ? undefined : finalResponse);
}, 1000);
self.logger.logRequest({
email: username,
time: requestTime,
mode: session.currentMode,
model: proxyModel,
completed: true,
unusualQueryVolume: unusualQueryVolumeTriggered,
});
}
break;
}
case "customEndMarkerEnabled":
customEndMarkerEnabled = true;
break;
case "done":
if (isEnding) return;
console.log("请求结束");
isEnding = true;
await cleanup(); // 清理
emitter.emit(stream ? "end" : "completion", traceId, stream ? undefined : finalResponse);
self.logger.logRequest({
email: username,
time: requestTime,
mode: session.currentMode,
model: proxyModel,
completed: true,
unusualQueryVolume: unusualQueryVolumeTriggered,
});
break;
case "error": {
if (isEnding) return; // 已结束则忽略
console.error("请求发生错误", data);
errorCount++;
if (errorCount >= 3) {
const errorMessage = "连接中断，未收到服务器响应";
if (errorTimer) {
clearTimeout(errorTimer);
errorTimer = null;
}
isEnding = true;
finalResponse += ` (${errorMessage})`;
await cleanup();
emitter.emit("completion", traceId, errorMessage);
emitter.emit("end", traceId);
// 记录日志
self.logger.logRequest({
email: username,
time: requestTime,
mode: session.currentMode,
model: proxyModel,
completed: false,
unusualQueryVolume: unusualQueryVolumeTriggered,
});
} else {
if (errorTimer) {
clearTimeout(errorTimer);
}
errorTimer = setTimeout(async () => {
console.log("连接超时，终止请求");
const errorMessage = "连接中断，未收到服务器响应";
emitter.emit("completion", traceId, errorMessage);
finalResponse += ` (${errorMessage})`;
isEnding = true;
await cleanup();
emitter.emit("end", traceId);
self.logger.logRequest({
email: username,
time: requestTime,
mode: session.currentMode,
model: proxyModel,
completed: false,
unusualQueryVolume: unusualQueryVolumeTriggered,
});
}, ERROR_TIMEOUT);
}
break;
}
}
});
// proxy response
const req_param = new URLSearchParams();
@@ -1582,6 +1434,183 @@ class YouProvider {
await page.goto(`https://you.com/search?q=&fromSearchBar=true&tbm=youchat&chatMode=${userChatModeId}&cid=c0_${traceId}`, {waitUntil: "domcontentloaded"});
}
await page.exposeFunction("callback" + traceId, async (event, data) => {
if (isEnding) return;
switch (event) {
case "youChatToken": {
data = JSON.parse(data);
let tokenContent = data.youChatToken;
buffer += tokenContent;
if (buffer.endsWith('\\') && !buffer.endsWith('\\\\')) {
// 等待下一个字符
break;
}
let processedContent = unescapeContent(buffer);
buffer = '';
if (!responseStarted) {
responseStarted = true;
startTime = Date.now();
clearTimeout(responseTimeout);
// 自定义终止符延迟触发
customEndMarkerTimer = setTimeout(() => {
customEndMarkerEnabled = true;
}, 20000);
// 停止
if (heartbeatInterval) {
clearInterval(heartbeatInterval);
heartbeatInterval = null;
}
}
// 重置错误计时器
if (errorTimer) {
clearTimeout(errorTimer);
errorTimer = null;
}
// 检测 'unusual query volume'
if (processedContent.includes('unusual query volume')) {
const warningMessage = "您在 you.com 账号的使用已达上限，当前(default/agent)模式已进入冷却期(CD)。请切换模式(default/agent[custom])或耐心等待冷却结束后再继续使用。";
emitter.emit("completion", traceId, warningMessage);
unusualQueryVolumeTriggered = true; // 更新标志位
if (self.isRotationEnabled) {
session.modeStatus[session.currentMode] = false;
self.checkAndSwitchMode();
if (Object.values(session.modeStatus).some(status => status)) {
console.log(`模式达到请求上限，已切换模式 ${session.currentMode}，请重试请求。`);
}
} else {
console.log("检测到请求量异常提示，请求终止。");
}
isEnding = true;
// 终止
setTimeout(async () => {
await cleanup();
emitter.emit("end", traceId);
}, 1000);
self.logger.logRequest({
email: username,
time: requestTime,
mode: session.currentMode,
model: proxyModel,
completed: true,
unusualQueryVolume: true,
});
break;
}
process.stdout.write(processedContent);
accumulatedResponse += processedContent;
if (Date.now() - startTime >= 20000) {
responseAfter20Seconds += processedContent;
}
if (stream) {
emitter.emit("completion", traceId, processedContent);
} else {
finalResponse += processedContent;
}
// 检查自定义结束标记
if (customEndMarkerEnabled && customEndMarker && checkEndMarker(responseAfter20Seconds, customEndMarker)) {
isEnding = true;
console.log("检测到自定义终止，关闭请求");
setTimeout(async () => {
await cleanup();
emitter.emit(stream ? "end" : "completion", traceId, stream ? undefined : finalResponse);
}, 1000);
self.logger.logRequest({
email: username,
time: requestTime,
mode: session.currentMode,
model: proxyModel,
completed: true,
unusualQueryVolume: unusualQueryVolumeTriggered,
});
}
break;
}
case "customEndMarkerEnabled":
customEndMarkerEnabled = true;
break;
case "done":
if (isEnding) return;
console.log("请求结束");
isEnding = true;
await cleanup(); // 清理
emitter.emit(stream ? "end" : "completion", traceId, stream ? undefined : finalResponse);
self.logger.logRequest({
email: username,
time: requestTime,
mode: session.currentMode,
model: proxyModel,
completed: true,
unusualQueryVolume: unusualQueryVolumeTriggered,
});
break;
case "error": {
if (isEnding) return; // 已结束则忽略
console.error("请求发生错误", data);
errorCount++;
if (errorCount >= 3) {
const errorMessage = "连接中断，未收到服务器响应";
if (errorTimer) {
clearTimeout(errorTimer);
errorTimer = null;
}
isEnding = true;
finalResponse += ` (${errorMessage})`;
await cleanup();
emitter.emit("completion", traceId, errorMessage);
emitter.emit("end", traceId);
// 记录日志
self.logger.logRequest({
email: username,
time: requestTime,
mode: session.currentMode,
model: proxyModel,
completed: false,
unusualQueryVolume: unusualQueryVolumeTriggered,
});
} else {
if (errorTimer) {
clearTimeout(errorTimer);
}
errorTimer = setTimeout(async () => {
console.log("连接超时，终止请求");
const errorMessage = "连接中断，未收到服务器响应";
emitter.emit("completion", traceId, errorMessage);
finalResponse += ` (${errorMessage})`;
isEnding = true;
await cleanup();
emitter.emit("end", traceId);
self.logger.logRequest({
email: username,
time: requestTime,
mode: session.currentMode,
model: proxyModel,
completed: false,
unusualQueryVolume: unusualQueryVolumeTriggered,
});
}, ERROR_TIMEOUT);
}
break;
}
}
});
responseTimeout = setTimeout(async () => {
if (!responseStarted && !clientState.isClosed()) {
console.log(`${responseTimeoutTimer / 1000}秒内没有收到响应，尝试重新发送请求`);
@@ -1710,3 +1739,20 @@ function randomSelect(input) {
return words[randomIndex];
});
}
/**
* 账号标记失效并保存
* @param {string} username - 账号邮箱
* @param {Object} config - 配置对象
*/
async function markAccountAsInvalid(username, config) {
if (!config.invalid_accounts) {
config.invalid_accounts = {};
}
config.invalid_accounts[username] = "已失效";
try {
fs.writeFileSync("./config.mjs", `export const config = ${JSON.stringify(config, null, 4)}`);
} catch (error) {
console.error(`保存失效账号信息失败:`, error);
}
}