From d882b036ad8d12fb241e3fb2047a92e508d028a2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=A2=AB=E9=81=97=E5=BF=98=E7=9A=84=E8=AE=B0=E5=BF=86?= <32097237+YIWANG-sketch@users.noreply.github.com> Date: Mon, 26 Aug 2024 22:28:59 +0800 Subject: [PATCH] add functionality --- provider.mjs | 136 ++++++++++++++++++++++++++++++++++++++++++++------- 1 file changed, 118 insertions(+), 18 deletions(-) diff --git a/provider.mjs b/provider.mjs index 63d2595..15c3c5f 100644 --- a/provider.mjs +++ b/provider.mjs @@ -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: () => { + } }; }