add functionality
This commit is contained in:
被遗忘的记忆
+118
-18
@@ -1,6 +1,6 @@
|
||||
import {EventEmitter} from "events";
|
||||
import {connect} from "puppeteer-real-browser";
|
||||
import {v4 as uuidv4} from "uuid";
|
||||
import {v4 as uuidV4} from "uuid";
|
||||
import path from "path";
|
||||
import fs from "fs";
|
||||
import {fileURLToPath} from "url";
|
||||
@@ -111,11 +111,10 @@ class YouProvider {
|
||||
let currentUsername = originalUsername;
|
||||
let session = this.sessions[currentUsername];
|
||||
createDirectoryIfNotExists(path.join(__dirname, "browser_profiles", currentUsername));
|
||||
const isWindows = os.platform() === 'win32';
|
||||
|
||||
try {
|
||||
const response = await connect({
|
||||
headless: isWindows ? false : 'auto',
|
||||
headless: "auto",
|
||||
turnstile: true,
|
||||
customConfig: {
|
||||
userDataDir: path.join(__dirname, "browser_profiles", currentUsername),
|
||||
@@ -123,14 +122,14 @@ class YouProvider {
|
||||
},
|
||||
});
|
||||
|
||||
const { page, browser } = 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 page.goto("https://you.com", {timeout: timeout});
|
||||
// 等待页面加载完毕
|
||||
await sleep(5000);
|
||||
console.log(`请在打开的浏览器窗口中手动登录 You.com (session #${session.configIndex})`);
|
||||
const { loginInfo, sessionCookie } = await this.waitForManualLogin(page);
|
||||
const {loginInfo, sessionCookie} = await this.waitForManualLogin(page);
|
||||
if (sessionCookie) {
|
||||
const email = loginInfo || sessionCookie.email;
|
||||
this.sessions[email] = {
|
||||
@@ -156,7 +155,7 @@ class YouProvider {
|
||||
session.ds,
|
||||
session.dsr
|
||||
));
|
||||
await page.goto("https://you.com", { timeout: timeout });
|
||||
await page.goto("https://you.com", {timeout: timeout});
|
||||
}
|
||||
|
||||
await sleep(5000); // 等待加载完毕
|
||||
@@ -177,19 +176,28 @@ class YouProvider {
|
||||
return fetch("https://you.com/api/user/getYouProState").then((res) => res.text());
|
||||
});
|
||||
const json = JSON.parse(content);
|
||||
if (json.subscriptions.length > 0) {
|
||||
if (json.subscriptions && json.subscriptions.length > 0) {
|
||||
console.log(`${currentUsername} 有效`);
|
||||
session.valid = true;
|
||||
session.browser = browser;
|
||||
session.page = page;
|
||||
|
||||
// 获取订阅信息
|
||||
const subscriptionInfo = await this.getSubscriptionInfo(page);
|
||||
if (subscriptionInfo) {
|
||||
session.subscriptionInfo = subscriptionInfo;
|
||||
}
|
||||
} else {
|
||||
console.log(`${currentUsername} 无有效订阅`);
|
||||
console.warn(`警告: ${currentUsername} 可能没有有效的订阅。请检查You是否有有效的Pro订阅。`);
|
||||
await this.clearYouCookies(page);
|
||||
await browser.close();
|
||||
}
|
||||
} catch (e) {
|
||||
console.log(`${currentUsername} 已失效`);
|
||||
console.warn(`警告: ${currentUsername} 验证失败。请检查cookie是否有效。`);
|
||||
console.error(e);
|
||||
await this.clearYouCookies(page);
|
||||
await browser.close();
|
||||
}
|
||||
} catch (e) {
|
||||
@@ -197,9 +205,99 @@ class YouProvider {
|
||||
console.error(e);
|
||||
}
|
||||
}
|
||||
|
||||
console.log("订阅信息汇总：");
|
||||
for (const [username, session] of Object.entries(this.sessions)) {
|
||||
if (session.valid) {
|
||||
console.log(`{${username}:`);
|
||||
if (session.subscriptionInfo) {
|
||||
console.log(` 订阅计划: ${session.subscriptionInfo.planName}`);
|
||||
console.log(` 到期日期: ${session.subscriptionInfo.expirationDate}`);
|
||||
console.log(` 剩余天数: ${session.subscriptionInfo.daysRemaining}天`);
|
||||
if (session.subscriptionInfo.cancelAtPeriodEnd) {
|
||||
console.log(' 注意: 该订阅已设置为在当前周期结束后取消');
|
||||
}
|
||||
} else {
|
||||
console.log(' 无法获取具体订阅信息，但账户有效');
|
||||
}
|
||||
console.log('}');
|
||||
}
|
||||
}
|
||||
console.log(`验证完毕，有效cookie数量 ${Object.keys(this.sessions).filter((username) => this.sessions[username].valid).length}`);
|
||||
}
|
||||
|
||||
async getSubscriptionInfo(page) {
|
||||
try {
|
||||
const response = await page.evaluate(async () => {
|
||||
const res = await fetch('https://you.com/api/user/getYouProState', {
|
||||
method: 'GET',
|
||||
credentials: 'include'
|
||||
});
|
||||
return await res.json();
|
||||
});
|
||||
if (response && response.subscriptions && response.subscriptions.length > 0) {
|
||||
const subscription = response.subscriptions[0];
|
||||
if (subscription.start_date && subscription.interval) {
|
||||
const startDate = new Date(subscription.start_date);
|
||||
const today = new Date();
|
||||
let expirationDate;
|
||||
|
||||
// 计算订阅结束日期
|
||||
if (subscription.interval === 'month') {
|
||||
expirationDate = new Date(startDate.getFullYear(), startDate.getMonth() + 1, startDate.getDate());
|
||||
} else if (subscription.interval === 'year') {
|
||||
expirationDate = new Date(startDate.getFullYear() + 1, startDate.getMonth(), startDate.getDate());
|
||||
} else {
|
||||
console.log(`未知的订阅间隔: ${subscription.interval}`);
|
||||
return null;
|
||||
}
|
||||
|
||||
// 如果计算出的结束日期已经过去，继续加月/年直到未来日期
|
||||
while (expirationDate <= today) {
|
||||
if (subscription.interval === 'month') {
|
||||
expirationDate.setMonth(expirationDate.getMonth() + 1);
|
||||
} else {
|
||||
expirationDate.setFullYear(expirationDate.getFullYear() + 1);
|
||||
}
|
||||
}
|
||||
|
||||
const daysRemaining = Math.ceil((expirationDate - today) / (1000 * 60 * 60 * 24));
|
||||
|
||||
return {
|
||||
expirationDate: expirationDate.toLocaleDateString('zh-CN', {
|
||||
year: 'numeric',
|
||||
month: 'long',
|
||||
day: 'numeric'
|
||||
}),
|
||||
daysRemaining: daysRemaining,
|
||||
planName: subscription.plan_name,
|
||||
cancelAtPeriodEnd: subscription.cancel_at_period_end
|
||||
};
|
||||
} else {
|
||||
console.log('订阅信息中缺少 start_date 或 interval 字段');
|
||||
return null;
|
||||
}
|
||||
} else {
|
||||
console.log('API 响应中没有有效的订阅信息');
|
||||
return null;
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('获取订阅信息时出错:', error);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
async clearYouCookies(page) {
|
||||
const client = await page.target().createCDPSession();
|
||||
await client.send('Network.clearBrowserCookies');
|
||||
await client.send('Network.clearBrowserCache');
|
||||
const cookies = await page.cookies('https://you.com');
|
||||
for (const cookie of cookies) {
|
||||
await page.deleteCookie(cookie);
|
||||
}
|
||||
console.log('已自动清理 cookie');
|
||||
}
|
||||
|
||||
async waitForManualLogin(page) {
|
||||
return new Promise((resolve) => {
|
||||
const checkLoginStatus = async () => {
|
||||
@@ -222,7 +320,7 @@ class YouProvider {
|
||||
await page.setCookie(...sessionCookie);
|
||||
}
|
||||
|
||||
resolve({ loginInfo, sessionCookie });
|
||||
resolve({loginInfo, sessionCookie});
|
||||
} else {
|
||||
setTimeout(checkLoginStatus, 1000);
|
||||
}
|
||||
@@ -238,7 +336,7 @@ class YouProvider {
|
||||
await page.setCookie(...sessionCookie);
|
||||
}
|
||||
|
||||
resolve({ loginInfo: null, sessionCookie });
|
||||
resolve({loginInfo: null, sessionCookie});
|
||||
}
|
||||
});
|
||||
|
||||
@@ -415,19 +513,19 @@ class YouProvider {
|
||||
let lastUpdate = true;
|
||||
|
||||
messages.forEach((msg) => {
|
||||
if (msg.role == "system" || msg.role == "user") {
|
||||
if (msg.role === "system" || msg.role === "user") {
|
||||
if (lastUpdate) {
|
||||
userMessage[userMessage.length - 1].question += msg.content + "\n";
|
||||
} else if (userMessage[userMessage.length - 1].question == "") {
|
||||
} else if (userMessage[userMessage.length - 1].question === "") {
|
||||
userMessage[userMessage.length - 1].question += msg.content + "\n";
|
||||
} else {
|
||||
userMessage.push({question: msg.content + "\n", answer: ""});
|
||||
}
|
||||
lastUpdate = true;
|
||||
} else if (msg.role == "assistant") {
|
||||
} else if (msg.role === "assistant") {
|
||||
if (!lastUpdate) {
|
||||
userMessage[userMessage.length - 1].answer += msg.content + "\n";
|
||||
} else if (userMessage[userMessage.length - 1].answer == "") {
|
||||
} else if (userMessage[userMessage.length - 1].answer === "") {
|
||||
userMessage[userMessage.length - 1].answer += msg.content + "\n";
|
||||
} else {
|
||||
userMessage.push({question: "", answer: msg.content + "\n"});
|
||||
@@ -467,7 +565,7 @@ class YouProvider {
|
||||
}).then((res) => res.json());
|
||||
},
|
||||
proxyModel,
|
||||
uuidv4().substring(0, 4)
|
||||
uuidV4().substring(0, 4)
|
||||
);
|
||||
if (userChatMode.chat_mode_id) {
|
||||
this.config.sessions[session.configIndex].user_chat_mode_id[proxyModel] = userChatMode.chat_mode_id;
|
||||
@@ -551,8 +649,8 @@ class YouProvider {
|
||||
if (uploadedFile.error) throw new Error(uploadedFile.error);
|
||||
}
|
||||
|
||||
let msgid = uuidv4();
|
||||
let traceId = uuidv4();
|
||||
let msgid = uuidV4();
|
||||
let traceId = uuidV4();
|
||||
|
||||
// expose function to receive youChatToken
|
||||
let finalResponse = "";
|
||||
@@ -606,6 +704,7 @@ class YouProvider {
|
||||
const 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.evaluate(
|
||||
@@ -659,7 +758,8 @@ class YouProvider {
|
||||
emitter.emit("error", error);
|
||||
return {
|
||||
completion: emitter,
|
||||
cancel: () => {}
|
||||
cancel: () => {
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user