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

adapt to the new version of cookie

Browse Source
This commit is contained in:
被遗忘的记忆
2024-08-25 09:00:16 +08:00
committed by GitHub
parent e63b43de52
commit f0e9d8e24b
1 changed files with 25 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
provider.mjs
+15 -40
View File
@@ -56,10 +56,8 @@ class YouProvider {
// extract essential jwt session and token from cookie // extract essential jwt session and token from cookie
for (let index = 0; index < config.sessions.length; index++) { for (let index = 0; index < config.sessions.length; index++) {
let session = config.sessions[index]; let session = config.sessions[index], {jwtSession, jwtToken, ds, dsr} = extractCookie(session.cookie);
var { jwtSession, jwtToken, ds, dsr, youproSubscription, youSubscription, aiModel } = extractCookie(session.cookie); if (jwtSession && jwtToken) {
if (jwtSession && jwtToken) {
// 旧版cookie处理 // 旧版cookie处理
try { try {
let jwt = JSON.parse(Buffer.from(jwtToken.split(".")[1], "base64").toString()); let jwt = JSON.parse(Buffer.from(jwtToken.split(".")[1], "base64").toString());
@@ -67,9 +65,6 @@ class YouProvider {
configIndex: index, configIndex: index,
jwtSession, jwtSession,
jwtToken, jwtToken,
youproSubscription,
youSubscription,
aiModel,
valid: false, valid: false,
}; };
console.log(`已添加 #${index} ${jwt.user.name} (旧版cookie)`); console.log(`已添加 #${index} ${jwt.user.name} (旧版cookie)`);
@@ -77,51 +72,32 @@ class YouProvider {
console.error(`解析第${index}个旧版cookie失败: ${e.message}`); console.error(`解析第${index}个旧版cookie失败: ${e.message}`);
} }
} else if (ds) { } else if (ds) {
// 新版cookie处理，只要有DS就处理 // 新版cookie处理
try { try {
let jwt = JSON.parse(Buffer.from(ds.split(".")[1], "base64").toString()); let jwt = JSON.parse(Buffer.from(ds.split(".")[1], "base64").toString());
this.sessions[jwt.email] = { this.sessions[jwt.email] = {
configIndex: index, configIndex: index,
ds, ds,
dsr, // 即使dsr为null也会被添加 dsr,
youproSubscription,
youSubscription,
aiModel,
valid: false, valid: false,
}; };
console.log(`已添加 #${index} ${jwt.email} (新版cookie)`); console.log(`已添加 #${index} ${jwt.email} (新版cookie)`);
if (!dsr) { if (!dsr) {
console.warn(`警告: 第${index}个cookie缺少DSR字段。程序继续执行。`); console.warn(`警告: 第${index}个cookie缺少DSR字段。`);
} }
} catch (e) { } catch (e) {
console.error(`解析第${index}个新版cookie失败: ${e.message}`); console.error(`解析第${index}个新版cookie失败: ${e.message}`);
} }
} else { } else {
// 无效cookie处理
console.error(`第${index}个cookie无效，请重新获取。`); console.error(`第${index}个cookie无效，请重新获取。`);
if (jwtSession || jwtToken) { console.error(`未检测到有效的DS或stytch_session字段。`);
console.error(`检测到部分旧版cookie字段，但不完整。`);
if (!jwtSession) console.error(`缺少: stytch_session`);
if (!jwtToken) console.error(`缺少: stytch_session_jwt`);
} else if (ds) {
console.error(`检测到新版cookie的DS字段，但缺少其他必要字段。`);
} else {
console.error(`未检测到任何有效的cookie字段。`);
}
console.error(`提示: 请确保登录You.com并成功获取完整的cookie。`);
continue; // 跳过无效的cookie，继续下一个
}
if (youproSubscription === "false" || youSubscription === "freemium") {
console.warn(`警告: 第${index}个cookie可能没有有效的订阅。请检查You是否有有效的Pro订阅。`);
} }
} }
console.log(`已添加 ${Object.keys(this.sessions).length} 个 cookie，开始验证有效性（是否有订阅）`); console.log(`已添加 ${Object.keys(this.sessions).length} 个 cookie，开始验证有效性`);
for (let username of Object.keys(this.sessions)) {
for (var username of Object.keys(this.sessions)) { let session = this.sessions[username];
var session = this.sessions[username]; createDirectoryIfNotExists(path.join(__dirname, "browser_profiles", username));
createDirectoryIfNotExists(path.join(__dirname, "browser_profiles", username));
await connect({ await connect({
headless: "auto", headless: "auto",
turnstile: true, turnstile: true,
@@ -136,16 +112,13 @@ class YouProvider {
session.jwtSession, session.jwtSession,
session.jwtToken, session.jwtToken,
session.ds, session.ds,
session.dsr, session.dsr
session.youproSubscription,
session.youSubscription,
session.aiModel
)); ));
page.goto("https://you.com", { timeout: 60000 }); page.goto("https://you.com", { timeout: 60000 });
await sleep(5000); // 等待加载完毕 await sleep(5000); // 等待加载完毕
// 如果遇到盾了就多等一段时间 // 如果遇到盾了就多等一段时间
var pageContent = await page.content(); let pageContent = await page.content();
if (pageContent.indexOf("https://challenges.cloudflare.com") > -1) { if (pageContent.indexOf("https://challenges.cloudflare.com") > -1) {
console.log(`请在30秒内完成人机验证`); console.log(`请在30秒内完成人机验证`);
page.evaluate(() => { page.evaluate(() => {
@@ -167,10 +140,12 @@ class YouProvider {
session.page = page; session.page = page;
} else { } else {
console.log(`${username} 无有效订阅`); console.log(`${username} 无有效订阅`);
console.warn(`警告: ${username} 可能没有有效的订阅。请检查You是否有有效的Pro订阅。`);
await browser.close(); await browser.close();
} }
} catch (e) { } catch (e) {
console.log(`${username} 已失效`); console.log(`${username} 已失效`);
console.warn(`警告: ${username} 验证失败。请检查cookie是否有效。`);
await browser.close(); await browser.close();
} }
}) })
@@ -284,7 +259,7 @@ class YouProvider {
} }
} }
async getCompletion(username, messages, stream = false, proxyModel, useCustomMode = false) { async getCompletion({username, messages, stream = false, proxyModel, useCustomMode = false}) {
const session = this.sessions[username]; const session = this.sessions[username];
if (!session || !session.valid) { if (!session || !session.valid) {
throw new Error(`用户 ${username} 的会话无效`); throw new Error(`用户 ${username} 的会话无效`);