From a8d408d1f2078f5d52c4ff93bc6d48644c997f18 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: Wed, 23 Oct 2024 01:35:50 +0800 Subject: [PATCH] Create happyApi.mjs --- happyapi_providers/happyApi.mjs | 262 ++++++++++++++++++++++++++++++++ 1 file changed, 262 insertions(+) create mode 100644 happyapi_providers/happyApi.mjs diff --git a/happyapi_providers/happyApi.mjs b/happyapi_providers/happyApi.mjs new file mode 100644 index 0000000..ccbab3d --- /dev/null +++ b/happyapi_providers/happyApi.mjs @@ -0,0 +1,262 @@ +import {EventEmitter} from "events"; +import {connect} from "puppeteer-real-browser"; +import {v4 as uuidV4} from "uuid"; +import path from "path"; +import {fileURLToPath} from "url"; +import {createDirectoryIfNotExists, sleep} from "../utils.mjs"; +import '../proxyAgent.mjs'; +import NetworkMonitor from "../networkMonitor.mjs"; +import {detectBrowser} from "../utils/browserDetector.mjs"; + +const __filename = fileURLToPath(import.meta.url); +const __dirname = path.dirname(__filename); + +class HappyApiProvider { + constructor(config) { + this.config = config; + this.sessions = {}; + this.preferredBrowser = 'auto'; + this.networkMonitor = new NetworkMonitor(); + } + + async init() { + console.log(`本项目依赖 Chrome 或 Edge 浏览器，请勿关闭弹出的浏览器窗口。如果出现错误请检查是否已安装 Chrome 或 Edge 浏览器。`); + + const browserPath = detectBrowser(this.preferredBrowser); + + this.sessions = {}; + + // 手动登录 + const currentUsername = 'manual_login'; + createDirectoryIfNotExists(path.join(__dirname, "browser_profiles", currentUsername)); + + try { + const response = await connect({ + headless: "auto", + turnstile: true, + customConfig: { + userDataDir: path.join(__dirname, "browser_profiles", currentUsername), + executablePath: browserPath, + }, + }); + + const {page, browser} = response; + + console.log(`请在打开的浏览器窗口中手动登录 happyapi.org`); + await page.goto("https://happyapi.org", {waitUntil: 'networkidle0'}); + await sleep(3000); // 等待页面加载完毕 + + const sessionCookie = await this.waitForManualLogin(page); + if (sessionCookie) { + this.sessions[currentUsername] = {browser, page, sessionCookie, valid: true}; + console.log(`成功获取会话 cookie`); + } else { + console.error(`未能检测到登录状态，请确保已成功登录`); + await browser.close(); + } + + } catch (error) { + console.error(`初始化浏览器失败`); + console.error(error); + } + + // 开始网络监控 + await this.networkMonitor.startMonitoring(); + } + + async waitForManualLogin(page) { + return new Promise((resolve) => { + const checkLoginStatus = async () => { + // 检查是否存在特定的登录成功标识元素 + const isLoggedIn = await this.checkLoginStatus(page); + if (isLoggedIn) { + console.log(`检测到登录成功`); + const cookies = await page.cookies(); + const sessionCookie = this.extractSessionCookie(cookies); + resolve(sessionCookie); + } else { + setTimeout(checkLoginStatus, 1000); + } + }; + + page.on('request', (request) => { + if (request.url() === 'https://happyapi.org/api/v1/auths/signin' && request.method() === 'POST') { + console.log('检测到登录请求'); + checkLoginStatus(); + } + }); + + checkLoginStatus(); + }); + } + + extractSessionCookie(cookies) { + const cfClearance = cookies.find(c => c.name === 'cf_clearance')?.value; + const token = cookies.find(c => c.name === 'token')?.value; + + if (cfClearance && token) { + return [ + { + name: 'cf_clearance', + value: cfClearance, + domain: 'happyapi.org', + path: '/', + httpOnly: true, + secure: true, + sameSite: 'Lax', + }, + { + name: 'token', + value: token, + domain: 'happyapi.org', + path: '/', + httpOnly: true, + secure: true, + sameSite: 'Lax', + } + ]; + } else { + console.error('无法提取有效的会话 cookie'); + return null; + } + } + + async checkLoginStatus(page) { + try { + const userNameElement = await page.waitForSelector('div.self-center.font-medium', {timeout: 5000}); + const userName = await page.evaluate(element => element.textContent, userNameElement); + return true; + } catch { + return false; + } + } + + async getCompletion({username, messages, stream = false, proxyModel}) { + const session = this.sessions[username]; + if (!session || !session.valid) { + throw new Error(`用户 ${username} 的会话无效`); + } + //刷新页面 + await session.page.goto("https://happyapi.org", {waitUntil: 'domcontentloaded'}); + + const {page} = session; + const emitter = new EventEmitter(); + + try { + + let userQuery = messages.map(msg => msg.content).join("\n\n"); + + const inputSelector = 'textarea#chat-textarea'; + const sendButtonSelector = 'button#send-message-button'; + + // 输入消息 + await page.waitForSelector(inputSelector, {visible: true}); + + await page.evaluate((selector, message) => { + const textarea = document.querySelector(selector); + textarea.value = message; + + // 手动触发事件 + textarea.dispatchEvent(new Event('input', {bubbles: true})); + textarea.dispatchEvent(new Event('change', {bubbles: true})); + }, inputSelector, userQuery.trim()); + + await page.waitForTimeout(100); + + // 点击发送按钮 + await page.waitForSelector(sendButtonSelector, {visible: true}); + await page.click(sendButtonSelector); + + // 监听请求 + const targetUrl = 'https://happyapi.org/api/chat/completions'; + + const traceId = uuidV4(); + let finalResponse = ""; // 存储最终响应 + let responseStarted = false; // 是否开始接收 + let isEnding = false; // 是否正在结束 + + // 清理函数 + const cleanup = async () => { + clearTimeout(responseTimeout); + page.removeListener('response', responseHandler); + }; + + const responseTimeout = setTimeout(() => { + if (!isEnding) { + isEnding = true; + emitter.emit('error', new Error('请求超时')); + cleanup(); + } + }, 60000); // 60秒，可根据需要调整 + + const responseHandler = async (response) => { + if (response.url() === targetUrl) { + try { + const buffer = await response.buffer(); + const text = buffer.toString('utf8'); + + // 处理流式数据 + const lines = text.split('\n\n'); + for (const line of lines) { + if (line.startsWith('data: ')) { + const data = line.substring(6).trim(); + if (data === '[DONE]') { + if (!isEnding) { + isEnding = true; + console.log("请求结束"); + emitter.emit('end'); + cleanup(); + } + return; + } + if (data.length === 0) { + continue; + } + const parsedData = JSON.parse(data); + const contentDelta = parsedData.choices[0].delta.content || ''; + + if (stream) { + emitter.emit('completion', traceId, contentDelta); + } else { + finalResponse += contentDelta; + } + } + } + } catch (e) { + if (!isEnding) { + isEnding = true; + console.error("请求发生错误", e); + emitter.emit('error', e); + cleanup(); + } + } + } + }; + + // 监听response + page.on('response', responseHandler); + + // 清除计时器和监听器 + emitter.on('end', () => { + cleanup(); + }); + emitter.on('error', () => { + cleanup(); + }); + + return { + completion: emitter, + cancel: () => { + } + }; + } catch (error) { + emitter.emit('error', error); + return { + completion: emitter, + cancel: () => { } + }; + } + } +} + +export default HappyApiProvider;