add manual login
This commit is contained in:
被遗忘的记忆
+492
-411
@@ -1,14 +1,14 @@
|
|||||||
import { EventEmitter } from "events";
|
import {EventEmitter} from "events";
|
||||||
import { connect } from "puppeteer-real-browser";
|
import {connect} from "puppeteer-real-browser";
|
||||||
import { v4 as uuidv4 } from "uuid";
|
import {v4 as uuidv4} from "uuid";
|
||||||
import path from "path";
|
import path from "path";
|
||||||
import fs from "fs";
|
import fs from "fs";
|
||||||
import { fileURLToPath } from "url";
|
import {fileURLToPath} from "url";
|
||||||
import { createDirectoryIfNotExists, sleep, extractCookie, getSessionCookie, createDocx } from "./utils.mjs";
|
import {createDirectoryIfNotExists, createDocx, extractCookie, getSessionCookie, sleep} from "./utils.mjs";
|
||||||
import { execSync } from 'child_process';
|
import {execSync} from 'child_process';
|
||||||
import os from 'os';
|
import os from 'os';
|
||||||
import './proxyAgent.mjs';
|
import './proxyAgent.mjs';
|
||||||
import { formatMessages } from './formatMessages.mjs';
|
import {formatMessages} from './formatMessages.mjs';
|
||||||
|
|
||||||
const __filename = fileURLToPath(import.meta.url);
|
const __filename = fileURLToPath(import.meta.url);
|
||||||
const __dirname = path.dirname(__filename);
|
const __dirname = path.dirname(__filename);
|
||||||
@@ -25,7 +25,7 @@ class YouProvider {
|
|||||||
this.switchCounter = 0;
|
this.switchCounter = 0;
|
||||||
this.requestsInCurrentMode = 0;
|
this.requestsInCurrentMode = 0;
|
||||||
this.switchThreshold = this.getRandomSwitchThreshold();
|
this.switchThreshold = this.getRandomSwitchThreshold();
|
||||||
this.lastDefaultThreshold = 0; // 记录上一次default模式的阈值
|
this.lastDefaultThreshold = 0; // 记录上一次default模式的阈值
|
||||||
}
|
}
|
||||||
|
|
||||||
getRandomSwitchThreshold() {
|
getRandomSwitchThreshold() {
|
||||||
@@ -54,52 +54,66 @@ class YouProvider {
|
|||||||
// 检测Chrome和Edge浏览器
|
// 检测Chrome和Edge浏览器
|
||||||
const browserPath = this.detectBrowser();
|
const browserPath = this.detectBrowser();
|
||||||
|
|
||||||
// extract essential jwt session and token from cookie
|
this.sessions = {};
|
||||||
for (let index = 0; index < config.sessions.length; index++) {
|
const timeout = 120000; // 120 秒超时
|
||||||
let session = config.sessions[index], {jwtSession, jwtToken, ds, dsr} = extractCookie(session.cookie);
|
|
||||||
if (jwtSession && jwtToken) {
|
if (process.env.USE_MANUAL_LOGIN === "true") {
|
||||||
// 旧版cookie处理
|
this.sessions['manual_login'] = {
|
||||||
try {
|
configIndex: 0,
|
||||||
let jwt = JSON.parse(Buffer.from(jwtToken.split(".")[1], "base64").toString());
|
valid: false,
|
||||||
this.sessions[jwt.user.name] = {
|
};
|
||||||
configIndex: index,
|
console.log("当前使用手动登录模式，跳过config.mjs文件中的 cookie 验证");
|
||||||
jwtSession,
|
} else {
|
||||||
jwtToken,
|
// 使用配置文件中的 cookie
|
||||||
valid: false,
|
for (let index = 0; index < config.sessions.length; index++) {
|
||||||
};
|
let session = config.sessions[index];
|
||||||
console.log(`已添加 #${index} ${jwt.user.name} (旧版cookie)`);
|
let {jwtSession, jwtToken, ds, dsr} = extractCookie(session.cookie);
|
||||||
} catch (e) {
|
if (jwtSession && jwtToken) {
|
||||||
console.error(`解析第${index}个旧版cookie失败: ${e.message}`);
|
// 旧版cookie处理
|
||||||
}
|
try {
|
||||||
} else if (ds) {
|
let jwt = JSON.parse(Buffer.from(jwtToken.split(".")[1], "base64").toString());
|
||||||
// 新版cookie处理
|
this.sessions[jwt.user.name] = {
|
||||||
try {
|
configIndex: index,
|
||||||
let jwt = JSON.parse(Buffer.from(ds.split(".")[1], "base64").toString());
|
jwtSession,
|
||||||
this.sessions[jwt.email] = {
|
jwtToken,
|
||||||
configIndex: index,
|
valid: false,
|
||||||
ds,
|
};
|
||||||
dsr,
|
console.log(`已添加 #${index} ${jwt.user.name} (旧版cookie)`);
|
||||||
valid: false,
|
} catch (e) {
|
||||||
};
|
console.error(`解析第${index}个旧版cookie失败: ${e.message}`);
|
||||||
console.log(`已添加 #${index} ${jwt.email} (新版cookie)`);
|
|
||||||
if (!dsr) {
|
|
||||||
console.warn(`警告: 第${index}个cookie缺少DSR字段。`);
|
|
||||||
}
|
}
|
||||||
} catch (e) {
|
} else if (ds) {
|
||||||
console.error(`解析第${index}个新版cookie失败: ${e.message}`);
|
// 新版cookie处理
|
||||||
|
try {
|
||||||
|
let jwt = JSON.parse(Buffer.from(ds.split(".")[1], "base64").toString());
|
||||||
|
this.sessions[jwt.email] = {
|
||||||
|
configIndex: index,
|
||||||
|
ds,
|
||||||
|
dsr,
|
||||||
|
valid: false,
|
||||||
|
};
|
||||||
|
console.log(`已添加 #${index} ${jwt.email} (新版cookie)`);
|
||||||
|
if (!dsr) {
|
||||||
|
console.warn(`警告: 第${index}个cookie缺少DSR字段。`);
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error(`解析第${index}个新版cookie失败: ${e.message}`);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
console.error(`第${index}个cookie无效，请重新获取。`);
|
||||||
|
console.error(`未检测到有效的DS或stytch_session字段。`);
|
||||||
}
|
}
|
||||||
} else {
|
|
||||||
console.error(`第${index}个cookie无效，请重新获取。`);
|
|
||||||
console.error(`未检测到有效的DS或stytch_session字段。`);
|
|
||||||
}
|
}
|
||||||
|
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 (let username of Object.keys(this.sessions)) {
|
||||||
let session = this.sessions[username];
|
let session = this.sessions[username];
|
||||||
createDirectoryIfNotExists(path.join(__dirname, "browser_profiles", username));
|
createDirectoryIfNotExists(path.join(__dirname, "browser_profiles", username));
|
||||||
|
const isWindows = os.platform() === 'win32';
|
||||||
|
|
||||||
await connect({
|
await connect({
|
||||||
headless: "auto",
|
headless: isWindows ? false : 'auto',
|
||||||
turnstile: true,
|
turnstile: true,
|
||||||
customConfig: {
|
customConfig: {
|
||||||
userDataDir: path.join(__dirname, "browser_profiles", username),
|
userDataDir: path.join(__dirname, "browser_profiles", username),
|
||||||
@@ -107,27 +121,53 @@ class YouProvider {
|
|||||||
},
|
},
|
||||||
})
|
})
|
||||||
.then(async (response) => {
|
.then(async (response) => {
|
||||||
const { page, browser, setTarget } = response;
|
const { page, browser } = response;
|
||||||
await page.setCookie(...getSessionCookie(
|
if (process.env.USE_MANUAL_LOGIN === "true") {
|
||||||
session.jwtSession,
|
console.log(`正在为 session #${session.configIndex} 进行手动登录...`);
|
||||||
session.jwtToken,
|
await page.goto("https://you.com", { timeout: timeout });
|
||||||
session.ds,
|
// 等待页面加载完毕
|
||||||
session.dsr
|
await sleep(5000);
|
||||||
));
|
console.log(`请在打开的浏览器窗口中手动登录 You.com (session #${session.configIndex})`);
|
||||||
|
await this.waitForManualLogin(page);
|
||||||
|
const cookies = await page.cookies();
|
||||||
|
const sessionCookie = this.extractSessionCookie(cookies);
|
||||||
|
if (sessionCookie) {
|
||||||
|
this.sessions[sessionCookie.email] = {
|
||||||
|
...session,
|
||||||
|
...sessionCookie,
|
||||||
|
};
|
||||||
|
delete this.sessions[username];
|
||||||
|
username = sessionCookie.email;
|
||||||
|
session = this.sessions[username];
|
||||||
|
console.log(`成功获取 ${sessionCookie.email} 登录的 cookie`);
|
||||||
|
} else {
|
||||||
|
console.error(`未能获取到 session #${session.configIndex} 有效登录的 cookie`);
|
||||||
|
await browser.close();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
await page.setCookie(...getSessionCookie(
|
||||||
|
session.jwtSession,
|
||||||
|
session.jwtToken,
|
||||||
|
session.ds,
|
||||||
|
session.dsr
|
||||||
|
));
|
||||||
|
await page.goto("https://you.com", { timeout: timeout });
|
||||||
|
}
|
||||||
|
|
||||||
page.goto("https://you.com", { timeout: 60000 });
|
|
||||||
await sleep(5000); // 等待加载完毕
|
await sleep(5000); // 等待加载完毕
|
||||||
|
|
||||||
// 如果遇到盾了就多等一段时间
|
// 如果遇到盾了就多等一段时间
|
||||||
let 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秒内完成人机验证 (${username})`);
|
||||||
page.evaluate(() => {
|
await page.evaluate(() => {
|
||||||
alert("请在30秒内完成人机验证");
|
alert("请在30秒内完成人机验证");
|
||||||
});
|
});
|
||||||
await sleep(30000);
|
await sleep(30000);
|
||||||
}
|
}
|
||||||
|
|
||||||
// get page content and try parse JSON
|
// 验证 cookie 有效性
|
||||||
try {
|
try {
|
||||||
let content = await page.evaluate(() => {
|
let content = await page.evaluate(() => {
|
||||||
return fetch("https://you.com/api/user/getYouProState").then((res) => res.text());
|
return fetch("https://you.com/api/user/getYouProState").then((res) => res.text());
|
||||||
@@ -150,406 +190,447 @@ class YouProvider {
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
.catch((e) => {
|
.catch((e) => {
|
||||||
console.error(`初始化浏览器失败`);
|
console.error(`初始化浏览器失败 (${username})`);
|
||||||
console.error(e);
|
console.error(e);
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
console.log(`验证完毕，有效cookie数量 ${Object.keys(this.sessions).filter((username) => this.sessions[username].valid).length}`);
|
console.log(`验证完毕，有效cookie数量 ${Object.keys(this.sessions).filter((username) => this.sessions[username].valid).length}`);
|
||||||
}
|
}
|
||||||
|
|
||||||
detectBrowser() {
|
async waitForManualLogin(page) {
|
||||||
const platform = os.platform();
|
return new Promise((resolve) => {
|
||||||
let browsers = {
|
const checkLoginStatus = async () => {
|
||||||
'chrome': null,
|
const isLoggedIn = await page.evaluate(() => {
|
||||||
'edge': null
|
return !!document.querySelector('button[aria-label="User menu"]');
|
||||||
};
|
});
|
||||||
|
if (isLoggedIn) {
|
||||||
|
console.log("检测到登录成功");
|
||||||
|
resolve();
|
||||||
|
} else {
|
||||||
|
setTimeout(checkLoginStatus, 1000);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
if (platform === 'win32') {
|
page.on('request', request => {
|
||||||
browsers.chrome = this.findWindowsBrowser('Chrome');
|
if (request.url().includes('https://you.com/api/instrumentation')) {
|
||||||
browsers.edge = this.findWindowsBrowser('Edge');
|
resolve();
|
||||||
} else if (platform === 'darwin') {
|
}
|
||||||
browsers.chrome = this.findMacOSBrowser('Google Chrome');
|
});
|
||||||
browsers.edge = this.findMacOSBrowser('Microsoft Edge');
|
|
||||||
} else if (platform === 'linux') {
|
|
||||||
browsers.chrome = this.findLinuxBrowser('google-chrome');
|
|
||||||
browsers.edge = this.findLinuxBrowser('microsoft-edge');
|
|
||||||
}
|
|
||||||
|
|
||||||
const preferredBrowser = this.preferredBrowser === 'auto' || this.preferredBrowser === undefined
|
checkLoginStatus();
|
||||||
? Object.keys(browsers).find(browser => browsers[browser])
|
});
|
||||||
: this.preferredBrowser;
|
}
|
||||||
|
|
||||||
if (browsers[preferredBrowser]) {
|
extractSessionCookie(cookies) {
|
||||||
console.log(`使用${preferredBrowser === 'chrome' ? 'Chrome' : 'Edge'}浏览器`);
|
const ds = cookies.find(c => c.name === 'DS')?.value;
|
||||||
return browsers[preferredBrowser];
|
const dsr = cookies.find(c => c.name === 'DSR')?.value;
|
||||||
}
|
if (ds) {
|
||||||
|
const jwt = JSON.parse(Buffer.from(ds.split(".")[1], "base64").toString());
|
||||||
|
return {ds, dsr, email: jwt.email};
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
console.error('未找到Chrome或Edge浏览器，请确保已安装其中之一');
|
detectBrowser() {
|
||||||
process.exit(1);
|
const platform = os.platform();
|
||||||
}
|
let browsers = {
|
||||||
|
'chrome': null,
|
||||||
|
'edge': null
|
||||||
|
};
|
||||||
|
|
||||||
findWindowsBrowser(browserName) {
|
if (platform === 'win32') {
|
||||||
const regKeys = {
|
browsers.chrome = this.findWindowsBrowser('Chrome');
|
||||||
'Chrome': ['chrome.exe', 'Google\\Chrome'],
|
browsers.edge = this.findWindowsBrowser('Edge');
|
||||||
'Edge': ['msedge.exe', 'Microsoft\\Edge']
|
} else if (platform === 'darwin') {
|
||||||
};
|
browsers.chrome = this.findMacOSBrowser('Google Chrome');
|
||||||
const [exeName, folderName] = regKeys[browserName];
|
browsers.edge = this.findMacOSBrowser('Microsoft Edge');
|
||||||
|
} else if (platform === 'linux') {
|
||||||
|
browsers.chrome = this.findLinuxBrowser('google-chrome');
|
||||||
|
browsers.edge = this.findLinuxBrowser('microsoft-edge');
|
||||||
|
}
|
||||||
|
|
||||||
const regQuery = (key) => {
|
const preferredBrowser = this.preferredBrowser === 'auto' || this.preferredBrowser === undefined
|
||||||
try {
|
? Object.keys(browsers).find(browser => browsers[browser])
|
||||||
return execSync(`reg query "${key}" /ve`).toString().trim().split('\r\n').pop().split(' ').pop();
|
: this.preferredBrowser;
|
||||||
} catch (error) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
let browserPath = regQuery(`HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\App Paths\\${exeName}`) ||
|
if (browsers[preferredBrowser]) {
|
||||||
regQuery(`HKEY_CURRENT_USER\\Software\\Microsoft\\Windows\\CurrentVersion\\App Paths\\${exeName}`);
|
console.log(`使用${preferredBrowser === 'chrome' ? 'Chrome' : 'Edge'}浏览器`);
|
||||||
|
return browsers[preferredBrowser];
|
||||||
|
}
|
||||||
|
|
||||||
if (browserPath && fs.existsSync(browserPath)) {
|
console.error('未找到Chrome或Edge浏览器，请确保已安装其中之一');
|
||||||
return browserPath;
|
process.exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
const commonPaths = [
|
findWindowsBrowser(browserName) {
|
||||||
`C:\\Program Files\\${browserName}\\Application\\${exeName}`,
|
const regKeys = {
|
||||||
`C:\\Program Files (x86)\\${browserName}\\Application\\${exeName}`,
|
'Chrome': ['chrome.exe', 'Google\\Chrome'],
|
||||||
`${process.env.LOCALAPPDATA}\\${browserName}\\Application\\${exeName}`,
|
'Edge': ['msedge.exe', 'Microsoft\\Edge']
|
||||||
`${process.env.USERPROFILE}\\AppData\\Local\\${browserName}\\Application\\${exeName}`,
|
};
|
||||||
];
|
const [exeName, folderName] = regKeys[browserName];
|
||||||
|
|
||||||
const foundPath = commonPaths.find(path => fs.existsSync(path));
|
const regQuery = (key) => {
|
||||||
if (foundPath) {
|
try {
|
||||||
return foundPath;
|
return execSync(`reg query "${key}" /ve`).toString().trim().split('\r\n').pop().split(' ').pop();
|
||||||
}
|
} catch (error) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
const userAppDataPath = process.env.LOCALAPPDATA || `${process.env.USERPROFILE}\\AppData\\Local`;
|
let browserPath = regQuery(`HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\App Paths\\${exeName}`) ||
|
||||||
const appDataPath = path.join(userAppDataPath, folderName, 'Application');
|
regQuery(`HKEY_CURRENT_USER\\Software\\Microsoft\\Windows\\CurrentVersion\\App Paths\\${exeName}`);
|
||||||
|
|
||||||
if (fs.existsSync(appDataPath)) {
|
if (browserPath && fs.existsSync(browserPath)) {
|
||||||
const files = fs.readdirSync(appDataPath);
|
return browserPath;
|
||||||
const exePath = files.find(file => file.toLowerCase() === exeName.toLowerCase());
|
}
|
||||||
if (exePath) {
|
|
||||||
return path.join(appDataPath, exePath);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return null;
|
const commonPaths = [
|
||||||
}
|
`C:\\Program Files\\${browserName}\\Application\\${exeName}`,
|
||||||
|
`C:\\Program Files (x86)\\${browserName}\\Application\\${exeName}`,
|
||||||
|
`${process.env.LOCALAPPDATA}\\${browserName}\\Application\\${exeName}`,
|
||||||
|
`${process.env.USERPROFILE}\\AppData\\Local\\${browserName}\\Application\\${exeName}`,
|
||||||
|
];
|
||||||
|
|
||||||
findMacOSBrowser(browserName) {
|
const foundPath = commonPaths.find(path => fs.existsSync(path));
|
||||||
const paths = [
|
if (foundPath) {
|
||||||
`/Applications/${browserName}.app/Contents/MacOS/${browserName}`,
|
return foundPath;
|
||||||
`${os.homedir()}/Applications/${browserName}.app/Contents/MacOS/${browserName}`,
|
}
|
||||||
];
|
|
||||||
|
|
||||||
for (const path of paths) {
|
const userAppDataPath = process.env.LOCALAPPDATA || `${process.env.USERPROFILE}\\AppData\\Local`;
|
||||||
if (fs.existsSync(path)) {
|
const appDataPath = path.join(userAppDataPath, folderName, 'Application');
|
||||||
return path;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return null;
|
if (fs.existsSync(appDataPath)) {
|
||||||
}
|
const files = fs.readdirSync(appDataPath);
|
||||||
|
const exePath = files.find(file => file.toLowerCase() === exeName.toLowerCase());
|
||||||
|
if (exePath) {
|
||||||
|
return path.join(appDataPath, exePath);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
findLinuxBrowser(browserName) {
|
return null;
|
||||||
try {
|
}
|
||||||
return execSync(`which ${browserName}`).toString().trim();
|
|
||||||
} catch (error) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async getCompletion({username, messages, stream = false, proxyModel, useCustomMode = false}) {
|
findMacOSBrowser(browserName) {
|
||||||
const session = this.sessions[username];
|
const paths = [
|
||||||
if (!session || !session.valid) {
|
`/Applications/${browserName}.app/Contents/MacOS/${browserName}`,
|
||||||
throw new Error(`用户 ${username} 的会话无效`);
|
`${os.homedir()}/Applications/${browserName}.app/Contents/MacOS/${browserName}`,
|
||||||
}
|
];
|
||||||
|
|
||||||
const { page, browser } = session;
|
for (const path of paths) {
|
||||||
const emitter = new EventEmitter();
|
if (fs.existsSync(path)) {
|
||||||
// 处理模式轮换逻辑
|
return path;
|
||||||
if (this.isCustomModeEnabled && this.isRotationEnabled) {
|
}
|
||||||
this.switchCounter++;
|
}
|
||||||
this.requestsInCurrentMode++;
|
|
||||||
console.log(`当前模式: ${this.currentMode}, 本模式下的请求次数: ${this.requestsInCurrentMode}, 距离下次切换还有 ${this.switchThreshold - this.switchCounter} 次请求`);
|
|
||||||
|
|
||||||
if (this.switchCounter >= this.switchThreshold) {
|
return null;
|
||||||
this.switchMode();
|
}
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 根据轮换状态决定是否使用自定义模式
|
findLinuxBrowser(browserName) {
|
||||||
const effectiveUseCustomMode = this.isRotationEnabled ? (this.currentMode === "custom") : useCustomMode;
|
try {
|
||||||
|
return execSync(`which ${browserName}`).toString().trim();
|
||||||
|
} catch (error) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// 检查页面是否已经加载完成
|
async getCompletion({username, messages, stream = false, proxyModel, useCustomMode = false}) {
|
||||||
const isLoaded = await page.evaluate(() => {
|
const session = this.sessions[username];
|
||||||
return document.readyState === 'complete' || document.readyState === 'interactive';
|
if (!session || !session.valid) {
|
||||||
});
|
throw new Error(`用户 ${username} 的会话无效`);
|
||||||
|
}
|
||||||
|
|
||||||
if (!isLoaded) {
|
const {page, browser} = session;
|
||||||
console.log('页面尚未加载完成，等待加载...');
|
const emitter = new EventEmitter();
|
||||||
await page.waitForNavigation({ waitUntil: 'domcontentloaded', timeout: 10000 }).catch(() => {
|
// 处理模式轮换逻辑
|
||||||
console.log('页面加载超时，继续执行');
|
if (this.isCustomModeEnabled && this.isRotationEnabled) {
|
||||||
});
|
this.switchCounter++;
|
||||||
}
|
this.requestsInCurrentMode++;
|
||||||
|
console.log(`当前模式: ${this.currentMode}, 本模式下的请求次数: ${this.requestsInCurrentMode}, 距离下次切换还有 ${this.switchThreshold - this.switchCounter} 次请求`);
|
||||||
|
|
||||||
await page.goto("https://you.com/?chatMode=default", { waitUntil: "domcontentloaded" });
|
if (this.switchCounter >= this.switchThreshold) {
|
||||||
|
this.switchMode();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// 计算用户消息长度
|
// 根据轮换状态决定是否使用自定义模式
|
||||||
let userMessage = [{ question: "", answer: "" }];
|
const effectiveUseCustomMode = this.isRotationEnabled ? (this.currentMode === "custom") : useCustomMode;
|
||||||
let userQuery = "";
|
|
||||||
let lastUpdate = true;
|
|
||||||
|
|
||||||
messages.forEach((msg) => {
|
// 检查页面是否已经加载完成
|
||||||
if (msg.role == "system" || msg.role == "user") {
|
const isLoaded = await page.evaluate(() => {
|
||||||
if (lastUpdate) {
|
return document.readyState === 'complete' || document.readyState === 'interactive';
|
||||||
userMessage[userMessage.length - 1].question += msg.content + "\n";
|
});
|
||||||
} 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") {
|
|
||||||
if (!lastUpdate) {
|
|
||||||
userMessage[userMessage.length - 1].answer += msg.content + "\n";
|
|
||||||
} else if (userMessage[userMessage.length - 1].answer == "") {
|
|
||||||
userMessage[userMessage.length - 1].answer += msg.content + "\n";
|
|
||||||
} else {
|
|
||||||
userMessage.push({ question: "", answer: msg.content + "\n" });
|
|
||||||
}
|
|
||||||
lastUpdate = false;
|
|
||||||
}
|
|
||||||
});
|
|
||||||
userQuery = userMessage[userMessage.length - 1].question;
|
|
||||||
|
|
||||||
// 检查该session是否已经创建对应模型的对应user chat mode
|
if (!isLoaded) {
|
||||||
let userChatModeId = "custom";
|
console.log('页面尚未加载完成，等待加载...');
|
||||||
if (effectiveUseCustomMode) {
|
await page.waitForNavigation({waitUntil: 'domcontentloaded', timeout: 10000}).catch(() => {
|
||||||
if (!this.config.sessions[session.configIndex].user_chat_mode_id) {
|
console.log('页面加载超时，继续执行');
|
||||||
this.config.sessions[session.configIndex].user_chat_mode_id = {};
|
});
|
||||||
}
|
}
|
||||||
if (!this.config.sessions[session.configIndex].user_chat_mode_id[proxyModel]) {
|
|
||||||
// 创建新的user chat mode
|
|
||||||
let userChatMode = await page.evaluate(
|
|
||||||
async (proxyModel, proxyModelName) => {
|
|
||||||
return fetch("https://you.com/api/custom_assistants/assistants", {
|
|
||||||
method: "POST",
|
|
||||||
body: JSON.stringify({
|
|
||||||
aiModel: proxyModel,
|
|
||||||
hasLiveWebAccess: false,
|
|
||||||
hasPersonalization: false,
|
|
||||||
hideInstructions: true,
|
|
||||||
includeFollowUps: false,
|
|
||||||
instructions: "Please review the attached prompt",
|
|
||||||
instructionsSummary:"",
|
|
||||||
isUserOwned: true,
|
|
||||||
name: proxyModelName,
|
|
||||||
visibility: "private",
|
|
||||||
}),
|
|
||||||
headers: {
|
|
||||||
"Content-Type": "application/json",
|
|
||||||
},
|
|
||||||
}).then((res) => res.json());
|
|
||||||
},
|
|
||||||
proxyModel,
|
|
||||||
uuidv4().substring(0, 4)
|
|
||||||
);
|
|
||||||
if(userChatMode.chat_mode_id){
|
|
||||||
this.config.sessions[session.configIndex].user_chat_mode_id[proxyModel] = userChatMode.chat_mode_id;
|
|
||||||
// 写回config
|
|
||||||
fs.writeFileSync("./config.mjs", "export const config = " + JSON.stringify(this.config, null, 4));
|
|
||||||
}else{
|
|
||||||
if(userChatMode.error) console.log(userChatMode.error)
|
|
||||||
console.log("Failed to create user chat mode, will use default mode instead.");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
userChatModeId = this.config.sessions[session.configIndex].user_chat_mode_id[proxyModel];
|
|
||||||
} else {
|
|
||||||
console.log("Custom mode is disabled, using default mode.");
|
|
||||||
}
|
|
||||||
|
|
||||||
// 生成随机文件名
|
await page.goto("https://you.com/?chatMode=default", {waitUntil: "domcontentloaded"});
|
||||||
function generateRandomFileName(length) {
|
|
||||||
const validChars = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789_-';
|
|
||||||
let result = '';
|
|
||||||
for (let i = 0; i < length; i++) {
|
|
||||||
result += validChars.charAt(Math.floor(Math.random() * validChars.length));
|
|
||||||
}
|
|
||||||
return result + '.docx';
|
|
||||||
}
|
|
||||||
|
|
||||||
// 生成随机长度（6-16）的文件名
|
// 计算用户消息长度
|
||||||
const randomFileName = generateRandomFileName(Math.floor(Math.random() * 11) + 6);
|
let userMessage = [{question: "", answer: ""}];
|
||||||
console.log(`Generated random file name: ${randomFileName}`);
|
let userQuery = "";
|
||||||
|
let lastUpdate = true;
|
||||||
|
|
||||||
// 试算用户消息长度
|
messages.forEach((msg) => {
|
||||||
if (encodeURIComponent(JSON.stringify(userMessage)).length + encodeURIComponent(userQuery).length > 32000) {
|
if (msg.role == "system" || msg.role == "user") {
|
||||||
console.log("Using file upload mode");
|
if (lastUpdate) {
|
||||||
|
userMessage[userMessage.length - 1].question += msg.content + "\n";
|
||||||
|
} 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") {
|
||||||
|
if (!lastUpdate) {
|
||||||
|
userMessage[userMessage.length - 1].answer += msg.content + "\n";
|
||||||
|
} else if (userMessage[userMessage.length - 1].answer == "") {
|
||||||
|
userMessage[userMessage.length - 1].answer += msg.content + "\n";
|
||||||
|
} else {
|
||||||
|
userMessage.push({question: "", answer: msg.content + "\n"});
|
||||||
|
}
|
||||||
|
lastUpdate = false;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
userQuery = userMessage[userMessage.length - 1].question;
|
||||||
|
|
||||||
// 应用格式化逻辑
|
// 检查该session是否已经创建对应模型的对应user chat mode
|
||||||
const formattedMessages = formatMessages(messages);
|
let userChatModeId = "custom";
|
||||||
|
if (effectiveUseCustomMode) {
|
||||||
|
if (!this.config.sessions[session.configIndex].user_chat_mode_id) {
|
||||||
|
this.config.sessions[session.configIndex].user_chat_mode_id = {};
|
||||||
|
}
|
||||||
|
if (!this.config.sessions[session.configIndex].user_chat_mode_id[proxyModel]) {
|
||||||
|
// 创建新的user chat mode
|
||||||
|
let userChatMode = await page.evaluate(
|
||||||
|
async (proxyModel, proxyModelName) => {
|
||||||
|
return fetch("https://you.com/api/custom_assistants/assistants", {
|
||||||
|
method: "POST",
|
||||||
|
body: JSON.stringify({
|
||||||
|
aiModel: proxyModel,
|
||||||
|
hasLiveWebAccess: false,
|
||||||
|
hasPersonalization: false,
|
||||||
|
hideInstructions: true,
|
||||||
|
includeFollowUps: false,
|
||||||
|
instructions: "Please review the attached prompt",
|
||||||
|
instructionsSummary: "",
|
||||||
|
isUserOwned: true,
|
||||||
|
name: proxyModelName,
|
||||||
|
visibility: "private",
|
||||||
|
}),
|
||||||
|
headers: {
|
||||||
|
"Content-Type": "application/json",
|
||||||
|
},
|
||||||
|
}).then((res) => res.json());
|
||||||
|
},
|
||||||
|
proxyModel,
|
||||||
|
uuidv4().substring(0, 4)
|
||||||
|
);
|
||||||
|
if (userChatMode.chat_mode_id) {
|
||||||
|
this.config.sessions[session.configIndex].user_chat_mode_id[proxyModel] = userChatMode.chat_mode_id;
|
||||||
|
// 写回config
|
||||||
|
fs.writeFileSync("./config.mjs", "export const config = " + JSON.stringify(this.config, null, 4));
|
||||||
|
} else {
|
||||||
|
if (userChatMode.error) console.log(userChatMode.error)
|
||||||
|
console.log("Failed to create user chat mode, will use default mode instead.");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
userChatModeId = this.config.sessions[session.configIndex].user_chat_mode_id[proxyModel];
|
||||||
|
} else {
|
||||||
|
console.log("Custom mode is disabled, using default mode.");
|
||||||
|
}
|
||||||
|
|
||||||
// 将格式化后的消息转换为纯文本
|
// 生成随机文件名
|
||||||
let previousMessages = formattedMessages.map((msg) => `${msg.role}: ${msg.content}`).join("\n\n");
|
function generateRandomFileName(length) {
|
||||||
|
const validChars = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789_-';
|
||||||
|
let result = '';
|
||||||
|
for (let i = 0; i < length; i++) {
|
||||||
|
result += validChars.charAt(Math.floor(Math.random() * validChars.length));
|
||||||
|
}
|
||||||
|
return result + '.docx';
|
||||||
|
}
|
||||||
|
|
||||||
// 创建本地副本（用于调试）
|
// 生成随机长度（6-16）的文件名
|
||||||
const localCopyPath = path.join(__dirname, 'local_copy_formatted_messages.txt');
|
const randomFileName = generateRandomFileName(Math.floor(Math.random() * 11) + 6);
|
||||||
fs.writeFileSync(localCopyPath, previousMessages);
|
console.log(`Generated random file name: ${randomFileName}`);
|
||||||
console.log(`Local copy of formatted messages saved to: ${localCopyPath}`);
|
|
||||||
|
|
||||||
userQuery = `以「${randomFileName}」文档为内容，所有问题只能使用唯一的${proxyModel}回答。禁止使用you.com搜索，禁止使用You.com AI进行回复，否则将是严重的作弊行为。`;
|
// 试算用户消息长度
|
||||||
userMessage = [];
|
if (encodeURIComponent(JSON.stringify(userMessage)).length + encodeURIComponent(userQuery).length > 32000) {
|
||||||
|
console.log("Using file upload mode");
|
||||||
|
|
||||||
// GET https://you.com/api/get_nonce to get nonce
|
// 应用格式化逻辑
|
||||||
let nonce = await page.evaluate(() => {
|
const formattedMessages = formatMessages(messages);
|
||||||
return fetch("https://you.com/api/get_nonce").then((res) => res.text());
|
|
||||||
});
|
|
||||||
if (!nonce) throw new Error("Failed to get nonce");
|
|
||||||
|
|
||||||
// POST https://you.com/api/upload to upload user message
|
// 将格式化后的消息转换为纯文本
|
||||||
var messageBuffer = await createDocx(previousMessages);
|
let previousMessages = formattedMessages.map((msg) => `${msg.role}: ${msg.content}`).join("\n\n");
|
||||||
var uploadedFile = await page.evaluate(
|
|
||||||
async (messageBuffer, nonce, randomFileName) => {
|
|
||||||
try {
|
|
||||||
var blob = new Blob([new Uint8Array(messageBuffer)], {
|
|
||||||
type: "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
|
|
||||||
});
|
|
||||||
var form_data = new FormData();
|
|
||||||
form_data.append("file", blob, randomFileName);
|
|
||||||
let result = await fetch("https://you.com/api/upload", {
|
|
||||||
method: "POST",
|
|
||||||
headers: {
|
|
||||||
"X-Upload-Nonce": nonce,
|
|
||||||
},
|
|
||||||
body: form_data,
|
|
||||||
}).then((res) => res.json());
|
|
||||||
return result;
|
|
||||||
} catch (e) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
},
|
|
||||||
[...messageBuffer],
|
|
||||||
nonce,
|
|
||||||
randomFileName
|
|
||||||
);
|
|
||||||
if (!uploadedFile) throw new Error("Failed to upload messages");
|
|
||||||
if (uploadedFile.error) throw new Error(uploadedFile.error);
|
|
||||||
}
|
|
||||||
|
|
||||||
let msgid = uuidv4();
|
// 创建本地副本（用于调试）
|
||||||
let traceId = uuidv4();
|
const localCopyPath = path.join(__dirname, 'local_copy_formatted_messages.txt');
|
||||||
|
fs.writeFileSync(localCopyPath, previousMessages);
|
||||||
|
console.log(`Local copy of formatted messages saved to: ${localCopyPath}`);
|
||||||
|
|
||||||
// expose function to receive youChatToken
|
userQuery = `以「${randomFileName}」文档为内容，所有问题只能使用唯一的${proxyModel}回答。禁止使用you.com搜索，禁止使用You.com AI进行回复，否则将是严重的作弊行为。`;
|
||||||
var finalResponse = "";
|
userMessage = [];
|
||||||
page.exposeFunction("callback" + traceId, async (event, data) => {
|
|
||||||
switch (event) {
|
|
||||||
case "youChatToken":
|
|
||||||
data = JSON.parse(data);
|
|
||||||
process.stdout.write(data.youChatToken);
|
|
||||||
if (stream) {
|
|
||||||
emitter.emit("completion", traceId, data.youChatToken);
|
|
||||||
} else {
|
|
||||||
finalResponse += data.youChatToken;
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case "done":
|
|
||||||
console.log("请求结束");
|
|
||||||
if (stream) {
|
|
||||||
emitter.emit("end");
|
|
||||||
} else {
|
|
||||||
emitter.emit("completion", traceId, finalResponse);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case "error":
|
|
||||||
throw new Error(data);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
// proxy response
|
// GET https://you.com/api/get_nonce to get nonce
|
||||||
var req_param = new URLSearchParams();
|
let nonce = await page.evaluate(() => {
|
||||||
req_param.append("page", "1");
|
return fetch("https://you.com/api/get_nonce").then((res) => res.text());
|
||||||
req_param.append("count", "10");
|
});
|
||||||
req_param.append("safeSearch", "Off");
|
if (!nonce) throw new Error("Failed to get nonce");
|
||||||
req_param.append("q", userQuery);
|
|
||||||
req_param.append("chatId", traceId);
|
|
||||||
req_param.append("traceId", `${traceId}|${msgid}|${new Date().toISOString()}`);
|
|
||||||
req_param.append("conversationTurnId", msgid);
|
|
||||||
if (userChatModeId == "custom") req_param.append("selectedAiModel", proxyModel);
|
|
||||||
req_param.append("selectedChatMode", userChatModeId);
|
|
||||||
req_param.append("pastChatLength", userMessage.length);
|
|
||||||
req_param.append("queryTraceId", traceId);
|
|
||||||
req_param.append("use_personalization_extraction", "false");
|
|
||||||
req_param.append("domain", "youchat");
|
|
||||||
req_param.append("mkt", "ja-JP");
|
|
||||||
if (uploadedFile)
|
|
||||||
req_param.append("userFiles", JSON.stringify([{ user_filename: randomFileName, filename: uploadedFile.filename, size: messageBuffer.length }]));
|
|
||||||
req_param.append("chat", JSON.stringify(userMessage));
|
|
||||||
var 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(
|
|
||||||
async (url, traceId) => {
|
|
||||||
var evtSource = new EventSource(url);
|
|
||||||
var callbackName = "callback" + traceId;
|
|
||||||
evtSource.onerror = (error) => {
|
|
||||||
window[callbackName]("error", error);
|
|
||||||
evtSource.close();
|
|
||||||
};
|
|
||||||
evtSource.addEventListener(
|
|
||||||
"youChatToken",
|
|
||||||
(event) => {
|
|
||||||
var data = event.data;
|
|
||||||
window[callbackName]("youChatToken", data);
|
|
||||||
},
|
|
||||||
false
|
|
||||||
);
|
|
||||||
evtSource.addEventListener(
|
|
||||||
"done",
|
|
||||||
(event) => {
|
|
||||||
window[callbackName]("done", "");
|
|
||||||
evtSource.close();
|
|
||||||
fetch("https://you.com/api/chat/deleteChat", {
|
|
||||||
headers: {
|
|
||||||
"content-type": "application/json",
|
|
||||||
},
|
|
||||||
body: JSON.stringify({ chatId: traceId }),
|
|
||||||
method: "DELETE",
|
|
||||||
});
|
|
||||||
},
|
|
||||||
false
|
|
||||||
);
|
|
||||||
|
|
||||||
evtSource.onmessage = (event) => {
|
// POST https://you.com/api/upload to upload user message
|
||||||
const data = JSON.parse(event.data);
|
var messageBuffer = await createDocx(previousMessages);
|
||||||
if (data.youChatToken) {
|
var uploadedFile = await page.evaluate(
|
||||||
window[callbackName](youChatToken);
|
async (messageBuffer, nonce, randomFileName) => {
|
||||||
}
|
try {
|
||||||
};
|
var blob = new Blob([new Uint8Array(messageBuffer)], {
|
||||||
// 注册退出函数
|
type: "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
|
||||||
window["exit" + traceId] = () => {
|
});
|
||||||
evtSource.close();
|
var form_data = new FormData();
|
||||||
};
|
form_data.append("file", blob, randomFileName);
|
||||||
},
|
let result = await fetch("https://you.com/api/upload", {
|
||||||
url,
|
method: "POST",
|
||||||
traceId
|
headers: {
|
||||||
);
|
"X-Upload-Nonce": nonce,
|
||||||
} catch (error) {
|
},
|
||||||
console.error("评估过程中出错:", error);
|
body: form_data,
|
||||||
emitter.emit("error", error);
|
}).then((res) => res.json());
|
||||||
return { completion: emitter, cancel: () => {} };
|
return result;
|
||||||
}
|
} catch (e) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
[...messageBuffer],
|
||||||
|
nonce,
|
||||||
|
randomFileName
|
||||||
|
);
|
||||||
|
if (!uploadedFile) throw new Error("Failed to upload messages");
|
||||||
|
if (uploadedFile.error) throw new Error(uploadedFile.error);
|
||||||
|
}
|
||||||
|
|
||||||
const cancel = () => {
|
let msgid = uuidv4();
|
||||||
page?.evaluate((traceId) => {
|
let traceId = uuidv4();
|
||||||
window["exit" + traceId]();
|
|
||||||
}, traceId).catch(console.error);
|
|
||||||
};
|
|
||||||
|
|
||||||
return { completion: emitter, cancel };
|
// expose function to receive youChatToken
|
||||||
}
|
var finalResponse = "";
|
||||||
|
page.exposeFunction("callback" + traceId, async (event, data) => {
|
||||||
|
switch (event) {
|
||||||
|
case "youChatToken":
|
||||||
|
data = JSON.parse(data);
|
||||||
|
process.stdout.write(data.youChatToken);
|
||||||
|
if (stream) {
|
||||||
|
emitter.emit("completion", traceId, data.youChatToken);
|
||||||
|
} else {
|
||||||
|
finalResponse += data.youChatToken;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case "done":
|
||||||
|
console.log("请求结束");
|
||||||
|
if (stream) {
|
||||||
|
emitter.emit("end");
|
||||||
|
} else {
|
||||||
|
emitter.emit("completion", traceId, finalResponse);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case "error":
|
||||||
|
throw new Error(data);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// proxy response
|
||||||
|
var req_param = new URLSearchParams();
|
||||||
|
req_param.append("page", "1");
|
||||||
|
req_param.append("count", "10");
|
||||||
|
req_param.append("safeSearch", "Off");
|
||||||
|
req_param.append("q", userQuery);
|
||||||
|
req_param.append("chatId", traceId);
|
||||||
|
req_param.append("traceId", `${traceId}|${msgid}|${new Date().toISOString()}`);
|
||||||
|
req_param.append("conversationTurnId", msgid);
|
||||||
|
if (userChatModeId == "custom") req_param.append("selectedAiModel", proxyModel);
|
||||||
|
req_param.append("selectedChatMode", userChatModeId);
|
||||||
|
req_param.append("pastChatLength", userMessage.length);
|
||||||
|
req_param.append("queryTraceId", traceId);
|
||||||
|
req_param.append("use_personalization_extraction", "false");
|
||||||
|
req_param.append("domain", "youchat");
|
||||||
|
req_param.append("mkt", "ja-JP");
|
||||||
|
if (uploadedFile)
|
||||||
|
req_param.append("userFiles", JSON.stringify([{
|
||||||
|
user_filename: randomFileName,
|
||||||
|
filename: uploadedFile.filename,
|
||||||
|
size: messageBuffer.length
|
||||||
|
}]));
|
||||||
|
req_param.append("chat", JSON.stringify(userMessage));
|
||||||
|
var 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(
|
||||||
|
async (url, traceId) => {
|
||||||
|
var evtSource = new EventSource(url);
|
||||||
|
var callbackName = "callback" + traceId;
|
||||||
|
evtSource.onerror = (error) => {
|
||||||
|
window[callbackName]("error", error);
|
||||||
|
evtSource.close();
|
||||||
|
};
|
||||||
|
evtSource.addEventListener(
|
||||||
|
"youChatToken",
|
||||||
|
(event) => {
|
||||||
|
var data = event.data;
|
||||||
|
window[callbackName]("youChatToken", data);
|
||||||
|
},
|
||||||
|
false
|
||||||
|
);
|
||||||
|
evtSource.addEventListener(
|
||||||
|
"done",
|
||||||
|
(event) => {
|
||||||
|
window[callbackName]("done", "");
|
||||||
|
evtSource.close();
|
||||||
|
fetch("https://you.com/api/chat/deleteChat", {
|
||||||
|
headers: {
|
||||||
|
"content-type": "application/json",
|
||||||
|
},
|
||||||
|
body: JSON.stringify({chatId: traceId}),
|
||||||
|
method: "DELETE",
|
||||||
|
});
|
||||||
|
},
|
||||||
|
false
|
||||||
|
);
|
||||||
|
|
||||||
|
evtSource.onmessage = (event) => {
|
||||||
|
const data = JSON.parse(event.data);
|
||||||
|
if (data.youChatToken) {
|
||||||
|
window[callbackName](youChatToken);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
// 注册退出函数
|
||||||
|
window["exit" + traceId] = () => {
|
||||||
|
evtSource.close();
|
||||||
|
};
|
||||||
|
},
|
||||||
|
url,
|
||||||
|
traceId
|
||||||
|
);
|
||||||
|
} catch (error) {
|
||||||
|
console.error("评估过程中出错:", error);
|
||||||
|
emitter.emit("error", error);
|
||||||
|
return {
|
||||||
|
completion: emitter, cancel: () => {
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
const cancel = () => {
|
||||||
|
page?.evaluate((traceId) => {
|
||||||
|
window["exit" + traceId]();
|
||||||
|
}, traceId).catch(console.error);
|
||||||
|
};
|
||||||
|
|
||||||
|
return {completion: emitter, cancel};
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
export default YouProvider;
|
export default YouProvider;
|
||||||
|
|||||||
Reference in New Issue
Block a user