Upload 20 files
Browse files- .github/workflows/publish-docker-image.yml +60 -0
- data/admin.example.json +7 -0
- data/api_keys.example.json +5 -0
- data/invalid_cookies.example.json +3 -0
- scripts/create-admin.js +71 -0
- src/app.js +154 -0
- src/config/config.js +50 -0
- src/middleware/auth.js +44 -0
- src/models/admin.js +114 -0
- src/proto/message.js +1918 -0
- src/proto/message.proto +44 -0
- src/public/index.html +921 -0
- src/public/login.html +243 -0
- src/routes/index.js +8 -0
- src/routes/v1.js +931 -0
- src/utils/cookieRefresher.js +750 -0
- src/utils/envChecker.js +76 -0
- src/utils/extractCookieFromCsv.js +171 -0
- src/utils/keyManager.js +406 -0
- src/utils/utils.js +200 -0
.github/workflows/publish-docker-image.yml
ADDED
@@ -0,0 +1,60 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
name: Publish Docker Image
|
2 |
+
|
3 |
+
on:
|
4 |
+
workflow_dispatch:
|
5 |
+
|
6 |
+
env:
|
7 |
+
REGISTRY: ghcr.io
|
8 |
+
IMAGE_NAME: ${{ github.repository }}
|
9 |
+
MAJOR: 1
|
10 |
+
MINOR: 0
|
11 |
+
PATCH: ${{github.run_number}}
|
12 |
+
|
13 |
+
jobs:
|
14 |
+
build-and-push-image:
|
15 |
+
runs-on: ubuntu-latest
|
16 |
+
|
17 |
+
# Sets the permissions granted to the `GITHUB_TOKEN` for the actions in this job.
|
18 |
+
permissions:
|
19 |
+
contents: read
|
20 |
+
packages: write
|
21 |
+
attestations: write
|
22 |
+
id-token: write
|
23 |
+
|
24 |
+
steps:
|
25 |
+
- name: Checkout repository
|
26 |
+
uses: actions/checkout@v4
|
27 |
+
|
28 |
+
# Uses the `docker/login-action` action to log in to the Container registry registry using the account and password that will publish the packages. Once published, the packages are scoped to the account defined here.
|
29 |
+
- name: Log in to the Container registry
|
30 |
+
uses: docker/login-action@v3
|
31 |
+
with:
|
32 |
+
registry: ${{ env.REGISTRY }}
|
33 |
+
username: ${{ github.actor }}
|
34 |
+
password: ${{ secrets.GITHUB_TOKEN }}
|
35 |
+
|
36 |
+
# This step uses [docker/metadata-action](https://github.com/docker/metadata-action#about) to extract tags and labels that will be applied to the specified image. The `id` "meta" allows the output of this step to be referenced in a subsequent step. The `images` value provides the base name for the tags and labels.
|
37 |
+
- name: Extract metadata (tags, labels) for Docker
|
38 |
+
id: meta
|
39 |
+
uses: docker/metadata-action@v5
|
40 |
+
with:
|
41 |
+
images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
|
42 |
+
tags: |
|
43 |
+
type=ref,event=branch
|
44 |
+
type=raw,value=latest,enable=${{ github.ref == 'refs/heads/main' }}
|
45 |
+
type=raw,value=v${{ env.MAJOR }}.${{ env.MINOR }}.${{ env.PATCH }}
|
46 |
+
|
47 |
+
# This step uses the `docker/build-push-action` action to build the image, based on your repository's `Dockerfile`. If the build succeeds, it pushes the image to GitHub Packages.
|
48 |
+
# It uses the `context` parameter to define the build's context as the set of files located in the specified path. For more information, see "[Usage](https://github.com/docker/build-push-action#usage)" in the README of the `docker/build-push-action` repository.
|
49 |
+
# It uses the `tags` and `labels` parameters to tag and label the image with the output from the "meta" step.
|
50 |
+
- name: Build and push Docker image
|
51 |
+
id: push
|
52 |
+
uses: docker/build-push-action@v6
|
53 |
+
with:
|
54 |
+
context: .
|
55 |
+
push: true
|
56 |
+
tags: ${{ steps.meta.outputs.tags }}
|
57 |
+
labels: ${{ steps.meta.outputs.labels }}
|
58 |
+
subject-digest: ${{ steps.push.outputs.digest }}
|
59 |
+
push-to-registry: true
|
60 |
+
|
data/admin.example.json
ADDED
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"admin": {
|
3 |
+
"username": "your_admin_username",
|
4 |
+
"salt": "your_generated_salt",
|
5 |
+
"hash": "your_password_hash"
|
6 |
+
}
|
7 |
+
}
|
data/api_keys.example.json
ADDED
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"sk-text@example": [
|
3 |
+
"user_XXXXXXXXXXXX%3A%3AeyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX"
|
4 |
+
]
|
5 |
+
}
|
data/invalid_cookies.example.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
[
|
2 |
+
"user_XXXXXXXXXXXX%3A%3AeyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX"
|
3 |
+
]
|
scripts/create-admin.js
ADDED
@@ -0,0 +1,71 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
const fs = require('fs');
|
2 |
+
const path = require('path');
|
3 |
+
const crypto = require('crypto');
|
4 |
+
const readline = require('readline');
|
5 |
+
|
6 |
+
const ADMIN_FILE = path.join(__dirname, '../data/admin.json');
|
7 |
+
|
8 |
+
// 创建readline接口
|
9 |
+
const rl = readline.createInterface({
|
10 |
+
input: process.stdin,
|
11 |
+
output: process.stdout
|
12 |
+
});
|
13 |
+
|
14 |
+
// 生成盐值
|
15 |
+
function generateSalt() {
|
16 |
+
return crypto.randomBytes(16).toString('hex');
|
17 |
+
}
|
18 |
+
|
19 |
+
// 使用盐值哈希密码
|
20 |
+
function hashPassword(password, salt) {
|
21 |
+
return crypto.pbkdf2Sync(password, salt, 1000, 64, 'sha512').toString('hex');
|
22 |
+
}
|
23 |
+
|
24 |
+
// 提示用户输入
|
25 |
+
function question(query) {
|
26 |
+
return new Promise((resolve) => {
|
27 |
+
rl.question(query, resolve);
|
28 |
+
});
|
29 |
+
}
|
30 |
+
|
31 |
+
async function main() {
|
32 |
+
try {
|
33 |
+
console.log('创建管理员账户\n');
|
34 |
+
|
35 |
+
// 获取用户输入
|
36 |
+
const username = await question('请输入管理员用户名: ');
|
37 |
+
const password = await question('请输入管理员密码: ');
|
38 |
+
|
39 |
+
// 生成盐值和密码哈希
|
40 |
+
const salt = generateSalt();
|
41 |
+
const hash = hashPassword(password, salt);
|
42 |
+
|
43 |
+
// 创建管理员数据
|
44 |
+
const adminData = {
|
45 |
+
admin: {
|
46 |
+
username,
|
47 |
+
salt,
|
48 |
+
hash
|
49 |
+
}
|
50 |
+
};
|
51 |
+
|
52 |
+
// 确保data目录存在
|
53 |
+
const dataDir = path.dirname(ADMIN_FILE);
|
54 |
+
if (!fs.existsSync(dataDir)) {
|
55 |
+
fs.mkdirSync(dataDir, { recursive: true });
|
56 |
+
}
|
57 |
+
|
58 |
+
// 写入文件
|
59 |
+
fs.writeFileSync(ADMIN_FILE, JSON.stringify(adminData, null, 2));
|
60 |
+
|
61 |
+
console.log('\n管理员账户创建成功!');
|
62 |
+
console.log('请妥善保管账户信息,不要将admin.json文件提交到版本控制系统。');
|
63 |
+
|
64 |
+
} catch (error) {
|
65 |
+
console.error('创建管理员账户失败:', error);
|
66 |
+
} finally {
|
67 |
+
rl.close();
|
68 |
+
}
|
69 |
+
}
|
70 |
+
|
71 |
+
main();
|
src/app.js
ADDED
@@ -0,0 +1,154 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
// 加载环境变量
|
2 |
+
require('dotenv').config();
|
3 |
+
|
4 |
+
// 环境检查
|
5 |
+
const envChecker = require('./utils/envChecker');
|
6 |
+
console.log('启动前检查环境配置...');
|
7 |
+
envChecker.enforceEnvCheck();
|
8 |
+
|
9 |
+
const express = require('express');
|
10 |
+
const morgan = require('morgan');
|
11 |
+
const path = require('path');
|
12 |
+
const cron = require('node-cron');
|
13 |
+
const app = express();
|
14 |
+
|
15 |
+
const config = require('./config/config');
|
16 |
+
const routes = require('./routes');
|
17 |
+
const keyManager = require('./utils/keyManager');
|
18 |
+
const cookieRefresher = require('./utils/cookieRefresher');
|
19 |
+
const authMiddleware = require('./middleware/auth');
|
20 |
+
|
21 |
+
// 初始化API Keys
|
22 |
+
console.log('初始化API Keys...');
|
23 |
+
keyManager.initializeApiKeys();
|
24 |
+
|
25 |
+
// 输出最终的API Keys配置
|
26 |
+
console.log('最终API Keys配置:', JSON.stringify(keyManager.getAllApiKeys().reduce((obj, key) => {
|
27 |
+
obj[key] = keyManager.getAllCookiesForApiKey(key);
|
28 |
+
return obj;
|
29 |
+
}, {}), null, 2));
|
30 |
+
|
31 |
+
// 添加CORS支持
|
32 |
+
app.use((req, res, next) => {
|
33 |
+
res.header('Access-Control-Allow-Origin', '*');
|
34 |
+
res.header('Access-Control-Allow-Headers', 'Origin, X-Requested-With, Content-Type, Accept, Authorization');
|
35 |
+
res.header('Access-Control-Allow-Methods', 'GET, POST, PUT, DELETE, OPTIONS');
|
36 |
+
|
37 |
+
if (req.method === 'OPTIONS') {
|
38 |
+
return res.status(200).end();
|
39 |
+
}
|
40 |
+
|
41 |
+
next();
|
42 |
+
});
|
43 |
+
|
44 |
+
app.use(express.json({ limit: '50mb' }));
|
45 |
+
app.use(express.urlencoded({ extended: true, limit: '50mb' }));
|
46 |
+
|
47 |
+
app.use(morgan(process.env.MORGAN_FORMAT ?? 'tiny'));
|
48 |
+
|
49 |
+
// 添加静态文件支持
|
50 |
+
app.use(express.static(path.join(__dirname, 'public')));
|
51 |
+
|
52 |
+
// 添加根路由,重定向到登录页面
|
53 |
+
app.get('/', (req, res) => {
|
54 |
+
res.redirect('/login.html');
|
55 |
+
});
|
56 |
+
|
57 |
+
// 添加认证中间件
|
58 |
+
app.use(authMiddleware);
|
59 |
+
|
60 |
+
app.use("/", routes)
|
61 |
+
|
62 |
+
// 设置定时任务,自动刷新 Cookie
|
63 |
+
if (config.refresh.enabled) {
|
64 |
+
const cronSchedule = config.refresh.cron;
|
65 |
+
const minCookieCount = config.refresh.minCookieCount;
|
66 |
+
|
67 |
+
console.log(`启用自动刷新 Cookie,定时规则: ${cronSchedule},最小 Cookie 数量: ${minCookieCount}`);
|
68 |
+
|
69 |
+
cron.schedule(cronSchedule, async () => {
|
70 |
+
console.log('===== 自动刷新 Cookie 开始 =====');
|
71 |
+
console.log(`最小 Cookie 数量: ${minCookieCount}`);
|
72 |
+
|
73 |
+
try {
|
74 |
+
// 获取所有 API Key
|
75 |
+
const apiKeys = keyManager.getAllApiKeys();
|
76 |
+
|
77 |
+
if (apiKeys.length === 0) {
|
78 |
+
console.log('警告: 系统中没有找到任何 API Key');
|
79 |
+
|
80 |
+
// 检查环境变量中是否有 API Keys
|
81 |
+
const envApiKeys = Object.keys(config.apiKeys);
|
82 |
+
if (envApiKeys.length > 0) {
|
83 |
+
console.log(`检测到环境变量中有 ${envApiKeys.length} 个 API Key,但尚未加载到系统中`);
|
84 |
+
console.log('正在重新初始化 API Keys...');
|
85 |
+
|
86 |
+
// 重新获取 API Keys
|
87 |
+
const refreshedApiKeys = keyManager.getAllApiKeys();
|
88 |
+
if (refreshedApiKeys.length > 0) {
|
89 |
+
console.log(`成功加载 ${refreshedApiKeys.length} 个 API Key,继续刷新流程`);
|
90 |
+
// 继续执行后续刷新逻辑
|
91 |
+
} else {
|
92 |
+
console.log('初始化后仍未找到 API Key,请检查配置');
|
93 |
+
console.log('===== 自动刷新 Cookie 结束 =====');
|
94 |
+
return;
|
95 |
+
}
|
96 |
+
} else {
|
97 |
+
console.log('环境变量中也没有配置 API Key,请先添加 API Key');
|
98 |
+
console.log('===== 自动刷新 Cookie 结束 =====');
|
99 |
+
return;
|
100 |
+
}
|
101 |
+
}
|
102 |
+
|
103 |
+
// 重新获取最新的 API Keys(可能已经通过上面的初始化更新了)
|
104 |
+
const updatedApiKeys = keyManager.getAllApiKeys();
|
105 |
+
console.log(`系统中共有 ${updatedApiKeys.length} 个 API Key`);
|
106 |
+
|
107 |
+
// 按 Cookie 数量排序,优先处理 Cookie 数量少的 API Key
|
108 |
+
const sortedKeys = updatedApiKeys.sort((a, b) => {
|
109 |
+
const aCount = keyManager.getAllCookiesForApiKey(a).length;
|
110 |
+
const bCount = keyManager.getAllCookiesForApiKey(b).length;
|
111 |
+
return aCount - bCount; // 升序排列,Cookie 数量少的排在前面
|
112 |
+
});
|
113 |
+
|
114 |
+
// 检查每个 API Key 是否需要刷新
|
115 |
+
let refreshedCount = 0;
|
116 |
+
let needRefreshCount = 0;
|
117 |
+
|
118 |
+
for (const apiKey of sortedKeys) {
|
119 |
+
const cookies = keyManager.getAllCookiesForApiKey(apiKey);
|
120 |
+
console.log(`API Key: ${apiKey}, Cookie 数量: ${cookies.length}`);
|
121 |
+
|
122 |
+
if (cookies.length < minCookieCount) {
|
123 |
+
needRefreshCount++;
|
124 |
+
console.log(`API Key ${apiKey} 的 Cookie 数量不足,需要刷新`);
|
125 |
+
|
126 |
+
// 执行刷新
|
127 |
+
console.log(`开始自动刷新 Cookie,目标 API Key: ${apiKey},最小 Cookie 数量: ${minCookieCount}`);
|
128 |
+
const result = await cookieRefresher.autoRefreshCookies(apiKey, minCookieCount);
|
129 |
+
|
130 |
+
if (result.success) {
|
131 |
+
refreshedCount++;
|
132 |
+
console.log(`刷新结果: ${result.message}`);
|
133 |
+
} else {
|
134 |
+
console.error(`刷新失败: ${result.message}`);
|
135 |
+
}
|
136 |
+
} else {
|
137 |
+
console.log(`API Key ${apiKey} 的 Cookie 数量足够,不需要刷新`);
|
138 |
+
}
|
139 |
+
}
|
140 |
+
|
141 |
+
console.log('===== 自动刷新 Cookie 完成 =====');
|
142 |
+
console.log(`共有 ${needRefreshCount} 个 API Key 需要刷新,成功刷新 ${refreshedCount} 个`);
|
143 |
+
} catch (error) {
|
144 |
+
console.error('自动刷新 Cookie 任务执行失败:', error);
|
145 |
+
console.log('===== 自动刷新 Cookie 异常结束 =====');
|
146 |
+
}
|
147 |
+
});
|
148 |
+
} else {
|
149 |
+
console.log('未启用自动刷新 Cookie,如需启用请设置环境变量 ENABLE_AUTO_REFRESH=true');
|
150 |
+
}
|
151 |
+
|
152 |
+
app.listen(config.port, () => {
|
153 |
+
console.log(`The server listens port: ${config.port}`);
|
154 |
+
});
|
src/config/config.js
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
// 读取并解析API_KEYS环境变量
|
2 |
+
// 解析API Keys配置
|
3 |
+
let apiKeysConfig = {};
|
4 |
+
try {
|
5 |
+
if (process.env.API_KEYS) {
|
6 |
+
// 解析API Keys字符串为对象
|
7 |
+
apiKeysConfig = JSON.parse(process.env.API_KEYS);
|
8 |
+
console.log('正在从环境变量加载API Keys...');
|
9 |
+
console.log(`成功解析API Keys,包含 ${Object.keys(apiKeysConfig).length} 个键`);
|
10 |
+
} else {
|
11 |
+
console.log('警告: 环境变量API_KEYS未设置,系统将无法正常工作');
|
12 |
+
}
|
13 |
+
} catch (error) {
|
14 |
+
console.error('解析API_KEYS环境变量失败:', error.message);
|
15 |
+
console.error('请确保API_KEYS是有效的JSON格式');
|
16 |
+
}
|
17 |
+
|
18 |
+
// 导出配置
|
19 |
+
module.exports = {
|
20 |
+
port: process.env.PORT || 3010,
|
21 |
+
apiKeys: apiKeysConfig,
|
22 |
+
defaultRotationStrategy: process.env.ROTATION_STRATEGY || 'round-robin',
|
23 |
+
|
24 |
+
// GitHub相关配置
|
25 |
+
github: {
|
26 |
+
token: process.env.GITHUB_TOKEN,
|
27 |
+
owner: process.env.GITHUB_OWNER,
|
28 |
+
repo: process.env.GITHUB_REPO,
|
29 |
+
workflowId: process.env.GITHUB_WORKFLOW_ID,
|
30 |
+
triggerWorkflow: process.env.TRIGGER_WORKFLOW === 'true'
|
31 |
+
},
|
32 |
+
|
33 |
+
// 工作流参数
|
34 |
+
workflowParams: {
|
35 |
+
number: parseInt(process.env.REGISTER_NUMBER || '2', 10),
|
36 |
+
maxWorkers: parseInt(process.env.REGISTER_MAX_WORKERS || '1', 10),
|
37 |
+
emailServer: process.env.REGISTER_EMAIL_SERVER || 'TempEmail',
|
38 |
+
ingestToOneapi: process.env.REGISTER_INGEST_TO_ONEAPI === 'true',
|
39 |
+
uploadArtifact: process.env.REGISTER_UPLOAD_ARTIFACT === 'true',
|
40 |
+
useConfigFile: process.env.REGISTER_USE_CONFIG_FILE !== 'false',
|
41 |
+
emailConfigs: process.env.REGISTER_EMAIL_CONFIGS || '[]'
|
42 |
+
},
|
43 |
+
|
44 |
+
// 刷新配置
|
45 |
+
refresh: {
|
46 |
+
cron: process.env.REFRESH_CRON || '0 */6 * * *',
|
47 |
+
minCookieCount: parseInt(process.env.MIN_COOKIE_COUNT || '2', 10),
|
48 |
+
enabled: process.env.ENABLE_AUTO_REFRESH === 'true'
|
49 |
+
}
|
50 |
+
};
|
src/middleware/auth.js
ADDED
@@ -0,0 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
const admin = require('../models/admin');
|
2 |
+
|
3 |
+
// 验证管理员权限的中间件
|
4 |
+
function authMiddleware(req, res, next) {
|
5 |
+
// 跳过登录相关的路由
|
6 |
+
if (req.path.startsWith('/v1/admin/')) {
|
7 |
+
return next();
|
8 |
+
}
|
9 |
+
|
10 |
+
// 修改为:只对管理相关的API进行认证
|
11 |
+
if (req.path.startsWith('/v1/api-keys') ||
|
12 |
+
req.path.startsWith('/v1/invalid-cookies') ||
|
13 |
+
req.path.startsWith('/v1/refresh-cookies')) {
|
14 |
+
// 获取Authorization头
|
15 |
+
const authHeader = req.headers.authorization;
|
16 |
+
if (!authHeader || !authHeader.startsWith('Bearer ')) {
|
17 |
+
return res.status(401).json({
|
18 |
+
success: false,
|
19 |
+
message: '未提供认证token'
|
20 |
+
});
|
21 |
+
}
|
22 |
+
|
23 |
+
// 提取token
|
24 |
+
const token = authHeader.split(' ')[1];
|
25 |
+
|
26 |
+
// 验证token
|
27 |
+
const result = admin.verifyToken(token);
|
28 |
+
if (!result.success) {
|
29 |
+
return res.status(401).json({
|
30 |
+
success: false,
|
31 |
+
message: '无效的token'
|
32 |
+
});
|
33 |
+
}
|
34 |
+
|
35 |
+
// 将用户信息添加到请求对象
|
36 |
+
req.admin = {
|
37 |
+
username: result.username
|
38 |
+
};
|
39 |
+
}
|
40 |
+
|
41 |
+
next();
|
42 |
+
}
|
43 |
+
|
44 |
+
module.exports = authMiddleware;
|
src/models/admin.js
ADDED
@@ -0,0 +1,114 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
const fs = require('fs');
|
2 |
+
const path = require('path');
|
3 |
+
const crypto = require('crypto');
|
4 |
+
const jwt = require('jsonwebtoken');
|
5 |
+
|
6 |
+
// 管理员数据文件路径
|
7 |
+
const ADMIN_FILE = path.join(__dirname, '../../data/admin.json');
|
8 |
+
const JWT_SECRET = process.env.JWT_SECRET || 'your-secret-key';
|
9 |
+
|
10 |
+
// 确保data目录存在
|
11 |
+
const dataDir = path.dirname(ADMIN_FILE);
|
12 |
+
if (!fs.existsSync(dataDir)) {
|
13 |
+
fs.mkdirSync(dataDir, { recursive: true });
|
14 |
+
}
|
15 |
+
|
16 |
+
// 确保admin.json文件存在
|
17 |
+
if (!fs.existsSync(ADMIN_FILE)) {
|
18 |
+
fs.writeFileSync(ADMIN_FILE, JSON.stringify({ admin: null }), 'utf8');
|
19 |
+
}
|
20 |
+
|
21 |
+
class Admin {
|
22 |
+
constructor() {
|
23 |
+
this.loadAdmin();
|
24 |
+
}
|
25 |
+
|
26 |
+
// 加载管理员数据
|
27 |
+
loadAdmin() {
|
28 |
+
try {
|
29 |
+
const data = fs.readFileSync(ADMIN_FILE, 'utf8');
|
30 |
+
this.admin = JSON.parse(data).admin;
|
31 |
+
} catch (error) {
|
32 |
+
console.error('加载管理员数据失败:', error);
|
33 |
+
this.admin = null;
|
34 |
+
}
|
35 |
+
}
|
36 |
+
|
37 |
+
// 保存管理员数据
|
38 |
+
saveAdmin() {
|
39 |
+
try {
|
40 |
+
fs.writeFileSync(ADMIN_FILE, JSON.stringify({ admin: this.admin }), 'utf8');
|
41 |
+
} catch (error) {
|
42 |
+
console.error('保存管理员数据失败:', error);
|
43 |
+
throw error;
|
44 |
+
}
|
45 |
+
}
|
46 |
+
|
47 |
+
// 检查是否已有管理员
|
48 |
+
hasAdmin() {
|
49 |
+
return !!this.admin;
|
50 |
+
}
|
51 |
+
|
52 |
+
// 注册管理员
|
53 |
+
register(username, password) {
|
54 |
+
if (this.hasAdmin()) {
|
55 |
+
throw new Error('已存在管理员账号');
|
56 |
+
}
|
57 |
+
|
58 |
+
// 生成盐值
|
59 |
+
const salt = crypto.randomBytes(16).toString('hex');
|
60 |
+
// 使用盐值加密密码
|
61 |
+
const hash = crypto.pbkdf2Sync(password, salt, 1000, 64, 'sha512').toString('hex');
|
62 |
+
|
63 |
+
this.admin = {
|
64 |
+
username,
|
65 |
+
salt,
|
66 |
+
hash
|
67 |
+
};
|
68 |
+
|
69 |
+
this.saveAdmin();
|
70 |
+
return this.generateToken(username);
|
71 |
+
}
|
72 |
+
|
73 |
+
// 验证密码
|
74 |
+
verifyPassword(password, salt, hash) {
|
75 |
+
const testHash = crypto.pbkdf2Sync(password, salt, 1000, 64, 'sha512').toString('hex');
|
76 |
+
return testHash === hash;
|
77 |
+
}
|
78 |
+
|
79 |
+
// 登录验证
|
80 |
+
login(username, password) {
|
81 |
+
if (!this.admin || username !== this.admin.username) {
|
82 |
+
throw new Error('用户名或密码错误');
|
83 |
+
}
|
84 |
+
|
85 |
+
if (!this.verifyPassword(password, this.admin.salt, this.admin.hash)) {
|
86 |
+
throw new Error('用户名或密码错误');
|
87 |
+
}
|
88 |
+
|
89 |
+
return this.generateToken(username);
|
90 |
+
}
|
91 |
+
|
92 |
+
// 生成JWT token
|
93 |
+
generateToken(username) {
|
94 |
+
return jwt.sign({ username }, JWT_SECRET, { expiresIn: '24h' });
|
95 |
+
}
|
96 |
+
|
97 |
+
// 验证JWT token
|
98 |
+
verifyToken(token) {
|
99 |
+
try {
|
100 |
+
const decoded = jwt.verify(token, JWT_SECRET);
|
101 |
+
return {
|
102 |
+
success: true,
|
103 |
+
username: decoded.username
|
104 |
+
};
|
105 |
+
} catch (error) {
|
106 |
+
return {
|
107 |
+
success: false,
|
108 |
+
error: 'Invalid token'
|
109 |
+
};
|
110 |
+
}
|
111 |
+
}
|
112 |
+
}
|
113 |
+
|
114 |
+
module.exports = new Admin();
|
src/proto/message.js
ADDED
@@ -0,0 +1,1918 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
/*eslint-disable block-scoped-var, id-length, no-control-regex, no-magic-numbers, no-prototype-builtins, no-redeclare, no-shadow, no-var, sort-vars*/
|
2 |
+
"use strict";
|
3 |
+
|
4 |
+
var $protobuf = require("protobufjs/minimal");
|
5 |
+
|
6 |
+
// Common aliases
|
7 |
+
var $Reader = $protobuf.Reader, $Writer = $protobuf.Writer, $util = $protobuf.util;
|
8 |
+
|
9 |
+
// Exported root namespace
|
10 |
+
var $root = $protobuf.roots["default"] || ($protobuf.roots["default"] = {});
|
11 |
+
|
12 |
+
$root.ChatMessage = (function() {
|
13 |
+
|
14 |
+
/**
|
15 |
+
* Properties of a ChatMessage.
|
16 |
+
* @exports IChatMessage
|
17 |
+
* @interface IChatMessage
|
18 |
+
* @property {Array.<ChatMessage.IUserMessage>|null} [userMessages] ChatMessage userMessages
|
19 |
+
* @property {ChatMessage.IInstructions|null} [instructions] ChatMessage instructions
|
20 |
+
* @property {ChatMessage.IModel|null} [model] ChatMessage model
|
21 |
+
* @property {number|null} [unknown13] ChatMessage unknown13
|
22 |
+
* @property {string|null} [conversationId] ChatMessage conversationId
|
23 |
+
* @property {number|null} [unknown16] ChatMessage unknown16
|
24 |
+
* @property {number|null} [unknown29] ChatMessage unknown29
|
25 |
+
* @property {number|null} [unknown31] ChatMessage unknown31
|
26 |
+
*/
|
27 |
+
|
28 |
+
/**
|
29 |
+
* Constructs a new ChatMessage.
|
30 |
+
* @exports ChatMessage
|
31 |
+
* @classdesc Represents a ChatMessage.
|
32 |
+
* @implements IChatMessage
|
33 |
+
* @constructor
|
34 |
+
* @param {IChatMessage=} [properties] Properties to set
|
35 |
+
*/
|
36 |
+
function ChatMessage(properties) {
|
37 |
+
this.userMessages = [];
|
38 |
+
if (properties)
|
39 |
+
for (var keys = Object.keys(properties), i = 0; i < keys.length; ++i)
|
40 |
+
if (properties[keys[i]] != null)
|
41 |
+
this[keys[i]] = properties[keys[i]];
|
42 |
+
}
|
43 |
+
|
44 |
+
/**
|
45 |
+
* ChatMessage userMessages.
|
46 |
+
* @member {Array.<ChatMessage.IUserMessage>} userMessages
|
47 |
+
* @memberof ChatMessage
|
48 |
+
* @instance
|
49 |
+
*/
|
50 |
+
ChatMessage.prototype.userMessages = $util.emptyArray;
|
51 |
+
|
52 |
+
/**
|
53 |
+
* ChatMessage instructions.
|
54 |
+
* @member {ChatMessage.IInstructions|null|undefined} instructions
|
55 |
+
* @memberof ChatMessage
|
56 |
+
* @instance
|
57 |
+
*/
|
58 |
+
ChatMessage.prototype.instructions = null;
|
59 |
+
|
60 |
+
/**
|
61 |
+
* ChatMessage model.
|
62 |
+
* @member {ChatMessage.IModel|null|undefined} model
|
63 |
+
* @memberof ChatMessage
|
64 |
+
* @instance
|
65 |
+
*/
|
66 |
+
ChatMessage.prototype.model = null;
|
67 |
+
|
68 |
+
/**
|
69 |
+
* ChatMessage unknown13.
|
70 |
+
* @member {number} unknown13
|
71 |
+
* @memberof ChatMessage
|
72 |
+
* @instance
|
73 |
+
*/
|
74 |
+
ChatMessage.prototype.unknown13 = 0;
|
75 |
+
|
76 |
+
/**
|
77 |
+
* ChatMessage conversationId.
|
78 |
+
* @member {string} conversationId
|
79 |
+
* @memberof ChatMessage
|
80 |
+
* @instance
|
81 |
+
*/
|
82 |
+
ChatMessage.prototype.conversationId = "";
|
83 |
+
|
84 |
+
/**
|
85 |
+
* ChatMessage unknown16.
|
86 |
+
* @member {number} unknown16
|
87 |
+
* @memberof ChatMessage
|
88 |
+
* @instance
|
89 |
+
*/
|
90 |
+
ChatMessage.prototype.unknown16 = 0;
|
91 |
+
|
92 |
+
/**
|
93 |
+
* ChatMessage unknown29.
|
94 |
+
* @member {number} unknown29
|
95 |
+
* @memberof ChatMessage
|
96 |
+
* @instance
|
97 |
+
*/
|
98 |
+
ChatMessage.prototype.unknown29 = 0;
|
99 |
+
|
100 |
+
/**
|
101 |
+
* ChatMessage unknown31.
|
102 |
+
* @member {number} unknown31
|
103 |
+
* @memberof ChatMessage
|
104 |
+
* @instance
|
105 |
+
*/
|
106 |
+
ChatMessage.prototype.unknown31 = 0;
|
107 |
+
|
108 |
+
/**
|
109 |
+
* Creates a new ChatMessage instance using the specified properties.
|
110 |
+
* @function create
|
111 |
+
* @memberof ChatMessage
|
112 |
+
* @static
|
113 |
+
* @param {IChatMessage=} [properties] Properties to set
|
114 |
+
* @returns {ChatMessage} ChatMessage instance
|
115 |
+
*/
|
116 |
+
ChatMessage.create = function create(properties) {
|
117 |
+
return new ChatMessage(properties);
|
118 |
+
};
|
119 |
+
|
120 |
+
/**
|
121 |
+
* Encodes the specified ChatMessage message. Does not implicitly {@link ChatMessage.verify|verify} messages.
|
122 |
+
* @function encode
|
123 |
+
* @memberof ChatMessage
|
124 |
+
* @static
|
125 |
+
* @param {IChatMessage} message ChatMessage message or plain object to encode
|
126 |
+
* @param {$protobuf.Writer} [writer] Writer to encode to
|
127 |
+
* @returns {$protobuf.Writer} Writer
|
128 |
+
*/
|
129 |
+
ChatMessage.encode = function encode(message, writer) {
|
130 |
+
if (!writer)
|
131 |
+
writer = $Writer.create();
|
132 |
+
if (message.userMessages != null && message.userMessages.length)
|
133 |
+
for (var i = 0; i < message.userMessages.length; ++i)
|
134 |
+
$root.ChatMessage.UserMessage.encode(message.userMessages[i], writer.uint32(/* id 2, wireType 2 =*/18).fork()).ldelim();
|
135 |
+
if (message.instructions != null && Object.hasOwnProperty.call(message, "instructions"))
|
136 |
+
$root.ChatMessage.Instructions.encode(message.instructions, writer.uint32(/* id 4, wireType 2 =*/34).fork()).ldelim();
|
137 |
+
if (message.model != null && Object.hasOwnProperty.call(message, "model"))
|
138 |
+
$root.ChatMessage.Model.encode(message.model, writer.uint32(/* id 7, wireType 2 =*/58).fork()).ldelim();
|
139 |
+
if (message.unknown13 != null && Object.hasOwnProperty.call(message, "unknown13"))
|
140 |
+
writer.uint32(/* id 13, wireType 0 =*/104).int32(message.unknown13);
|
141 |
+
if (message.conversationId != null && Object.hasOwnProperty.call(message, "conversationId"))
|
142 |
+
writer.uint32(/* id 15, wireType 2 =*/122).string(message.conversationId);
|
143 |
+
if (message.unknown16 != null && Object.hasOwnProperty.call(message, "unknown16"))
|
144 |
+
writer.uint32(/* id 16, wireType 0 =*/128).int32(message.unknown16);
|
145 |
+
if (message.unknown29 != null && Object.hasOwnProperty.call(message, "unknown29"))
|
146 |
+
writer.uint32(/* id 29, wireType 0 =*/232).int32(message.unknown29);
|
147 |
+
if (message.unknown31 != null && Object.hasOwnProperty.call(message, "unknown31"))
|
148 |
+
writer.uint32(/* id 31, wireType 0 =*/248).int32(message.unknown31);
|
149 |
+
return writer;
|
150 |
+
};
|
151 |
+
|
152 |
+
/**
|
153 |
+
* Encodes the specified ChatMessage message, length delimited. Does not implicitly {@link ChatMessage.verify|verify} messages.
|
154 |
+
* @function encodeDelimited
|
155 |
+
* @memberof ChatMessage
|
156 |
+
* @static
|
157 |
+
* @param {IChatMessage} message ChatMessage message or plain object to encode
|
158 |
+
* @param {$protobuf.Writer} [writer] Writer to encode to
|
159 |
+
* @returns {$protobuf.Writer} Writer
|
160 |
+
*/
|
161 |
+
ChatMessage.encodeDelimited = function encodeDelimited(message, writer) {
|
162 |
+
return this.encode(message, writer).ldelim();
|
163 |
+
};
|
164 |
+
|
165 |
+
/**
|
166 |
+
* Decodes a ChatMessage message from the specified reader or buffer.
|
167 |
+
* @function decode
|
168 |
+
* @memberof ChatMessage
|
169 |
+
* @static
|
170 |
+
* @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
|
171 |
+
* @param {number} [length] Message length if known beforehand
|
172 |
+
* @returns {ChatMessage} ChatMessage
|
173 |
+
* @throws {Error} If the payload is not a reader or valid buffer
|
174 |
+
* @throws {$protobuf.util.ProtocolError} If required fields are missing
|
175 |
+
*/
|
176 |
+
ChatMessage.decode = function decode(reader, length) {
|
177 |
+
if (!(reader instanceof $Reader))
|
178 |
+
reader = $Reader.create(reader);
|
179 |
+
var end = length === undefined ? reader.len : reader.pos + length, message = new $root.ChatMessage();
|
180 |
+
while (reader.pos < end) {
|
181 |
+
var tag = reader.uint32();
|
182 |
+
switch (tag >>> 3) {
|
183 |
+
case 2: {
|
184 |
+
if (!(message.userMessages && message.userMessages.length))
|
185 |
+
message.userMessages = [];
|
186 |
+
message.userMessages.push($root.ChatMessage.UserMessage.decode(reader, reader.uint32()));
|
187 |
+
break;
|
188 |
+
}
|
189 |
+
case 4: {
|
190 |
+
message.instructions = $root.ChatMessage.Instructions.decode(reader, reader.uint32());
|
191 |
+
break;
|
192 |
+
}
|
193 |
+
case 7: {
|
194 |
+
message.model = $root.ChatMessage.Model.decode(reader, reader.uint32());
|
195 |
+
break;
|
196 |
+
}
|
197 |
+
case 13: {
|
198 |
+
message.unknown13 = reader.int32();
|
199 |
+
break;
|
200 |
+
}
|
201 |
+
case 15: {
|
202 |
+
message.conversationId = reader.string();
|
203 |
+
break;
|
204 |
+
}
|
205 |
+
case 16: {
|
206 |
+
message.unknown16 = reader.int32();
|
207 |
+
break;
|
208 |
+
}
|
209 |
+
case 29: {
|
210 |
+
message.unknown29 = reader.int32();
|
211 |
+
break;
|
212 |
+
}
|
213 |
+
case 31: {
|
214 |
+
message.unknown31 = reader.int32();
|
215 |
+
break;
|
216 |
+
}
|
217 |
+
default:
|
218 |
+
reader.skipType(tag & 7);
|
219 |
+
break;
|
220 |
+
}
|
221 |
+
}
|
222 |
+
return message;
|
223 |
+
};
|
224 |
+
|
225 |
+
/**
|
226 |
+
* Decodes a ChatMessage message from the specified reader or buffer, length delimited.
|
227 |
+
* @function decodeDelimited
|
228 |
+
* @memberof ChatMessage
|
229 |
+
* @static
|
230 |
+
* @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
|
231 |
+
* @returns {ChatMessage} ChatMessage
|
232 |
+
* @throws {Error} If the payload is not a reader or valid buffer
|
233 |
+
* @throws {$protobuf.util.ProtocolError} If required fields are missing
|
234 |
+
*/
|
235 |
+
ChatMessage.decodeDelimited = function decodeDelimited(reader) {
|
236 |
+
if (!(reader instanceof $Reader))
|
237 |
+
reader = new $Reader(reader);
|
238 |
+
return this.decode(reader, reader.uint32());
|
239 |
+
};
|
240 |
+
|
241 |
+
/**
|
242 |
+
* Verifies a ChatMessage message.
|
243 |
+
* @function verify
|
244 |
+
* @memberof ChatMessage
|
245 |
+
* @static
|
246 |
+
* @param {Object.<string,*>} message Plain object to verify
|
247 |
+
* @returns {string|null} `null` if valid, otherwise the reason why it is not
|
248 |
+
*/
|
249 |
+
ChatMessage.verify = function verify(message) {
|
250 |
+
if (typeof message !== "object" || message === null)
|
251 |
+
return "object expected";
|
252 |
+
if (message.userMessages != null && message.hasOwnProperty("userMessages")) {
|
253 |
+
if (!Array.isArray(message.userMessages))
|
254 |
+
return "userMessages: array expected";
|
255 |
+
for (var i = 0; i < message.userMessages.length; ++i) {
|
256 |
+
var error = $root.ChatMessage.UserMessage.verify(message.userMessages[i]);
|
257 |
+
if (error)
|
258 |
+
return "userMessages." + error;
|
259 |
+
}
|
260 |
+
}
|
261 |
+
if (message.instructions != null && message.hasOwnProperty("instructions")) {
|
262 |
+
var error = $root.ChatMessage.Instructions.verify(message.instructions);
|
263 |
+
if (error)
|
264 |
+
return "instructions." + error;
|
265 |
+
}
|
266 |
+
if (message.model != null && message.hasOwnProperty("model")) {
|
267 |
+
var error = $root.ChatMessage.Model.verify(message.model);
|
268 |
+
if (error)
|
269 |
+
return "model." + error;
|
270 |
+
}
|
271 |
+
if (message.unknown13 != null && message.hasOwnProperty("unknown13"))
|
272 |
+
if (!$util.isInteger(message.unknown13))
|
273 |
+
return "unknown13: integer expected";
|
274 |
+
if (message.conversationId != null && message.hasOwnProperty("conversationId"))
|
275 |
+
if (!$util.isString(message.conversationId))
|
276 |
+
return "conversationId: string expected";
|
277 |
+
if (message.unknown16 != null && message.hasOwnProperty("unknown16"))
|
278 |
+
if (!$util.isInteger(message.unknown16))
|
279 |
+
return "unknown16: integer expected";
|
280 |
+
if (message.unknown29 != null && message.hasOwnProperty("unknown29"))
|
281 |
+
if (!$util.isInteger(message.unknown29))
|
282 |
+
return "unknown29: integer expected";
|
283 |
+
if (message.unknown31 != null && message.hasOwnProperty("unknown31"))
|
284 |
+
if (!$util.isInteger(message.unknown31))
|
285 |
+
return "unknown31: integer expected";
|
286 |
+
return null;
|
287 |
+
};
|
288 |
+
|
289 |
+
/**
|
290 |
+
* Creates a ChatMessage message from a plain object. Also converts values to their respective internal types.
|
291 |
+
* @function fromObject
|
292 |
+
* @memberof ChatMessage
|
293 |
+
* @static
|
294 |
+
* @param {Object.<string,*>} object Plain object
|
295 |
+
* @returns {ChatMessage} ChatMessage
|
296 |
+
*/
|
297 |
+
ChatMessage.fromObject = function fromObject(object) {
|
298 |
+
if (object instanceof $root.ChatMessage)
|
299 |
+
return object;
|
300 |
+
var message = new $root.ChatMessage();
|
301 |
+
if (object.userMessages) {
|
302 |
+
if (!Array.isArray(object.userMessages))
|
303 |
+
throw TypeError(".ChatMessage.userMessages: array expected");
|
304 |
+
message.userMessages = [];
|
305 |
+
for (var i = 0; i < object.userMessages.length; ++i) {
|
306 |
+
if (typeof object.userMessages[i] !== "object")
|
307 |
+
throw TypeError(".ChatMessage.userMessages: object expected");
|
308 |
+
message.userMessages[i] = $root.ChatMessage.UserMessage.fromObject(object.userMessages[i]);
|
309 |
+
}
|
310 |
+
}
|
311 |
+
if (object.instructions != null) {
|
312 |
+
if (typeof object.instructions !== "object")
|
313 |
+
throw TypeError(".ChatMessage.instructions: object expected");
|
314 |
+
message.instructions = $root.ChatMessage.Instructions.fromObject(object.instructions);
|
315 |
+
}
|
316 |
+
if (object.model != null) {
|
317 |
+
if (typeof object.model !== "object")
|
318 |
+
throw TypeError(".ChatMessage.model: object expected");
|
319 |
+
message.model = $root.ChatMessage.Model.fromObject(object.model);
|
320 |
+
}
|
321 |
+
if (object.unknown13 != null)
|
322 |
+
message.unknown13 = object.unknown13 | 0;
|
323 |
+
if (object.conversationId != null)
|
324 |
+
message.conversationId = String(object.conversationId);
|
325 |
+
if (object.unknown16 != null)
|
326 |
+
message.unknown16 = object.unknown16 | 0;
|
327 |
+
if (object.unknown29 != null)
|
328 |
+
message.unknown29 = object.unknown29 | 0;
|
329 |
+
if (object.unknown31 != null)
|
330 |
+
message.unknown31 = object.unknown31 | 0;
|
331 |
+
return message;
|
332 |
+
};
|
333 |
+
|
334 |
+
/**
|
335 |
+
* Creates a plain object from a ChatMessage message. Also converts values to other types if specified.
|
336 |
+
* @function toObject
|
337 |
+
* @memberof ChatMessage
|
338 |
+
* @static
|
339 |
+
* @param {ChatMessage} message ChatMessage
|
340 |
+
* @param {$protobuf.IConversionOptions} [options] Conversion options
|
341 |
+
* @returns {Object.<string,*>} Plain object
|
342 |
+
*/
|
343 |
+
ChatMessage.toObject = function toObject(message, options) {
|
344 |
+
if (!options)
|
345 |
+
options = {};
|
346 |
+
var object = {};
|
347 |
+
if (options.arrays || options.defaults)
|
348 |
+
object.userMessages = [];
|
349 |
+
if (options.defaults) {
|
350 |
+
object.instructions = null;
|
351 |
+
object.model = null;
|
352 |
+
object.unknown13 = 0;
|
353 |
+
object.conversationId = "";
|
354 |
+
object.unknown16 = 0;
|
355 |
+
object.unknown29 = 0;
|
356 |
+
object.unknown31 = 0;
|
357 |
+
}
|
358 |
+
if (message.userMessages && message.userMessages.length) {
|
359 |
+
object.userMessages = [];
|
360 |
+
for (var j = 0; j < message.userMessages.length; ++j)
|
361 |
+
object.userMessages[j] = $root.ChatMessage.UserMessage.toObject(message.userMessages[j], options);
|
362 |
+
}
|
363 |
+
if (message.instructions != null && message.hasOwnProperty("instructions"))
|
364 |
+
object.instructions = $root.ChatMessage.Instructions.toObject(message.instructions, options);
|
365 |
+
if (message.model != null && message.hasOwnProperty("model"))
|
366 |
+
object.model = $root.ChatMessage.Model.toObject(message.model, options);
|
367 |
+
if (message.unknown13 != null && message.hasOwnProperty("unknown13"))
|
368 |
+
object.unknown13 = message.unknown13;
|
369 |
+
if (message.conversationId != null && message.hasOwnProperty("conversationId"))
|
370 |
+
object.conversationId = message.conversationId;
|
371 |
+
if (message.unknown16 != null && message.hasOwnProperty("unknown16"))
|
372 |
+
object.unknown16 = message.unknown16;
|
373 |
+
if (message.unknown29 != null && message.hasOwnProperty("unknown29"))
|
374 |
+
object.unknown29 = message.unknown29;
|
375 |
+
if (message.unknown31 != null && message.hasOwnProperty("unknown31"))
|
376 |
+
object.unknown31 = message.unknown31;
|
377 |
+
return object;
|
378 |
+
};
|
379 |
+
|
380 |
+
/**
|
381 |
+
* Converts this ChatMessage to JSON.
|
382 |
+
* @function toJSON
|
383 |
+
* @memberof ChatMessage
|
384 |
+
* @instance
|
385 |
+
* @returns {Object.<string,*>} JSON object
|
386 |
+
*/
|
387 |
+
ChatMessage.prototype.toJSON = function toJSON() {
|
388 |
+
return this.constructor.toObject(this, $protobuf.util.toJSONOptions);
|
389 |
+
};
|
390 |
+
|
391 |
+
/**
|
392 |
+
* Gets the default type url for ChatMessage
|
393 |
+
* @function getTypeUrl
|
394 |
+
* @memberof ChatMessage
|
395 |
+
* @static
|
396 |
+
* @param {string} [typeUrlPrefix] your custom typeUrlPrefix(default "type.googleapis.com")
|
397 |
+
* @returns {string} The default type url
|
398 |
+
*/
|
399 |
+
ChatMessage.getTypeUrl = function getTypeUrl(typeUrlPrefix) {
|
400 |
+
if (typeUrlPrefix === undefined) {
|
401 |
+
typeUrlPrefix = "type.googleapis.com";
|
402 |
+
}
|
403 |
+
return typeUrlPrefix + "/ChatMessage";
|
404 |
+
};
|
405 |
+
|
406 |
+
ChatMessage.UserMessage = (function() {
|
407 |
+
|
408 |
+
/**
|
409 |
+
* Properties of a UserMessage.
|
410 |
+
* @memberof ChatMessage
|
411 |
+
* @interface IUserMessage
|
412 |
+
* @property {string|null} [content] UserMessage content
|
413 |
+
* @property {number|null} [role] UserMessage role
|
414 |
+
* @property {string|null} [messageId] UserMessage messageId
|
415 |
+
*/
|
416 |
+
|
417 |
+
/**
|
418 |
+
* Constructs a new UserMessage.
|
419 |
+
* @memberof ChatMessage
|
420 |
+
* @classdesc Represents a UserMessage.
|
421 |
+
* @implements IUserMessage
|
422 |
+
* @constructor
|
423 |
+
* @param {ChatMessage.IUserMessage=} [properties] Properties to set
|
424 |
+
*/
|
425 |
+
function UserMessage(properties) {
|
426 |
+
if (properties)
|
427 |
+
for (var keys = Object.keys(properties), i = 0; i < keys.length; ++i)
|
428 |
+
if (properties[keys[i]] != null)
|
429 |
+
this[keys[i]] = properties[keys[i]];
|
430 |
+
}
|
431 |
+
|
432 |
+
/**
|
433 |
+
* UserMessage content.
|
434 |
+
* @member {string} content
|
435 |
+
* @memberof ChatMessage.UserMessage
|
436 |
+
* @instance
|
437 |
+
*/
|
438 |
+
UserMessage.prototype.content = "";
|
439 |
+
|
440 |
+
/**
|
441 |
+
* UserMessage role.
|
442 |
+
* @member {number} role
|
443 |
+
* @memberof ChatMessage.UserMessage
|
444 |
+
* @instance
|
445 |
+
*/
|
446 |
+
UserMessage.prototype.role = 0;
|
447 |
+
|
448 |
+
/**
|
449 |
+
* UserMessage messageId.
|
450 |
+
* @member {string} messageId
|
451 |
+
* @memberof ChatMessage.UserMessage
|
452 |
+
* @instance
|
453 |
+
*/
|
454 |
+
UserMessage.prototype.messageId = "";
|
455 |
+
|
456 |
+
/**
|
457 |
+
* Creates a new UserMessage instance using the specified properties.
|
458 |
+
* @function create
|
459 |
+
* @memberof ChatMessage.UserMessage
|
460 |
+
* @static
|
461 |
+
* @param {ChatMessage.IUserMessage=} [properties] Properties to set
|
462 |
+
* @returns {ChatMessage.UserMessage} UserMessage instance
|
463 |
+
*/
|
464 |
+
UserMessage.create = function create(properties) {
|
465 |
+
return new UserMessage(properties);
|
466 |
+
};
|
467 |
+
|
468 |
+
/**
|
469 |
+
* Encodes the specified UserMessage message. Does not implicitly {@link ChatMessage.UserMessage.verify|verify} messages.
|
470 |
+
* @function encode
|
471 |
+
* @memberof ChatMessage.UserMessage
|
472 |
+
* @static
|
473 |
+
* @param {ChatMessage.IUserMessage} message UserMessage message or plain object to encode
|
474 |
+
* @param {$protobuf.Writer} [writer] Writer to encode to
|
475 |
+
* @returns {$protobuf.Writer} Writer
|
476 |
+
*/
|
477 |
+
UserMessage.encode = function encode(message, writer) {
|
478 |
+
if (!writer)
|
479 |
+
writer = $Writer.create();
|
480 |
+
if (message.content != null && Object.hasOwnProperty.call(message, "content"))
|
481 |
+
writer.uint32(/* id 1, wireType 2 =*/10).string(message.content);
|
482 |
+
if (message.role != null && Object.hasOwnProperty.call(message, "role"))
|
483 |
+
writer.uint32(/* id 2, wireType 0 =*/16).int32(message.role);
|
484 |
+
if (message.messageId != null && Object.hasOwnProperty.call(message, "messageId"))
|
485 |
+
writer.uint32(/* id 13, wireType 2 =*/106).string(message.messageId);
|
486 |
+
return writer;
|
487 |
+
};
|
488 |
+
|
489 |
+
/**
|
490 |
+
* Encodes the specified UserMessage message, length delimited. Does not implicitly {@link ChatMessage.UserMessage.verify|verify} messages.
|
491 |
+
* @function encodeDelimited
|
492 |
+
* @memberof ChatMessage.UserMessage
|
493 |
+
* @static
|
494 |
+
* @param {ChatMessage.IUserMessage} message UserMessage message or plain object to encode
|
495 |
+
* @param {$protobuf.Writer} [writer] Writer to encode to
|
496 |
+
* @returns {$protobuf.Writer} Writer
|
497 |
+
*/
|
498 |
+
UserMessage.encodeDelimited = function encodeDelimited(message, writer) {
|
499 |
+
return this.encode(message, writer).ldelim();
|
500 |
+
};
|
501 |
+
|
502 |
+
/**
|
503 |
+
* Decodes a UserMessage message from the specified reader or buffer.
|
504 |
+
* @function decode
|
505 |
+
* @memberof ChatMessage.UserMessage
|
506 |
+
* @static
|
507 |
+
* @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
|
508 |
+
* @param {number} [length] Message length if known beforehand
|
509 |
+
* @returns {ChatMessage.UserMessage} UserMessage
|
510 |
+
* @throws {Error} If the payload is not a reader or valid buffer
|
511 |
+
* @throws {$protobuf.util.ProtocolError} If required fields are missing
|
512 |
+
*/
|
513 |
+
UserMessage.decode = function decode(reader, length) {
|
514 |
+
if (!(reader instanceof $Reader))
|
515 |
+
reader = $Reader.create(reader);
|
516 |
+
var end = length === undefined ? reader.len : reader.pos + length, message = new $root.ChatMessage.UserMessage();
|
517 |
+
while (reader.pos < end) {
|
518 |
+
var tag = reader.uint32();
|
519 |
+
switch (tag >>> 3) {
|
520 |
+
case 1: {
|
521 |
+
message.content = reader.string();
|
522 |
+
break;
|
523 |
+
}
|
524 |
+
case 2: {
|
525 |
+
message.role = reader.int32();
|
526 |
+
break;
|
527 |
+
}
|
528 |
+
case 13: {
|
529 |
+
message.messageId = reader.string();
|
530 |
+
break;
|
531 |
+
}
|
532 |
+
default:
|
533 |
+
reader.skipType(tag & 7);
|
534 |
+
break;
|
535 |
+
}
|
536 |
+
}
|
537 |
+
return message;
|
538 |
+
};
|
539 |
+
|
540 |
+
/**
|
541 |
+
* Decodes a UserMessage message from the specified reader or buffer, length delimited.
|
542 |
+
* @function decodeDelimited
|
543 |
+
* @memberof ChatMessage.UserMessage
|
544 |
+
* @static
|
545 |
+
* @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
|
546 |
+
* @returns {ChatMessage.UserMessage} UserMessage
|
547 |
+
* @throws {Error} If the payload is not a reader or valid buffer
|
548 |
+
* @throws {$protobuf.util.ProtocolError} If required fields are missing
|
549 |
+
*/
|
550 |
+
UserMessage.decodeDelimited = function decodeDelimited(reader) {
|
551 |
+
if (!(reader instanceof $Reader))
|
552 |
+
reader = new $Reader(reader);
|
553 |
+
return this.decode(reader, reader.uint32());
|
554 |
+
};
|
555 |
+
|
556 |
+
/**
|
557 |
+
* Verifies a UserMessage message.
|
558 |
+
* @function verify
|
559 |
+
* @memberof ChatMessage.UserMessage
|
560 |
+
* @static
|
561 |
+
* @param {Object.<string,*>} message Plain object to verify
|
562 |
+
* @returns {string|null} `null` if valid, otherwise the reason why it is not
|
563 |
+
*/
|
564 |
+
UserMessage.verify = function verify(message) {
|
565 |
+
if (typeof message !== "object" || message === null)
|
566 |
+
return "object expected";
|
567 |
+
if (message.content != null && message.hasOwnProperty("content"))
|
568 |
+
if (!$util.isString(message.content))
|
569 |
+
return "content: string expected";
|
570 |
+
if (message.role != null && message.hasOwnProperty("role"))
|
571 |
+
if (!$util.isInteger(message.role))
|
572 |
+
return "role: integer expected";
|
573 |
+
if (message.messageId != null && message.hasOwnProperty("messageId"))
|
574 |
+
if (!$util.isString(message.messageId))
|
575 |
+
return "messageId: string expected";
|
576 |
+
return null;
|
577 |
+
};
|
578 |
+
|
579 |
+
/**
|
580 |
+
* Creates a UserMessage message from a plain object. Also converts values to their respective internal types.
|
581 |
+
* @function fromObject
|
582 |
+
* @memberof ChatMessage.UserMessage
|
583 |
+
* @static
|
584 |
+
* @param {Object.<string,*>} object Plain object
|
585 |
+
* @returns {ChatMessage.UserMessage} UserMessage
|
586 |
+
*/
|
587 |
+
UserMessage.fromObject = function fromObject(object) {
|
588 |
+
if (object instanceof $root.ChatMessage.UserMessage)
|
589 |
+
return object;
|
590 |
+
var message = new $root.ChatMessage.UserMessage();
|
591 |
+
if (object.content != null)
|
592 |
+
message.content = String(object.content);
|
593 |
+
if (object.role != null)
|
594 |
+
message.role = object.role | 0;
|
595 |
+
if (object.messageId != null)
|
596 |
+
message.messageId = String(object.messageId);
|
597 |
+
return message;
|
598 |
+
};
|
599 |
+
|
600 |
+
/**
|
601 |
+
* Creates a plain object from a UserMessage message. Also converts values to other types if specified.
|
602 |
+
* @function toObject
|
603 |
+
* @memberof ChatMessage.UserMessage
|
604 |
+
* @static
|
605 |
+
* @param {ChatMessage.UserMessage} message UserMessage
|
606 |
+
* @param {$protobuf.IConversionOptions} [options] Conversion options
|
607 |
+
* @returns {Object.<string,*>} Plain object
|
608 |
+
*/
|
609 |
+
UserMessage.toObject = function toObject(message, options) {
|
610 |
+
if (!options)
|
611 |
+
options = {};
|
612 |
+
var object = {};
|
613 |
+
if (options.defaults) {
|
614 |
+
object.content = "";
|
615 |
+
object.role = 0;
|
616 |
+
object.messageId = "";
|
617 |
+
}
|
618 |
+
if (message.content != null && message.hasOwnProperty("content"))
|
619 |
+
object.content = message.content;
|
620 |
+
if (message.role != null && message.hasOwnProperty("role"))
|
621 |
+
object.role = message.role;
|
622 |
+
if (message.messageId != null && message.hasOwnProperty("messageId"))
|
623 |
+
object.messageId = message.messageId;
|
624 |
+
return object;
|
625 |
+
};
|
626 |
+
|
627 |
+
/**
|
628 |
+
* Converts this UserMessage to JSON.
|
629 |
+
* @function toJSON
|
630 |
+
* @memberof ChatMessage.UserMessage
|
631 |
+
* @instance
|
632 |
+
* @returns {Object.<string,*>} JSON object
|
633 |
+
*/
|
634 |
+
UserMessage.prototype.toJSON = function toJSON() {
|
635 |
+
return this.constructor.toObject(this, $protobuf.util.toJSONOptions);
|
636 |
+
};
|
637 |
+
|
638 |
+
/**
|
639 |
+
* Gets the default type url for UserMessage
|
640 |
+
* @function getTypeUrl
|
641 |
+
* @memberof ChatMessage.UserMessage
|
642 |
+
* @static
|
643 |
+
* @param {string} [typeUrlPrefix] your custom typeUrlPrefix(default "type.googleapis.com")
|
644 |
+
* @returns {string} The default type url
|
645 |
+
*/
|
646 |
+
UserMessage.getTypeUrl = function getTypeUrl(typeUrlPrefix) {
|
647 |
+
if (typeUrlPrefix === undefined) {
|
648 |
+
typeUrlPrefix = "type.googleapis.com";
|
649 |
+
}
|
650 |
+
return typeUrlPrefix + "/ChatMessage.UserMessage";
|
651 |
+
};
|
652 |
+
|
653 |
+
return UserMessage;
|
654 |
+
})();
|
655 |
+
|
656 |
+
ChatMessage.Instructions = (function() {
|
657 |
+
|
658 |
+
/**
|
659 |
+
* Properties of an Instructions.
|
660 |
+
* @memberof ChatMessage
|
661 |
+
* @interface IInstructions
|
662 |
+
* @property {string|null} [instruction] Instructions instruction
|
663 |
+
*/
|
664 |
+
|
665 |
+
/**
|
666 |
+
* Constructs a new Instructions.
|
667 |
+
* @memberof ChatMessage
|
668 |
+
* @classdesc Represents an Instructions.
|
669 |
+
* @implements IInstructions
|
670 |
+
* @constructor
|
671 |
+
* @param {ChatMessage.IInstructions=} [properties] Properties to set
|
672 |
+
*/
|
673 |
+
function Instructions(properties) {
|
674 |
+
if (properties)
|
675 |
+
for (var keys = Object.keys(properties), i = 0; i < keys.length; ++i)
|
676 |
+
if (properties[keys[i]] != null)
|
677 |
+
this[keys[i]] = properties[keys[i]];
|
678 |
+
}
|
679 |
+
|
680 |
+
/**
|
681 |
+
* Instructions instruction.
|
682 |
+
* @member {string} instruction
|
683 |
+
* @memberof ChatMessage.Instructions
|
684 |
+
* @instance
|
685 |
+
*/
|
686 |
+
Instructions.prototype.instruction = "";
|
687 |
+
|
688 |
+
/**
|
689 |
+
* Creates a new Instructions instance using the specified properties.
|
690 |
+
* @function create
|
691 |
+
* @memberof ChatMessage.Instructions
|
692 |
+
* @static
|
693 |
+
* @param {ChatMessage.IInstructions=} [properties] Properties to set
|
694 |
+
* @returns {ChatMessage.Instructions} Instructions instance
|
695 |
+
*/
|
696 |
+
Instructions.create = function create(properties) {
|
697 |
+
return new Instructions(properties);
|
698 |
+
};
|
699 |
+
|
700 |
+
/**
|
701 |
+
* Encodes the specified Instructions message. Does not implicitly {@link ChatMessage.Instructions.verify|verify} messages.
|
702 |
+
* @function encode
|
703 |
+
* @memberof ChatMessage.Instructions
|
704 |
+
* @static
|
705 |
+
* @param {ChatMessage.IInstructions} message Instructions message or plain object to encode
|
706 |
+
* @param {$protobuf.Writer} [writer] Writer to encode to
|
707 |
+
* @returns {$protobuf.Writer} Writer
|
708 |
+
*/
|
709 |
+
Instructions.encode = function encode(message, writer) {
|
710 |
+
if (!writer)
|
711 |
+
writer = $Writer.create();
|
712 |
+
if (message.instruction != null && Object.hasOwnProperty.call(message, "instruction"))
|
713 |
+
writer.uint32(/* id 1, wireType 2 =*/10).string(message.instruction);
|
714 |
+
return writer;
|
715 |
+
};
|
716 |
+
|
717 |
+
/**
|
718 |
+
* Encodes the specified Instructions message, length delimited. Does not implicitly {@link ChatMessage.Instructions.verify|verify} messages.
|
719 |
+
* @function encodeDelimited
|
720 |
+
* @memberof ChatMessage.Instructions
|
721 |
+
* @static
|
722 |
+
* @param {ChatMessage.IInstructions} message Instructions message or plain object to encode
|
723 |
+
* @param {$protobuf.Writer} [writer] Writer to encode to
|
724 |
+
* @returns {$protobuf.Writer} Writer
|
725 |
+
*/
|
726 |
+
Instructions.encodeDelimited = function encodeDelimited(message, writer) {
|
727 |
+
return this.encode(message, writer).ldelim();
|
728 |
+
};
|
729 |
+
|
730 |
+
/**
|
731 |
+
* Decodes an Instructions message from the specified reader or buffer.
|
732 |
+
* @function decode
|
733 |
+
* @memberof ChatMessage.Instructions
|
734 |
+
* @static
|
735 |
+
* @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
|
736 |
+
* @param {number} [length] Message length if known beforehand
|
737 |
+
* @returns {ChatMessage.Instructions} Instructions
|
738 |
+
* @throws {Error} If the payload is not a reader or valid buffer
|
739 |
+
* @throws {$protobuf.util.ProtocolError} If required fields are missing
|
740 |
+
*/
|
741 |
+
Instructions.decode = function decode(reader, length) {
|
742 |
+
if (!(reader instanceof $Reader))
|
743 |
+
reader = $Reader.create(reader);
|
744 |
+
var end = length === undefined ? reader.len : reader.pos + length, message = new $root.ChatMessage.Instructions();
|
745 |
+
while (reader.pos < end) {
|
746 |
+
var tag = reader.uint32();
|
747 |
+
switch (tag >>> 3) {
|
748 |
+
case 1: {
|
749 |
+
message.instruction = reader.string();
|
750 |
+
break;
|
751 |
+
}
|
752 |
+
default:
|
753 |
+
reader.skipType(tag & 7);
|
754 |
+
break;
|
755 |
+
}
|
756 |
+
}
|
757 |
+
return message;
|
758 |
+
};
|
759 |
+
|
760 |
+
/**
|
761 |
+
* Decodes an Instructions message from the specified reader or buffer, length delimited.
|
762 |
+
* @function decodeDelimited
|
763 |
+
* @memberof ChatMessage.Instructions
|
764 |
+
* @static
|
765 |
+
* @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
|
766 |
+
* @returns {ChatMessage.Instructions} Instructions
|
767 |
+
* @throws {Error} If the payload is not a reader or valid buffer
|
768 |
+
* @throws {$protobuf.util.ProtocolError} If required fields are missing
|
769 |
+
*/
|
770 |
+
Instructions.decodeDelimited = function decodeDelimited(reader) {
|
771 |
+
if (!(reader instanceof $Reader))
|
772 |
+
reader = new $Reader(reader);
|
773 |
+
return this.decode(reader, reader.uint32());
|
774 |
+
};
|
775 |
+
|
776 |
+
/**
|
777 |
+
* Verifies an Instructions message.
|
778 |
+
* @function verify
|
779 |
+
* @memberof ChatMessage.Instructions
|
780 |
+
* @static
|
781 |
+
* @param {Object.<string,*>} message Plain object to verify
|
782 |
+
* @returns {string|null} `null` if valid, otherwise the reason why it is not
|
783 |
+
*/
|
784 |
+
Instructions.verify = function verify(message) {
|
785 |
+
if (typeof message !== "object" || message === null)
|
786 |
+
return "object expected";
|
787 |
+
if (message.instruction != null && message.hasOwnProperty("instruction"))
|
788 |
+
if (!$util.isString(message.instruction))
|
789 |
+
return "instruction: string expected";
|
790 |
+
return null;
|
791 |
+
};
|
792 |
+
|
793 |
+
/**
|
794 |
+
* Creates an Instructions message from a plain object. Also converts values to their respective internal types.
|
795 |
+
* @function fromObject
|
796 |
+
* @memberof ChatMessage.Instructions
|
797 |
+
* @static
|
798 |
+
* @param {Object.<string,*>} object Plain object
|
799 |
+
* @returns {ChatMessage.Instructions} Instructions
|
800 |
+
*/
|
801 |
+
Instructions.fromObject = function fromObject(object) {
|
802 |
+
if (object instanceof $root.ChatMessage.Instructions)
|
803 |
+
return object;
|
804 |
+
var message = new $root.ChatMessage.Instructions();
|
805 |
+
if (object.instruction != null)
|
806 |
+
message.instruction = String(object.instruction);
|
807 |
+
return message;
|
808 |
+
};
|
809 |
+
|
810 |
+
/**
|
811 |
+
* Creates a plain object from an Instructions message. Also converts values to other types if specified.
|
812 |
+
* @function toObject
|
813 |
+
* @memberof ChatMessage.Instructions
|
814 |
+
* @static
|
815 |
+
* @param {ChatMessage.Instructions} message Instructions
|
816 |
+
* @param {$protobuf.IConversionOptions} [options] Conversion options
|
817 |
+
* @returns {Object.<string,*>} Plain object
|
818 |
+
*/
|
819 |
+
Instructions.toObject = function toObject(message, options) {
|
820 |
+
if (!options)
|
821 |
+
options = {};
|
822 |
+
var object = {};
|
823 |
+
if (options.defaults)
|
824 |
+
object.instruction = "";
|
825 |
+
if (message.instruction != null && message.hasOwnProperty("instruction"))
|
826 |
+
object.instruction = message.instruction;
|
827 |
+
return object;
|
828 |
+
};
|
829 |
+
|
830 |
+
/**
|
831 |
+
* Converts this Instructions to JSON.
|
832 |
+
* @function toJSON
|
833 |
+
* @memberof ChatMessage.Instructions
|
834 |
+
* @instance
|
835 |
+
* @returns {Object.<string,*>} JSON object
|
836 |
+
*/
|
837 |
+
Instructions.prototype.toJSON = function toJSON() {
|
838 |
+
return this.constructor.toObject(this, $protobuf.util.toJSONOptions);
|
839 |
+
};
|
840 |
+
|
841 |
+
/**
|
842 |
+
* Gets the default type url for Instructions
|
843 |
+
* @function getTypeUrl
|
844 |
+
* @memberof ChatMessage.Instructions
|
845 |
+
* @static
|
846 |
+
* @param {string} [typeUrlPrefix] your custom typeUrlPrefix(default "type.googleapis.com")
|
847 |
+
* @returns {string} The default type url
|
848 |
+
*/
|
849 |
+
Instructions.getTypeUrl = function getTypeUrl(typeUrlPrefix) {
|
850 |
+
if (typeUrlPrefix === undefined) {
|
851 |
+
typeUrlPrefix = "type.googleapis.com";
|
852 |
+
}
|
853 |
+
return typeUrlPrefix + "/ChatMessage.Instructions";
|
854 |
+
};
|
855 |
+
|
856 |
+
return Instructions;
|
857 |
+
})();
|
858 |
+
|
859 |
+
ChatMessage.Model = (function() {
|
860 |
+
|
861 |
+
/**
|
862 |
+
* Properties of a Model.
|
863 |
+
* @memberof ChatMessage
|
864 |
+
* @interface IModel
|
865 |
+
* @property {string|null} [name] Model name
|
866 |
+
* @property {Uint8Array|null} [empty] Model empty
|
867 |
+
*/
|
868 |
+
|
869 |
+
/**
|
870 |
+
* Constructs a new Model.
|
871 |
+
* @memberof ChatMessage
|
872 |
+
* @classdesc Represents a Model.
|
873 |
+
* @implements IModel
|
874 |
+
* @constructor
|
875 |
+
* @param {ChatMessage.IModel=} [properties] Properties to set
|
876 |
+
*/
|
877 |
+
function Model(properties) {
|
878 |
+
if (properties)
|
879 |
+
for (var keys = Object.keys(properties), i = 0; i < keys.length; ++i)
|
880 |
+
if (properties[keys[i]] != null)
|
881 |
+
this[keys[i]] = properties[keys[i]];
|
882 |
+
}
|
883 |
+
|
884 |
+
/**
|
885 |
+
* Model name.
|
886 |
+
* @member {string} name
|
887 |
+
* @memberof ChatMessage.Model
|
888 |
+
* @instance
|
889 |
+
*/
|
890 |
+
Model.prototype.name = "";
|
891 |
+
|
892 |
+
/**
|
893 |
+
* Model empty.
|
894 |
+
* @member {Uint8Array} empty
|
895 |
+
* @memberof ChatMessage.Model
|
896 |
+
* @instance
|
897 |
+
*/
|
898 |
+
Model.prototype.empty = $util.newBuffer([]);
|
899 |
+
|
900 |
+
/**
|
901 |
+
* Creates a new Model instance using the specified properties.
|
902 |
+
* @function create
|
903 |
+
* @memberof ChatMessage.Model
|
904 |
+
* @static
|
905 |
+
* @param {ChatMessage.IModel=} [properties] Properties to set
|
906 |
+
* @returns {ChatMessage.Model} Model instance
|
907 |
+
*/
|
908 |
+
Model.create = function create(properties) {
|
909 |
+
return new Model(properties);
|
910 |
+
};
|
911 |
+
|
912 |
+
/**
|
913 |
+
* Encodes the specified Model message. Does not implicitly {@link ChatMessage.Model.verify|verify} messages.
|
914 |
+
* @function encode
|
915 |
+
* @memberof ChatMessage.Model
|
916 |
+
* @static
|
917 |
+
* @param {ChatMessage.IModel} message Model message or plain object to encode
|
918 |
+
* @param {$protobuf.Writer} [writer] Writer to encode to
|
919 |
+
* @returns {$protobuf.Writer} Writer
|
920 |
+
*/
|
921 |
+
Model.encode = function encode(message, writer) {
|
922 |
+
if (!writer)
|
923 |
+
writer = $Writer.create();
|
924 |
+
if (message.name != null && Object.hasOwnProperty.call(message, "name"))
|
925 |
+
writer.uint32(/* id 1, wireType 2 =*/10).string(message.name);
|
926 |
+
if (message.empty != null && Object.hasOwnProperty.call(message, "empty"))
|
927 |
+
writer.uint32(/* id 4, wireType 2 =*/34).bytes(message.empty);
|
928 |
+
return writer;
|
929 |
+
};
|
930 |
+
|
931 |
+
/**
|
932 |
+
* Encodes the specified Model message, length delimited. Does not implicitly {@link ChatMessage.Model.verify|verify} messages.
|
933 |
+
* @function encodeDelimited
|
934 |
+
* @memberof ChatMessage.Model
|
935 |
+
* @static
|
936 |
+
* @param {ChatMessage.IModel} message Model message or plain object to encode
|
937 |
+
* @param {$protobuf.Writer} [writer] Writer to encode to
|
938 |
+
* @returns {$protobuf.Writer} Writer
|
939 |
+
*/
|
940 |
+
Model.encodeDelimited = function encodeDelimited(message, writer) {
|
941 |
+
return this.encode(message, writer).ldelim();
|
942 |
+
};
|
943 |
+
|
944 |
+
/**
|
945 |
+
* Decodes a Model message from the specified reader or buffer.
|
946 |
+
* @function decode
|
947 |
+
* @memberof ChatMessage.Model
|
948 |
+
* @static
|
949 |
+
* @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
|
950 |
+
* @param {number} [length] Message length if known beforehand
|
951 |
+
* @returns {ChatMessage.Model} Model
|
952 |
+
* @throws {Error} If the payload is not a reader or valid buffer
|
953 |
+
* @throws {$protobuf.util.ProtocolError} If required fields are missing
|
954 |
+
*/
|
955 |
+
Model.decode = function decode(reader, length) {
|
956 |
+
if (!(reader instanceof $Reader))
|
957 |
+
reader = $Reader.create(reader);
|
958 |
+
var end = length === undefined ? reader.len : reader.pos + length, message = new $root.ChatMessage.Model();
|
959 |
+
while (reader.pos < end) {
|
960 |
+
var tag = reader.uint32();
|
961 |
+
switch (tag >>> 3) {
|
962 |
+
case 1: {
|
963 |
+
message.name = reader.string();
|
964 |
+
break;
|
965 |
+
}
|
966 |
+
case 4: {
|
967 |
+
message.empty = reader.bytes();
|
968 |
+
break;
|
969 |
+
}
|
970 |
+
default:
|
971 |
+
reader.skipType(tag & 7);
|
972 |
+
break;
|
973 |
+
}
|
974 |
+
}
|
975 |
+
return message;
|
976 |
+
};
|
977 |
+
|
978 |
+
/**
|
979 |
+
* Decodes a Model message from the specified reader or buffer, length delimited.
|
980 |
+
* @function decodeDelimited
|
981 |
+
* @memberof ChatMessage.Model
|
982 |
+
* @static
|
983 |
+
* @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
|
984 |
+
* @returns {ChatMessage.Model} Model
|
985 |
+
* @throws {Error} If the payload is not a reader or valid buffer
|
986 |
+
* @throws {$protobuf.util.ProtocolError} If required fields are missing
|
987 |
+
*/
|
988 |
+
Model.decodeDelimited = function decodeDelimited(reader) {
|
989 |
+
if (!(reader instanceof $Reader))
|
990 |
+
reader = new $Reader(reader);
|
991 |
+
return this.decode(reader, reader.uint32());
|
992 |
+
};
|
993 |
+
|
994 |
+
/**
|
995 |
+
* Verifies a Model message.
|
996 |
+
* @function verify
|
997 |
+
* @memberof ChatMessage.Model
|
998 |
+
* @static
|
999 |
+
* @param {Object.<string,*>} message Plain object to verify
|
1000 |
+
* @returns {string|null} `null` if valid, otherwise the reason why it is not
|
1001 |
+
*/
|
1002 |
+
Model.verify = function verify(message) {
|
1003 |
+
if (typeof message !== "object" || message === null)
|
1004 |
+
return "object expected";
|
1005 |
+
if (message.name != null && message.hasOwnProperty("name"))
|
1006 |
+
if (!$util.isString(message.name))
|
1007 |
+
return "name: string expected";
|
1008 |
+
if (message.empty != null && message.hasOwnProperty("empty"))
|
1009 |
+
if (!(message.empty && typeof message.empty.length === "number" || $util.isString(message.empty)))
|
1010 |
+
return "empty: buffer expected";
|
1011 |
+
return null;
|
1012 |
+
};
|
1013 |
+
|
1014 |
+
/**
|
1015 |
+
* Creates a Model message from a plain object. Also converts values to their respective internal types.
|
1016 |
+
* @function fromObject
|
1017 |
+
* @memberof ChatMessage.Model
|
1018 |
+
* @static
|
1019 |
+
* @param {Object.<string,*>} object Plain object
|
1020 |
+
* @returns {ChatMessage.Model} Model
|
1021 |
+
*/
|
1022 |
+
Model.fromObject = function fromObject(object) {
|
1023 |
+
if (object instanceof $root.ChatMessage.Model)
|
1024 |
+
return object;
|
1025 |
+
var message = new $root.ChatMessage.Model();
|
1026 |
+
if (object.name != null)
|
1027 |
+
message.name = String(object.name);
|
1028 |
+
if (object.empty != null)
|
1029 |
+
if (typeof object.empty === "string")
|
1030 |
+
$util.base64.decode(object.empty, message.empty = $util.newBuffer($util.base64.length(object.empty)), 0);
|
1031 |
+
else if (object.empty.length >= 0)
|
1032 |
+
message.empty = object.empty;
|
1033 |
+
return message;
|
1034 |
+
};
|
1035 |
+
|
1036 |
+
/**
|
1037 |
+
* Creates a plain object from a Model message. Also converts values to other types if specified.
|
1038 |
+
* @function toObject
|
1039 |
+
* @memberof ChatMessage.Model
|
1040 |
+
* @static
|
1041 |
+
* @param {ChatMessage.Model} message Model
|
1042 |
+
* @param {$protobuf.IConversionOptions} [options] Conversion options
|
1043 |
+
* @returns {Object.<string,*>} Plain object
|
1044 |
+
*/
|
1045 |
+
Model.toObject = function toObject(message, options) {
|
1046 |
+
if (!options)
|
1047 |
+
options = {};
|
1048 |
+
var object = {};
|
1049 |
+
if (options.defaults) {
|
1050 |
+
object.name = "";
|
1051 |
+
if (options.bytes === String)
|
1052 |
+
object.empty = "";
|
1053 |
+
else {
|
1054 |
+
object.empty = [];
|
1055 |
+
if (options.bytes !== Array)
|
1056 |
+
object.empty = $util.newBuffer(object.empty);
|
1057 |
+
}
|
1058 |
+
}
|
1059 |
+
if (message.name != null && message.hasOwnProperty("name"))
|
1060 |
+
object.name = message.name;
|
1061 |
+
if (message.empty != null && message.hasOwnProperty("empty"))
|
1062 |
+
object.empty = options.bytes === String ? $util.base64.encode(message.empty, 0, message.empty.length) : options.bytes === Array ? Array.prototype.slice.call(message.empty) : message.empty;
|
1063 |
+
return object;
|
1064 |
+
};
|
1065 |
+
|
1066 |
+
/**
|
1067 |
+
* Converts this Model to JSON.
|
1068 |
+
* @function toJSON
|
1069 |
+
* @memberof ChatMessage.Model
|
1070 |
+
* @instance
|
1071 |
+
* @returns {Object.<string,*>} JSON object
|
1072 |
+
*/
|
1073 |
+
Model.prototype.toJSON = function toJSON() {
|
1074 |
+
return this.constructor.toObject(this, $protobuf.util.toJSONOptions);
|
1075 |
+
};
|
1076 |
+
|
1077 |
+
/**
|
1078 |
+
* Gets the default type url for Model
|
1079 |
+
* @function getTypeUrl
|
1080 |
+
* @memberof ChatMessage.Model
|
1081 |
+
* @static
|
1082 |
+
* @param {string} [typeUrlPrefix] your custom typeUrlPrefix(default "type.googleapis.com")
|
1083 |
+
* @returns {string} The default type url
|
1084 |
+
*/
|
1085 |
+
Model.getTypeUrl = function getTypeUrl(typeUrlPrefix) {
|
1086 |
+
if (typeUrlPrefix === undefined) {
|
1087 |
+
typeUrlPrefix = "type.googleapis.com";
|
1088 |
+
}
|
1089 |
+
return typeUrlPrefix + "/ChatMessage.Model";
|
1090 |
+
};
|
1091 |
+
|
1092 |
+
return Model;
|
1093 |
+
})();
|
1094 |
+
|
1095 |
+
return ChatMessage;
|
1096 |
+
})();
|
1097 |
+
|
1098 |
+
$root.ResMessage = (function() {
|
1099 |
+
|
1100 |
+
/**
|
1101 |
+
* Properties of a ResMessage.
|
1102 |
+
* @exports IResMessage
|
1103 |
+
* @interface IResMessage
|
1104 |
+
* @property {string|null} [content] ResMessage content
|
1105 |
+
* @property {Uint8Array|null} [empty] ResMessage empty
|
1106 |
+
* @property {string|null} [prompt] ResMessage prompt
|
1107 |
+
*/
|
1108 |
+
|
1109 |
+
/**
|
1110 |
+
* Constructs a new ResMessage.
|
1111 |
+
* @exports ResMessage
|
1112 |
+
* @classdesc Represents a ResMessage.
|
1113 |
+
* @implements IResMessage
|
1114 |
+
* @constructor
|
1115 |
+
* @param {IResMessage=} [properties] Properties to set
|
1116 |
+
*/
|
1117 |
+
function ResMessage(properties) {
|
1118 |
+
if (properties)
|
1119 |
+
for (var keys = Object.keys(properties), i = 0; i < keys.length; ++i)
|
1120 |
+
if (properties[keys[i]] != null)
|
1121 |
+
this[keys[i]] = properties[keys[i]];
|
1122 |
+
}
|
1123 |
+
|
1124 |
+
/**
|
1125 |
+
* ResMessage content.
|
1126 |
+
* @member {string} content
|
1127 |
+
* @memberof ResMessage
|
1128 |
+
* @instance
|
1129 |
+
*/
|
1130 |
+
ResMessage.prototype.content = "";
|
1131 |
+
|
1132 |
+
/**
|
1133 |
+
* ResMessage empty.
|
1134 |
+
* @member {Uint8Array} empty
|
1135 |
+
* @memberof ResMessage
|
1136 |
+
* @instance
|
1137 |
+
*/
|
1138 |
+
ResMessage.prototype.empty = $util.newBuffer([]);
|
1139 |
+
|
1140 |
+
/**
|
1141 |
+
* ResMessage prompt.
|
1142 |
+
* @member {string} prompt
|
1143 |
+
* @memberof ResMessage
|
1144 |
+
* @instance
|
1145 |
+
*/
|
1146 |
+
ResMessage.prototype.prompt = "";
|
1147 |
+
|
1148 |
+
/**
|
1149 |
+
* Creates a new ResMessage instance using the specified properties.
|
1150 |
+
* @function create
|
1151 |
+
* @memberof ResMessage
|
1152 |
+
* @static
|
1153 |
+
* @param {IResMessage=} [properties] Properties to set
|
1154 |
+
* @returns {ResMessage} ResMessage instance
|
1155 |
+
*/
|
1156 |
+
ResMessage.create = function create(properties) {
|
1157 |
+
return new ResMessage(properties);
|
1158 |
+
};
|
1159 |
+
|
1160 |
+
/**
|
1161 |
+
* Encodes the specified ResMessage message. Does not implicitly {@link ResMessage.verify|verify} messages.
|
1162 |
+
* @function encode
|
1163 |
+
* @memberof ResMessage
|
1164 |
+
* @static
|
1165 |
+
* @param {IResMessage} message ResMessage message or plain object to encode
|
1166 |
+
* @param {$protobuf.Writer} [writer] Writer to encode to
|
1167 |
+
* @returns {$protobuf.Writer} Writer
|
1168 |
+
*/
|
1169 |
+
ResMessage.encode = function encode(message, writer) {
|
1170 |
+
if (!writer)
|
1171 |
+
writer = $Writer.create();
|
1172 |
+
if (message.content != null && Object.hasOwnProperty.call(message, "content"))
|
1173 |
+
writer.uint32(/* id 1, wireType 2 =*/10).string(message.content);
|
1174 |
+
if (message.empty != null && Object.hasOwnProperty.call(message, "empty"))
|
1175 |
+
writer.uint32(/* id 4, wireType 2 =*/34).bytes(message.empty);
|
1176 |
+
if (message.prompt != null && Object.hasOwnProperty.call(message, "prompt"))
|
1177 |
+
writer.uint32(/* id 5, wireType 2 =*/42).string(message.prompt);
|
1178 |
+
return writer;
|
1179 |
+
};
|
1180 |
+
|
1181 |
+
/**
|
1182 |
+
* Encodes the specified ResMessage message, length delimited. Does not implicitly {@link ResMessage.verify|verify} messages.
|
1183 |
+
* @function encodeDelimited
|
1184 |
+
* @memberof ResMessage
|
1185 |
+
* @static
|
1186 |
+
* @param {IResMessage} message ResMessage message or plain object to encode
|
1187 |
+
* @param {$protobuf.Writer} [writer] Writer to encode to
|
1188 |
+
* @returns {$protobuf.Writer} Writer
|
1189 |
+
*/
|
1190 |
+
ResMessage.encodeDelimited = function encodeDelimited(message, writer) {
|
1191 |
+
return this.encode(message, writer).ldelim();
|
1192 |
+
};
|
1193 |
+
|
1194 |
+
/**
|
1195 |
+
* Decodes a ResMessage message from the specified reader or buffer.
|
1196 |
+
* @function decode
|
1197 |
+
* @memberof ResMessage
|
1198 |
+
* @static
|
1199 |
+
* @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
|
1200 |
+
* @param {number} [length] Message length if known beforehand
|
1201 |
+
* @returns {ResMessage} ResMessage
|
1202 |
+
* @throws {Error} If the payload is not a reader or valid buffer
|
1203 |
+
* @throws {$protobuf.util.ProtocolError} If required fields are missing
|
1204 |
+
*/
|
1205 |
+
ResMessage.decode = function decode(reader, length) {
|
1206 |
+
if (!(reader instanceof $Reader))
|
1207 |
+
reader = $Reader.create(reader);
|
1208 |
+
var end = length === undefined ? reader.len : reader.pos + length, message = new $root.ResMessage();
|
1209 |
+
while (reader.pos < end) {
|
1210 |
+
var tag = reader.uint32();
|
1211 |
+
switch (tag >>> 3) {
|
1212 |
+
case 1: {
|
1213 |
+
message.content = reader.string();
|
1214 |
+
break;
|
1215 |
+
}
|
1216 |
+
case 4: {
|
1217 |
+
message.empty = reader.bytes();
|
1218 |
+
break;
|
1219 |
+
}
|
1220 |
+
case 5: {
|
1221 |
+
message.prompt = reader.string();
|
1222 |
+
break;
|
1223 |
+
}
|
1224 |
+
default:
|
1225 |
+
reader.skipType(tag & 7);
|
1226 |
+
break;
|
1227 |
+
}
|
1228 |
+
}
|
1229 |
+
return message;
|
1230 |
+
};
|
1231 |
+
|
1232 |
+
/**
|
1233 |
+
* Decodes a ResMessage message from the specified reader or buffer, length delimited.
|
1234 |
+
* @function decodeDelimited
|
1235 |
+
* @memberof ResMessage
|
1236 |
+
* @static
|
1237 |
+
* @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
|
1238 |
+
* @returns {ResMessage} ResMessage
|
1239 |
+
* @throws {Error} If the payload is not a reader or valid buffer
|
1240 |
+
* @throws {$protobuf.util.ProtocolError} If required fields are missing
|
1241 |
+
*/
|
1242 |
+
ResMessage.decodeDelimited = function decodeDelimited(reader) {
|
1243 |
+
if (!(reader instanceof $Reader))
|
1244 |
+
reader = new $Reader(reader);
|
1245 |
+
return this.decode(reader, reader.uint32());
|
1246 |
+
};
|
1247 |
+
|
1248 |
+
/**
|
1249 |
+
* Verifies a ResMessage message.
|
1250 |
+
* @function verify
|
1251 |
+
* @memberof ResMessage
|
1252 |
+
* @static
|
1253 |
+
* @param {Object.<string,*>} message Plain object to verify
|
1254 |
+
* @returns {string|null} `null` if valid, otherwise the reason why it is not
|
1255 |
+
*/
|
1256 |
+
ResMessage.verify = function verify(message) {
|
1257 |
+
if (typeof message !== "object" || message === null)
|
1258 |
+
return "object expected";
|
1259 |
+
if (message.content != null && message.hasOwnProperty("content"))
|
1260 |
+
if (!$util.isString(message.content))
|
1261 |
+
return "content: string expected";
|
1262 |
+
if (message.empty != null && message.hasOwnProperty("empty"))
|
1263 |
+
if (!(message.empty && typeof message.empty.length === "number" || $util.isString(message.empty)))
|
1264 |
+
return "empty: buffer expected";
|
1265 |
+
if (message.prompt != null && message.hasOwnProperty("prompt"))
|
1266 |
+
if (!$util.isString(message.prompt))
|
1267 |
+
return "prompt: string expected";
|
1268 |
+
return null;
|
1269 |
+
};
|
1270 |
+
|
1271 |
+
/**
|
1272 |
+
* Creates a ResMessage message from a plain object. Also converts values to their respective internal types.
|
1273 |
+
* @function fromObject
|
1274 |
+
* @memberof ResMessage
|
1275 |
+
* @static
|
1276 |
+
* @param {Object.<string,*>} object Plain object
|
1277 |
+
* @returns {ResMessage} ResMessage
|
1278 |
+
*/
|
1279 |
+
ResMessage.fromObject = function fromObject(object) {
|
1280 |
+
if (object instanceof $root.ResMessage)
|
1281 |
+
return object;
|
1282 |
+
var message = new $root.ResMessage();
|
1283 |
+
if (object.content != null)
|
1284 |
+
message.content = String(object.content);
|
1285 |
+
if (object.empty != null)
|
1286 |
+
if (typeof object.empty === "string")
|
1287 |
+
$util.base64.decode(object.empty, message.empty = $util.newBuffer($util.base64.length(object.empty)), 0);
|
1288 |
+
else if (object.empty.length >= 0)
|
1289 |
+
message.empty = object.empty;
|
1290 |
+
if (object.prompt != null)
|
1291 |
+
message.prompt = String(object.prompt);
|
1292 |
+
return message;
|
1293 |
+
};
|
1294 |
+
|
1295 |
+
/**
|
1296 |
+
* Creates a plain object from a ResMessage message. Also converts values to other types if specified.
|
1297 |
+
* @function toObject
|
1298 |
+
* @memberof ResMessage
|
1299 |
+
* @static
|
1300 |
+
* @param {ResMessage} message ResMessage
|
1301 |
+
* @param {$protobuf.IConversionOptions} [options] Conversion options
|
1302 |
+
* @returns {Object.<string,*>} Plain object
|
1303 |
+
*/
|
1304 |
+
ResMessage.toObject = function toObject(message, options) {
|
1305 |
+
if (!options)
|
1306 |
+
options = {};
|
1307 |
+
var object = {};
|
1308 |
+
if (options.defaults) {
|
1309 |
+
object.content = "";
|
1310 |
+
if (options.bytes === String)
|
1311 |
+
object.empty = "";
|
1312 |
+
else {
|
1313 |
+
object.empty = [];
|
1314 |
+
if (options.bytes !== Array)
|
1315 |
+
object.empty = $util.newBuffer(object.empty);
|
1316 |
+
}
|
1317 |
+
object.prompt = "";
|
1318 |
+
}
|
1319 |
+
if (message.content != null && message.hasOwnProperty("content"))
|
1320 |
+
object.content = message.content;
|
1321 |
+
if (message.empty != null && message.hasOwnProperty("empty"))
|
1322 |
+
object.empty = options.bytes === String ? $util.base64.encode(message.empty, 0, message.empty.length) : options.bytes === Array ? Array.prototype.slice.call(message.empty) : message.empty;
|
1323 |
+
if (message.prompt != null && message.hasOwnProperty("prompt"))
|
1324 |
+
object.prompt = message.prompt;
|
1325 |
+
return object;
|
1326 |
+
};
|
1327 |
+
|
1328 |
+
/**
|
1329 |
+
* Converts this ResMessage to JSON.
|
1330 |
+
* @function toJSON
|
1331 |
+
* @memberof ResMessage
|
1332 |
+
* @instance
|
1333 |
+
* @returns {Object.<string,*>} JSON object
|
1334 |
+
*/
|
1335 |
+
ResMessage.prototype.toJSON = function toJSON() {
|
1336 |
+
return this.constructor.toObject(this, $protobuf.util.toJSONOptions);
|
1337 |
+
};
|
1338 |
+
|
1339 |
+
/**
|
1340 |
+
* Gets the default type url for ResMessage
|
1341 |
+
* @function getTypeUrl
|
1342 |
+
* @memberof ResMessage
|
1343 |
+
* @static
|
1344 |
+
* @param {string} [typeUrlPrefix] your custom typeUrlPrefix(default "type.googleapis.com")
|
1345 |
+
* @returns {string} The default type url
|
1346 |
+
*/
|
1347 |
+
ResMessage.getTypeUrl = function getTypeUrl(typeUrlPrefix) {
|
1348 |
+
if (typeUrlPrefix === undefined) {
|
1349 |
+
typeUrlPrefix = "type.googleapis.com";
|
1350 |
+
}
|
1351 |
+
return typeUrlPrefix + "/ResMessage";
|
1352 |
+
};
|
1353 |
+
|
1354 |
+
return ResMessage;
|
1355 |
+
})();
|
1356 |
+
|
1357 |
+
$root.AvailableModelsResponse = (function() {
|
1358 |
+
|
1359 |
+
/**
|
1360 |
+
* Properties of an AvailableModelsResponse.
|
1361 |
+
* @exports IAvailableModelsResponse
|
1362 |
+
* @interface IAvailableModelsResponse
|
1363 |
+
* @property {Array.<AvailableModelsResponse.IAvailableModel>|null} [models] AvailableModelsResponse models
|
1364 |
+
* @property {Array.<string>|null} [modelNames] AvailableModelsResponse modelNames
|
1365 |
+
*/
|
1366 |
+
|
1367 |
+
/**
|
1368 |
+
* Constructs a new AvailableModelsResponse.
|
1369 |
+
* @exports AvailableModelsResponse
|
1370 |
+
* @classdesc Represents an AvailableModelsResponse.
|
1371 |
+
* @implements IAvailableModelsResponse
|
1372 |
+
* @constructor
|
1373 |
+
* @param {IAvailableModelsResponse=} [properties] Properties to set
|
1374 |
+
*/
|
1375 |
+
function AvailableModelsResponse(properties) {
|
1376 |
+
this.models = [];
|
1377 |
+
this.modelNames = [];
|
1378 |
+
if (properties)
|
1379 |
+
for (var keys = Object.keys(properties), i = 0; i < keys.length; ++i)
|
1380 |
+
if (properties[keys[i]] != null)
|
1381 |
+
this[keys[i]] = properties[keys[i]];
|
1382 |
+
}
|
1383 |
+
|
1384 |
+
/**
|
1385 |
+
* AvailableModelsResponse models.
|
1386 |
+
* @member {Array.<AvailableModelsResponse.IAvailableModel>} models
|
1387 |
+
* @memberof AvailableModelsResponse
|
1388 |
+
* @instance
|
1389 |
+
*/
|
1390 |
+
AvailableModelsResponse.prototype.models = $util.emptyArray;
|
1391 |
+
|
1392 |
+
/**
|
1393 |
+
* AvailableModelsResponse modelNames.
|
1394 |
+
* @member {Array.<string>} modelNames
|
1395 |
+
* @memberof AvailableModelsResponse
|
1396 |
+
* @instance
|
1397 |
+
*/
|
1398 |
+
AvailableModelsResponse.prototype.modelNames = $util.emptyArray;
|
1399 |
+
|
1400 |
+
/**
|
1401 |
+
* Creates a new AvailableModelsResponse instance using the specified properties.
|
1402 |
+
* @function create
|
1403 |
+
* @memberof AvailableModelsResponse
|
1404 |
+
* @static
|
1405 |
+
* @param {IAvailableModelsResponse=} [properties] Properties to set
|
1406 |
+
* @returns {AvailableModelsResponse} AvailableModelsResponse instance
|
1407 |
+
*/
|
1408 |
+
AvailableModelsResponse.create = function create(properties) {
|
1409 |
+
return new AvailableModelsResponse(properties);
|
1410 |
+
};
|
1411 |
+
|
1412 |
+
/**
|
1413 |
+
* Encodes the specified AvailableModelsResponse message. Does not implicitly {@link AvailableModelsResponse.verify|verify} messages.
|
1414 |
+
* @function encode
|
1415 |
+
* @memberof AvailableModelsResponse
|
1416 |
+
* @static
|
1417 |
+
* @param {IAvailableModelsResponse} message AvailableModelsResponse message or plain object to encode
|
1418 |
+
* @param {$protobuf.Writer} [writer] Writer to encode to
|
1419 |
+
* @returns {$protobuf.Writer} Writer
|
1420 |
+
*/
|
1421 |
+
AvailableModelsResponse.encode = function encode(message, writer) {
|
1422 |
+
if (!writer)
|
1423 |
+
writer = $Writer.create();
|
1424 |
+
if (message.modelNames != null && message.modelNames.length)
|
1425 |
+
for (var i = 0; i < message.modelNames.length; ++i)
|
1426 |
+
writer.uint32(/* id 1, wireType 2 =*/10).string(message.modelNames[i]);
|
1427 |
+
if (message.models != null && message.models.length)
|
1428 |
+
for (var i = 0; i < message.models.length; ++i)
|
1429 |
+
$root.AvailableModelsResponse.AvailableModel.encode(message.models[i], writer.uint32(/* id 2, wireType 2 =*/18).fork()).ldelim();
|
1430 |
+
return writer;
|
1431 |
+
};
|
1432 |
+
|
1433 |
+
/**
|
1434 |
+
* Encodes the specified AvailableModelsResponse message, length delimited. Does not implicitly {@link AvailableModelsResponse.verify|verify} messages.
|
1435 |
+
* @function encodeDelimited
|
1436 |
+
* @memberof AvailableModelsResponse
|
1437 |
+
* @static
|
1438 |
+
* @param {IAvailableModelsResponse} message AvailableModelsResponse message or plain object to encode
|
1439 |
+
* @param {$protobuf.Writer} [writer] Writer to encode to
|
1440 |
+
* @returns {$protobuf.Writer} Writer
|
1441 |
+
*/
|
1442 |
+
AvailableModelsResponse.encodeDelimited = function encodeDelimited(message, writer) {
|
1443 |
+
return this.encode(message, writer).ldelim();
|
1444 |
+
};
|
1445 |
+
|
1446 |
+
/**
|
1447 |
+
* Decodes an AvailableModelsResponse message from the specified reader or buffer.
|
1448 |
+
* @function decode
|
1449 |
+
* @memberof AvailableModelsResponse
|
1450 |
+
* @static
|
1451 |
+
* @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
|
1452 |
+
* @param {number} [length] Message length if known beforehand
|
1453 |
+
* @returns {AvailableModelsResponse} AvailableModelsResponse
|
1454 |
+
* @throws {Error} If the payload is not a reader or valid buffer
|
1455 |
+
* @throws {$protobuf.util.ProtocolError} If required fields are missing
|
1456 |
+
*/
|
1457 |
+
AvailableModelsResponse.decode = function decode(reader, length) {
|
1458 |
+
if (!(reader instanceof $Reader))
|
1459 |
+
reader = $Reader.create(reader);
|
1460 |
+
var end = length === undefined ? reader.len : reader.pos + length, message = new $root.AvailableModelsResponse();
|
1461 |
+
while (reader.pos < end) {
|
1462 |
+
var tag = reader.uint32();
|
1463 |
+
switch (tag >>> 3) {
|
1464 |
+
case 2: {
|
1465 |
+
if (!(message.models && message.models.length))
|
1466 |
+
message.models = [];
|
1467 |
+
message.models.push($root.AvailableModelsResponse.AvailableModel.decode(reader, reader.uint32()));
|
1468 |
+
break;
|
1469 |
+
}
|
1470 |
+
case 1: {
|
1471 |
+
if (!(message.modelNames && message.modelNames.length))
|
1472 |
+
message.modelNames = [];
|
1473 |
+
message.modelNames.push(reader.string());
|
1474 |
+
break;
|
1475 |
+
}
|
1476 |
+
default:
|
1477 |
+
reader.skipType(tag & 7);
|
1478 |
+
break;
|
1479 |
+
}
|
1480 |
+
}
|
1481 |
+
return message;
|
1482 |
+
};
|
1483 |
+
|
1484 |
+
/**
|
1485 |
+
* Decodes an AvailableModelsResponse message from the specified reader or buffer, length delimited.
|
1486 |
+
* @function decodeDelimited
|
1487 |
+
* @memberof AvailableModelsResponse
|
1488 |
+
* @static
|
1489 |
+
* @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
|
1490 |
+
* @returns {AvailableModelsResponse} AvailableModelsResponse
|
1491 |
+
* @throws {Error} If the payload is not a reader or valid buffer
|
1492 |
+
* @throws {$protobuf.util.ProtocolError} If required fields are missing
|
1493 |
+
*/
|
1494 |
+
AvailableModelsResponse.decodeDelimited = function decodeDelimited(reader) {
|
1495 |
+
if (!(reader instanceof $Reader))
|
1496 |
+
reader = new $Reader(reader);
|
1497 |
+
return this.decode(reader, reader.uint32());
|
1498 |
+
};
|
1499 |
+
|
1500 |
+
/**
|
1501 |
+
* Verifies an AvailableModelsResponse message.
|
1502 |
+
* @function verify
|
1503 |
+
* @memberof AvailableModelsResponse
|
1504 |
+
* @static
|
1505 |
+
* @param {Object.<string,*>} message Plain object to verify
|
1506 |
+
* @returns {string|null} `null` if valid, otherwise the reason why it is not
|
1507 |
+
*/
|
1508 |
+
AvailableModelsResponse.verify = function verify(message) {
|
1509 |
+
if (typeof message !== "object" || message === null)
|
1510 |
+
return "object expected";
|
1511 |
+
if (message.models != null && message.hasOwnProperty("models")) {
|
1512 |
+
if (!Array.isArray(message.models))
|
1513 |
+
return "models: array expected";
|
1514 |
+
for (var i = 0; i < message.models.length; ++i) {
|
1515 |
+
var error = $root.AvailableModelsResponse.AvailableModel.verify(message.models[i]);
|
1516 |
+
if (error)
|
1517 |
+
return "models." + error;
|
1518 |
+
}
|
1519 |
+
}
|
1520 |
+
if (message.modelNames != null && message.hasOwnProperty("modelNames")) {
|
1521 |
+
if (!Array.isArray(message.modelNames))
|
1522 |
+
return "modelNames: array expected";
|
1523 |
+
for (var i = 0; i < message.modelNames.length; ++i)
|
1524 |
+
if (!$util.isString(message.modelNames[i]))
|
1525 |
+
return "modelNames: string[] expected";
|
1526 |
+
}
|
1527 |
+
return null;
|
1528 |
+
};
|
1529 |
+
|
1530 |
+
/**
|
1531 |
+
* Creates an AvailableModelsResponse message from a plain object. Also converts values to their respective internal types.
|
1532 |
+
* @function fromObject
|
1533 |
+
* @memberof AvailableModelsResponse
|
1534 |
+
* @static
|
1535 |
+
* @param {Object.<string,*>} object Plain object
|
1536 |
+
* @returns {AvailableModelsResponse} AvailableModelsResponse
|
1537 |
+
*/
|
1538 |
+
AvailableModelsResponse.fromObject = function fromObject(object) {
|
1539 |
+
if (object instanceof $root.AvailableModelsResponse)
|
1540 |
+
return object;
|
1541 |
+
var message = new $root.AvailableModelsResponse();
|
1542 |
+
if (object.models) {
|
1543 |
+
if (!Array.isArray(object.models))
|
1544 |
+
throw TypeError(".AvailableModelsResponse.models: array expected");
|
1545 |
+
message.models = [];
|
1546 |
+
for (var i = 0; i < object.models.length; ++i) {
|
1547 |
+
if (typeof object.models[i] !== "object")
|
1548 |
+
throw TypeError(".AvailableModelsResponse.models: object expected");
|
1549 |
+
message.models[i] = $root.AvailableModelsResponse.AvailableModel.fromObject(object.models[i]);
|
1550 |
+
}
|
1551 |
+
}
|
1552 |
+
if (object.modelNames) {
|
1553 |
+
if (!Array.isArray(object.modelNames))
|
1554 |
+
throw TypeError(".AvailableModelsResponse.modelNames: array expected");
|
1555 |
+
message.modelNames = [];
|
1556 |
+
for (var i = 0; i < object.modelNames.length; ++i)
|
1557 |
+
message.modelNames[i] = String(object.modelNames[i]);
|
1558 |
+
}
|
1559 |
+
return message;
|
1560 |
+
};
|
1561 |
+
|
1562 |
+
/**
|
1563 |
+
* Creates a plain object from an AvailableModelsResponse message. Also converts values to other types if specified.
|
1564 |
+
* @function toObject
|
1565 |
+
* @memberof AvailableModelsResponse
|
1566 |
+
* @static
|
1567 |
+
* @param {AvailableModelsResponse} message AvailableModelsResponse
|
1568 |
+
* @param {$protobuf.IConversionOptions} [options] Conversion options
|
1569 |
+
* @returns {Object.<string,*>} Plain object
|
1570 |
+
*/
|
1571 |
+
AvailableModelsResponse.toObject = function toObject(message, options) {
|
1572 |
+
if (!options)
|
1573 |
+
options = {};
|
1574 |
+
var object = {};
|
1575 |
+
if (options.arrays || options.defaults) {
|
1576 |
+
object.modelNames = [];
|
1577 |
+
object.models = [];
|
1578 |
+
}
|
1579 |
+
if (message.modelNames && message.modelNames.length) {
|
1580 |
+
object.modelNames = [];
|
1581 |
+
for (var j = 0; j < message.modelNames.length; ++j)
|
1582 |
+
object.modelNames[j] = message.modelNames[j];
|
1583 |
+
}
|
1584 |
+
if (message.models && message.models.length) {
|
1585 |
+
object.models = [];
|
1586 |
+
for (var j = 0; j < message.models.length; ++j)
|
1587 |
+
object.models[j] = $root.AvailableModelsResponse.AvailableModel.toObject(message.models[j], options);
|
1588 |
+
}
|
1589 |
+
return object;
|
1590 |
+
};
|
1591 |
+
|
1592 |
+
/**
|
1593 |
+
* Converts this AvailableModelsResponse to JSON.
|
1594 |
+
* @function toJSON
|
1595 |
+
* @memberof AvailableModelsResponse
|
1596 |
+
* @instance
|
1597 |
+
* @returns {Object.<string,*>} JSON object
|
1598 |
+
*/
|
1599 |
+
AvailableModelsResponse.prototype.toJSON = function toJSON() {
|
1600 |
+
return this.constructor.toObject(this, $protobuf.util.toJSONOptions);
|
1601 |
+
};
|
1602 |
+
|
1603 |
+
/**
|
1604 |
+
* Gets the default type url for AvailableModelsResponse
|
1605 |
+
* @function getTypeUrl
|
1606 |
+
* @memberof AvailableModelsResponse
|
1607 |
+
* @static
|
1608 |
+
* @param {string} [typeUrlPrefix] your custom typeUrlPrefix(default "type.googleapis.com")
|
1609 |
+
* @returns {string} The default type url
|
1610 |
+
*/
|
1611 |
+
AvailableModelsResponse.getTypeUrl = function getTypeUrl(typeUrlPrefix) {
|
1612 |
+
if (typeUrlPrefix === undefined) {
|
1613 |
+
typeUrlPrefix = "type.googleapis.com";
|
1614 |
+
}
|
1615 |
+
return typeUrlPrefix + "/AvailableModelsResponse";
|
1616 |
+
};
|
1617 |
+
|
1618 |
+
AvailableModelsResponse.AvailableModel = (function() {
|
1619 |
+
|
1620 |
+
/**
|
1621 |
+
* Properties of an AvailableModel.
|
1622 |
+
* @memberof AvailableModelsResponse
|
1623 |
+
* @interface IAvailableModel
|
1624 |
+
* @property {string|null} [name] AvailableModel name
|
1625 |
+
* @property {boolean|null} [defaultOn] AvailableModel defaultOn
|
1626 |
+
* @property {boolean|null} [isLongContextOnly] AvailableModel isLongContextOnly
|
1627 |
+
* @property {boolean|null} [isChatOnly] AvailableModel isChatOnly
|
1628 |
+
*/
|
1629 |
+
|
1630 |
+
/**
|
1631 |
+
* Constructs a new AvailableModel.
|
1632 |
+
* @memberof AvailableModelsResponse
|
1633 |
+
* @classdesc Represents an AvailableModel.
|
1634 |
+
* @implements IAvailableModel
|
1635 |
+
* @constructor
|
1636 |
+
* @param {AvailableModelsResponse.IAvailableModel=} [properties] Properties to set
|
1637 |
+
*/
|
1638 |
+
function AvailableModel(properties) {
|
1639 |
+
if (properties)
|
1640 |
+
for (var keys = Object.keys(properties), i = 0; i < keys.length; ++i)
|
1641 |
+
if (properties[keys[i]] != null)
|
1642 |
+
this[keys[i]] = properties[keys[i]];
|
1643 |
+
}
|
1644 |
+
|
1645 |
+
/**
|
1646 |
+
* AvailableModel name.
|
1647 |
+
* @member {string} name
|
1648 |
+
* @memberof AvailableModelsResponse.AvailableModel
|
1649 |
+
* @instance
|
1650 |
+
*/
|
1651 |
+
AvailableModel.prototype.name = "";
|
1652 |
+
|
1653 |
+
/**
|
1654 |
+
* AvailableModel defaultOn.
|
1655 |
+
* @member {boolean} defaultOn
|
1656 |
+
* @memberof AvailableModelsResponse.AvailableModel
|
1657 |
+
* @instance
|
1658 |
+
*/
|
1659 |
+
AvailableModel.prototype.defaultOn = false;
|
1660 |
+
|
1661 |
+
/**
|
1662 |
+
* AvailableModel isLongContextOnly.
|
1663 |
+
* @member {boolean|null|undefined} isLongContextOnly
|
1664 |
+
* @memberof AvailableModelsResponse.AvailableModel
|
1665 |
+
* @instance
|
1666 |
+
*/
|
1667 |
+
AvailableModel.prototype.isLongContextOnly = null;
|
1668 |
+
|
1669 |
+
/**
|
1670 |
+
* AvailableModel isChatOnly.
|
1671 |
+
* @member {boolean|null|undefined} isChatOnly
|
1672 |
+
* @memberof AvailableModelsResponse.AvailableModel
|
1673 |
+
* @instance
|
1674 |
+
*/
|
1675 |
+
AvailableModel.prototype.isChatOnly = null;
|
1676 |
+
|
1677 |
+
// OneOf field names bound to virtual getters and setters
|
1678 |
+
var $oneOfFields;
|
1679 |
+
|
1680 |
+
// Virtual OneOf for proto3 optional field
|
1681 |
+
Object.defineProperty(AvailableModel.prototype, "_isLongContextOnly", {
|
1682 |
+
get: $util.oneOfGetter($oneOfFields = ["isLongContextOnly"]),
|
1683 |
+
set: $util.oneOfSetter($oneOfFields)
|
1684 |
+
});
|
1685 |
+
|
1686 |
+
// Virtual OneOf for proto3 optional field
|
1687 |
+
Object.defineProperty(AvailableModel.prototype, "_isChatOnly", {
|
1688 |
+
get: $util.oneOfGetter($oneOfFields = ["isChatOnly"]),
|
1689 |
+
set: $util.oneOfSetter($oneOfFields)
|
1690 |
+
});
|
1691 |
+
|
1692 |
+
/**
|
1693 |
+
* Creates a new AvailableModel instance using the specified properties.
|
1694 |
+
* @function create
|
1695 |
+
* @memberof AvailableModelsResponse.AvailableModel
|
1696 |
+
* @static
|
1697 |
+
* @param {AvailableModelsResponse.IAvailableModel=} [properties] Properties to set
|
1698 |
+
* @returns {AvailableModelsResponse.AvailableModel} AvailableModel instance
|
1699 |
+
*/
|
1700 |
+
AvailableModel.create = function create(properties) {
|
1701 |
+
return new AvailableModel(properties);
|
1702 |
+
};
|
1703 |
+
|
1704 |
+
/**
|
1705 |
+
* Encodes the specified AvailableModel message. Does not implicitly {@link AvailableModelsResponse.AvailableModel.verify|verify} messages.
|
1706 |
+
* @function encode
|
1707 |
+
* @memberof AvailableModelsResponse.AvailableModel
|
1708 |
+
* @static
|
1709 |
+
* @param {AvailableModelsResponse.IAvailableModel} message AvailableModel message or plain object to encode
|
1710 |
+
* @param {$protobuf.Writer} [writer] Writer to encode to
|
1711 |
+
* @returns {$protobuf.Writer} Writer
|
1712 |
+
*/
|
1713 |
+
AvailableModel.encode = function encode(message, writer) {
|
1714 |
+
if (!writer)
|
1715 |
+
writer = $Writer.create();
|
1716 |
+
if (message.name != null && Object.hasOwnProperty.call(message, "name"))
|
1717 |
+
writer.uint32(/* id 1, wireType 2 =*/10).string(message.name);
|
1718 |
+
if (message.defaultOn != null && Object.hasOwnProperty.call(message, "defaultOn"))
|
1719 |
+
writer.uint32(/* id 2, wireType 0 =*/16).bool(message.defaultOn);
|
1720 |
+
if (message.isLongContextOnly != null && Object.hasOwnProperty.call(message, "isLongContextOnly"))
|
1721 |
+
writer.uint32(/* id 3, wireType 0 =*/24).bool(message.isLongContextOnly);
|
1722 |
+
if (message.isChatOnly != null && Object.hasOwnProperty.call(message, "isChatOnly"))
|
1723 |
+
writer.uint32(/* id 4, wireType 0 =*/32).bool(message.isChatOnly);
|
1724 |
+
return writer;
|
1725 |
+
};
|
1726 |
+
|
1727 |
+
/**
|
1728 |
+
* Encodes the specified AvailableModel message, length delimited. Does not implicitly {@link AvailableModelsResponse.AvailableModel.verify|verify} messages.
|
1729 |
+
* @function encodeDelimited
|
1730 |
+
* @memberof AvailableModelsResponse.AvailableModel
|
1731 |
+
* @static
|
1732 |
+
* @param {AvailableModelsResponse.IAvailableModel} message AvailableModel message or plain object to encode
|
1733 |
+
* @param {$protobuf.Writer} [writer] Writer to encode to
|
1734 |
+
* @returns {$protobuf.Writer} Writer
|
1735 |
+
*/
|
1736 |
+
AvailableModel.encodeDelimited = function encodeDelimited(message, writer) {
|
1737 |
+
return this.encode(message, writer).ldelim();
|
1738 |
+
};
|
1739 |
+
|
1740 |
+
/**
|
1741 |
+
* Decodes an AvailableModel message from the specified reader or buffer.
|
1742 |
+
* @function decode
|
1743 |
+
* @memberof AvailableModelsResponse.AvailableModel
|
1744 |
+
* @static
|
1745 |
+
* @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
|
1746 |
+
* @param {number} [length] Message length if known beforehand
|
1747 |
+
* @returns {AvailableModelsResponse.AvailableModel} AvailableModel
|
1748 |
+
* @throws {Error} If the payload is not a reader or valid buffer
|
1749 |
+
* @throws {$protobuf.util.ProtocolError} If required fields are missing
|
1750 |
+
*/
|
1751 |
+
AvailableModel.decode = function decode(reader, length) {
|
1752 |
+
if (!(reader instanceof $Reader))
|
1753 |
+
reader = $Reader.create(reader);
|
1754 |
+
var end = length === undefined ? reader.len : reader.pos + length, message = new $root.AvailableModelsResponse.AvailableModel();
|
1755 |
+
while (reader.pos < end) {
|
1756 |
+
var tag = reader.uint32();
|
1757 |
+
switch (tag >>> 3) {
|
1758 |
+
case 1: {
|
1759 |
+
message.name = reader.string();
|
1760 |
+
break;
|
1761 |
+
}
|
1762 |
+
case 2: {
|
1763 |
+
message.defaultOn = reader.bool();
|
1764 |
+
break;
|
1765 |
+
}
|
1766 |
+
case 3: {
|
1767 |
+
message.isLongContextOnly = reader.bool();
|
1768 |
+
break;
|
1769 |
+
}
|
1770 |
+
case 4: {
|
1771 |
+
message.isChatOnly = reader.bool();
|
1772 |
+
break;
|
1773 |
+
}
|
1774 |
+
default:
|
1775 |
+
reader.skipType(tag & 7);
|
1776 |
+
break;
|
1777 |
+
}
|
1778 |
+
}
|
1779 |
+
return message;
|
1780 |
+
};
|
1781 |
+
|
1782 |
+
/**
|
1783 |
+
* Decodes an AvailableModel message from the specified reader or buffer, length delimited.
|
1784 |
+
* @function decodeDelimited
|
1785 |
+
* @memberof AvailableModelsResponse.AvailableModel
|
1786 |
+
* @static
|
1787 |
+
* @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
|
1788 |
+
* @returns {AvailableModelsResponse.AvailableModel} AvailableModel
|
1789 |
+
* @throws {Error} If the payload is not a reader or valid buffer
|
1790 |
+
* @throws {$protobuf.util.ProtocolError} If required fields are missing
|
1791 |
+
*/
|
1792 |
+
AvailableModel.decodeDelimited = function decodeDelimited(reader) {
|
1793 |
+
if (!(reader instanceof $Reader))
|
1794 |
+
reader = new $Reader(reader);
|
1795 |
+
return this.decode(reader, reader.uint32());
|
1796 |
+
};
|
1797 |
+
|
1798 |
+
/**
|
1799 |
+
* Verifies an AvailableModel message.
|
1800 |
+
* @function verify
|
1801 |
+
* @memberof AvailableModelsResponse.AvailableModel
|
1802 |
+
* @static
|
1803 |
+
* @param {Object.<string,*>} message Plain object to verify
|
1804 |
+
* @returns {string|null} `null` if valid, otherwise the reason why it is not
|
1805 |
+
*/
|
1806 |
+
AvailableModel.verify = function verify(message) {
|
1807 |
+
if (typeof message !== "object" || message === null)
|
1808 |
+
return "object expected";
|
1809 |
+
var properties = {};
|
1810 |
+
if (message.name != null && message.hasOwnProperty("name"))
|
1811 |
+
if (!$util.isString(message.name))
|
1812 |
+
return "name: string expected";
|
1813 |
+
if (message.defaultOn != null && message.hasOwnProperty("defaultOn"))
|
1814 |
+
if (typeof message.defaultOn !== "boolean")
|
1815 |
+
return "defaultOn: boolean expected";
|
1816 |
+
if (message.isLongContextOnly != null && message.hasOwnProperty("isLongContextOnly")) {
|
1817 |
+
properties._isLongContextOnly = 1;
|
1818 |
+
if (typeof message.isLongContextOnly !== "boolean")
|
1819 |
+
return "isLongContextOnly: boolean expected";
|
1820 |
+
}
|
1821 |
+
if (message.isChatOnly != null && message.hasOwnProperty("isChatOnly")) {
|
1822 |
+
properties._isChatOnly = 1;
|
1823 |
+
if (typeof message.isChatOnly !== "boolean")
|
1824 |
+
return "isChatOnly: boolean expected";
|
1825 |
+
}
|
1826 |
+
return null;
|
1827 |
+
};
|
1828 |
+
|
1829 |
+
/**
|
1830 |
+
* Creates an AvailableModel message from a plain object. Also converts values to their respective internal types.
|
1831 |
+
* @function fromObject
|
1832 |
+
* @memberof AvailableModelsResponse.AvailableModel
|
1833 |
+
* @static
|
1834 |
+
* @param {Object.<string,*>} object Plain object
|
1835 |
+
* @returns {AvailableModelsResponse.AvailableModel} AvailableModel
|
1836 |
+
*/
|
1837 |
+
AvailableModel.fromObject = function fromObject(object) {
|
1838 |
+
if (object instanceof $root.AvailableModelsResponse.AvailableModel)
|
1839 |
+
return object;
|
1840 |
+
var message = new $root.AvailableModelsResponse.AvailableModel();
|
1841 |
+
if (object.name != null)
|
1842 |
+
message.name = String(object.name);
|
1843 |
+
if (object.defaultOn != null)
|
1844 |
+
message.defaultOn = Boolean(object.defaultOn);
|
1845 |
+
if (object.isLongContextOnly != null)
|
1846 |
+
message.isLongContextOnly = Boolean(object.isLongContextOnly);
|
1847 |
+
if (object.isChatOnly != null)
|
1848 |
+
message.isChatOnly = Boolean(object.isChatOnly);
|
1849 |
+
return message;
|
1850 |
+
};
|
1851 |
+
|
1852 |
+
/**
|
1853 |
+
* Creates a plain object from an AvailableModel message. Also converts values to other types if specified.
|
1854 |
+
* @function toObject
|
1855 |
+
* @memberof AvailableModelsResponse.AvailableModel
|
1856 |
+
* @static
|
1857 |
+
* @param {AvailableModelsResponse.AvailableModel} message AvailableModel
|
1858 |
+
* @param {$protobuf.IConversionOptions} [options] Conversion options
|
1859 |
+
* @returns {Object.<string,*>} Plain object
|
1860 |
+
*/
|
1861 |
+
AvailableModel.toObject = function toObject(message, options) {
|
1862 |
+
if (!options)
|
1863 |
+
options = {};
|
1864 |
+
var object = {};
|
1865 |
+
if (options.defaults) {
|
1866 |
+
object.name = "";
|
1867 |
+
object.defaultOn = false;
|
1868 |
+
}
|
1869 |
+
if (message.name != null && message.hasOwnProperty("name"))
|
1870 |
+
object.name = message.name;
|
1871 |
+
if (message.defaultOn != null && message.hasOwnProperty("defaultOn"))
|
1872 |
+
object.defaultOn = message.defaultOn;
|
1873 |
+
if (message.isLongContextOnly != null && message.hasOwnProperty("isLongContextOnly")) {
|
1874 |
+
object.isLongContextOnly = message.isLongContextOnly;
|
1875 |
+
if (options.oneofs)
|
1876 |
+
object._isLongContextOnly = "isLongContextOnly";
|
1877 |
+
}
|
1878 |
+
if (message.isChatOnly != null && message.hasOwnProperty("isChatOnly")) {
|
1879 |
+
object.isChatOnly = message.isChatOnly;
|
1880 |
+
if (options.oneofs)
|
1881 |
+
object._isChatOnly = "isChatOnly";
|
1882 |
+
}
|
1883 |
+
return object;
|
1884 |
+
};
|
1885 |
+
|
1886 |
+
/**
|
1887 |
+
* Converts this AvailableModel to JSON.
|
1888 |
+
* @function toJSON
|
1889 |
+
* @memberof AvailableModelsResponse.AvailableModel
|
1890 |
+
* @instance
|
1891 |
+
* @returns {Object.<string,*>} JSON object
|
1892 |
+
*/
|
1893 |
+
AvailableModel.prototype.toJSON = function toJSON() {
|
1894 |
+
return this.constructor.toObject(this, $protobuf.util.toJSONOptions);
|
1895 |
+
};
|
1896 |
+
|
1897 |
+
/**
|
1898 |
+
* Gets the default type url for AvailableModel
|
1899 |
+
* @function getTypeUrl
|
1900 |
+
* @memberof AvailableModelsResponse.AvailableModel
|
1901 |
+
* @static
|
1902 |
+
* @param {string} [typeUrlPrefix] your custom typeUrlPrefix(default "type.googleapis.com")
|
1903 |
+
* @returns {string} The default type url
|
1904 |
+
*/
|
1905 |
+
AvailableModel.getTypeUrl = function getTypeUrl(typeUrlPrefix) {
|
1906 |
+
if (typeUrlPrefix === undefined) {
|
1907 |
+
typeUrlPrefix = "type.googleapis.com";
|
1908 |
+
}
|
1909 |
+
return typeUrlPrefix + "/AvailableModelsResponse.AvailableModel";
|
1910 |
+
};
|
1911 |
+
|
1912 |
+
return AvailableModel;
|
1913 |
+
})();
|
1914 |
+
|
1915 |
+
return AvailableModelsResponse;
|
1916 |
+
})();
|
1917 |
+
|
1918 |
+
module.exports = $root;
|
src/proto/message.proto
ADDED
@@ -0,0 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
syntax = "proto3";
|
2 |
+
|
3 |
+
message ChatMessage {
|
4 |
+
message UserMessage {
|
5 |
+
string content = 1;
|
6 |
+
int32 role = 2;
|
7 |
+
string messageId = 13;
|
8 |
+
}
|
9 |
+
|
10 |
+
message Instructions {
|
11 |
+
string instruction = 1;
|
12 |
+
}
|
13 |
+
|
14 |
+
message Model {
|
15 |
+
string name = 1;
|
16 |
+
bytes empty = 4;
|
17 |
+
}
|
18 |
+
|
19 |
+
repeated UserMessage userMessages = 2;
|
20 |
+
Instructions instructions = 4;
|
21 |
+
Model model = 7;
|
22 |
+
int32 unknown13 = 13;
|
23 |
+
string conversationId = 15;
|
24 |
+
int32 unknown16 = 16;
|
25 |
+
int32 unknown29 = 29;
|
26 |
+
int32 unknown31 = 31;
|
27 |
+
}
|
28 |
+
|
29 |
+
message ResMessage {
|
30 |
+
string content = 1;
|
31 |
+
bytes empty = 4;
|
32 |
+
string prompt = 5;
|
33 |
+
}
|
34 |
+
|
35 |
+
message AvailableModelsResponse {
|
36 |
+
message AvailableModel {
|
37 |
+
string name = 1;
|
38 |
+
bool default_on = 2;
|
39 |
+
optional bool is_long_context_only = 3;
|
40 |
+
optional bool is_chat_only = 4;
|
41 |
+
}
|
42 |
+
repeated AvailableModel models = 2;
|
43 |
+
repeated string model_names = 1;
|
44 |
+
}
|
src/public/index.html
ADDED
@@ -0,0 +1,921 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
<!DOCTYPE html>
|
2 |
+
<html lang="zh-CN">
|
3 |
+
<head>
|
4 |
+
<meta charset="UTF-8">
|
5 |
+
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
6 |
+
<title>Cursor To OpenAI - API Key 管理</title>
|
7 |
+
<style>
|
8 |
+
body {
|
9 |
+
font-family: 'Arial', sans-serif;
|
10 |
+
line-height: 1.6;
|
11 |
+
margin: 0;
|
12 |
+
padding: 20px;
|
13 |
+
color: #333;
|
14 |
+
max-width: 1200px;
|
15 |
+
margin: 0 auto;
|
16 |
+
}
|
17 |
+
h1, h2 {
|
18 |
+
color: #2c3e50;
|
19 |
+
}
|
20 |
+
.container {
|
21 |
+
display: flex;
|
22 |
+
flex-direction: column;
|
23 |
+
gap: 20px;
|
24 |
+
}
|
25 |
+
.card {
|
26 |
+
background: #fff;
|
27 |
+
border-radius: 8px;
|
28 |
+
box-shadow: 0 2px 10px rgba(0, 0, 0, 0.1);
|
29 |
+
padding: 20px;
|
30 |
+
}
|
31 |
+
.form-group {
|
32 |
+
margin-bottom: 15px;
|
33 |
+
}
|
34 |
+
label {
|
35 |
+
display: block;
|
36 |
+
margin-bottom: 5px;
|
37 |
+
font-weight: bold;
|
38 |
+
}
|
39 |
+
input, textarea {
|
40 |
+
width: 100%;
|
41 |
+
padding: 8px;
|
42 |
+
border: 1px solid #ddd;
|
43 |
+
border-radius: 4px;
|
44 |
+
font-size: 16px;
|
45 |
+
}
|
46 |
+
textarea {
|
47 |
+
min-height: 100px;
|
48 |
+
font-family: monospace;
|
49 |
+
}
|
50 |
+
button {
|
51 |
+
background: #3498db;
|
52 |
+
color: white;
|
53 |
+
border: none;
|
54 |
+
padding: 10px 15px;
|
55 |
+
border-radius: 4px;
|
56 |
+
cursor: pointer;
|
57 |
+
font-size: 16px;
|
58 |
+
}
|
59 |
+
button:hover {
|
60 |
+
background: #2980b9;
|
61 |
+
}
|
62 |
+
table {
|
63 |
+
width: 100%;
|
64 |
+
border-collapse: collapse;
|
65 |
+
}
|
66 |
+
th, td {
|
67 |
+
padding: 12px;
|
68 |
+
text-align: left;
|
69 |
+
border-bottom: 1px solid #ddd;
|
70 |
+
}
|
71 |
+
th {
|
72 |
+
background-color: #f2f2f2;
|
73 |
+
}
|
74 |
+
.action-btn {
|
75 |
+
background: #e74c3c;
|
76 |
+
margin-right: 5px;
|
77 |
+
}
|
78 |
+
.action-btn:hover {
|
79 |
+
background: #c0392b;
|
80 |
+
}
|
81 |
+
.edit-btn {
|
82 |
+
background: #f39c12;
|
83 |
+
margin-right: 5px;
|
84 |
+
}
|
85 |
+
.edit-btn:hover {
|
86 |
+
background: #d35400;
|
87 |
+
}
|
88 |
+
.info {
|
89 |
+
background-color: #d4edda;
|
90 |
+
color: #155724;
|
91 |
+
padding: 10px;
|
92 |
+
border-radius: 4px;
|
93 |
+
margin-bottom: 15px;
|
94 |
+
}
|
95 |
+
.error {
|
96 |
+
background-color: #f8d7da;
|
97 |
+
color: #721c24;
|
98 |
+
padding: 10px;
|
99 |
+
border-radius: 4px;
|
100 |
+
margin-bottom: 15px;
|
101 |
+
}
|
102 |
+
.modal {
|
103 |
+
display: none;
|
104 |
+
position: fixed;
|
105 |
+
z-index: 1;
|
106 |
+
left: 0;
|
107 |
+
top: 0;
|
108 |
+
width: 100%;
|
109 |
+
height: 100%;
|
110 |
+
overflow: auto;
|
111 |
+
background-color: rgba(0,0,0,0.4);
|
112 |
+
}
|
113 |
+
.modal-content {
|
114 |
+
background-color: #fefefe;
|
115 |
+
margin: 15% auto;
|
116 |
+
padding: 20px;
|
117 |
+
border: 1px solid #888;
|
118 |
+
width: 80%;
|
119 |
+
max-width: 600px;
|
120 |
+
border-radius: 8px;
|
121 |
+
}
|
122 |
+
.close {
|
123 |
+
color: #aaa;
|
124 |
+
float: right;
|
125 |
+
font-size: 28px;
|
126 |
+
font-weight: bold;
|
127 |
+
cursor: pointer;
|
128 |
+
}
|
129 |
+
.close:hover,
|
130 |
+
.close:focus {
|
131 |
+
color: black;
|
132 |
+
text-decoration: none;
|
133 |
+
}
|
134 |
+
/* 无效Cookie样式 */
|
135 |
+
.cookie-text {
|
136 |
+
max-width: 80%;
|
137 |
+
word-break: break-all;
|
138 |
+
}
|
139 |
+
</style>
|
140 |
+
</head>
|
141 |
+
<body>
|
142 |
+
<div class="container">
|
143 |
+
<div class="card">
|
144 |
+
<h1>Cursor To OpenAI - API Key 管理</h1>
|
145 |
+
<p>在此页面上,您可以管理自定义 API Key 与 Cursor Cookie 的映射关系。</p>
|
146 |
+
<div style="margin-top: 10px; display: flex; justify-content: space-between; align-items: center;">
|
147 |
+
<div>
|
148 |
+
<button id="testApiBtn" style="margin-right: 10px;">测试API连接</button>
|
149 |
+
<button id="clearCacheBtn">清除缓存并刷新</button>
|
150 |
+
</div>
|
151 |
+
<div>
|
152 |
+
<span id="adminUsername" style="margin-right: 10px;"></span>
|
153 |
+
<button id="logoutBtn" style="background: #e74c3c;">退出登录</button>
|
154 |
+
</div>
|
155 |
+
</div>
|
156 |
+
<div id="testApiResult" style="margin-top: 10px;"></div>
|
157 |
+
</div>
|
158 |
+
|
159 |
+
<div class="card">
|
160 |
+
<h2>添加/更新 API Key</h2>
|
161 |
+
<div id="addKeyMessage"></div>
|
162 |
+
<form id="addKeyForm">
|
163 |
+
<div class="form-group">
|
164 |
+
<label for="apiKey">API Key(自定义)</label>
|
165 |
+
<input type="text" id="apiKey" name="apiKey" placeholder="输入您想使用的自定义 API Key" required>
|
166 |
+
</div>
|
167 |
+
<div class="form-group">
|
168 |
+
<label for="cookieValues">Cursor Cookie 值(多个值请用逗号分隔)</label>
|
169 |
+
<textarea id="cookieValues" name="cookieValues" placeholder="输入 WorkosCursorSessionToken 值,多个值请用逗号分隔" required></textarea>
|
170 |
+
</div>
|
171 |
+
<button type="submit">保存</button>
|
172 |
+
</form>
|
173 |
+
</div>
|
174 |
+
|
175 |
+
<div class="card">
|
176 |
+
<h2>现有 API Key</h2>
|
177 |
+
<div id="keyListMessage"></div>
|
178 |
+
<table id="keyTable">
|
179 |
+
<thead>
|
180 |
+
<tr>
|
181 |
+
<th>API Key</th>
|
182 |
+
<th>Cookie 数量</th>
|
183 |
+
<th>操作</th>
|
184 |
+
</tr>
|
185 |
+
</thead>
|
186 |
+
<tbody id="keyList">
|
187 |
+
<!-- 数据将通过 JavaScript 动态加载 -->
|
188 |
+
</tbody>
|
189 |
+
</table>
|
190 |
+
</div>
|
191 |
+
|
192 |
+
<div class="card">
|
193 |
+
<h2>使用说明</h2>
|
194 |
+
<ol>
|
195 |
+
<li>添加自定义 API Key 和对应的 Cursor Cookie 值。</li>
|
196 |
+
<li>使用自定义 API Key 作为 OpenAI API 的认证凭证。</li>
|
197 |
+
<li>系统将自动在多个 Cookie 之间进行轮询。</li>
|
198 |
+
<li>API 端点:
|
199 |
+
<ul>
|
200 |
+
<li>模型列表:<code>/v1/models</code></li>
|
201 |
+
<li>聊天补全:<code>/v1/chat/completions</code></li>
|
202 |
+
</ul>
|
203 |
+
</li>
|
204 |
+
</ol>
|
205 |
+
</div>
|
206 |
+
</div>
|
207 |
+
|
208 |
+
<!-- 修改 Cookie 的模态框 -->
|
209 |
+
<div id="editModal" class="modal">
|
210 |
+
<div class="modal-content">
|
211 |
+
<span class="close">×</span>
|
212 |
+
<h2>修改 API Key 的 Cookie</h2>
|
213 |
+
<div id="editModalMessage"></div>
|
214 |
+
<form id="editCookieForm">
|
215 |
+
<input type="hidden" id="editApiKey" name="editApiKey">
|
216 |
+
<div class="form-group">
|
217 |
+
<label for="editCookieValues">Cursor Cookie 值(多个值请用逗号分隔)</label>
|
218 |
+
<textarea id="editCookieValues" name="editCookieValues" placeholder="输入 WorkosCursorSessionToken 值,多个值请用逗号分隔" required></textarea>
|
219 |
+
</div>
|
220 |
+
<button type="submit">保存修改</button>
|
221 |
+
</form>
|
222 |
+
</div>
|
223 |
+
</div>
|
224 |
+
|
225 |
+
<div class="card">
|
226 |
+
<h2>无效Cookie管理</h2>
|
227 |
+
<div class="form-group">
|
228 |
+
<div class="info">
|
229 |
+
以下是系统自动检测到的无效Cookie列表。这些Cookie在请求过程中被发现无效,已被自动从API Key中移除。
|
230 |
+
</div>
|
231 |
+
<div id="invalidCookiesContainer">
|
232 |
+
<div style="text-align: center; padding: 20px;">
|
233 |
+
<div>加载中...</div>
|
234 |
+
</div>
|
235 |
+
</div>
|
236 |
+
<button id="clearAllInvalidCookies" style="background: #e74c3c;">清除所有无效Cookie</button>
|
237 |
+
</div>
|
238 |
+
</div>
|
239 |
+
|
240 |
+
<!-- 新增:Cookie刷新功能 -->
|
241 |
+
<div class="card">
|
242 |
+
<h2>Cookie自动刷新</h2>
|
243 |
+
<div class="form-group">
|
244 |
+
<div class="info">
|
245 |
+
系统支持自动刷新Cookie,确保API Key始终有足够的可用Cookie。您可以在此手动触发刷新操作。
|
246 |
+
</div>
|
247 |
+
<div id="refreshCookieMessage"></div>
|
248 |
+
<div style="margin-top: 15px;">
|
249 |
+
<div class="form-group">
|
250 |
+
<label for="refreshApiKey">选择要刷新的API Key(不选则刷新所有)</label>
|
251 |
+
<select id="refreshApiKey" style="width: 100%; padding: 8px; border: 1px solid #ddd; border-radius: 4px; font-size: 16px;">
|
252 |
+
<option value="">所有API Key</option>
|
253 |
+
<!-- 选项将通过JavaScript动态加载 -->
|
254 |
+
</select>
|
255 |
+
</div>
|
256 |
+
<button id="refreshCookieBtn" style="background: #27ae60;">刷新Cookie</button>
|
257 |
+
</div>
|
258 |
+
<div id="refreshStatusContainer" style="margin-top: 15px; display: none;">
|
259 |
+
<div class="info">
|
260 |
+
<div>刷新状态:<span id="refreshStatus">准备中...</span></div>
|
261 |
+
<div style="margin-top: 10px;">
|
262 |
+
<progress id="refreshProgress" value="0" max="100" style="width: 100%;"></progress>
|
263 |
+
</div>
|
264 |
+
</div>
|
265 |
+
</div>
|
266 |
+
</div>
|
267 |
+
</div>
|
268 |
+
|
269 |
+
<script>
|
270 |
+
// 获取模态框元素
|
271 |
+
const modal = document.getElementById('editModal');
|
272 |
+
const closeBtn = document.getElementsByClassName('close')[0];
|
273 |
+
|
274 |
+
// 关闭模态框
|
275 |
+
closeBtn.onclick = function() {
|
276 |
+
modal.style.display = 'none';
|
277 |
+
}
|
278 |
+
|
279 |
+
// 点击模态框外部关闭
|
280 |
+
window.onclick = function(event) {
|
281 |
+
if (event.target == modal) {
|
282 |
+
modal.style.display = 'none';
|
283 |
+
}
|
284 |
+
}
|
285 |
+
|
286 |
+
// 加载现有 API Key
|
287 |
+
async function loadApiKeys() {
|
288 |
+
try {
|
289 |
+
console.log('开始加载API Keys...');
|
290 |
+
const response = await fetch('/v1/api-keys', {
|
291 |
+
method: 'GET',
|
292 |
+
headers: {
|
293 |
+
'Content-Type': 'application/json',
|
294 |
+
'Cache-Control': 'no-cache'
|
295 |
+
}
|
296 |
+
});
|
297 |
+
|
298 |
+
if (!response.ok) {
|
299 |
+
throw new Error(`HTTP错误: ${response.status} ${response.statusText}`);
|
300 |
+
}
|
301 |
+
|
302 |
+
console.log('API响应状态:', response.status);
|
303 |
+
const data = await response.json();
|
304 |
+
console.log('获取到的数据:', data);
|
305 |
+
|
306 |
+
const keyList = document.getElementById('keyList');
|
307 |
+
keyList.innerHTML = '';
|
308 |
+
|
309 |
+
if (data.success && data.apiKeys.length > 0) {
|
310 |
+
data.apiKeys.forEach(key => {
|
311 |
+
const row = document.createElement('tr');
|
312 |
+
row.innerHTML = `
|
313 |
+
<td>${key.key}</td>
|
314 |
+
<td>${key.cookieCount}</td>
|
315 |
+
<td>
|
316 |
+
<button class="edit-btn" onclick="editApiKey('${key.key}')">修改</button>
|
317 |
+
<button class="action-btn" onclick="deleteApiKey('${key.key}')">删除</button>
|
318 |
+
</td>
|
319 |
+
`;
|
320 |
+
keyList.appendChild(row);
|
321 |
+
});
|
322 |
+
} else {
|
323 |
+
keyList.innerHTML = '<tr><td colspan="3">暂无 API Key</td></tr>';
|
324 |
+
}
|
325 |
+
} catch (error) {
|
326 |
+
console.error('加载 API Key 失败:', error);
|
327 |
+
document.getElementById('keyListMessage').innerHTML = `
|
328 |
+
<div class="error">加载 API Key 失败: ${error.message}</div>
|
329 |
+
`;
|
330 |
+
}
|
331 |
+
}
|
332 |
+
|
333 |
+
// 添加/更新 API Key
|
334 |
+
document.getElementById('addKeyForm').addEventListener('submit', async function(e) {
|
335 |
+
e.preventDefault();
|
336 |
+
|
337 |
+
const apiKey = document.getElementById('apiKey').value.trim();
|
338 |
+
const cookieValuesText = document.getElementById('cookieValues').value.trim();
|
339 |
+
|
340 |
+
if (!apiKey || !cookieValuesText) {
|
341 |
+
document.getElementById('addKeyMessage').innerHTML = `
|
342 |
+
<div class="error">API Key 和 Cookie 值不能为空</div>
|
343 |
+
`;
|
344 |
+
return;
|
345 |
+
}
|
346 |
+
|
347 |
+
// 将逗号分隔的 Cookie 值转换为数组
|
348 |
+
const cookieValues = cookieValuesText.split(',').map(cookie => cookie.trim()).filter(cookie => cookie);
|
349 |
+
|
350 |
+
try {
|
351 |
+
const response = await fetch('/v1/api-keys', {
|
352 |
+
method: 'POST',
|
353 |
+
headers: {
|
354 |
+
'Content-Type': 'application/json',
|
355 |
+
},
|
356 |
+
body: JSON.stringify({
|
357 |
+
apiKey,
|
358 |
+
cookieValues,
|
359 |
+
}),
|
360 |
+
});
|
361 |
+
|
362 |
+
const data = await response.json();
|
363 |
+
|
364 |
+
if (data.success) {
|
365 |
+
document.getElementById('addKeyMessage').innerHTML = `
|
366 |
+
<div class="info">API Key 添加/更新成功</div>
|
367 |
+
`;
|
368 |
+
document.getElementById('apiKey').value = '';
|
369 |
+
document.getElementById('cookieValues').value = '';
|
370 |
+
loadApiKeys();
|
371 |
+
} else {
|
372 |
+
document.getElementById('addKeyMessage').innerHTML = `
|
373 |
+
<div class="error">API Key 添加/更新失败: ${data.error}</div>
|
374 |
+
`;
|
375 |
+
}
|
376 |
+
} catch (error) {
|
377 |
+
console.error('添加/更新 API Key 失败:', error);
|
378 |
+
document.getElementById('addKeyMessage').innerHTML = `
|
379 |
+
<div class="error">添加/更新 API Key 失败: ${error.message}</div>
|
380 |
+
`;
|
381 |
+
}
|
382 |
+
});
|
383 |
+
|
384 |
+
// 删除 API Key
|
385 |
+
async function deleteApiKey(apiKey) {
|
386 |
+
if (!confirm(`确定要删除 API Key "${apiKey}" 吗?`)) {
|
387 |
+
return;
|
388 |
+
}
|
389 |
+
|
390 |
+
try {
|
391 |
+
const response = await fetch(`/v1/api-keys/${encodeURIComponent(apiKey)}`, {
|
392 |
+
method: 'DELETE',
|
393 |
+
});
|
394 |
+
|
395 |
+
const data = await response.json();
|
396 |
+
|
397 |
+
if (data.success) {
|
398 |
+
document.getElementById('keyListMessage').innerHTML = `
|
399 |
+
<div class="info">API Key 删除成功</div>
|
400 |
+
`;
|
401 |
+
loadApiKeys();
|
402 |
+
} else {
|
403 |
+
document.getElementById('keyListMessage').innerHTML = `
|
404 |
+
<div class="error">API Key ���除失败: ${data.error}</div>
|
405 |
+
`;
|
406 |
+
}
|
407 |
+
} catch (error) {
|
408 |
+
console.error('删除 API Key 失败:', error);
|
409 |
+
document.getElementById('keyListMessage').innerHTML = `
|
410 |
+
<div class="error">删除 API Key 失败: ${error.message}</div>
|
411 |
+
`;
|
412 |
+
}
|
413 |
+
}
|
414 |
+
|
415 |
+
// 获取API Key的Cookie值
|
416 |
+
async function getCookiesForApiKey(apiKey) {
|
417 |
+
try {
|
418 |
+
const response = await fetch(`/v1/api-keys/${encodeURIComponent(apiKey)}/cookies`, {
|
419 |
+
method: 'GET',
|
420 |
+
headers: {
|
421 |
+
'Content-Type': 'application/json',
|
422 |
+
'Cache-Control': 'no-cache'
|
423 |
+
}
|
424 |
+
});
|
425 |
+
|
426 |
+
if (!response.ok) {
|
427 |
+
throw new Error(`HTTP错误: ${response.status} ${response.statusText}`);
|
428 |
+
}
|
429 |
+
|
430 |
+
const data = await response.json();
|
431 |
+
return data.cookies;
|
432 |
+
} catch (error) {
|
433 |
+
console.error(`获取 ${apiKey} 的Cookie值失败:`, error);
|
434 |
+
throw error;
|
435 |
+
}
|
436 |
+
}
|
437 |
+
|
438 |
+
// 修改 API Key
|
439 |
+
async function editApiKey(apiKey) {
|
440 |
+
try {
|
441 |
+
document.getElementById('editModalMessage').innerHTML = '';
|
442 |
+
document.getElementById('editApiKey').value = apiKey;
|
443 |
+
|
444 |
+
// 获取当前Cookie值
|
445 |
+
const cookies = await getCookiesForApiKey(apiKey);
|
446 |
+
document.getElementById('editCookieValues').value = cookies.join(',');
|
447 |
+
|
448 |
+
// 显示模态框
|
449 |
+
modal.style.display = 'block';
|
450 |
+
} catch (error) {
|
451 |
+
alert(`获取 ${apiKey} 的Cookie值失败: ${error.message}`);
|
452 |
+
}
|
453 |
+
}
|
454 |
+
|
455 |
+
// 提交修改表单
|
456 |
+
document.getElementById('editCookieForm').addEventListener('submit', async function(e) {
|
457 |
+
e.preventDefault();
|
458 |
+
|
459 |
+
const apiKey = document.getElementById('editApiKey').value.trim();
|
460 |
+
const cookieValuesText = document.getElementById('editCookieValues').value.trim();
|
461 |
+
|
462 |
+
if (!apiKey || !cookieValuesText) {
|
463 |
+
document.getElementById('editModalMessage').innerHTML = `
|
464 |
+
<div class="error">API Key 和 Cookie 值不能为空</div>
|
465 |
+
`;
|
466 |
+
return;
|
467 |
+
}
|
468 |
+
|
469 |
+
// 将逗号分隔的 Cookie 值转换为数组
|
470 |
+
const cookieValues = cookieValuesText.split(',').map(cookie => cookie.trim()).filter(cookie => cookie);
|
471 |
+
|
472 |
+
try {
|
473 |
+
const response = await fetch('/v1/api-keys', {
|
474 |
+
method: 'POST',
|
475 |
+
headers: {
|
476 |
+
'Content-Type': 'application/json',
|
477 |
+
},
|
478 |
+
body: JSON.stringify({
|
479 |
+
apiKey,
|
480 |
+
cookieValues,
|
481 |
+
}),
|
482 |
+
});
|
483 |
+
|
484 |
+
const data = await response.json();
|
485 |
+
|
486 |
+
if (data.success) {
|
487 |
+
document.getElementById('editModalMessage').innerHTML = `
|
488 |
+
<div class="info">Cookie 修改成功</div>
|
489 |
+
`;
|
490 |
+
setTimeout(() => {
|
491 |
+
modal.style.display = 'none';
|
492 |
+
loadApiKeys();
|
493 |
+
}, 1500);
|
494 |
+
} else {
|
495 |
+
document.getElementById('editModalMessage').innerHTML = `
|
496 |
+
<div class="error">Cookie 修改失败: ${data.error}</div>
|
497 |
+
`;
|
498 |
+
}
|
499 |
+
} catch (error) {
|
500 |
+
console.error('修改 Cookie 失败:', error);
|
501 |
+
document.getElementById('editModalMessage').innerHTML = `
|
502 |
+
<div class="error">修改 Cookie 失败: ${error.message}</div>
|
503 |
+
`;
|
504 |
+
}
|
505 |
+
});
|
506 |
+
|
507 |
+
// 测试API连接
|
508 |
+
document.getElementById('testApiBtn').addEventListener('click', async function() {
|
509 |
+
const resultDiv = document.getElementById('testApiResult');
|
510 |
+
resultDiv.innerHTML = '<div class="info">正在测试API连接...</div>';
|
511 |
+
|
512 |
+
try {
|
513 |
+
const response = await fetch('/v1/api-keys', {
|
514 |
+
method: 'GET',
|
515 |
+
headers: {
|
516 |
+
'Content-Type': 'application/json',
|
517 |
+
'Cache-Control': 'no-cache'
|
518 |
+
}
|
519 |
+
});
|
520 |
+
|
521 |
+
resultDiv.innerHTML = `<div class="info">API响应状态: ${response.status}</div>`;
|
522 |
+
|
523 |
+
if (!response.ok) {
|
524 |
+
throw new Error(`HTTP错误: ${response.status} ${response.statusText}`);
|
525 |
+
}
|
526 |
+
|
527 |
+
const data = await response.json();
|
528 |
+
resultDiv.innerHTML += `<div class="info">获取到的数据: ${JSON.stringify(data)}</div>`;
|
529 |
+
} catch (error) {
|
530 |
+
console.error('测试API失败:', error);
|
531 |
+
resultDiv.innerHTML = `<div class="error">测试API失败: ${error.message}</div>`;
|
532 |
+
}
|
533 |
+
});
|
534 |
+
|
535 |
+
// 清除缓存并刷新
|
536 |
+
document.getElementById('clearCacheBtn').addEventListener('click', function() {
|
537 |
+
// 清除缓存
|
538 |
+
if ('caches' in window) {
|
539 |
+
caches.keys().then(function(names) {
|
540 |
+
for (let name of names) {
|
541 |
+
caches.delete(name);
|
542 |
+
}
|
543 |
+
});
|
544 |
+
}
|
545 |
+
|
546 |
+
// 强制刷新页面(绕过缓存)
|
547 |
+
window.location.reload(true);
|
548 |
+
});
|
549 |
+
|
550 |
+
// 获取无效Cookie列表
|
551 |
+
async function getInvalidCookies() {
|
552 |
+
try {
|
553 |
+
const response = await fetch('/v1/invalid-cookies', {
|
554 |
+
method: 'GET',
|
555 |
+
headers: {
|
556 |
+
'Content-Type': 'application/json',
|
557 |
+
'Cache-Control': 'no-cache'
|
558 |
+
}
|
559 |
+
});
|
560 |
+
|
561 |
+
if (!response.ok) {
|
562 |
+
throw new Error(`HTTP错误: ${response.status} ${response.statusText}`);
|
563 |
+
}
|
564 |
+
|
565 |
+
const data = await response.json();
|
566 |
+
return data.invalidCookies;
|
567 |
+
} catch (error) {
|
568 |
+
console.error('获取无效Cookie失败:', error);
|
569 |
+
throw error;
|
570 |
+
}
|
571 |
+
}
|
572 |
+
|
573 |
+
// 清除特定无效Cookie
|
574 |
+
async function clearInvalidCookie(cookie) {
|
575 |
+
try {
|
576 |
+
const response = await fetch(`/v1/invalid-cookies/${encodeURIComponent(cookie)}`, {
|
577 |
+
method: 'DELETE',
|
578 |
+
headers: {
|
579 |
+
'Content-Type': 'application/json',
|
580 |
+
'Cache-Control': 'no-cache'
|
581 |
+
}
|
582 |
+
});
|
583 |
+
|
584 |
+
if (!response.ok) {
|
585 |
+
throw new Error(`HTTP错误: ${response.status} ${response.statusText}`);
|
586 |
+
}
|
587 |
+
|
588 |
+
const data = await response.json();
|
589 |
+
return data.success;
|
590 |
+
} catch (error) {
|
591 |
+
console.error(`清除无效Cookie失败:`, error);
|
592 |
+
throw error;
|
593 |
+
}
|
594 |
+
}
|
595 |
+
|
596 |
+
// 清除所有无效Cookie
|
597 |
+
async function clearAllInvalidCookies() {
|
598 |
+
try {
|
599 |
+
const response = await fetch('/v1/invalid-cookies', {
|
600 |
+
method: 'DELETE',
|
601 |
+
headers: {
|
602 |
+
'Content-Type': 'application/json',
|
603 |
+
'Cache-Control': 'no-cache'
|
604 |
+
}
|
605 |
+
});
|
606 |
+
|
607 |
+
if (!response.ok) {
|
608 |
+
throw new Error(`HTTP错误: ${response.status} ${response.statusText}`);
|
609 |
+
}
|
610 |
+
|
611 |
+
const data = await response.json();
|
612 |
+
return data.success;
|
613 |
+
} catch (error) {
|
614 |
+
console.error('清除所有无效Cookie失败:', error);
|
615 |
+
throw error;
|
616 |
+
}
|
617 |
+
}
|
618 |
+
|
619 |
+
// 渲染无效Cookie列表
|
620 |
+
async function renderInvalidCookies() {
|
621 |
+
const container = document.getElementById('invalidCookiesContainer');
|
622 |
+
|
623 |
+
try {
|
624 |
+
const invalidCookies = await getInvalidCookies();
|
625 |
+
|
626 |
+
if (invalidCookies.length === 0) {
|
627 |
+
container.innerHTML = '<div class="info">没有检测到无效Cookie</div>';
|
628 |
+
return;
|
629 |
+
}
|
630 |
+
|
631 |
+
let html = '<table><thead><tr><th>无效Cookie</th><th>操作</th></tr></thead><tbody>';
|
632 |
+
|
633 |
+
invalidCookies.forEach(cookie => {
|
634 |
+
// 截断显示cookie,避免页面过长
|
635 |
+
const displayCookie = cookie.length > 50 ? cookie.substring(0, 50) + '...' : cookie;
|
636 |
+
|
637 |
+
html += `
|
638 |
+
<tr>
|
639 |
+
<td class="cookie-text" title="${cookie}">${displayCookie}</td>
|
640 |
+
<td>
|
641 |
+
<button class="action-btn clear-invalid-cookie" data-cookie="${cookie}">
|
642 |
+
清除
|
643 |
+
</button>
|
644 |
+
</td>
|
645 |
+
</tr>
|
646 |
+
`;
|
647 |
+
});
|
648 |
+
|
649 |
+
html += '</tbody></table>';
|
650 |
+
container.innerHTML = html;
|
651 |
+
|
652 |
+
// 添加清除按钮事件监听
|
653 |
+
document.querySelectorAll('.clear-invalid-cookie').forEach(button => {
|
654 |
+
button.addEventListener('click', async function() {
|
655 |
+
const cookie = this.getAttribute('data-cookie');
|
656 |
+
|
657 |
+
try {
|
658 |
+
await clearInvalidCookie(cookie);
|
659 |
+
showMessage('invalidCookiesContainer', '无效Cookie已清除', 'info');
|
660 |
+
renderInvalidCookies(); // 重新渲染列表
|
661 |
+
} catch (error) {
|
662 |
+
showMessage('invalidCookiesContainer', `清除失败: ${error.message}`, 'error');
|
663 |
+
}
|
664 |
+
});
|
665 |
+
});
|
666 |
+
|
667 |
+
} catch (error) {
|
668 |
+
container.innerHTML = `<div class="error">加载失败: ${error.message}</div>`;
|
669 |
+
}
|
670 |
+
}
|
671 |
+
|
672 |
+
// 清除所有无效Cookie按钮事件
|
673 |
+
document.getElementById('clearAllInvalidCookies').addEventListener('click', async function() {
|
674 |
+
try {
|
675 |
+
await clearAllInvalidCookies();
|
676 |
+
showMessage('invalidCookiesContainer', '所有无效Cookie已清除', 'info');
|
677 |
+
renderInvalidCookies(); // 重新渲染列表
|
678 |
+
} catch (error) {
|
679 |
+
showMessage('invalidCookiesContainer', `清除失败: ${error.message}`, 'error');
|
680 |
+
}
|
681 |
+
});
|
682 |
+
|
683 |
+
// 页面加载时获取 API Key 列表和无效Cookie列表
|
684 |
+
document.addEventListener('DOMContentLoaded', function() {
|
685 |
+
checkAuth();
|
686 |
+
loadApiKeys();
|
687 |
+
renderInvalidCookies();
|
688 |
+
populateRefreshApiKeySelect();
|
689 |
+
});
|
690 |
+
|
691 |
+
// 显示消息的通用函数
|
692 |
+
function showMessage(containerId, message, type) {
|
693 |
+
const container = document.getElementById(containerId);
|
694 |
+
container.innerHTML = `<div class="${type}">${message}</div>`;
|
695 |
+
}
|
696 |
+
|
697 |
+
// 填充刷新API Key的下拉选择框
|
698 |
+
async function populateRefreshApiKeySelect() {
|
699 |
+
try {
|
700 |
+
const apiKeys = await getApiKeys();
|
701 |
+
const select = document.getElementById('refreshApiKey');
|
702 |
+
|
703 |
+
// 清空现有选项(保留"所有API Key"选项)
|
704 |
+
while (select.options.length > 1) {
|
705 |
+
select.remove(1);
|
706 |
+
}
|
707 |
+
|
708 |
+
// 添加API Key选项
|
709 |
+
apiKeys.forEach(key => {
|
710 |
+
const option = document.createElement('option');
|
711 |
+
option.value = key.key;
|
712 |
+
option.textContent = `${key.key} (${key.cookieCount} 个Cookie)`;
|
713 |
+
select.appendChild(option);
|
714 |
+
});
|
715 |
+
} catch (error) {
|
716 |
+
console.error('加载API Key选项失败:', error);
|
717 |
+
}
|
718 |
+
}
|
719 |
+
|
720 |
+
// 获取API Keys的辅助函数
|
721 |
+
async function getApiKeys() {
|
722 |
+
const response = await fetch('/v1/api-keys', {
|
723 |
+
method: 'GET',
|
724 |
+
headers: {
|
725 |
+
'Content-Type': 'application/json',
|
726 |
+
'Cache-Control': 'no-cache'
|
727 |
+
}
|
728 |
+
});
|
729 |
+
|
730 |
+
if (!response.ok) {
|
731 |
+
throw new Error(`HTTP错误: ${response.status} ${response.statusText}`);
|
732 |
+
}
|
733 |
+
|
734 |
+
const data = await response.json();
|
735 |
+
return data.success ? data.apiKeys : [];
|
736 |
+
}
|
737 |
+
|
738 |
+
// 刷新Cookie按钮事件
|
739 |
+
document.getElementById('refreshCookieBtn').addEventListener('click', async function() {
|
740 |
+
const refreshBtn = this;
|
741 |
+
const apiKey = document.getElementById('refreshApiKey').value;
|
742 |
+
const statusContainer = document.getElementById('refreshStatusContainer');
|
743 |
+
const statusText = document.getElementById('refreshStatus');
|
744 |
+
const progressBar = document.getElementById('refreshProgress');
|
745 |
+
|
746 |
+
// 禁用按钮,显示状态容器
|
747 |
+
refreshBtn.disabled = true;
|
748 |
+
statusContainer.style.display = 'block';
|
749 |
+
statusText.textContent = '正在发送刷新请求...';
|
750 |
+
progressBar.value = 10;
|
751 |
+
|
752 |
+
try {
|
753 |
+
// 构建请求URL
|
754 |
+
let url = '/v1/refresh-cookies';
|
755 |
+
if (apiKey) {
|
756 |
+
url += `?apiKey=${encodeURIComponent(apiKey)}`;
|
757 |
+
}
|
758 |
+
|
759 |
+
// 发送刷新请求
|
760 |
+
statusText.textContent = '正在发送刷新请求...';
|
761 |
+
progressBar.value = 20;
|
762 |
+
|
763 |
+
const response = await fetch(url, {
|
764 |
+
method: 'POST',
|
765 |
+
headers: {
|
766 |
+
'Content-Type': 'application/json',
|
767 |
+
'Cache-Control': 'no-cache'
|
768 |
+
}
|
769 |
+
});
|
770 |
+
|
771 |
+
if (!response.ok) {
|
772 |
+
throw new Error(`HTTP错误: ${response.status} ${response.statusText}`);
|
773 |
+
}
|
774 |
+
|
775 |
+
// 显示长时间等待提示
|
776 |
+
statusText.textContent = '刷新请求已发送,请耐心等待2-12分钟...';
|
777 |
+
progressBar.value = 50;
|
778 |
+
showMessage('refreshCookieMessage', '刷新请求已发送,由于需要访问Cursor官网获取新Cookie,整个过程可能需要2-12分钟,请耐心等待。您可以关闭此页面,稍后再来查看结果。', 'info');
|
779 |
+
|
780 |
+
// 启动定时检查刷新状态
|
781 |
+
let checkInterval = setInterval(async () => {
|
782 |
+
try {
|
783 |
+
const statusResponse = await fetch('/v1/refresh-status', {
|
784 |
+
method: 'GET',
|
785 |
+
headers: {
|
786 |
+
'Cache-Control': 'no-cache'
|
787 |
+
}
|
788 |
+
});
|
789 |
+
|
790 |
+
if (!statusResponse.ok) {
|
791 |
+
throw new Error(`HTTP错误: ${statusResponse.status} ${statusResponse.statusText}`);
|
792 |
+
}
|
793 |
+
|
794 |
+
const statusData = await statusResponse.json();
|
795 |
+
const refreshData = statusData.data;
|
796 |
+
|
797 |
+
// 更新状态信息
|
798 |
+
statusText.textContent = refreshData.message || '正在刷新...';
|
799 |
+
|
800 |
+
// 根据状态更新进度条和UI
|
801 |
+
if (refreshData.status === 'completed') {
|
802 |
+
// 刷新完成
|
803 |
+
progressBar.value = 100;
|
804 |
+
statusText.textContent = `刷新完成: ${refreshData.message}`;
|
805 |
+
clearInterval(checkInterval);
|
806 |
+
|
807 |
+
// 重新加载API Key列表
|
808 |
+
await loadApiKeys();
|
809 |
+
await populateRefreshApiKeySelect();
|
810 |
+
|
811 |
+
// 显示成功消息
|
812 |
+
showMessage('refreshCookieMessage', `刷新完成: ${refreshData.message}`, 'success');
|
813 |
+
|
814 |
+
// 启用按钮
|
815 |
+
refreshBtn.disabled = false;
|
816 |
+
|
817 |
+
// 3秒后隐藏状态容器
|
818 |
+
setTimeout(() => {
|
819 |
+
statusContainer.style.display = 'none';
|
820 |
+
}, 3000);
|
821 |
+
} else if (refreshData.status === 'failed') {
|
822 |
+
// 刷新失败
|
823 |
+
progressBar.value = 0;
|
824 |
+
statusText.textContent = `刷新失败: ${refreshData.message}`;
|
825 |
+
clearInterval(checkInterval);
|
826 |
+
|
827 |
+
// 显示错误消息
|
828 |
+
showMessage('refreshCookieMessage', `刷新失败: ${refreshData.message}`, 'error');
|
829 |
+
|
830 |
+
// 启用按钮
|
831 |
+
refreshBtn.disabled = false;
|
832 |
+
} else if (refreshData.status === 'running') {
|
833 |
+
// 正在刷新
|
834 |
+
progressBar.value = 75;
|
835 |
+
} else if (!refreshData.isRunning) {
|
836 |
+
// 未知状态但不在运行
|
837 |
+
clearInterval(checkInterval);
|
838 |
+
refreshBtn.disabled = false;
|
839 |
+
}
|
840 |
+
} catch (error) {
|
841 |
+
console.error('检查刷新状态失败:', error);
|
842 |
+
}
|
843 |
+
}, 5000); // 每5秒检查一次
|
844 |
+
|
845 |
+
// 设置超时检查,12分钟后如果还没完成就停止检查
|
846 |
+
setTimeout(() => {
|
847 |
+
if (checkInterval) {
|
848 |
+
clearInterval(checkInterval);
|
849 |
+
refreshBtn.disabled = false;
|
850 |
+
statusContainer.style.display = 'none';
|
851 |
+
}
|
852 |
+
}, 720000);
|
853 |
+
} catch (error) {
|
854 |
+
console.error('刷新Cookie失败:', error);
|
855 |
+
statusText.textContent = '刷新请求发送失败';
|
856 |
+
progressBar.value = 0;
|
857 |
+
showMessage('refreshCookieMessage', `刷新请求发送失败: ${error.message}`, 'error');
|
858 |
+
refreshBtn.disabled = false;
|
859 |
+
}
|
860 |
+
});
|
861 |
+
|
862 |
+
// 检查登录状态
|
863 |
+
function checkAuth() {
|
864 |
+
const token = localStorage.getItem('adminToken');
|
865 |
+
if (!token) {
|
866 |
+
window.location.href = '/login.html';
|
867 |
+
return;
|
868 |
+
}
|
869 |
+
|
870 |
+
// 验证token
|
871 |
+
fetch('/v1/admin/verify', {
|
872 |
+
headers: {
|
873 |
+
'Authorization': `Bearer ${token}`
|
874 |
+
}
|
875 |
+
})
|
876 |
+
.then(response => response.json())
|
877 |
+
.then(data => {
|
878 |
+
if (!data.success) {
|
879 |
+
localStorage.removeItem('adminToken');
|
880 |
+
window.location.href = '/login.html';
|
881 |
+
} else {
|
882 |
+
// 显示管理员用户名
|
883 |
+
document.getElementById('adminUsername').textContent = `管理员:${data.username}`;
|
884 |
+
}
|
885 |
+
})
|
886 |
+
.catch(error => {
|
887 |
+
console.error('验证失败:', error);
|
888 |
+
localStorage.removeItem('adminToken');
|
889 |
+
window.location.href = '/login.html';
|
890 |
+
});
|
891 |
+
}
|
892 |
+
|
893 |
+
// 退出登录
|
894 |
+
document.getElementById('logoutBtn').addEventListener('click', () => {
|
895 |
+
localStorage.removeItem('adminToken');
|
896 |
+
window.location.href = '/login.html';
|
897 |
+
});
|
898 |
+
|
899 |
+
// 添加token到所有API请求
|
900 |
+
function addAuthHeader(headers = {}) {
|
901 |
+
const token = localStorage.getItem('adminToken');
|
902 |
+
return {
|
903 |
+
...headers,
|
904 |
+
'Authorization': `Bearer ${token}`
|
905 |
+
};
|
906 |
+
}
|
907 |
+
|
908 |
+
// 修改所有fetch请求,添加token
|
909 |
+
const originalFetch = window.fetch;
|
910 |
+
window.fetch = function(url, options = {}) {
|
911 |
+
// 只对管理页面的API请求添加token
|
912 |
+
if (url.includes('/v1/api-keys') ||
|
913 |
+
url.includes('/v1/invalid-cookies') ||
|
914 |
+
url.includes('/v1/refresh-cookies')) {
|
915 |
+
options.headers = addAuthHeader(options.headers);
|
916 |
+
}
|
917 |
+
return originalFetch(url, options);
|
918 |
+
};
|
919 |
+
</script>
|
920 |
+
</body>
|
921 |
+
</html>
|
src/public/login.html
ADDED
@@ -0,0 +1,243 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
<!DOCTYPE html>
|
2 |
+
<html lang="zh-CN">
|
3 |
+
<head>
|
4 |
+
<meta charset="UTF-8">
|
5 |
+
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
6 |
+
<title>Cursor To OpenAI - 管理员登录</title>
|
7 |
+
<style>
|
8 |
+
body {
|
9 |
+
font-family: 'Arial', sans-serif;
|
10 |
+
line-height: 1.6;
|
11 |
+
margin: 0;
|
12 |
+
padding: 20px;
|
13 |
+
color: #333;
|
14 |
+
background-color: #f5f5f5;
|
15 |
+
display: flex;
|
16 |
+
justify-content: center;
|
17 |
+
align-items: center;
|
18 |
+
min-height: 100vh;
|
19 |
+
}
|
20 |
+
.container {
|
21 |
+
background: #fff;
|
22 |
+
padding: 30px;
|
23 |
+
border-radius: 8px;
|
24 |
+
box-shadow: 0 2px 10px rgba(0, 0, 0, 0.1);
|
25 |
+
width: 100%;
|
26 |
+
max-width: 400px;
|
27 |
+
}
|
28 |
+
h1 {
|
29 |
+
color: #2c3e50;
|
30 |
+
text-align: center;
|
31 |
+
margin-bottom: 30px;
|
32 |
+
}
|
33 |
+
.form-group {
|
34 |
+
margin-bottom: 20px;
|
35 |
+
}
|
36 |
+
label {
|
37 |
+
display: block;
|
38 |
+
margin-bottom: 5px;
|
39 |
+
font-weight: bold;
|
40 |
+
}
|
41 |
+
input {
|
42 |
+
width: 100%;
|
43 |
+
padding: 10px;
|
44 |
+
border: 1px solid #ddd;
|
45 |
+
border-radius: 4px;
|
46 |
+
font-size: 16px;
|
47 |
+
box-sizing: border-box;
|
48 |
+
}
|
49 |
+
button {
|
50 |
+
background: #3498db;
|
51 |
+
color: white;
|
52 |
+
border: none;
|
53 |
+
padding: 12px 20px;
|
54 |
+
border-radius: 4px;
|
55 |
+
cursor: pointer;
|
56 |
+
font-size: 16px;
|
57 |
+
width: 100%;
|
58 |
+
margin-top: 10px;
|
59 |
+
}
|
60 |
+
button:hover {
|
61 |
+
background: #2980b9;
|
62 |
+
}
|
63 |
+
.message {
|
64 |
+
margin-top: 20px;
|
65 |
+
padding: 10px;
|
66 |
+
border-radius: 4px;
|
67 |
+
text-align: center;
|
68 |
+
}
|
69 |
+
.error {
|
70 |
+
background-color: #f8d7da;
|
71 |
+
color: #721c24;
|
72 |
+
}
|
73 |
+
.info {
|
74 |
+
background-color: #d4edda;
|
75 |
+
color: #155724;
|
76 |
+
}
|
77 |
+
.toggle-form {
|
78 |
+
text-align: center;
|
79 |
+
margin-top: 20px;
|
80 |
+
color: #3498db;
|
81 |
+
cursor: pointer;
|
82 |
+
}
|
83 |
+
#registerForm {
|
84 |
+
display: none;
|
85 |
+
}
|
86 |
+
</style>
|
87 |
+
</head>
|
88 |
+
<body>
|
89 |
+
<div class="container">
|
90 |
+
<h1>管理员登录</h1>
|
91 |
+
|
92 |
+
<!-- 登录表单 -->
|
93 |
+
<form id="loginForm">
|
94 |
+
<div class="form-group">
|
95 |
+
<label for="loginUsername">用户名</label>
|
96 |
+
<input type="text" id="loginUsername" required>
|
97 |
+
</div>
|
98 |
+
<div class="form-group">
|
99 |
+
<label for="loginPassword">密码</label>
|
100 |
+
<input type="password" id="loginPassword" required>
|
101 |
+
</div>
|
102 |
+
<button type="submit">登录</button>
|
103 |
+
<div id="loginMessage" class="message"></div>
|
104 |
+
</form>
|
105 |
+
|
106 |
+
<!-- 注册表单 -->
|
107 |
+
<form id="registerForm">
|
108 |
+
<div class="form-group">
|
109 |
+
<label for="registerUsername">用户名</label>
|
110 |
+
<input type="text" id="registerUsername" required>
|
111 |
+
</div>
|
112 |
+
<div class="form-group">
|
113 |
+
<label for="registerPassword">密码</label>
|
114 |
+
<input type="password" id="registerPassword" required>
|
115 |
+
</div>
|
116 |
+
<div class="form-group">
|
117 |
+
<label for="confirmPassword">确认密码</label>
|
118 |
+
<input type="password" id="confirmPassword" required>
|
119 |
+
</div>
|
120 |
+
<button type="submit">注册</button>
|
121 |
+
<div id="registerMessage" class="message"></div>
|
122 |
+
</form>
|
123 |
+
|
124 |
+
<div class="toggle-form" id="toggleForm">
|
125 |
+
还没有账号?点击注册
|
126 |
+
</div>
|
127 |
+
</div>
|
128 |
+
|
129 |
+
<script>
|
130 |
+
// 获取元素
|
131 |
+
const loginForm = document.getElementById('loginForm');
|
132 |
+
const registerForm = document.getElementById('registerForm');
|
133 |
+
const toggleForm = document.getElementById('toggleForm');
|
134 |
+
const loginMessage = document.getElementById('loginMessage');
|
135 |
+
const registerMessage = document.getElementById('registerMessage');
|
136 |
+
|
137 |
+
// 切换表单显示
|
138 |
+
let isLoginForm = true;
|
139 |
+
toggleForm.addEventListener('click', () => {
|
140 |
+
isLoginForm = !isLoginForm;
|
141 |
+
loginForm.style.display = isLoginForm ? 'block' : 'none';
|
142 |
+
registerForm.style.display = isLoginForm ? 'none' : 'block';
|
143 |
+
toggleForm.textContent = isLoginForm ? '还没有账号?点击注册' : '已有账号?点击登录';
|
144 |
+
loginMessage.textContent = '';
|
145 |
+
registerMessage.textContent = '';
|
146 |
+
});
|
147 |
+
|
148 |
+
// 检查是否已有管理员账号
|
149 |
+
async function checkAdminExists() {
|
150 |
+
try {
|
151 |
+
const response = await fetch('/v1/admin/check');
|
152 |
+
const data = await response.json();
|
153 |
+
|
154 |
+
if (data.exists) {
|
155 |
+
// 如果已有管理员,显示登录表单
|
156 |
+
loginForm.style.display = 'block';
|
157 |
+
registerForm.style.display = 'none';
|
158 |
+
toggleForm.style.display = 'none';
|
159 |
+
} else {
|
160 |
+
// 如果没有管理员,显示注册表单
|
161 |
+
loginForm.style.display = 'none';
|
162 |
+
registerForm.style.display = 'block';
|
163 |
+
toggleForm.style.display = 'none';
|
164 |
+
registerMessage.innerHTML = '<div class="info">首次使用,请注册管理员账号</div>';
|
165 |
+
}
|
166 |
+
} catch (error) {
|
167 |
+
console.error('检查管理员账号失败:', error);
|
168 |
+
}
|
169 |
+
}
|
170 |
+
|
171 |
+
// 登录处理
|
172 |
+
loginForm.addEventListener('submit', async (e) => {
|
173 |
+
e.preventDefault();
|
174 |
+
|
175 |
+
const username = document.getElementById('loginUsername').value;
|
176 |
+
const password = document.getElementById('loginPassword').value;
|
177 |
+
|
178 |
+
try {
|
179 |
+
const response = await fetch('/v1/admin/login', {
|
180 |
+
method: 'POST',
|
181 |
+
headers: {
|
182 |
+
'Content-Type': 'application/json',
|
183 |
+
},
|
184 |
+
body: JSON.stringify({ username, password }),
|
185 |
+
});
|
186 |
+
|
187 |
+
const data = await response.json();
|
188 |
+
|
189 |
+
if (data.success) {
|
190 |
+
// 登录成功,保存token并跳转
|
191 |
+
localStorage.setItem('adminToken', data.token);
|
192 |
+
window.location.href = '/index.html';
|
193 |
+
} else {
|
194 |
+
loginMessage.innerHTML = `<div class="error">${data.message}</div>`;
|
195 |
+
}
|
196 |
+
} catch (error) {
|
197 |
+
console.error('登录失败:', error);
|
198 |
+
loginMessage.innerHTML = '<div class="error">登录失败,请稍后重试</div>';
|
199 |
+
}
|
200 |
+
});
|
201 |
+
|
202 |
+
// 注册处理
|
203 |
+
registerForm.addEventListener('submit', async (e) => {
|
204 |
+
e.preventDefault();
|
205 |
+
|
206 |
+
const username = document.getElementById('registerUsername').value;
|
207 |
+
const password = document.getElementById('registerPassword').value;
|
208 |
+
const confirmPassword = document.getElementById('confirmPassword').value;
|
209 |
+
|
210 |
+
if (password !== confirmPassword) {
|
211 |
+
registerMessage.innerHTML = '<div class="error">两次输入的密码不一致</div>';
|
212 |
+
return;
|
213 |
+
}
|
214 |
+
|
215 |
+
try {
|
216 |
+
const response = await fetch('/v1/admin/register', {
|
217 |
+
method: 'POST',
|
218 |
+
headers: {
|
219 |
+
'Content-Type': 'application/json',
|
220 |
+
},
|
221 |
+
body: JSON.stringify({ username, password }),
|
222 |
+
});
|
223 |
+
|
224 |
+
const data = await response.json();
|
225 |
+
|
226 |
+
if (data.success) {
|
227 |
+
// 注册成功,保存token并跳转
|
228 |
+
localStorage.setItem('adminToken', data.token);
|
229 |
+
window.location.href = '/index.html';
|
230 |
+
} else {
|
231 |
+
registerMessage.innerHTML = `<div class="error">${data.message}</div>`;
|
232 |
+
}
|
233 |
+
} catch (error) {
|
234 |
+
console.error('注册失败:', error);
|
235 |
+
registerMessage.innerHTML = '<div class="error">注册失败,请稍后重试</div>';
|
236 |
+
}
|
237 |
+
});
|
238 |
+
|
239 |
+
// 页面加载时检查管理员账号
|
240 |
+
document.addEventListener('DOMContentLoaded', checkAdminExists);
|
241 |
+
</script>
|
242 |
+
</body>
|
243 |
+
</html>
|
src/routes/index.js
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
const express = require('express');
|
2 |
+
const router = express.Router();
|
3 |
+
const v1Routes = require('./v1');
|
4 |
+
|
5 |
+
// OpenAI v1 API routes
|
6 |
+
router.use('/v1', v1Routes);
|
7 |
+
|
8 |
+
module.exports = router;
|
src/routes/v1.js
ADDED
@@ -0,0 +1,931 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
const express = require('express');
|
2 |
+
const router = express.Router();
|
3 |
+
|
4 |
+
const $root = require('../proto/message.js');
|
5 |
+
const { v4: uuidv4, v5: uuidv5 } = require('uuid');
|
6 |
+
const { generateCursorBody, chunkToUtf8String, generateHashed64Hex, generateCursorChecksum } = require('../utils/utils.js');
|
7 |
+
const keyManager = require('../utils/keyManager.js');
|
8 |
+
const { spawn } = require('child_process');
|
9 |
+
const path = require('path');
|
10 |
+
const admin = require('../models/admin');
|
11 |
+
|
12 |
+
// 存储刷新状态的变量
|
13 |
+
let refreshStatus = {
|
14 |
+
isRunning: false,
|
15 |
+
status: 'idle', // idle, running, completed, failed
|
16 |
+
message: '',
|
17 |
+
startTime: null,
|
18 |
+
endTime: null,
|
19 |
+
error: null
|
20 |
+
};
|
21 |
+
|
22 |
+
// 检查是否已有管理员账号
|
23 |
+
router.get('/admin/check', (req, res) => {
|
24 |
+
try {
|
25 |
+
return res.json({
|
26 |
+
success: true,
|
27 |
+
exists: admin.hasAdmin()
|
28 |
+
});
|
29 |
+
} catch (error) {
|
30 |
+
console.error('检查管理员账号失败:', error);
|
31 |
+
return res.status(500).json({
|
32 |
+
success: false,
|
33 |
+
message: error.message
|
34 |
+
});
|
35 |
+
}
|
36 |
+
});
|
37 |
+
|
38 |
+
// 注册管理员
|
39 |
+
router.post('/admin/register', (req, res) => {
|
40 |
+
try {
|
41 |
+
const { username, password } = req.body;
|
42 |
+
|
43 |
+
if (!username || !password) {
|
44 |
+
return res.status(400).json({
|
45 |
+
success: false,
|
46 |
+
message: '用户名和密码不能为空'
|
47 |
+
});
|
48 |
+
}
|
49 |
+
|
50 |
+
const token = admin.register(username, password);
|
51 |
+
|
52 |
+
return res.json({
|
53 |
+
success: true,
|
54 |
+
message: '注册成功',
|
55 |
+
token
|
56 |
+
});
|
57 |
+
} catch (error) {
|
58 |
+
console.error('注册管理员失败:', error);
|
59 |
+
return res.status(400).json({
|
60 |
+
success: false,
|
61 |
+
message: error.message
|
62 |
+
});
|
63 |
+
}
|
64 |
+
});
|
65 |
+
|
66 |
+
// 管理员登录
|
67 |
+
router.post('/admin/login', (req, res) => {
|
68 |
+
try {
|
69 |
+
const { username, password } = req.body;
|
70 |
+
|
71 |
+
if (!username || !password) {
|
72 |
+
return res.status(400).json({
|
73 |
+
success: false,
|
74 |
+
message: '用户名和密码不能为空'
|
75 |
+
});
|
76 |
+
}
|
77 |
+
|
78 |
+
const token = admin.login(username, password);
|
79 |
+
|
80 |
+
return res.json({
|
81 |
+
success: true,
|
82 |
+
message: '登录成功',
|
83 |
+
token
|
84 |
+
});
|
85 |
+
} catch (error) {
|
86 |
+
console.error('登录失败:', error);
|
87 |
+
return res.status(400).json({
|
88 |
+
success: false,
|
89 |
+
message: error.message
|
90 |
+
});
|
91 |
+
}
|
92 |
+
});
|
93 |
+
|
94 |
+
// 验证token
|
95 |
+
router.get('/admin/verify', (req, res) => {
|
96 |
+
try {
|
97 |
+
const authHeader = req.headers.authorization;
|
98 |
+
if (!authHeader || !authHeader.startsWith('Bearer ')) {
|
99 |
+
return res.status(401).json({
|
100 |
+
success: false,
|
101 |
+
message: '未提供认证token'
|
102 |
+
});
|
103 |
+
}
|
104 |
+
|
105 |
+
const token = authHeader.split(' ')[1];
|
106 |
+
const result = admin.verifyToken(token);
|
107 |
+
|
108 |
+
return res.json(result);
|
109 |
+
} catch (error) {
|
110 |
+
console.error('验证token失败:', error);
|
111 |
+
return res.status(401).json({
|
112 |
+
success: false,
|
113 |
+
message: error.message
|
114 |
+
});
|
115 |
+
}
|
116 |
+
});
|
117 |
+
|
118 |
+
// 添加API key管理路由
|
119 |
+
router.post("/api-keys", async (req, res) => {
|
120 |
+
try {
|
121 |
+
const { apiKey, cookieValues } = req.body;
|
122 |
+
|
123 |
+
if (!apiKey || !cookieValues) {
|
124 |
+
return res.status(400).json({
|
125 |
+
error: 'API key and cookie values are required',
|
126 |
+
});
|
127 |
+
}
|
128 |
+
|
129 |
+
keyManager.addOrUpdateApiKey(apiKey, cookieValues);
|
130 |
+
|
131 |
+
return res.json({
|
132 |
+
success: true,
|
133 |
+
message: 'API key added or updated successfully',
|
134 |
+
});
|
135 |
+
} catch (error) {
|
136 |
+
console.error(error);
|
137 |
+
return res.status(500).json({
|
138 |
+
error: 'Internal server error',
|
139 |
+
});
|
140 |
+
}
|
141 |
+
});
|
142 |
+
|
143 |
+
// 获取所有API Keys
|
144 |
+
router.get("/api-keys", async (req, res) => {
|
145 |
+
try {
|
146 |
+
console.log('收到获取API Keys请求');
|
147 |
+
const apiKeys = keyManager.getAllApiKeys();
|
148 |
+
console.log('获取到的API Keys:', apiKeys);
|
149 |
+
|
150 |
+
const result = {
|
151 |
+
success: true,
|
152 |
+
apiKeys: apiKeys.map(apiKey => ({
|
153 |
+
key: apiKey,
|
154 |
+
cookieCount: keyManager.getAllCookiesForApiKey(apiKey).length,
|
155 |
+
})),
|
156 |
+
};
|
157 |
+
console.log('返回结果:', result);
|
158 |
+
|
159 |
+
return res.json(result);
|
160 |
+
} catch (error) {
|
161 |
+
console.error('获取API Keys失败:', error);
|
162 |
+
return res.status(500).json({
|
163 |
+
error: 'Internal server error',
|
164 |
+
message: error.message
|
165 |
+
});
|
166 |
+
}
|
167 |
+
});
|
168 |
+
|
169 |
+
// 删除API key
|
170 |
+
router.delete("/api-keys/:apiKey", async (req, res) => {
|
171 |
+
try {
|
172 |
+
const { apiKey } = req.params;
|
173 |
+
|
174 |
+
keyManager.removeApiKey(apiKey);
|
175 |
+
|
176 |
+
return res.json({
|
177 |
+
success: true,
|
178 |
+
message: 'API key removed successfully',
|
179 |
+
});
|
180 |
+
} catch (error) {
|
181 |
+
console.error(error);
|
182 |
+
return res.status(500).json({
|
183 |
+
error: 'Internal server error',
|
184 |
+
});
|
185 |
+
}
|
186 |
+
});
|
187 |
+
|
188 |
+
// 获取特定API Key的Cookie值
|
189 |
+
router.get("/api-keys/:apiKey/cookies", async (req, res) => {
|
190 |
+
try {
|
191 |
+
const { apiKey } = req.params;
|
192 |
+
console.log(`收到获取API Key ${apiKey}的Cookie值请求`);
|
193 |
+
|
194 |
+
const cookies = keyManager.getAllCookiesForApiKey(apiKey);
|
195 |
+
console.log(`API Key ${apiKey}的Cookie值:`, cookies);
|
196 |
+
|
197 |
+
return res.json({
|
198 |
+
success: true,
|
199 |
+
cookies: cookies
|
200 |
+
});
|
201 |
+
} catch (error) {
|
202 |
+
console.error(`获取API Key ${req.params.apiKey}的Cookie值失败:`, error);
|
203 |
+
return res.status(500).json({
|
204 |
+
error: 'Internal server error',
|
205 |
+
message: error.message
|
206 |
+
});
|
207 |
+
}
|
208 |
+
});
|
209 |
+
|
210 |
+
// 获取所有无效的cookie
|
211 |
+
router.get("/invalid-cookies", async (req, res) => {
|
212 |
+
try {
|
213 |
+
const invalidCookies = keyManager.getInvalidCookies();
|
214 |
+
|
215 |
+
return res.json({
|
216 |
+
success: true,
|
217 |
+
invalidCookies: Array.from(invalidCookies)
|
218 |
+
});
|
219 |
+
} catch (error) {
|
220 |
+
console.error('获取无效cookie失败:', error);
|
221 |
+
return res.status(500).json({
|
222 |
+
error: 'Internal server error',
|
223 |
+
message: error.message
|
224 |
+
});
|
225 |
+
}
|
226 |
+
});
|
227 |
+
|
228 |
+
// 清除特定的无效cookie
|
229 |
+
router.delete("/invalid-cookies/:cookie", async (req, res) => {
|
230 |
+
try {
|
231 |
+
const { cookie } = req.params;
|
232 |
+
const success = keyManager.clearInvalidCookie(cookie);
|
233 |
+
|
234 |
+
return res.json({
|
235 |
+
success: success,
|
236 |
+
message: success ? '无效cookie已清除' : '未找到指定的无效cookie'
|
237 |
+
});
|
238 |
+
} catch (error) {
|
239 |
+
console.error('清除无效cookie失败:', error);
|
240 |
+
return res.status(500).json({
|
241 |
+
error: 'Internal server error',
|
242 |
+
message: error.message
|
243 |
+
});
|
244 |
+
}
|
245 |
+
});
|
246 |
+
|
247 |
+
// 清除所有无效cookie
|
248 |
+
router.delete("/invalid-cookies", async (req, res) => {
|
249 |
+
try {
|
250 |
+
keyManager.clearAllInvalidCookies();
|
251 |
+
|
252 |
+
return res.json({
|
253 |
+
success: true,
|
254 |
+
message: '所有无效cookie已清除'
|
255 |
+
});
|
256 |
+
} catch (error) {
|
257 |
+
console.error('清除所有无效cookie失败:', error);
|
258 |
+
return res.status(500).json({
|
259 |
+
error: 'Internal server error',
|
260 |
+
message: error.message
|
261 |
+
});
|
262 |
+
}
|
263 |
+
});
|
264 |
+
|
265 |
+
router.get("/models", async (req, res) => {
|
266 |
+
try{
|
267 |
+
let bearerToken = req.headers.authorization?.replace('Bearer ', '');
|
268 |
+
|
269 |
+
// 使用keyManager获取实际的cookie
|
270 |
+
let authToken = keyManager.getCookieForApiKey(bearerToken);
|
271 |
+
|
272 |
+
if (authToken && authToken.includes('%3A%3A')) {
|
273 |
+
authToken = authToken.split('%3A%3A')[1];
|
274 |
+
}
|
275 |
+
else if (authToken && authToken.includes('::')) {
|
276 |
+
authToken = authToken.split('::')[1];
|
277 |
+
}
|
278 |
+
|
279 |
+
const checksum = req.headers['x-cursor-checksum']
|
280 |
+
?? process.env['x-cursor-checksum']
|
281 |
+
?? generateCursorChecksum(authToken.trim());
|
282 |
+
const cursorClientVersion = "0.45.11"
|
283 |
+
|
284 |
+
const availableModelsResponse = await fetch("https://api2.cursor.sh/aiserver.v1.AiService/AvailableModels", {
|
285 |
+
method: 'POST',
|
286 |
+
headers: {
|
287 |
+
'accept-encoding': 'gzip',
|
288 |
+
'authorization': `Bearer ${authToken}`,
|
289 |
+
'connect-protocol-version': '1',
|
290 |
+
'content-type': 'application/proto',
|
291 |
+
'user-agent': 'connect-es/1.6.1',
|
292 |
+
'x-cursor-checksum': checksum,
|
293 |
+
'x-cursor-client-version': cursorClientVersion,
|
294 |
+
'x-cursor-timezone': 'Asia/Shanghai',
|
295 |
+
'x-ghost-mode': 'true',
|
296 |
+
'Host': 'api2.cursor.sh',
|
297 |
+
},
|
298 |
+
})
|
299 |
+
const data = await availableModelsResponse.arrayBuffer();
|
300 |
+
const buffer = Buffer.from(data);
|
301 |
+
try{
|
302 |
+
const models = $root.AvailableModelsResponse.decode(buffer).models;
|
303 |
+
|
304 |
+
return res.json({
|
305 |
+
object: "list",
|
306 |
+
data: models.map(model => ({
|
307 |
+
id: model.name,
|
308 |
+
created: Date.now(),
|
309 |
+
object: 'model',
|
310 |
+
owned_by: 'cursor'
|
311 |
+
}))
|
312 |
+
})
|
313 |
+
} catch (error) {
|
314 |
+
const text = buffer.toString('utf-8');
|
315 |
+
throw new Error(text);
|
316 |
+
}
|
317 |
+
}
|
318 |
+
catch (error) {
|
319 |
+
console.error(error);
|
320 |
+
return res.status(500).json({
|
321 |
+
error: 'Internal server error',
|
322 |
+
});
|
323 |
+
}
|
324 |
+
})
|
325 |
+
|
326 |
+
router.post('/chat/completions', async (req, res) => {
|
327 |
+
// o1开头的模型,不支持流式输出
|
328 |
+
if (req.body.model.startsWith('o1-') && req.body.stream) {
|
329 |
+
return res.status(400).json({
|
330 |
+
error: 'Model not supported stream',
|
331 |
+
});
|
332 |
+
}
|
333 |
+
|
334 |
+
try {
|
335 |
+
const { model, messages, stream = false } = req.body;
|
336 |
+
let bearerToken = req.headers.authorization?.replace('Bearer ', '');
|
337 |
+
|
338 |
+
// 使用keyManager获取实际的cookie
|
339 |
+
let authToken = keyManager.getCookieForApiKey(bearerToken);
|
340 |
+
// 保存原始cookie,用于后续可能的错误处理
|
341 |
+
const originalAuthToken = authToken;
|
342 |
+
//console.log('原始cookie:', originalAuthToken);
|
343 |
+
|
344 |
+
if (authToken && authToken.includes('%3A%3A')) {
|
345 |
+
authToken = authToken.split('%3A%3A')[1];
|
346 |
+
}
|
347 |
+
else if (authToken && authToken.includes('::')) {
|
348 |
+
authToken = authToken.split('::')[1];
|
349 |
+
}
|
350 |
+
|
351 |
+
if (!messages || !Array.isArray(messages) || messages.length === 0 || !authToken) {
|
352 |
+
return res.status(400).json({
|
353 |
+
error: 'Invalid request. Messages should be a non-empty array and authorization is required',
|
354 |
+
});
|
355 |
+
}
|
356 |
+
|
357 |
+
const checksum = req.headers['x-cursor-checksum']
|
358 |
+
?? process.env['x-cursor-checksum']
|
359 |
+
?? generateCursorChecksum(authToken.trim());
|
360 |
+
|
361 |
+
const sessionid = uuidv5(authToken, uuidv5.DNS);
|
362 |
+
const clientKey = generateHashed64Hex(authToken)
|
363 |
+
const cursorClientVersion = "0.45.11"
|
364 |
+
|
365 |
+
// Request the AvailableModels before StreamChat.
|
366 |
+
const availableModelsResponse = await fetch("https://api2.cursor.sh/aiserver.v1.AiService/AvailableModels", {
|
367 |
+
method: 'POST',
|
368 |
+
headers: {
|
369 |
+
'accept-encoding': 'gzip',
|
370 |
+
'authorization': `Bearer ${authToken}`,
|
371 |
+
'connect-protocol-version': '1',
|
372 |
+
'content-type': 'application/proto',
|
373 |
+
'user-agent': 'connect-es/1.6.1',
|
374 |
+
'x-amzn-trace-id': `Root=${uuidv4()}`,
|
375 |
+
'x-client-key': clientKey,
|
376 |
+
'x-cursor-checksum': checksum,
|
377 |
+
'x-cursor-client-version': cursorClientVersion,
|
378 |
+
'x-cursor-timezone': 'Asia/Shanghai',
|
379 |
+
'x-ghost-mode': 'true',
|
380 |
+
"x-request-id": uuidv4(),
|
381 |
+
"x-session-id": sessionid,
|
382 |
+
'Host': 'api2.cursor.sh',
|
383 |
+
},
|
384 |
+
})
|
385 |
+
|
386 |
+
const cursorBody = generateCursorBody(messages, model);
|
387 |
+
const response = await fetch('https://api2.cursor.sh/aiserver.v1.AiService/StreamChat', {
|
388 |
+
method: 'POST',
|
389 |
+
headers: {
|
390 |
+
'authorization': `Bearer ${authToken}`,
|
391 |
+
'connect-accept-encoding': 'gzip',
|
392 |
+
'connect-content-encoding': 'gzip',
|
393 |
+
'connect-protocol-version': '1',
|
394 |
+
'content-type': 'application/connect+proto',
|
395 |
+
'user-agent': 'connect-es/1.6.1',
|
396 |
+
'x-amzn-trace-id': `Root=${uuidv4()}`,
|
397 |
+
'x-client-key': clientKey,
|
398 |
+
'x-cursor-checksum': checksum,
|
399 |
+
'x-cursor-client-version': cursorClientVersion,
|
400 |
+
'x-cursor-timezone': 'Asia/Shanghai',
|
401 |
+
'x-ghost-mode': 'true',
|
402 |
+
'x-request-id': uuidv4(),
|
403 |
+
'x-session-id': sessionid,
|
404 |
+
'Host': 'api2.cursor.sh',
|
405 |
+
},
|
406 |
+
body: cursorBody,
|
407 |
+
timeout: {
|
408 |
+
connect: 5000,
|
409 |
+
read: 30000
|
410 |
+
}
|
411 |
+
});
|
412 |
+
|
413 |
+
// 处理响应
|
414 |
+
if (stream) {
|
415 |
+
res.setHeader('Content-Type', 'text/event-stream');
|
416 |
+
res.setHeader('Cache-Control', 'no-cache');
|
417 |
+
res.setHeader('Connection', 'keep-alive');
|
418 |
+
|
419 |
+
const responseId = `chatcmpl-${uuidv4()}`;
|
420 |
+
|
421 |
+
|
422 |
+
try {
|
423 |
+
let responseEnded = false; // 添加标志,标记响应是否已结束
|
424 |
+
|
425 |
+
for await (const chunk of response.body) {
|
426 |
+
// 如果响应已结束,不再处理后续数据
|
427 |
+
if (responseEnded) {
|
428 |
+
continue;
|
429 |
+
}
|
430 |
+
|
431 |
+
let text = chunkToUtf8String(chunk);
|
432 |
+
// 输出完整的text内容和类型,便于调试
|
433 |
+
//console.log("收到的响应:", typeof text, text && typeof text === 'object' ? JSON.stringify(text) : text);
|
434 |
+
|
435 |
+
|
436 |
+
// 检查是否返回了错误对象
|
437 |
+
if (text && typeof text === 'object' && text.error) {
|
438 |
+
// 检查是否包含特定的无效cookie错误信息
|
439 |
+
const errorStr = typeof text.error === 'string' ? text.error : JSON.stringify(text.error);
|
440 |
+
|
441 |
+
// 处理错误并获取结果
|
442 |
+
const errorResult = handleCursorError(errorStr, bearerToken, originalAuthToken);
|
443 |
+
|
444 |
+
// 如果是需要移除的cookie,从API Key中移除
|
445 |
+
if (errorResult.shouldRemoveCookie) {
|
446 |
+
const removed = keyManager.removeCookieFromApiKey(bearerToken, originalAuthToken);
|
447 |
+
console.log(`Cookie移除${removed ? '成功' : '失败'}`);
|
448 |
+
|
449 |
+
// 如果成功移除,在错误消息中添加明确提示
|
450 |
+
if (removed) {
|
451 |
+
errorResult.message = `⚠️ 目前Cookie已从API Key中移除 ⚠️\n\n${errorResult.message}`;
|
452 |
+
}
|
453 |
+
}
|
454 |
+
|
455 |
+
// 返回错误信息给客户端,作为assistant消息
|
456 |
+
res.write(
|
457 |
+
`data: ${JSON.stringify({
|
458 |
+
id: responseId,
|
459 |
+
object: 'chat.completion.chunk',
|
460 |
+
created: Math.floor(Date.now() / 1000),
|
461 |
+
model: req.body.model,
|
462 |
+
choices: [
|
463 |
+
{
|
464 |
+
index: 0,
|
465 |
+
delta: {
|
466 |
+
content: errorResult.message,
|
467 |
+
},
|
468 |
+
},
|
469 |
+
],
|
470 |
+
})}\n\n`
|
471 |
+
);
|
472 |
+
|
473 |
+
res.write('data: [DONE]\n\n');
|
474 |
+
responseEnded = true; // 标记响应已结束
|
475 |
+
break; // 跳出循环,不再处理后续数据
|
476 |
+
}
|
477 |
+
|
478 |
+
if (text && text.length > 0) {
|
479 |
+
res.write(
|
480 |
+
`data: ${JSON.stringify({
|
481 |
+
id: responseId,
|
482 |
+
object: 'chat.completion.chunk',
|
483 |
+
created: Math.floor(Date.now() / 1000),
|
484 |
+
model: req.body.model,
|
485 |
+
choices: [
|
486 |
+
{
|
487 |
+
index: 0,
|
488 |
+
delta: {
|
489 |
+
content: text,
|
490 |
+
},
|
491 |
+
},
|
492 |
+
],
|
493 |
+
})}\n\n`
|
494 |
+
);
|
495 |
+
}
|
496 |
+
}
|
497 |
+
|
498 |
+
// 只有在响应尚未结束的情况下,才发送结束标记
|
499 |
+
if (!responseEnded) {
|
500 |
+
res.write('data: [DONE]\n\n');
|
501 |
+
res.end();
|
502 |
+
}
|
503 |
+
} catch (streamError) {
|
504 |
+
console.error('Stream error:', streamError);
|
505 |
+
// 确保在发送错误信息前检查响应是否已结束
|
506 |
+
if (!res.writableEnded) {
|
507 |
+
if (streamError.name === 'TimeoutError') {
|
508 |
+
// 将超时错误作为assistant消息发送
|
509 |
+
const errorMessage = `⚠️ 请求超时 ⚠️\n\n错误:服务器响应超时,请稍后重试。`;
|
510 |
+
res.write(
|
511 |
+
`data: ${JSON.stringify({
|
512 |
+
id: responseId,
|
513 |
+
object: 'chat.completion.chunk',
|
514 |
+
created: Math.floor(Date.now() / 1000),
|
515 |
+
model: req.body.model,
|
516 |
+
choices: [
|
517 |
+
{
|
518 |
+
index: 0,
|
519 |
+
delta: {
|
520 |
+
content: errorMessage,
|
521 |
+
},
|
522 |
+
},
|
523 |
+
],
|
524 |
+
})}\n\n`
|
525 |
+
);
|
526 |
+
} else {
|
527 |
+
// 将处理错误作为assistant消息发送
|
528 |
+
const errorMessage = `⚠️ 处理错误 ⚠️\n\n错误:流处理出错,请稍后重试。\n\n${streamError.message || ''}`;
|
529 |
+
res.write(
|
530 |
+
`data: ${JSON.stringify({
|
531 |
+
id: responseId,
|
532 |
+
object: 'chat.completion.chunk',
|
533 |
+
created: Math.floor(Date.now() / 1000),
|
534 |
+
model: req.body.model,
|
535 |
+
choices: [
|
536 |
+
{
|
537 |
+
index: 0,
|
538 |
+
delta: {
|
539 |
+
content: errorMessage,
|
540 |
+
},
|
541 |
+
},
|
542 |
+
],
|
543 |
+
})}\n\n`
|
544 |
+
);
|
545 |
+
}
|
546 |
+
res.write('data: [DONE]\n\n');
|
547 |
+
res.end();
|
548 |
+
}
|
549 |
+
}
|
550 |
+
} else {
|
551 |
+
try {
|
552 |
+
let text = '';
|
553 |
+
let responseEnded = false; // 添加标志,标记响应是否已结束
|
554 |
+
|
555 |
+
for await (const chunk of response.body) {
|
556 |
+
// 如果响应已结束,不再处理后续数据
|
557 |
+
if (responseEnded) {
|
558 |
+
continue;
|
559 |
+
}
|
560 |
+
|
561 |
+
const chunkText = chunkToUtf8String(chunk);
|
562 |
+
// 输出完整的chunkText内容和类型,便于调试
|
563 |
+
//console.log("收到的非流式响应:", typeof chunkText, chunkText && typeof chunkText === 'object' ? JSON.stringify(chunkText) : chunkText);
|
564 |
+
|
565 |
+
// 检查是否返回了错误对象
|
566 |
+
if (chunkText && typeof chunkText === 'object' && chunkText.error) {
|
567 |
+
//console.error('检测到错误响应:', chunkText.error);
|
568 |
+
|
569 |
+
// 检查是否包含特定的无效cookie错误信息
|
570 |
+
const errorStr = typeof chunkText.error === 'string' ? chunkText.error : JSON.stringify(chunkText.error);
|
571 |
+
|
572 |
+
// 处理错误并获取结果
|
573 |
+
const errorResult = handleCursorError(errorStr, bearerToken, originalAuthToken);
|
574 |
+
|
575 |
+
// 如果是需要移除的cookie,从API Key中移除
|
576 |
+
if (errorResult.shouldRemoveCookie) {
|
577 |
+
const removed = keyManager.removeCookieFromApiKey(bearerToken, originalAuthToken);
|
578 |
+
console.log(`Cookie移除${removed ? '成功' : '失败'}`);
|
579 |
+
|
580 |
+
// 如果成功移除,在错误消息中添加明确提示
|
581 |
+
if (removed) {
|
582 |
+
errorResult.message = `⚠️ 目前Cookie已从API Key中移除 ⚠️\n\n${errorResult.message}`;
|
583 |
+
}
|
584 |
+
}
|
585 |
+
|
586 |
+
// 无效cookie错误,格式化为assistant消息
|
587 |
+
res.json({
|
588 |
+
id: `chatcmpl-${uuidv4()}`,
|
589 |
+
object: 'chat.completion',
|
590 |
+
created: Math.floor(Date.now() / 1000),
|
591 |
+
model,
|
592 |
+
choices: [
|
593 |
+
{
|
594 |
+
index: 0,
|
595 |
+
message: {
|
596 |
+
role: 'assistant',
|
597 |
+
content: errorResult.message,
|
598 |
+
},
|
599 |
+
finish_reason: 'stop',
|
600 |
+
},
|
601 |
+
],
|
602 |
+
usage: {
|
603 |
+
prompt_tokens: 0,
|
604 |
+
completion_tokens: 0,
|
605 |
+
total_tokens: 0,
|
606 |
+
},
|
607 |
+
});
|
608 |
+
|
609 |
+
responseEnded = true; // 标记响应已结束
|
610 |
+
break; // 跳出循环,不再处理后续数据
|
611 |
+
}
|
612 |
+
|
613 |
+
// 正常文本,添加到结果中
|
614 |
+
if (chunkText && typeof chunkText === 'string') {
|
615 |
+
text += chunkText;
|
616 |
+
}
|
617 |
+
}
|
618 |
+
|
619 |
+
// 只有在响应尚未结束的情况下,才处理和返回结果
|
620 |
+
if (!responseEnded) {
|
621 |
+
// 对解析后的字符串进行进一步处理
|
622 |
+
text = text.replace(/^.*<\|END_USER\|>/s, '');
|
623 |
+
text = text.replace(/^\n[a-zA-Z]?/, '').trim();
|
624 |
+
// console.log(text)
|
625 |
+
|
626 |
+
res.json({
|
627 |
+
id: `chatcmpl-${uuidv4()}`,
|
628 |
+
object: 'chat.completion',
|
629 |
+
created: Math.floor(Date.now() / 1000),
|
630 |
+
model,
|
631 |
+
choices: [
|
632 |
+
{
|
633 |
+
index: 0,
|
634 |
+
message: {
|
635 |
+
role: 'assistant',
|
636 |
+
content: text,
|
637 |
+
},
|
638 |
+
finish_reason: 'stop',
|
639 |
+
},
|
640 |
+
],
|
641 |
+
usage: {
|
642 |
+
prompt_tokens: 0,
|
643 |
+
completion_tokens: 0,
|
644 |
+
total_tokens: 0,
|
645 |
+
},
|
646 |
+
});
|
647 |
+
}
|
648 |
+
} catch (error) {
|
649 |
+
console.error('Non-stream error:', error);
|
650 |
+
// 确保在发送错误信息前检查响应是否已结束
|
651 |
+
if (!res.headersSent) {
|
652 |
+
if (error.name === 'TimeoutError') {
|
653 |
+
// 使用统一的错误格式
|
654 |
+
const errorMessage = `⚠️ 请求超时 ⚠️\n\n错误:服务器响应超时,请稍后重试。`;
|
655 |
+
return res.json({
|
656 |
+
id: `chatcmpl-${uuidv4()}`,
|
657 |
+
object: 'chat.completion',
|
658 |
+
created: Math.floor(Date.now() / 1000),
|
659 |
+
model: req.body.model || 'unknown',
|
660 |
+
choices: [
|
661 |
+
{
|
662 |
+
index: 0,
|
663 |
+
message: {
|
664 |
+
role: 'assistant',
|
665 |
+
content: errorMessage,
|
666 |
+
},
|
667 |
+
finish_reason: 'stop',
|
668 |
+
},
|
669 |
+
],
|
670 |
+
usage: {
|
671 |
+
prompt_tokens: 0,
|
672 |
+
completion_tokens: 0,
|
673 |
+
total_tokens: 0,
|
674 |
+
},
|
675 |
+
});
|
676 |
+
}
|
677 |
+
throw error;
|
678 |
+
}
|
679 |
+
}
|
680 |
+
}
|
681 |
+
} catch (error) {
|
682 |
+
console.error('Error:', error);
|
683 |
+
if (!res.headersSent) {
|
684 |
+
const errorText = error.name === 'TimeoutError' ? '请求超时' : '服务器内部错误';
|
685 |
+
|
686 |
+
if (req.body.stream) {
|
687 |
+
// 流式响应格式的错误
|
688 |
+
const responseId = `chatcmpl-${uuidv4()}`;
|
689 |
+
// 添加清晰的错误提示
|
690 |
+
const errorMessage = `⚠️ 请求失败 ⚠️\n\n错误:${errorText},请稍后重试。\n\n${error.message || ''}`;
|
691 |
+
res.write(
|
692 |
+
`data: ${JSON.stringify({
|
693 |
+
id: responseId,
|
694 |
+
object: 'chat.completion.chunk',
|
695 |
+
created: Math.floor(Date.now() / 1000),
|
696 |
+
model: req.body.model || 'unknown',
|
697 |
+
choices: [
|
698 |
+
{
|
699 |
+
index: 0,
|
700 |
+
delta: {
|
701 |
+
content: errorMessage,
|
702 |
+
},
|
703 |
+
},
|
704 |
+
],
|
705 |
+
})}\n\n`
|
706 |
+
);
|
707 |
+
res.write('data: [DONE]\n\n');
|
708 |
+
res.end();
|
709 |
+
} else {
|
710 |
+
// 非流式响应格式的错误
|
711 |
+
// 添加清晰的错误提示
|
712 |
+
const errorMessage = `⚠️ 请求失败 ⚠️\n\n错误:${errorText},请稍后重试。\n\n${error.message || ''}`;
|
713 |
+
res.json({
|
714 |
+
id: `chatcmpl-${uuidv4()}`,
|
715 |
+
object: 'chat.completion',
|
716 |
+
created: Math.floor(Date.now() / 1000),
|
717 |
+
model: req.body.model || 'unknown',
|
718 |
+
choices: [
|
719 |
+
{
|
720 |
+
index: 0,
|
721 |
+
message: {
|
722 |
+
role: 'assistant',
|
723 |
+
content: errorMessage,
|
724 |
+
},
|
725 |
+
finish_reason: 'stop',
|
726 |
+
},
|
727 |
+
],
|
728 |
+
usage: {
|
729 |
+
prompt_tokens: 0,
|
730 |
+
completion_tokens: 0,
|
731 |
+
total_tokens: 0,
|
732 |
+
},
|
733 |
+
});
|
734 |
+
}
|
735 |
+
}
|
736 |
+
}
|
737 |
+
});
|
738 |
+
|
739 |
+
// 触发Cookie刷新
|
740 |
+
router.post("/refresh-cookies", async (req, res) => {
|
741 |
+
try {
|
742 |
+
// 如果已经有刷新进程在运行,则返回错误
|
743 |
+
if (refreshStatus.isRunning) {
|
744 |
+
return res.status(409).json({
|
745 |
+
success: false,
|
746 |
+
message: '已有刷新进程在运行,请等待完成后再试'
|
747 |
+
});
|
748 |
+
}
|
749 |
+
|
750 |
+
// 获取请求参数
|
751 |
+
const apiKey = req.query.apiKey || '';
|
752 |
+
|
753 |
+
// 重置刷新状态
|
754 |
+
refreshStatus = {
|
755 |
+
isRunning: true,
|
756 |
+
status: 'running',
|
757 |
+
message: '正在启动刷新进程...',
|
758 |
+
startTime: new Date(),
|
759 |
+
endTime: null,
|
760 |
+
error: null
|
761 |
+
};
|
762 |
+
|
763 |
+
console.log(`收到刷新Cookie请求,API Key: ${apiKey || '所有'}`);
|
764 |
+
|
765 |
+
// 构建命令行参数
|
766 |
+
const args = [];
|
767 |
+
if (apiKey) {
|
768 |
+
args.push(apiKey);
|
769 |
+
}
|
770 |
+
|
771 |
+
// 获取auto-refresh-cookies.js的绝对路径
|
772 |
+
const scriptPath = path.resolve(__dirname, '../../auto-refresh-cookies.js');
|
773 |
+
|
774 |
+
// 启动子进程执行刷新脚本
|
775 |
+
const refreshProcess = spawn('node', [scriptPath, ...args], {
|
776 |
+
stdio: ['ignore', 'pipe', 'pipe']
|
777 |
+
});
|
778 |
+
|
779 |
+
// 收集输出
|
780 |
+
let output = '';
|
781 |
+
|
782 |
+
refreshProcess.stdout.on('data', (data) => {
|
783 |
+
const text = data.toString();
|
784 |
+
output += text;
|
785 |
+
console.log(`刷新进程输出: ${text}`);
|
786 |
+
|
787 |
+
// 更新状态消息
|
788 |
+
if (text.includes('开始自动刷新')) {
|
789 |
+
refreshStatus.message = '正在刷新Cookie...';
|
790 |
+
} else if (text.includes('刷新结果:')) {
|
791 |
+
refreshStatus.message = text.trim();
|
792 |
+
}
|
793 |
+
});
|
794 |
+
|
795 |
+
refreshProcess.stderr.on('data', (data) => {
|
796 |
+
const text = data.toString();
|
797 |
+
output += text;
|
798 |
+
console.error(`刷新进程错误: ${text}`);
|
799 |
+
|
800 |
+
// 更新错误信息
|
801 |
+
refreshStatus.error = text.trim();
|
802 |
+
refreshStatus.message = `发生错误: ${text.trim()}`;
|
803 |
+
});
|
804 |
+
|
805 |
+
refreshProcess.on('close', (code) => {
|
806 |
+
console.log(`刷新进程退出,代码: ${code}`);
|
807 |
+
|
808 |
+
refreshStatus.isRunning = false;
|
809 |
+
refreshStatus.endTime = new Date();
|
810 |
+
|
811 |
+
if (code === 0) {
|
812 |
+
refreshStatus.status = 'completed';
|
813 |
+
|
814 |
+
// 提取成功信息
|
815 |
+
const successMatch = output.match(/成功刷新 (\d+) 个/);
|
816 |
+
if (successMatch) {
|
817 |
+
refreshStatus.message = `成功刷新 ${successMatch[1]} 个API Key的Cookie`;
|
818 |
+
} else {
|
819 |
+
refreshStatus.message = '刷新完成';
|
820 |
+
}
|
821 |
+
|
822 |
+
// 子进程执行完成后,重新初始化API Keys来加载新的Cookie
|
823 |
+
try {
|
824 |
+
const keyManager = require('../utils/keyManager');
|
825 |
+
console.log('子进程刷新Cookie完成,重新初始化主进程中的API Keys...');
|
826 |
+
keyManager.initializeApiKeys();
|
827 |
+
console.log('主进程API Keys重新加载完成');
|
828 |
+
} catch (initError) {
|
829 |
+
console.error('重新初始化API Keys失败:', initError);
|
830 |
+
}
|
831 |
+
} else {
|
832 |
+
refreshStatus.status = 'failed';
|
833 |
+
refreshStatus.message = refreshStatus.error || '刷新失败,请查看服务器日志';
|
834 |
+
}
|
835 |
+
});
|
836 |
+
|
837 |
+
// 立即返回响应,不等待刷新完成
|
838 |
+
return res.json({
|
839 |
+
success: true,
|
840 |
+
message: '刷新请求已接受,正在后台处理'
|
841 |
+
});
|
842 |
+
} catch (error) {
|
843 |
+
console.error('触发刷新Cookie失败:', error);
|
844 |
+
|
845 |
+
// 更新刷新状态
|
846 |
+
refreshStatus.isRunning = false;
|
847 |
+
refreshStatus.status = 'failed';
|
848 |
+
refreshStatus.endTime = new Date();
|
849 |
+
refreshStatus.error = error.message;
|
850 |
+
refreshStatus.message = `触发刷新失败: ${error.message}`;
|
851 |
+
|
852 |
+
return res.status(500).json({
|
853 |
+
success: false,
|
854 |
+
message: `触发刷新失败: ${error.message}`
|
855 |
+
});
|
856 |
+
}
|
857 |
+
});
|
858 |
+
|
859 |
+
// 查询Cookie刷新状态
|
860 |
+
router.get("/refresh-status", (req, res) => {
|
861 |
+
try {
|
862 |
+
// 返回当前刷新状态
|
863 |
+
return res.json({
|
864 |
+
success: true,
|
865 |
+
data: {
|
866 |
+
...refreshStatus,
|
867 |
+
isRunning: refreshStatus.isRunning || false,
|
868 |
+
status: refreshStatus.status || 'unknown',
|
869 |
+
message: refreshStatus.message || '未触发刷新',
|
870 |
+
startTime: refreshStatus.startTime || null,
|
871 |
+
endTime: refreshStatus.endTime || null
|
872 |
+
}
|
873 |
+
});
|
874 |
+
} catch (error) {
|
875 |
+
console.error('获取刷新状态失败:', error);
|
876 |
+
return res.status(500).json({
|
877 |
+
success: false,
|
878 |
+
message: `获取刷新状态失败: ${error.message}`
|
879 |
+
});
|
880 |
+
}
|
881 |
+
});
|
882 |
+
|
883 |
+
// 在文件末尾添加错误处理函数
|
884 |
+
function handleCursorError(errorStr, bearerToken, originalAuthToken) {
|
885 |
+
let message = '';
|
886 |
+
let shouldRemoveCookie = false;
|
887 |
+
|
888 |
+
if (errorStr.includes('Not logged in')) {
|
889 |
+
// 更明确的错误日志
|
890 |
+
if (originalAuthToken === bearerToken) {
|
891 |
+
console.error(`检测到API Key "${bearerToken}" 中没有可用Cookie,正在尝试以向后兼容模式使用API Key本身`);
|
892 |
+
message = `错误:API Key "${bearerToken}" 中没有可用的Cookie。请添加有效的Cookie到此API Key,或使用其他有效的API Key。\n\n详细信息:${errorStr}`;
|
893 |
+
} else {
|
894 |
+
console.error('检测到无效cookie:', originalAuthToken);
|
895 |
+
message = `错误:Cookie无效或已过期,请更新Cookie。\n\n详细信息:${errorStr}`;
|
896 |
+
}
|
897 |
+
shouldRemoveCookie = true;
|
898 |
+
} else if (errorStr.includes('You\'ve reached your trial request limit')) {
|
899 |
+
console.error('检测到额度用尽cookie:', originalAuthToken);
|
900 |
+
message = `错误:Cookie使用额度已用完,请更换Cookie或等待刷新。\n\n详细信息:${errorStr}`;
|
901 |
+
shouldRemoveCookie = true;
|
902 |
+
} else if (errorStr.includes('User is unauthorized')) {
|
903 |
+
console.error('检测到未授权cookie:', originalAuthToken);
|
904 |
+
message = `错误:Cookie已被封禁或失效,请更换Cookie。\n\n详细信息:${errorStr}`;
|
905 |
+
shouldRemoveCookie = true;
|
906 |
+
} else if (errorStr.includes('suspicious activity checks')) {
|
907 |
+
console.error('检测到IP黑名单:', originalAuthToken);
|
908 |
+
message = `错误:IP可能被列入黑名单,请尝试更换网络环境或使用代理。\n\n详细信息:${errorStr}`;
|
909 |
+
shouldRemoveCookie = false;
|
910 |
+
} else if (errorStr.includes('Too many computers')) {
|
911 |
+
console.error('检测到账户暂时被封禁:', originalAuthToken);
|
912 |
+
message = `错误:账户因在多台设备登录而暂时被封禁,请稍后再试或更换账户。\n\n详细信息:${errorStr}`;
|
913 |
+
shouldRemoveCookie = true;
|
914 |
+
} else if (errorStr.includes('Login expired') || errorStr.includes('login expired')) {
|
915 |
+
console.error('检测到登录过期cookie:', originalAuthToken);
|
916 |
+
message = `错误:Cookie登录已过期,请更新Cookie。\n\n详细信息:${errorStr}`;
|
917 |
+
shouldRemoveCookie = true;
|
918 |
+
} else {
|
919 |
+
// 非Cookie相关错误
|
920 |
+
console.error('检测到其他错误:', errorStr);
|
921 |
+
message = `错误:请求失败。\n\n详细信息:${errorStr}`;
|
922 |
+
shouldRemoveCookie = false;
|
923 |
+
}
|
924 |
+
|
925 |
+
return {
|
926 |
+
message,
|
927 |
+
shouldRemoveCookie
|
928 |
+
};
|
929 |
+
}
|
930 |
+
|
931 |
+
module.exports = router;
|
src/utils/cookieRefresher.js
ADDED
@@ -0,0 +1,750 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
const fs = require('fs');
|
2 |
+
const path = require('path');
|
3 |
+
const csv = require('csv-parser');
|
4 |
+
const axios = require('axios');
|
5 |
+
const AdmZip = require('adm-zip');
|
6 |
+
const { Octokit } = require('@octokit/rest');
|
7 |
+
const keyManager = require('./keyManager');
|
8 |
+
const config = require('../config/config');
|
9 |
+
const { extractCookiesFromCsv } = require('./extractCookieFromCsv');
|
10 |
+
|
11 |
+
// GitHub 仓库信息从环境变量中获取
|
12 |
+
const GITHUB_OWNER = process.env.GITHUB_OWNER || 'liuw1535';
|
13 |
+
const GITHUB_REPO = process.env.GITHUB_REPO || 'Cursor-Register';
|
14 |
+
const GITHUB_TOKEN = process.env.GITHUB_TOKEN; // 需要在环境变量中设置
|
15 |
+
const GITHUB_WORKFLOW_ID = process.env.GITHUB_WORKFLOW_ID || 'register.yml';
|
16 |
+
const TRIGGER_WORKFLOW = process.env.TRIGGER_WORKFLOW === 'true';
|
17 |
+
|
18 |
+
// 下载目录
|
19 |
+
const DOWNLOAD_DIR = path.join(__dirname, '../../downloads');
|
20 |
+
const EXTRACT_DIR = path.join(__dirname, '../../extracted');
|
21 |
+
|
22 |
+
// 确保目录存在
|
23 |
+
function ensureDirectoryExists(dir) {
|
24 |
+
if (!fs.existsSync(dir)) {
|
25 |
+
try {
|
26 |
+
fs.mkdirSync(dir, { recursive: true });
|
27 |
+
console.log(`创建目录成功: ${dir}`);
|
28 |
+
} catch (err) {
|
29 |
+
console.error(`创建目录失败: ${dir}`, err);
|
30 |
+
throw err;
|
31 |
+
}
|
32 |
+
}
|
33 |
+
}
|
34 |
+
|
35 |
+
// 触发 GitHub Actions 工作流
|
36 |
+
async function triggerWorkflow() {
|
37 |
+
try {
|
38 |
+
if (!GITHUB_TOKEN) {
|
39 |
+
console.error('未设置 GITHUB_TOKEN,无法触发工作流');
|
40 |
+
return null;
|
41 |
+
}
|
42 |
+
|
43 |
+
console.log(`正在触发 GitHub Actions 工作流: ${GITHUB_WORKFLOW_ID}...`);
|
44 |
+
const octokit = new Octokit({
|
45 |
+
auth: GITHUB_TOKEN
|
46 |
+
});
|
47 |
+
|
48 |
+
// 从环境变量获取工作流参数
|
49 |
+
const number = process.env.REGISTER_NUMBER || '2';
|
50 |
+
const maxWorkers = process.env.REGISTER_MAX_WORKERS || '1';
|
51 |
+
const emailServer = process.env.REGISTER_EMAIL_SERVER || 'TempEmail';
|
52 |
+
const ingestToOneapi = process.env.REGISTER_INGEST_TO_ONEAPI === 'true';
|
53 |
+
const uploadArtifact = process.env.REGISTER_UPLOAD_ARTIFACT !== 'false'; // 默认为true
|
54 |
+
const useConfigFile = process.env.REGISTER_USE_CONFIG_FILE !== 'false'; // 默认为true
|
55 |
+
const emailConfigs = process.env.REGISTER_EMAIL_CONFIGS || '[]';
|
56 |
+
|
57 |
+
console.log(`工作流参数: number=${number}, maxWorkers=${maxWorkers}, emailServer=${emailServer}, ingestToOneapi=${ingestToOneapi}, uploadArtifact=${uploadArtifact}, useConfigFile=${useConfigFile}`);
|
58 |
+
|
59 |
+
// 获取触发前的最新工作流ID,用于后续识别新触发的工作流
|
60 |
+
const { data: beforeWorkflowRuns } = await octokit.actions.listWorkflowRuns({
|
61 |
+
owner: GITHUB_OWNER,
|
62 |
+
repo: GITHUB_REPO,
|
63 |
+
workflow_id: GITHUB_WORKFLOW_ID,
|
64 |
+
per_page: 1
|
65 |
+
});
|
66 |
+
|
67 |
+
const latestWorkflowIdBefore = beforeWorkflowRuns.workflow_runs && beforeWorkflowRuns.workflow_runs.length > 0
|
68 |
+
? beforeWorkflowRuns.workflow_runs[0].id
|
69 |
+
: 0;
|
70 |
+
|
71 |
+
console.log(`触发前最新工作流ID: ${latestWorkflowIdBefore}`);
|
72 |
+
|
73 |
+
// 触发工作流
|
74 |
+
const response = await octokit.actions.createWorkflowDispatch({
|
75 |
+
owner: GITHUB_OWNER,
|
76 |
+
repo: GITHUB_REPO,
|
77 |
+
workflow_id: GITHUB_WORKFLOW_ID,
|
78 |
+
ref: 'main', // 默认使用 main 分支,可以根据需要修改
|
79 |
+
inputs: {
|
80 |
+
number: number,
|
81 |
+
max_workers: maxWorkers,
|
82 |
+
email_server: emailServer,
|
83 |
+
ingest_to_oneapi: ingestToOneapi.toString(),
|
84 |
+
upload_artifact: uploadArtifact.toString(),
|
85 |
+
use_config_file: useConfigFile.toString(),
|
86 |
+
email_configs: emailConfigs
|
87 |
+
}
|
88 |
+
});
|
89 |
+
|
90 |
+
console.log('工作流触发成功,等待工作流开始运行...');
|
91 |
+
|
92 |
+
// 等待新工作流出现并获取其ID
|
93 |
+
let newWorkflowRunId = null;
|
94 |
+
let findAttempts = 0;
|
95 |
+
const maxFindAttempts = 30; // 最多等待30次,每次5秒
|
96 |
+
|
97 |
+
while (findAttempts < maxFindAttempts && !newWorkflowRunId) {
|
98 |
+
findAttempts++;
|
99 |
+
console.log(`查找新触发的工作流,尝试 ${findAttempts}/${maxFindAttempts}...`);
|
100 |
+
|
101 |
+
try {
|
102 |
+
const { data: afterWorkflowRuns } = await octokit.actions.listWorkflowRuns({
|
103 |
+
owner: GITHUB_OWNER,
|
104 |
+
repo: GITHUB_REPO,
|
105 |
+
workflow_id: GITHUB_WORKFLOW_ID,
|
106 |
+
per_page: 5
|
107 |
+
});
|
108 |
+
|
109 |
+
if (afterWorkflowRuns.workflow_runs && afterWorkflowRuns.workflow_runs.length > 0) {
|
110 |
+
// 查找ID大于之前最新工作流ID的工作流(即新触发的工作流)
|
111 |
+
const newWorkflow = afterWorkflowRuns.workflow_runs.find(run => run.id > latestWorkflowIdBefore);
|
112 |
+
if (newWorkflow) {
|
113 |
+
newWorkflowRunId = newWorkflow.id;
|
114 |
+
console.log(`找到新触发的工作流,ID: ${newWorkflowRunId}, 状态: ${newWorkflow.status}`);
|
115 |
+
}
|
116 |
+
}
|
117 |
+
} catch (error) {
|
118 |
+
console.error(`查找工作流时出错 (尝试 ${findAttempts}/${maxFindAttempts}): ${error.message}`);
|
119 |
+
// 出错时继续尝试,不中断循环
|
120 |
+
}
|
121 |
+
|
122 |
+
if (!newWorkflowRunId) {
|
123 |
+
// 等待5秒后再次检查
|
124 |
+
await new Promise(resolve => setTimeout(resolve, 5000));
|
125 |
+
}
|
126 |
+
}
|
127 |
+
|
128 |
+
if (!newWorkflowRunId) {
|
129 |
+
console.log('未能找到新触发的工作流,可能触发失败');
|
130 |
+
return null;
|
131 |
+
}
|
132 |
+
|
133 |
+
// 等待工作流完成
|
134 |
+
let attempts = 0;
|
135 |
+
const maxAttempts = 120; // 最多等待120次,每次30秒,总共60分钟
|
136 |
+
let consecutiveErrors = 0;
|
137 |
+
const maxConsecutiveErrors = 5; // 最多允许连续5次错误
|
138 |
+
|
139 |
+
while (attempts < maxAttempts) {
|
140 |
+
attempts++;
|
141 |
+
console.log(`等待工作流完成,尝试 ${attempts}/${maxAttempts}...`);
|
142 |
+
|
143 |
+
try {
|
144 |
+
// 获取工作流状态
|
145 |
+
const { data: workflowRun } = await octokit.actions.getWorkflowRun({
|
146 |
+
owner: GITHUB_OWNER,
|
147 |
+
repo: GITHUB_REPO,
|
148 |
+
run_id: newWorkflowRunId
|
149 |
+
});
|
150 |
+
|
151 |
+
// 重置连续错误计数
|
152 |
+
consecutiveErrors = 0;
|
153 |
+
|
154 |
+
console.log(`工作流状态: ${workflowRun.status}, 结果: ${workflowRun.conclusion || '进行中'}`);
|
155 |
+
|
156 |
+
// 检查工作流是否完成
|
157 |
+
if (workflowRun.status === 'completed') {
|
158 |
+
if (workflowRun.conclusion === 'success') {
|
159 |
+
console.log(`工作流运行成功,ID: ${newWorkflowRunId}`);
|
160 |
+
return workflowRun;
|
161 |
+
} else {
|
162 |
+
console.log(`工作流运行失败,结果: ${workflowRun.conclusion}`);
|
163 |
+
return null;
|
164 |
+
}
|
165 |
+
}
|
166 |
+
} catch (error) {
|
167 |
+
consecutiveErrors++;
|
168 |
+
console.error(`获取工作流状态时出错 (尝试 ${attempts}/${maxAttempts}, 连续错误 ${consecutiveErrors}/${maxConsecutiveErrors}): ${error.message}`);
|
169 |
+
|
170 |
+
// 如果连续错误次数超过阈值,则放弃
|
171 |
+
if (consecutiveErrors >= maxConsecutiveErrors) {
|
172 |
+
console.error(`连续错误次数超过阈值 (${maxConsecutiveErrors}),放弃等待`);
|
173 |
+
throw new Error(`连续 ${maxConsecutiveErrors} 次获取工作流状态失败: ${error.message}`);
|
174 |
+
}
|
175 |
+
|
176 |
+
// 错误后等待时间稍微延长
|
177 |
+
await new Promise(resolve => setTimeout(resolve, 10000));
|
178 |
+
// 继续循环,不中断
|
179 |
+
continue;
|
180 |
+
}
|
181 |
+
|
182 |
+
// 等待30秒后再次检查
|
183 |
+
await new Promise(resolve => setTimeout(resolve, 30000));
|
184 |
+
}
|
185 |
+
|
186 |
+
console.log('等待工作流完成超时');
|
187 |
+
return null;
|
188 |
+
} catch (error) {
|
189 |
+
console.error('触发工作流失败:', error);
|
190 |
+
throw error; // 重新抛出错误,让调用者处理
|
191 |
+
}
|
192 |
+
}
|
193 |
+
|
194 |
+
// 从 GitHub Actions 获取最新的 Artifact
|
195 |
+
async function getLatestArtifact() {
|
196 |
+
try {
|
197 |
+
console.log('正在连接 GitHub API...');
|
198 |
+
const octokit = new Octokit({
|
199 |
+
auth: GITHUB_TOKEN
|
200 |
+
});
|
201 |
+
|
202 |
+
// 如果配置了自动触发工作流,则先触发工作流
|
203 |
+
let workflowRun = null;
|
204 |
+
if (TRIGGER_WORKFLOW) {
|
205 |
+
console.log('配置了自动触发工作流,正在触发...');
|
206 |
+
try {
|
207 |
+
workflowRun = await triggerWorkflow();
|
208 |
+
} catch (error) {
|
209 |
+
console.error('触发工作流过程中出现错误:', error.message);
|
210 |
+
console.log('尝试继续使用已找到的工作流ID...');
|
211 |
+
|
212 |
+
// 尝试获取最新的工作流,看是否有正在运行的工作流
|
213 |
+
const { data: runningWorkflows } = await octokit.actions.listWorkflowRuns({
|
214 |
+
owner: GITHUB_OWNER,
|
215 |
+
repo: GITHUB_REPO,
|
216 |
+
workflow_id: GITHUB_WORKFLOW_ID,
|
217 |
+
status: 'in_progress',
|
218 |
+
per_page: 5
|
219 |
+
});
|
220 |
+
|
221 |
+
if (runningWorkflows.workflow_runs && runningWorkflows.workflow_runs.length > 0) {
|
222 |
+
// 找到正在运行的工作流
|
223 |
+
const runningWorkflow = runningWorkflows.workflow_runs[0];
|
224 |
+
console.log(`找到正在运行的工作流,ID: ${runningWorkflow.id}, 状态: ${runningWorkflow.status}`);
|
225 |
+
|
226 |
+
// 等待工作流完成
|
227 |
+
let attempts = 0;
|
228 |
+
const maxAttempts = 120; // 最多等待120次,每次30秒,总共60分钟
|
229 |
+
let consecutiveErrors = 0;
|
230 |
+
const maxConsecutiveErrors = 5; // 最多允许连续5次错误
|
231 |
+
|
232 |
+
while (attempts < maxAttempts) {
|
233 |
+
attempts++;
|
234 |
+
console.log(`等待工作流完成,尝试 ${attempts}/${maxAttempts}...`);
|
235 |
+
|
236 |
+
try {
|
237 |
+
// 获取工作流状态
|
238 |
+
const { data: currentWorkflow } = await octokit.actions.getWorkflowRun({
|
239 |
+
owner: GITHUB_OWNER,
|
240 |
+
repo: GITHUB_REPO,
|
241 |
+
run_id: runningWorkflow.id
|
242 |
+
});
|
243 |
+
|
244 |
+
// 重置连续错误计数
|
245 |
+
consecutiveErrors = 0;
|
246 |
+
|
247 |
+
console.log(`工作流状态: ${currentWorkflow.status}, 结果: ${currentWorkflow.conclusion || '进行中'}`);
|
248 |
+
|
249 |
+
// 检查工作流是否完成
|
250 |
+
if (currentWorkflow.status === 'completed') {
|
251 |
+
if (currentWorkflow.conclusion === 'success') {
|
252 |
+
console.log(`工作流运行成功,ID: ${currentWorkflow.id}`);
|
253 |
+
workflowRun = currentWorkflow;
|
254 |
+
break;
|
255 |
+
} else {
|
256 |
+
console.log(`工作流运行失败,结果: ${currentWorkflow.conclusion}`);
|
257 |
+
break;
|
258 |
+
}
|
259 |
+
}
|
260 |
+
} catch (err) {
|
261 |
+
consecutiveErrors++;
|
262 |
+
console.error(`获取工作流状态时出错 (尝试 ${attempts}/${maxAttempts}, 连续错误 ${consecutiveErrors}/${maxConsecutiveErrors}): ${err.message}`);
|
263 |
+
|
264 |
+
// 如果连续错误次数超过阈值,则放弃
|
265 |
+
if (consecutiveErrors >= maxConsecutiveErrors) {
|
266 |
+
console.error(`连续错误次数超过阈值 (${maxConsecutiveErrors}),放弃等待`);
|
267 |
+
break;
|
268 |
+
}
|
269 |
+
|
270 |
+
// 错误后等待时间稍微延长
|
271 |
+
await new Promise(resolve => setTimeout(resolve, 10000));
|
272 |
+
// 继续循环,不中断
|
273 |
+
continue;
|
274 |
+
}
|
275 |
+
|
276 |
+
// 等待30秒后再次检查
|
277 |
+
await new Promise(resolve => setTimeout(resolve, 30000));
|
278 |
+
}
|
279 |
+
}
|
280 |
+
}
|
281 |
+
|
282 |
+
if (!workflowRun) {
|
283 |
+
console.log('触发工作流失败或等待超时,尝试获取最新的工作流运行');
|
284 |
+
}
|
285 |
+
}
|
286 |
+
|
287 |
+
// 如果没有触发工作流或触发失败,则获取最新的工作流运行
|
288 |
+
if (!workflowRun) {
|
289 |
+
console.log('获取最新的工作流运行...');
|
290 |
+
const { data: workflowRuns } = await octokit.actions.listWorkflowRunsForRepo({
|
291 |
+
owner: GITHUB_OWNER,
|
292 |
+
repo: GITHUB_REPO,
|
293 |
+
status: 'success',
|
294 |
+
per_page: 5
|
295 |
+
});
|
296 |
+
|
297 |
+
if (!workflowRuns.workflow_runs || workflowRuns.workflow_runs.length === 0) {
|
298 |
+
console.log('没有找到成功的工作流运行');
|
299 |
+
return null;
|
300 |
+
}
|
301 |
+
|
302 |
+
// 获取最新成功运行的 Artifacts
|
303 |
+
workflowRun = workflowRuns.workflow_runs[0];
|
304 |
+
}
|
305 |
+
|
306 |
+
console.log(`找到最新的工作流运行: ${workflowRun.id}`);
|
307 |
+
|
308 |
+
// 等待一段时间,确保Artifact已经上传完成
|
309 |
+
console.log('等待Artifact上传完成...');
|
310 |
+
await new Promise(resolve => setTimeout(resolve, 10000));
|
311 |
+
|
312 |
+
// 获取工作流的Artifacts
|
313 |
+
let artifacts = null;
|
314 |
+
let artifactAttempts = 0;
|
315 |
+
const maxArtifactAttempts = 10; // 最多尝试10次,每次10秒
|
316 |
+
|
317 |
+
while (artifactAttempts < maxArtifactAttempts && (!artifacts || !artifacts.artifacts || artifacts.artifacts.length === 0)) {
|
318 |
+
artifactAttempts++;
|
319 |
+
console.log(`尝试获取Artifacts,尝试 ${artifactAttempts}/${maxArtifactAttempts}...`);
|
320 |
+
|
321 |
+
try {
|
322 |
+
const response = await octokit.actions.listWorkflowRunArtifacts({
|
323 |
+
owner: GITHUB_OWNER,
|
324 |
+
repo: GITHUB_REPO,
|
325 |
+
run_id: workflowRun.id
|
326 |
+
});
|
327 |
+
|
328 |
+
artifacts = response.data;
|
329 |
+
} catch (error) {
|
330 |
+
console.error(`获取Artifacts时出错 (尝试 ${artifactAttempts}/${maxArtifactAttempts}): ${error.message}`);
|
331 |
+
// 出错时继续尝试,不中断循环
|
332 |
+
}
|
333 |
+
|
334 |
+
if (!artifacts || !artifacts.artifacts || artifacts.artifacts.length === 0) {
|
335 |
+
console.log('暂时没有找到Artifacts,等待10秒后重试...');
|
336 |
+
await new Promise(resolve => setTimeout(resolve, 10000));
|
337 |
+
}
|
338 |
+
}
|
339 |
+
|
340 |
+
if (!artifacts || !artifacts.artifacts || artifacts.artifacts.length === 0) {
|
341 |
+
console.log('没有找到Artifacts,可能工作流没有生成Artifact');
|
342 |
+
return null;
|
343 |
+
}
|
344 |
+
|
345 |
+
console.log(`找到 ${artifacts.artifacts.length} 个Artifacts`);
|
346 |
+
|
347 |
+
// 查找 Account info Artifact
|
348 |
+
const accountInfoArtifact = artifacts.artifacts.find(artifact =>
|
349 |
+
artifact.name.toLowerCase().includes('account info'));
|
350 |
+
|
351 |
+
if (!accountInfoArtifact) {
|
352 |
+
console.log('没有找到 Account info Artifact');
|
353 |
+
return null;
|
354 |
+
}
|
355 |
+
|
356 |
+
console.log(`找到 Account info Artifact: ${accountInfoArtifact.id}`);
|
357 |
+
return accountInfoArtifact;
|
358 |
+
} catch (error) {
|
359 |
+
console.error('获取 Artifact 失败:', error);
|
360 |
+
return null;
|
361 |
+
}
|
362 |
+
}
|
363 |
+
|
364 |
+
// 下载 Artifact
|
365 |
+
async function downloadArtifact(artifact) {
|
366 |
+
let downloadAttempts = 0;
|
367 |
+
const maxDownloadAttempts = 5; // 最多尝试5次下载
|
368 |
+
|
369 |
+
while (downloadAttempts < maxDownloadAttempts) {
|
370 |
+
downloadAttempts++;
|
371 |
+
try {
|
372 |
+
console.log(`开始下载 Artifact: ${artifact.id}... (尝试 ${downloadAttempts}/${maxDownloadAttempts})`);
|
373 |
+
ensureDirectoryExists(DOWNLOAD_DIR);
|
374 |
+
|
375 |
+
const octokit = new Octokit({
|
376 |
+
auth: GITHUB_TOKEN
|
377 |
+
});
|
378 |
+
|
379 |
+
// 获取下载 URL
|
380 |
+
const { url } = await octokit.actions.downloadArtifact({
|
381 |
+
owner: GITHUB_OWNER,
|
382 |
+
repo: GITHUB_REPO,
|
383 |
+
artifact_id: artifact.id,
|
384 |
+
archive_format: 'zip'
|
385 |
+
});
|
386 |
+
|
387 |
+
// 下载 zip 文件
|
388 |
+
const zipFilePath = path.join(DOWNLOAD_DIR, `${artifact.id}.zip`);
|
389 |
+
const response = await axios({
|
390 |
+
method: 'get',
|
391 |
+
url: url,
|
392 |
+
responseType: 'arraybuffer',
|
393 |
+
timeout: 60000 // 设置60秒超时
|
394 |
+
});
|
395 |
+
|
396 |
+
fs.writeFileSync(zipFilePath, response.data);
|
397 |
+
console.log(`Artifact 下载完成: ${zipFilePath}`);
|
398 |
+
return zipFilePath;
|
399 |
+
} catch (error) {
|
400 |
+
console.error(`下载 Artifact 失败 (尝试 ${downloadAttempts}/${maxDownloadAttempts}): ${error.message}`);
|
401 |
+
|
402 |
+
if (downloadAttempts >= maxDownloadAttempts) {
|
403 |
+
console.error('达到最大尝试次数,放弃下载');
|
404 |
+
return null;
|
405 |
+
}
|
406 |
+
|
407 |
+
// 等待一段时间后重试
|
408 |
+
const retryDelay = 10000; // 10秒
|
409 |
+
console.log(`等待 ${retryDelay/1000} 秒后重试...`);
|
410 |
+
await new Promise(resolve => setTimeout(resolve, retryDelay));
|
411 |
+
}
|
412 |
+
}
|
413 |
+
|
414 |
+
return null;
|
415 |
+
}
|
416 |
+
|
417 |
+
// 解压 Artifact
|
418 |
+
async function extractArtifact(zipFilePath) {
|
419 |
+
let extractAttempts = 0;
|
420 |
+
const maxExtractAttempts = 3; // 最多尝试3次解压
|
421 |
+
|
422 |
+
while (extractAttempts < maxExtractAttempts) {
|
423 |
+
extractAttempts++;
|
424 |
+
try {
|
425 |
+
console.log(`开始解压 Artifact: ${zipFilePath}... (尝试 ${extractAttempts}/${maxExtractAttempts})`);
|
426 |
+
ensureDirectoryExists(EXTRACT_DIR);
|
427 |
+
|
428 |
+
const zip = new AdmZip(zipFilePath);
|
429 |
+
zip.extractAllTo(EXTRACT_DIR, true);
|
430 |
+
console.log(`Artifact 解压完成: ${EXTRACT_DIR}`);
|
431 |
+
|
432 |
+
// 查找 token CSV 文件
|
433 |
+
const files = fs.readdirSync(EXTRACT_DIR);
|
434 |
+
const tokenFile = files.find(file => file.startsWith('token_') && file.endsWith('.csv'));
|
435 |
+
|
436 |
+
if (!tokenFile) {
|
437 |
+
console.log('没有找到 token CSV 文件');
|
438 |
+
|
439 |
+
if (extractAttempts >= maxExtractAttempts) {
|
440 |
+
return null;
|
441 |
+
}
|
442 |
+
|
443 |
+
// 等待一段时间后重试
|
444 |
+
const retryDelay = 5000; // 5秒
|
445 |
+
console.log(`等待 ${retryDelay/1000} 秒后重试...`);
|
446 |
+
await new Promise(resolve => setTimeout(resolve, retryDelay));
|
447 |
+
continue;
|
448 |
+
}
|
449 |
+
|
450 |
+
console.log(`找到 token CSV 文件: ${tokenFile}`);
|
451 |
+
return path.join(EXTRACT_DIR, tokenFile);
|
452 |
+
} catch (error) {
|
453 |
+
console.error(`解压 Artifact 失败 (尝试 ${extractAttempts}/${maxExtractAttempts}): ${error.message}`);
|
454 |
+
|
455 |
+
if (extractAttempts >= maxExtractAttempts) {
|
456 |
+
console.error('达到最大尝试次数,放弃解压');
|
457 |
+
return null;
|
458 |
+
}
|
459 |
+
|
460 |
+
// 等待一段时间后重试
|
461 |
+
const retryDelay = 5000; // 5秒
|
462 |
+
console.log(`等待 ${retryDelay/1000} 秒后重试...`);
|
463 |
+
await new Promise(resolve => setTimeout(resolve, retryDelay));
|
464 |
+
}
|
465 |
+
}
|
466 |
+
|
467 |
+
return null;
|
468 |
+
}
|
469 |
+
|
470 |
+
/**
|
471 |
+
* 从CSV文件中提取cookies
|
472 |
+
* @param {string} csvFilePath - CSV文件路径
|
473 |
+
* @returns {Promise<string[]>} - 提取到的cookie数组
|
474 |
+
*/
|
475 |
+
async function extractCookiesFromCsvFile(csvFilePath) {
|
476 |
+
const maxExtractAttempts = 3;
|
477 |
+
let attempt = 1;
|
478 |
+
|
479 |
+
while (attempt <= maxExtractAttempts) {
|
480 |
+
console.log(`尝试从CSV文件提取cookies (尝试 ${attempt}/${maxExtractAttempts})...`);
|
481 |
+
|
482 |
+
try {
|
483 |
+
// 使用新的提取模块
|
484 |
+
const cookies = await extractCookiesFromCsv(csvFilePath);
|
485 |
+
|
486 |
+
if (cookies && cookies.length > 0) {
|
487 |
+
console.log(`成功从CSV文件提取到 ${cookies.length} 个cookies`);
|
488 |
+
return cookies;
|
489 |
+
} else {
|
490 |
+
console.warn(`未能从CSV文件提取到cookies (尝试 ${attempt}/${maxExtractAttempts})`);
|
491 |
+
}
|
492 |
+
} catch (error) {
|
493 |
+
console.error(`从CSV文件提取cookies时出错 (尝试 ${attempt}/${maxExtractAttempts}):`, error);
|
494 |
+
}
|
495 |
+
|
496 |
+
attempt++;
|
497 |
+
if (attempt <= maxExtractAttempts) {
|
498 |
+
console.log(`等待5秒后重试...`);
|
499 |
+
await new Promise(resolve => setTimeout(resolve, 5000));
|
500 |
+
}
|
501 |
+
}
|
502 |
+
|
503 |
+
console.error(`在 ${maxExtractAttempts} 次尝试后未能从CSV文件提取到cookies`);
|
504 |
+
return [];
|
505 |
+
}
|
506 |
+
|
507 |
+
// 将新的有效cookie添加到系统中
|
508 |
+
function addNewCookiesToSystem(apiKey, newCookies) {
|
509 |
+
try {
|
510 |
+
console.log(`准备添加 ${newCookies.length} 个新cookie到系统中`);
|
511 |
+
|
512 |
+
// 获取当前的cookies
|
513 |
+
const currentCookies = keyManager.getAllCookiesForApiKey(apiKey) || [];
|
514 |
+
console.log(`当前API密钥 ${apiKey} 有 ${currentCookies.length} 个cookies`);
|
515 |
+
|
516 |
+
// 获取无效的cookies
|
517 |
+
const invalidCookies = keyManager.getInvalidCookies() || [];
|
518 |
+
console.log(`系统中有 ${invalidCookies.length || 0} 个无效cookies`);
|
519 |
+
|
520 |
+
// 过滤出新的有效cookie
|
521 |
+
let newValidCookies = [];
|
522 |
+
|
523 |
+
// 检查invalidCookies的类型并相应处理
|
524 |
+
if (invalidCookies instanceof Set) {
|
525 |
+
newValidCookies = newCookies.filter(cookie =>
|
526 |
+
!currentCookies.includes(cookie) && !invalidCookies.has(cookie)
|
527 |
+
);
|
528 |
+
} else if (Array.isArray(invalidCookies)) {
|
529 |
+
newValidCookies = newCookies.filter(cookie =>
|
530 |
+
!currentCookies.includes(cookie) && !invalidCookies.includes(cookie)
|
531 |
+
);
|
532 |
+
} else if (invalidCookies && typeof invalidCookies === 'object') {
|
533 |
+
// 如果是普通对象,检查cookie是否作为键存在
|
534 |
+
newValidCookies = newCookies.filter(cookie =>
|
535 |
+
!currentCookies.includes(cookie) && !(cookie in invalidCookies)
|
536 |
+
);
|
537 |
+
} else {
|
538 |
+
// 如果invalidCookies不是预期的类型,只过滤当前cookies
|
539 |
+
newValidCookies = newCookies.filter(cookie => !currentCookies.includes(cookie));
|
540 |
+
}
|
541 |
+
|
542 |
+
console.log(`过滤后有 ${newValidCookies.length} 个新的有效cookies`);
|
543 |
+
|
544 |
+
// 验证cookie是否完整
|
545 |
+
const validatedCookies = newValidCookies.filter(cookie => {
|
546 |
+
// 检查cookie是否包含JWT的三个部分
|
547 |
+
if (cookie.includes('%3A%3A')) {
|
548 |
+
const parts = cookie.split('%3A%3A');
|
549 |
+
if (parts.length === 2) {
|
550 |
+
const jwt = parts[1];
|
551 |
+
// 检查JWT是否包含点(表示JWT的三个部分)
|
552 |
+
if (!jwt.includes('.') || jwt.split('.').length !== 3) {
|
553 |
+
console.warn(`跳过不完整的cookie: ${cookie}`);
|
554 |
+
return false;
|
555 |
+
}
|
556 |
+
}
|
557 |
+
}
|
558 |
+
return true;
|
559 |
+
});
|
560 |
+
|
561 |
+
console.log(`验证完整性后有 ${validatedCookies.length} 个有效cookies`);
|
562 |
+
|
563 |
+
if (validatedCookies.length > 0) {
|
564 |
+
// 添加新的有效cookie到系统
|
565 |
+
keyManager.addOrUpdateApiKey(apiKey, [...currentCookies, ...validatedCookies]);
|
566 |
+
console.log(`成功添加 ${validatedCookies.length} 个新cookie到API密钥 ${apiKey}`);
|
567 |
+
return validatedCookies.length; // 返回添加的cookie数量
|
568 |
+
} else {
|
569 |
+
console.log(`没有新的有效cookie需要添加到API密钥 ${apiKey}`);
|
570 |
+
return 0; // 没有添加cookie,返回0
|
571 |
+
}
|
572 |
+
} catch (error) {
|
573 |
+
console.error('添加新cookie到系统时出错:', error);
|
574 |
+
return 0; // 出错时返回0
|
575 |
+
}
|
576 |
+
}
|
577 |
+
|
578 |
+
// 清理临时文件
|
579 |
+
function cleanupTempFiles() {
|
580 |
+
try {
|
581 |
+
console.log('开始清理临时文件...');
|
582 |
+
|
583 |
+
// 清理下载目录
|
584 |
+
if (fs.existsSync(DOWNLOAD_DIR)) {
|
585 |
+
fs.readdirSync(DOWNLOAD_DIR).forEach(file => {
|
586 |
+
fs.unlinkSync(path.join(DOWNLOAD_DIR, file));
|
587 |
+
});
|
588 |
+
}
|
589 |
+
|
590 |
+
// 清理解压目录
|
591 |
+
if (fs.existsSync(EXTRACT_DIR)) {
|
592 |
+
fs.readdirSync(EXTRACT_DIR).forEach(file => {
|
593 |
+
fs.unlinkSync(path.join(EXTRACT_DIR, file));
|
594 |
+
});
|
595 |
+
}
|
596 |
+
|
597 |
+
console.log('临时文件清理完成');
|
598 |
+
} catch (error) {
|
599 |
+
console.error('清理临时文件失败:', error);
|
600 |
+
}
|
601 |
+
}
|
602 |
+
|
603 |
+
// 检查 API Key 是否需要补充 Cookie
|
604 |
+
function checkApiKeyNeedRefresh(apiKey, minCookieCount = config.refresh.minCookieCount) {
|
605 |
+
const cookies = keyManager.getAllCookiesForApiKey(apiKey);
|
606 |
+
return cookies.length < minCookieCount;
|
607 |
+
}
|
608 |
+
|
609 |
+
// 将现有cookie全部设为无效并从API Key中移除
|
610 |
+
function markExistingCookiesAsInvalid(apiKey) {
|
611 |
+
try {
|
612 |
+
// 获取当前API Key的所有cookie
|
613 |
+
const currentCookies = keyManager.getAllCookiesForApiKey(apiKey) || [];
|
614 |
+
console.log(`正在将API Key ${apiKey} 的 ${currentCookies.length} 个现有cookie标记为无效...`);
|
615 |
+
|
616 |
+
// 如果没有cookie,直接返回
|
617 |
+
if (currentCookies.length === 0) {
|
618 |
+
console.log(`API Key ${apiKey} 没有现有cookie,无需标记为无效`);
|
619 |
+
return 0;
|
620 |
+
}
|
621 |
+
|
622 |
+
// 获取无效cookie列表
|
623 |
+
const invalidCookies = keyManager.getInvalidCookies();
|
624 |
+
let markedCount = 0;
|
625 |
+
|
626 |
+
// 遍历cookie并添加到无效列表
|
627 |
+
for (const cookie of currentCookies) {
|
628 |
+
// 将cookie添加到无效集合中
|
629 |
+
if (invalidCookies instanceof Set) {
|
630 |
+
invalidCookies.add(cookie);
|
631 |
+
}
|
632 |
+
markedCount++;
|
633 |
+
}
|
634 |
+
|
635 |
+
// 保存无效cookie到文件
|
636 |
+
keyManager.saveInvalidCookiesToFile();
|
637 |
+
|
638 |
+
// 清空当前API Key的cookie列表
|
639 |
+
keyManager.addOrUpdateApiKey(apiKey, []);
|
640 |
+
|
641 |
+
// 保存更新后的API Keys
|
642 |
+
keyManager.saveApiKeysToFile();
|
643 |
+
|
644 |
+
console.log(`已将API Key ${apiKey} 的 ${markedCount} 个cookie标记为无效并从API Key中移除`);
|
645 |
+
return markedCount;
|
646 |
+
} catch (error) {
|
647 |
+
console.error(`标记现有cookie为无效时出错:`, error);
|
648 |
+
return 0;
|
649 |
+
}
|
650 |
+
}
|
651 |
+
|
652 |
+
// 主函数:自动刷新 Cookie
|
653 |
+
async function autoRefreshCookies(apiKey, minCookieCount = config.refresh.minCookieCount) {
|
654 |
+
console.log(`开始自动刷新 Cookie,目标 API Key: ${apiKey},最小 Cookie 数量: ${minCookieCount}`);
|
655 |
+
|
656 |
+
try {
|
657 |
+
// 检查是否需要刷新
|
658 |
+
if (!checkApiKeyNeedRefresh(apiKey, minCookieCount)) {
|
659 |
+
console.log(`API Key ${apiKey} 的 Cookie 数量足够,不需要刷新`);
|
660 |
+
return {
|
661 |
+
success: true,
|
662 |
+
message: '当前 Cookie 数量足够,不需要刷新',
|
663 |
+
refreshed: 0
|
664 |
+
};
|
665 |
+
}
|
666 |
+
|
667 |
+
// 获取最新的 Artifact
|
668 |
+
const artifact = await getLatestArtifact();
|
669 |
+
if (!artifact) {
|
670 |
+
return {
|
671 |
+
success: false,
|
672 |
+
message: '获取 Artifact 失败',
|
673 |
+
refreshed: 0
|
674 |
+
};
|
675 |
+
}
|
676 |
+
|
677 |
+
// 下载 Artifact
|
678 |
+
const zipFilePath = await downloadArtifact(artifact);
|
679 |
+
if (!zipFilePath) {
|
680 |
+
return {
|
681 |
+
success: false,
|
682 |
+
message: '下载 Artifact 失败',
|
683 |
+
refreshed: 0
|
684 |
+
};
|
685 |
+
}
|
686 |
+
|
687 |
+
// 解压 Artifact
|
688 |
+
const csvFilePath = await extractArtifact(zipFilePath);
|
689 |
+
if (!csvFilePath) {
|
690 |
+
return {
|
691 |
+
success: false,
|
692 |
+
message: '解压 Artifact 失败',
|
693 |
+
refreshed: 0
|
694 |
+
};
|
695 |
+
}
|
696 |
+
|
697 |
+
// 提取 Cookie
|
698 |
+
const cookies = await extractCookiesFromCsvFile(csvFilePath);
|
699 |
+
if (cookies.length === 0) {
|
700 |
+
return {
|
701 |
+
success: false,
|
702 |
+
message: '没有找到有效的 Cookie',
|
703 |
+
refreshed: 0
|
704 |
+
};
|
705 |
+
}
|
706 |
+
|
707 |
+
// 根据配置决定是否将现有cookie标记为无效
|
708 |
+
const refreshMode = process.env.COOKIE_REFRESH_MODE || 'append';
|
709 |
+
|
710 |
+
if (refreshMode === 'replace') {
|
711 |
+
// 将现有cookie标记为无效并从API Key中移除
|
712 |
+
console.log('使用替换模式: 将现有cookie标记为无效');
|
713 |
+
markExistingCookiesAsInvalid(apiKey);
|
714 |
+
} else {
|
715 |
+
console.log('使用追加模式: 保留现有cookie,只添加新cookie');
|
716 |
+
}
|
717 |
+
|
718 |
+
// 添加新的 Cookie 到系统
|
719 |
+
const addedCount = addNewCookiesToSystem(apiKey, cookies);
|
720 |
+
|
721 |
+
// 清理临时文件
|
722 |
+
cleanupTempFiles();
|
723 |
+
|
724 |
+
return {
|
725 |
+
success: true,
|
726 |
+
message: `成功添加 ${addedCount} 个新 Cookie`,
|
727 |
+
refreshed: addedCount
|
728 |
+
};
|
729 |
+
} catch (error) {
|
730 |
+
console.error('自动刷新 Cookie 失败:', error);
|
731 |
+
return {
|
732 |
+
success: false,
|
733 |
+
message: `刷新失败: ${error.message}`,
|
734 |
+
refreshed: 0
|
735 |
+
};
|
736 |
+
}
|
737 |
+
}
|
738 |
+
|
739 |
+
module.exports = {
|
740 |
+
autoRefreshCookies,
|
741 |
+
checkApiKeyNeedRefresh,
|
742 |
+
getLatestArtifact,
|
743 |
+
downloadArtifact,
|
744 |
+
extractArtifact,
|
745 |
+
extractCookiesFromCsvFile,
|
746 |
+
addNewCookiesToSystem,
|
747 |
+
cleanupTempFiles,
|
748 |
+
triggerWorkflow,
|
749 |
+
markExistingCookiesAsInvalid
|
750 |
+
};
|
src/utils/envChecker.js
ADDED
@@ -0,0 +1,76 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
const fs = require('fs');
|
2 |
+
const path = require('path');
|
3 |
+
|
4 |
+
/**
|
5 |
+
* 检查 .env 文件是否存在
|
6 |
+
* @returns {boolean} 文件是否存在
|
7 |
+
*/
|
8 |
+
function checkEnvFileExists() {
|
9 |
+
const envPath = path.resolve(process.cwd(), '.env');
|
10 |
+
return fs.existsSync(envPath);
|
11 |
+
}
|
12 |
+
|
13 |
+
/**
|
14 |
+
* 检查必要的环境变量是否已设置
|
15 |
+
* @returns {Object} 检查结果,包含是否通过和缺失的变量列表
|
16 |
+
*/
|
17 |
+
function checkRequiredEnvVars() {
|
18 |
+
// 定义必要的环境变量列表
|
19 |
+
const requiredVars = [
|
20 |
+
'API_KEYS', // API Keys 配置
|
21 |
+
];
|
22 |
+
|
23 |
+
// 如果启用了自动刷新,则需要检查相关配置
|
24 |
+
if (process.env.ENABLE_AUTO_REFRESH === 'true') {
|
25 |
+
requiredVars.push(
|
26 |
+
'GITHUB_TOKEN',
|
27 |
+
'GITHUB_OWNER',
|
28 |
+
'GITHUB_REPO',
|
29 |
+
'GITHUB_WORKFLOW_ID',
|
30 |
+
'TRIGGER_WORKFLOW'
|
31 |
+
);
|
32 |
+
}
|
33 |
+
|
34 |
+
// 检查每个必要的环境变量
|
35 |
+
const missingVars = requiredVars.filter(varName => !process.env[varName]);
|
36 |
+
|
37 |
+
return {
|
38 |
+
passed: missingVars.length === 0,
|
39 |
+
missingVars
|
40 |
+
};
|
41 |
+
}
|
42 |
+
|
43 |
+
/**
|
44 |
+
* 执行环境检查,如果不符合要求则退出程序
|
45 |
+
*/
|
46 |
+
function enforceEnvCheck() {
|
47 |
+
console.log('正在检查环境配置...');
|
48 |
+
|
49 |
+
// 检查 .env 文件是否存在
|
50 |
+
const envFileExists = checkEnvFileExists();
|
51 |
+
if (!envFileExists) {
|
52 |
+
console.error('\n错误: 未找到 .env 文件!');
|
53 |
+
console.error('请根据 .env.example 创建 .env 文件并配置必要的环境变量。');
|
54 |
+
console.error('执行以下命令复制示例文件: cp .env.example .env,或执行npm run setup\n');
|
55 |
+
process.exit(1); // 退出程序,状态码 1 表示错误
|
56 |
+
}
|
57 |
+
|
58 |
+
// 检查必要的环境变量
|
59 |
+
const { passed, missingVars } = checkRequiredEnvVars();
|
60 |
+
if (!passed) {
|
61 |
+
console.error('\n错误: 以下必要的环境变量未在 .env 文件中设置:');
|
62 |
+
missingVars.forEach(varName => {
|
63 |
+
console.error(` - ${varName}`);
|
64 |
+
});
|
65 |
+
console.error('\n请在 .env 文件中配置这些变量后重新启动程序。\n');
|
66 |
+
process.exit(1); // 退出程序,状态码 1 表示错误
|
67 |
+
}
|
68 |
+
|
69 |
+
console.log('环境检查通过,继续启动程序...');
|
70 |
+
}
|
71 |
+
|
72 |
+
module.exports = {
|
73 |
+
checkEnvFileExists,
|
74 |
+
checkRequiredEnvVars,
|
75 |
+
enforceEnvCheck
|
76 |
+
};
|
src/utils/extractCookieFromCsv.js
ADDED
@@ -0,0 +1,171 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
const fs = require('fs');
|
2 |
+
const path = require('path');
|
3 |
+
const csv = require('csv-parser');
|
4 |
+
|
5 |
+
/**
|
6 |
+
* 从CSV文件中提取完整的cookie
|
7 |
+
* @param {string} csvFilePath - CSV文件路径
|
8 |
+
* @returns {Promise<string[]>} - 提取到的cookie数组
|
9 |
+
*/
|
10 |
+
async function extractCookiesFromCsv(csvFilePath) {
|
11 |
+
return new Promise((resolve, reject) => {
|
12 |
+
try {
|
13 |
+
// 检查文件是否存在
|
14 |
+
if (!fs.existsSync(csvFilePath)) {
|
15 |
+
console.error(`CSV文件不存在: ${csvFilePath}`);
|
16 |
+
return resolve([]);
|
17 |
+
}
|
18 |
+
|
19 |
+
// 读取文件内容
|
20 |
+
const fileContent = fs.readFileSync(csvFilePath, 'utf8');
|
21 |
+
console.log(`文件内容前200个字符: ${fileContent.substring(0, 200)}`);
|
22 |
+
|
23 |
+
// 检查文件是否为空
|
24 |
+
if (!fileContent || fileContent.trim() === '') {
|
25 |
+
console.error('CSV文件为空');
|
26 |
+
return resolve([]);
|
27 |
+
}
|
28 |
+
|
29 |
+
// 检查文件内容是否包含关键字
|
30 |
+
const hasTokenKeyword = fileContent.includes('token');
|
31 |
+
const hasUserPrefix = fileContent.includes('user_');
|
32 |
+
console.log(`文件包含"token"关键字: ${hasTokenKeyword}`);
|
33 |
+
console.log(`文件包含"user_"前缀: ${hasUserPrefix}`);
|
34 |
+
|
35 |
+
// 如果文件包含user_前缀,尝试直接从内容中提取cookie
|
36 |
+
if (hasUserPrefix) {
|
37 |
+
const cookies = extractCookiesFromText(fileContent);
|
38 |
+
if (cookies.length > 0) {
|
39 |
+
console.log(`直接从文件内容中提取到 ${cookies.length} 个Cookie`);
|
40 |
+
return resolve(cookies);
|
41 |
+
}
|
42 |
+
}
|
43 |
+
|
44 |
+
// 使用csv-parser解析CSV文件
|
45 |
+
const cookies = [];
|
46 |
+
const possibleTokenFields = ['token', 'cookie', 'value', 'Token', 'Cookie', 'Value'];
|
47 |
+
|
48 |
+
fs.createReadStream(csvFilePath)
|
49 |
+
.pipe(csv())
|
50 |
+
.on('data', (row) => {
|
51 |
+
// 检查所有可能的字段名
|
52 |
+
for (const field of possibleTokenFields) {
|
53 |
+
if (row[field] && row[field].includes('user_')) {
|
54 |
+
cookies.push(row[field]);
|
55 |
+
break;
|
56 |
+
}
|
57 |
+
}
|
58 |
+
|
59 |
+
// 如果没有找到预定义的字段,遍历所有字段
|
60 |
+
if (cookies.length === 0) {
|
61 |
+
for (const field in row) {
|
62 |
+
if (row[field] && typeof row[field] === 'string' && row[field].includes('user_')) {
|
63 |
+
cookies.push(row[field]);
|
64 |
+
break;
|
65 |
+
}
|
66 |
+
}
|
67 |
+
}
|
68 |
+
})
|
69 |
+
.on('end', () => {
|
70 |
+
console.log(`从CSV解析中提取到 ${cookies.length} 个Cookie`);
|
71 |
+
|
72 |
+
// 如果通过CSV解析没有找到cookie,尝试按行读取
|
73 |
+
if (cookies.length === 0) {
|
74 |
+
console.log('尝试按行读取文件...');
|
75 |
+
const lines = fileContent.split('\n');
|
76 |
+
for (const line of lines) {
|
77 |
+
if (line.includes('user_')) {
|
78 |
+
const extractedCookies = extractCookiesFromText(line);
|
79 |
+
extractedCookies.forEach(cookie => {
|
80 |
+
if (!cookies.includes(cookie)) {
|
81 |
+
cookies.push(cookie);
|
82 |
+
}
|
83 |
+
});
|
84 |
+
}
|
85 |
+
}
|
86 |
+
console.log(`按行读取后提取到 ${cookies.length} 个Cookie`);
|
87 |
+
}
|
88 |
+
|
89 |
+
// 如果仍然没有找到cookie,尝试从整个文件内容中提取
|
90 |
+
if (cookies.length === 0) {
|
91 |
+
console.log('尝试从整个文件内容中提取Cookie...');
|
92 |
+
const extractedCookies = extractCookiesFromText(fileContent);
|
93 |
+
extractedCookies.forEach(cookie => {
|
94 |
+
if (!cookies.includes(cookie)) {
|
95 |
+
cookies.push(cookie);
|
96 |
+
}
|
97 |
+
});
|
98 |
+
console.log(`从整个文件内容中提取到 ${cookies.length} 个Cookie`);
|
99 |
+
}
|
100 |
+
|
101 |
+
// 验证提取的cookie是否完整
|
102 |
+
const validatedCookies = validateCookies(cookies);
|
103 |
+
|
104 |
+
resolve(validatedCookies);
|
105 |
+
})
|
106 |
+
.on('error', (error) => {
|
107 |
+
console.error('解析CSV文件时出错:', error);
|
108 |
+
// 如果解析出错,尝试从整个文件内容中提取
|
109 |
+
const extractedCookies = extractCookiesFromText(fileContent);
|
110 |
+
console.log(`解析出错后从文件内容中提取到 ${extractedCookies.length} 个Cookie`);
|
111 |
+
resolve(validateCookies(extractedCookies));
|
112 |
+
});
|
113 |
+
} catch (error) {
|
114 |
+
console.error('提取Cookie时出错:', error);
|
115 |
+
reject(error);
|
116 |
+
}
|
117 |
+
});
|
118 |
+
}
|
119 |
+
|
120 |
+
/**
|
121 |
+
* 从文本中提取cookie
|
122 |
+
* @param {string} text - 要提取cookie的文本
|
123 |
+
* @returns {string[]} - 提取到的cookie数组
|
124 |
+
*/
|
125 |
+
function extractCookiesFromText(text) {
|
126 |
+
const cookies = [];
|
127 |
+
|
128 |
+
// 使用正则表达式匹配user_开头的cookie
|
129 |
+
const regex = /user_[a-zA-Z0-9%]+%3A%3A[a-zA-Z0-9%\.\_\-]+/g;
|
130 |
+
const matches = text.match(regex);
|
131 |
+
|
132 |
+
if (matches) {
|
133 |
+
matches.forEach(match => {
|
134 |
+
if (!cookies.includes(match)) {
|
135 |
+
cookies.push(match);
|
136 |
+
}
|
137 |
+
});
|
138 |
+
}
|
139 |
+
|
140 |
+
return cookies;
|
141 |
+
}
|
142 |
+
|
143 |
+
/**
|
144 |
+
* 验证cookie是否完整
|
145 |
+
* @param {string[]} cookies - 要验证的cookie数组
|
146 |
+
* @returns {string[]} - 验证后的cookie数组
|
147 |
+
*/
|
148 |
+
function validateCookies(cookies) {
|
149 |
+
return cookies.map(cookie => {
|
150 |
+
// 检查cookie是否包含JWT的三个部分(header.payload.signature)
|
151 |
+
if (cookie.includes('%3A%3A')) {
|
152 |
+
const parts = cookie.split('%3A%3A');
|
153 |
+
if (parts.length === 2) {
|
154 |
+
const jwt = parts[1];
|
155 |
+
// 检查JWT是否包含两个点(表示三个部分)
|
156 |
+
if (jwt.includes('.') && jwt.split('.').length === 3) {
|
157 |
+
return cookie; // cookie完整
|
158 |
+
} else {
|
159 |
+
console.warn(`检测到不完整的JWT: ${cookie}`);
|
160 |
+
// 尝试修复不完整的JWT(如果可能)
|
161 |
+
return cookie;
|
162 |
+
}
|
163 |
+
}
|
164 |
+
}
|
165 |
+
return cookie;
|
166 |
+
});
|
167 |
+
}
|
168 |
+
|
169 |
+
module.exports = {
|
170 |
+
extractCookiesFromCsv
|
171 |
+
};
|
src/utils/keyManager.js
ADDED
@@ -0,0 +1,406 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
const config = require('../config/config');
|
2 |
+
const fs = require('fs');
|
3 |
+
const path = require('path');
|
4 |
+
|
5 |
+
// 定义无效cookie的存储文件路径
|
6 |
+
const INVALID_COOKIES_FILE = path.join(__dirname, '../../data/invalid_cookies.json');
|
7 |
+
// 定义API Keys的存储文件路径
|
8 |
+
const API_KEYS_FILE = path.join(__dirname, '../../data/api_keys.json');
|
9 |
+
|
10 |
+
// 确保data目录存在
|
11 |
+
function ensureDataDirExists() {
|
12 |
+
const dataDir = path.join(__dirname, '../../data');
|
13 |
+
if (!fs.existsSync(dataDir)) {
|
14 |
+
try {
|
15 |
+
fs.mkdirSync(dataDir, { recursive: true });
|
16 |
+
console.log('创建data目录成功');
|
17 |
+
} catch (err) {
|
18 |
+
console.error('创建data目录失败:', err);
|
19 |
+
}
|
20 |
+
}
|
21 |
+
}
|
22 |
+
|
23 |
+
// 存储API key与Cursor cookie的映射关系
|
24 |
+
let apiKeyMap = new Map();
|
25 |
+
|
26 |
+
// 存储每个API key对应的cookie轮询索引
|
27 |
+
let rotationIndexes = new Map();
|
28 |
+
|
29 |
+
// 存储被标记为无效的cookie
|
30 |
+
let invalidCookies = new Set();
|
31 |
+
|
32 |
+
// 从文件加载无效cookie
|
33 |
+
function loadInvalidCookiesFromFile() {
|
34 |
+
ensureDataDirExists();
|
35 |
+
|
36 |
+
try {
|
37 |
+
if (fs.existsSync(INVALID_COOKIES_FILE)) {
|
38 |
+
const data = fs.readFileSync(INVALID_COOKIES_FILE, 'utf8');
|
39 |
+
const cookiesArray = JSON.parse(data);
|
40 |
+
|
41 |
+
// 清空当前集合并添加从文件加载的cookie
|
42 |
+
invalidCookies.clear();
|
43 |
+
cookiesArray.forEach(cookie => invalidCookies.add(cookie));
|
44 |
+
|
45 |
+
console.log(`从文件加载了 ${cookiesArray.length} 个无效cookie`);
|
46 |
+
} else {
|
47 |
+
saveInvalidCookiesToFile(); // 如果文件不存在,创建新文件
|
48 |
+
}
|
49 |
+
} catch (err) {
|
50 |
+
console.error('加载无效cookie文件失败:', err);
|
51 |
+
saveInvalidCookiesToFile(); // 如果加载失败,尝试创建新文件
|
52 |
+
}
|
53 |
+
}
|
54 |
+
|
55 |
+
// 将无效cookie保存到文件
|
56 |
+
function saveInvalidCookiesToFile() {
|
57 |
+
ensureDataDirExists();
|
58 |
+
|
59 |
+
try {
|
60 |
+
const cookiesArray = Array.from(invalidCookies);
|
61 |
+
fs.writeFileSync(INVALID_COOKIES_FILE, JSON.stringify(cookiesArray, null, 2), 'utf8');
|
62 |
+
console.log(`已将 ${cookiesArray.length} 个无效cookie保存到文件`);
|
63 |
+
} catch (err) {
|
64 |
+
console.error('保存无效cookie文件失败:', err);
|
65 |
+
}
|
66 |
+
}
|
67 |
+
|
68 |
+
// 从文件加载API Keys
|
69 |
+
function loadApiKeysFromFile() {
|
70 |
+
ensureDataDirExists();
|
71 |
+
|
72 |
+
try {
|
73 |
+
if (fs.existsSync(API_KEYS_FILE)) {
|
74 |
+
const data = fs.readFileSync(API_KEYS_FILE, 'utf8');
|
75 |
+
const apiKeysObj = JSON.parse(data);
|
76 |
+
|
77 |
+
// 清空现有映射
|
78 |
+
apiKeyMap.clear();
|
79 |
+
rotationIndexes.clear();
|
80 |
+
|
81 |
+
// 统计总cookie数量
|
82 |
+
let totalCookies = 0;
|
83 |
+
|
84 |
+
// 添加从文件加载的API Keys
|
85 |
+
for (const [apiKey, cookies] of Object.entries(apiKeysObj)) {
|
86 |
+
if (Array.isArray(cookies)) {
|
87 |
+
apiKeyMap.set(apiKey, cookies);
|
88 |
+
rotationIndexes.set(apiKey, 0);
|
89 |
+
totalCookies += cookies.length;
|
90 |
+
} else {
|
91 |
+
console.error(`API Key ${apiKey} 的cookies不是数组,跳过`);
|
92 |
+
}
|
93 |
+
}
|
94 |
+
|
95 |
+
const apiKeyCount = Object.keys(apiKeysObj).length;
|
96 |
+
console.log(`从文件加载了 ${apiKeyCount} 个API Key,共 ${totalCookies} 个Cookie`);
|
97 |
+
return apiKeyCount > 0;
|
98 |
+
} else {
|
99 |
+
console.log('API Keys文件不存在,将使用配置中的API Keys');
|
100 |
+
return false;
|
101 |
+
}
|
102 |
+
} catch (err) {
|
103 |
+
console.error('加载API Keys文件失败:', err);
|
104 |
+
return false;
|
105 |
+
}
|
106 |
+
}
|
107 |
+
|
108 |
+
// 将API Keys保存到文件
|
109 |
+
function saveApiKeysToFile() {
|
110 |
+
ensureDataDirExists();
|
111 |
+
|
112 |
+
try {
|
113 |
+
// 将Map转换为普通对象
|
114 |
+
const apiKeysObj = {};
|
115 |
+
for (const [apiKey, cookies] of apiKeyMap.entries()) {
|
116 |
+
apiKeysObj[apiKey] = cookies;
|
117 |
+
}
|
118 |
+
|
119 |
+
// 使用JSON.stringify时避免特殊字符处理问题
|
120 |
+
const jsonString = JSON.stringify(apiKeysObj, null, 2);
|
121 |
+
fs.writeFileSync(API_KEYS_FILE, jsonString, 'utf8');
|
122 |
+
console.log(`已将 ${Object.keys(apiKeysObj).length} 个API Key保存到文件`);
|
123 |
+
|
124 |
+
// 简化验证过程
|
125 |
+
try {
|
126 |
+
const savedContent = fs.readFileSync(API_KEYS_FILE, 'utf8');
|
127 |
+
JSON.parse(savedContent); // 只验证JSON格式是否正确
|
128 |
+
console.log('验证通过: 所有cookie都被完整保存');
|
129 |
+
} catch (verifyErr) {
|
130 |
+
console.error('验证保存内容时出错:', verifyErr);
|
131 |
+
}
|
132 |
+
} catch (err) {
|
133 |
+
console.error('保存API Keys文件失败:', err);
|
134 |
+
}
|
135 |
+
}
|
136 |
+
|
137 |
+
// API Keys初始化函数
|
138 |
+
function initializeApiKeys() {
|
139 |
+
// 首先从文件加载现有的API Keys
|
140 |
+
const loadedFromFile = loadApiKeysFromFile();
|
141 |
+
|
142 |
+
// 检查环境变量中是否有API Keys配置
|
143 |
+
const configApiKeys = config.apiKeys;
|
144 |
+
const hasEnvApiKeys = Object.keys(configApiKeys).length > 0;
|
145 |
+
|
146 |
+
if (hasEnvApiKeys) {
|
147 |
+
console.log('从环境变量检测到API Keys配置,将合并到现有配置...');
|
148 |
+
|
149 |
+
// 记录合并前的Cookie数量
|
150 |
+
let beforeMergeCookies = 0;
|
151 |
+
for (const cookies of apiKeyMap.values()) {
|
152 |
+
beforeMergeCookies += cookies.length;
|
153 |
+
}
|
154 |
+
|
155 |
+
// 合并环境变量中的API Keys到现有映射
|
156 |
+
for (const [apiKey, cookieValue] of Object.entries(configApiKeys)) {
|
157 |
+
// 获取现有的cookies(如果有)
|
158 |
+
const existingCookies = apiKeyMap.get(apiKey) || [];
|
159 |
+
|
160 |
+
// 准备要添加的新cookies
|
161 |
+
let newCookies = [];
|
162 |
+
if (typeof cookieValue === 'string') {
|
163 |
+
newCookies = [cookieValue];
|
164 |
+
} else if (Array.isArray(cookieValue)) {
|
165 |
+
newCookies = cookieValue;
|
166 |
+
}
|
167 |
+
|
168 |
+
// 合并cookies,确保不重复
|
169 |
+
const mergedCookies = [...existingCookies];
|
170 |
+
for (const cookie of newCookies) {
|
171 |
+
if (!mergedCookies.includes(cookie)) {
|
172 |
+
mergedCookies.push(cookie);
|
173 |
+
}
|
174 |
+
}
|
175 |
+
|
176 |
+
// 更新映射
|
177 |
+
apiKeyMap.set(apiKey, mergedCookies);
|
178 |
+
|
179 |
+
// 确保轮询索引存在
|
180 |
+
if (!rotationIndexes.has(apiKey)) {
|
181 |
+
rotationIndexes.set(apiKey, 0);
|
182 |
+
}
|
183 |
+
}
|
184 |
+
|
185 |
+
// 记录合并后的Cookie数量
|
186 |
+
let afterMergeCookies = 0;
|
187 |
+
for (const cookies of apiKeyMap.values()) {
|
188 |
+
afterMergeCookies += cookies.length;
|
189 |
+
}
|
190 |
+
|
191 |
+
console.log(`合并前共有 ${beforeMergeCookies} 个Cookie,合并后共有 ${afterMergeCookies} 个Cookie`);
|
192 |
+
|
193 |
+
// 保存合并后的结果到文件
|
194 |
+
saveApiKeysToFile();
|
195 |
+
} else if (!loadedFromFile) {
|
196 |
+
console.log('警告: 未能从文件加载API Keys,且环境变量中也没有配置API Keys');
|
197 |
+
}
|
198 |
+
|
199 |
+
// 统计API Keys和Cookies数量
|
200 |
+
let totalCookies = 0;
|
201 |
+
for (const cookies of apiKeyMap.values()) {
|
202 |
+
totalCookies += cookies.length;
|
203 |
+
}
|
204 |
+
|
205 |
+
console.log(`API Keys初始化完成,共有 ${apiKeyMap.size} 个API Key,${totalCookies} 个Cookie`);
|
206 |
+
|
207 |
+
// 加载无效cookie
|
208 |
+
loadInvalidCookiesFromFile();
|
209 |
+
|
210 |
+
// 从API Key中移除已知的无效cookie
|
211 |
+
console.log('开始从API Keys中移除无效cookie...');
|
212 |
+
removeInvalidCookiesFromApiKeys();
|
213 |
+
}
|
214 |
+
|
215 |
+
// 从所有API Key中移除已知的无效cookie
|
216 |
+
function removeInvalidCookiesFromApiKeys() {
|
217 |
+
let totalRemoved = 0;
|
218 |
+
|
219 |
+
for (const [apiKey, cookies] of apiKeyMap.entries()) {
|
220 |
+
const initialLength = cookies.length;
|
221 |
+
|
222 |
+
// 过滤掉无效的cookie
|
223 |
+
const filteredCookies = cookies.filter(cookie => !invalidCookies.has(cookie));
|
224 |
+
|
225 |
+
// 如果有cookie被移除,更新API Key的cookie列表
|
226 |
+
if (filteredCookies.length < initialLength) {
|
227 |
+
const removedCount = initialLength - filteredCookies.length;
|
228 |
+
totalRemoved += removedCount;
|
229 |
+
|
230 |
+
apiKeyMap.set(apiKey, filteredCookies);
|
231 |
+
rotationIndexes.set(apiKey, 0);
|
232 |
+
|
233 |
+
console.log(`从API Key ${apiKey} 中移除了 ${removedCount} 个无效cookie,剩余 ${filteredCookies.length} 个`);
|
234 |
+
}
|
235 |
+
}
|
236 |
+
|
237 |
+
console.log(`总共从API Keys中移除了 ${totalRemoved} 个无效cookie`);
|
238 |
+
|
239 |
+
// 如果有cookie被移除,保存更新后的API Keys
|
240 |
+
if (totalRemoved > 0) {
|
241 |
+
saveApiKeysToFile();
|
242 |
+
}
|
243 |
+
}
|
244 |
+
|
245 |
+
// 添加或更新API key映射
|
246 |
+
function addOrUpdateApiKey(apiKey, cookieValues) {
|
247 |
+
if (!Array.isArray(cookieValues)) {
|
248 |
+
cookieValues = [cookieValues];
|
249 |
+
}
|
250 |
+
|
251 |
+
// 过滤掉已知的无效cookie
|
252 |
+
const validCookies = cookieValues.filter(cookie => !invalidCookies.has(cookie));
|
253 |
+
|
254 |
+
if (validCookies.length < cookieValues.length) {
|
255 |
+
console.log(`API Key ${apiKey} 中有 ${cookieValues.length - validCookies.length} 个无效cookie被过滤`);
|
256 |
+
}
|
257 |
+
|
258 |
+
apiKeyMap.set(apiKey, validCookies);
|
259 |
+
rotationIndexes.set(apiKey, 0);
|
260 |
+
|
261 |
+
// 保存更新后的API Keys
|
262 |
+
saveApiKeysToFile();
|
263 |
+
}
|
264 |
+
|
265 |
+
// 删除API key映射
|
266 |
+
function removeApiKey(apiKey) {
|
267 |
+
apiKeyMap.delete(apiKey);
|
268 |
+
rotationIndexes.delete(apiKey);
|
269 |
+
|
270 |
+
// 保存更新后的API Keys
|
271 |
+
saveApiKeysToFile();
|
272 |
+
}
|
273 |
+
|
274 |
+
// 获取API key对应的cookie值(根据轮询策略)
|
275 |
+
function getCookieForApiKey(apiKey, strategy = config.defaultRotationStrategy) {
|
276 |
+
// 如果API key不存在,也许是cookie本身,直接返回API key本身(向后兼容)
|
277 |
+
if (!apiKeyMap.has(apiKey)) {
|
278 |
+
return apiKey;
|
279 |
+
}
|
280 |
+
const cookies = apiKeyMap.get(apiKey);
|
281 |
+
|
282 |
+
if (!cookies || cookies.length === 0) {
|
283 |
+
return apiKey;
|
284 |
+
}
|
285 |
+
|
286 |
+
if (cookies.length === 1) {
|
287 |
+
return cookies[0];
|
288 |
+
}
|
289 |
+
|
290 |
+
// 根据策略选择cookie
|
291 |
+
if (strategy === 'random') {
|
292 |
+
// 随机策略
|
293 |
+
const randomIndex = Math.floor(Math.random() * cookies.length);
|
294 |
+
return cookies[randomIndex];
|
295 |
+
} else {
|
296 |
+
// 轮询策略(round-robin)
|
297 |
+
let currentIndex = rotationIndexes.get(apiKey) || 0;
|
298 |
+
const cookie = cookies[currentIndex];
|
299 |
+
|
300 |
+
// 更新索引
|
301 |
+
currentIndex = (currentIndex + 1) % cookies.length;
|
302 |
+
rotationIndexes.set(apiKey, currentIndex);
|
303 |
+
|
304 |
+
return cookie;
|
305 |
+
}
|
306 |
+
}
|
307 |
+
|
308 |
+
// 获取所有API key
|
309 |
+
function getAllApiKeys() {
|
310 |
+
return Array.from(apiKeyMap.keys());
|
311 |
+
}
|
312 |
+
|
313 |
+
// 获取API key对应的所有cookie
|
314 |
+
function getAllCookiesForApiKey(apiKey) {
|
315 |
+
return apiKeyMap.get(apiKey) || [];
|
316 |
+
}
|
317 |
+
|
318 |
+
// 从API key的cookie列表中移除特定cookie
|
319 |
+
function removeCookieFromApiKey(apiKey, cookieToRemove) {
|
320 |
+
if (!apiKeyMap.has(apiKey)) {
|
321 |
+
console.log(`API Key ${apiKey} 不存在,无法移除cookie`);
|
322 |
+
return false;
|
323 |
+
}
|
324 |
+
|
325 |
+
const cookies = apiKeyMap.get(apiKey);
|
326 |
+
const initialLength = cookies.length;
|
327 |
+
|
328 |
+
// 检查是否尝试移除与API Key相同的值(可能是向后兼容模式)
|
329 |
+
if (cookieToRemove === apiKey && initialLength === 0) {
|
330 |
+
console.log(`API Key ${apiKey} 中没有任何cookie,系统正在尝试以向后兼容模式使用API Key本身`);
|
331 |
+
return false;
|
332 |
+
}
|
333 |
+
|
334 |
+
// 过滤掉要移除的cookie
|
335 |
+
const filteredCookies = cookies.filter(cookie => cookie !== cookieToRemove);
|
336 |
+
|
337 |
+
// 如果长度没变,说明没有找到要移除的cookie
|
338 |
+
if (filteredCookies.length === initialLength) {
|
339 |
+
console.log(`未找到要移除的cookie: ${cookieToRemove}`);
|
340 |
+
return false;
|
341 |
+
}
|
342 |
+
|
343 |
+
// 更新cookie列表
|
344 |
+
apiKeyMap.set(apiKey, filteredCookies);
|
345 |
+
|
346 |
+
// 重置轮询索引
|
347 |
+
rotationIndexes.set(apiKey, 0);
|
348 |
+
|
349 |
+
// 将移除的cookie添加到无效cookie集合中
|
350 |
+
invalidCookies.add(cookieToRemove);
|
351 |
+
|
352 |
+
// 保存无效cookie到文件
|
353 |
+
saveInvalidCookiesToFile();
|
354 |
+
|
355 |
+
// 保存更新后的API Keys
|
356 |
+
saveApiKeysToFile();
|
357 |
+
|
358 |
+
console.log(`已从API Key ${apiKey} 中移除cookie: ${cookieToRemove}`);
|
359 |
+
console.log(`剩余cookie数量: ${filteredCookies.length}`);
|
360 |
+
|
361 |
+
return true;
|
362 |
+
}
|
363 |
+
|
364 |
+
// 获取所有被标记为无效的cookie
|
365 |
+
function getInvalidCookies() {
|
366 |
+
return invalidCookies;
|
367 |
+
}
|
368 |
+
|
369 |
+
// 清除特定的无效cookie记录
|
370 |
+
function clearInvalidCookie(cookie) {
|
371 |
+
const result = invalidCookies.delete(cookie);
|
372 |
+
|
373 |
+
if (result) {
|
374 |
+
// 保存更新后的无效cookie到文件
|
375 |
+
saveInvalidCookiesToFile();
|
376 |
+
}
|
377 |
+
|
378 |
+
return result;
|
379 |
+
}
|
380 |
+
|
381 |
+
// 清除所有无效cookie记录
|
382 |
+
function clearAllInvalidCookies() {
|
383 |
+
invalidCookies.clear();
|
384 |
+
|
385 |
+
// 保存更新后的无效cookie到文件
|
386 |
+
saveInvalidCookiesToFile();
|
387 |
+
|
388 |
+
return true;
|
389 |
+
}
|
390 |
+
|
391 |
+
module.exports = {
|
392 |
+
addOrUpdateApiKey,
|
393 |
+
removeApiKey,
|
394 |
+
getCookieForApiKey,
|
395 |
+
getAllApiKeys,
|
396 |
+
getAllCookiesForApiKey,
|
397 |
+
initializeApiKeys,
|
398 |
+
removeCookieFromApiKey,
|
399 |
+
getInvalidCookies,
|
400 |
+
clearInvalidCookie,
|
401 |
+
clearAllInvalidCookies,
|
402 |
+
loadInvalidCookiesFromFile,
|
403 |
+
saveInvalidCookiesToFile,
|
404 |
+
loadApiKeysFromFile,
|
405 |
+
saveApiKeysToFile
|
406 |
+
};
|
src/utils/utils.js
ADDED
@@ -0,0 +1,200 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
const os = require('os');
|
2 |
+
const zlib = require('zlib');
|
3 |
+
const crypto = require('crypto');
|
4 |
+
const { v4: uuidv4 } = require('uuid');
|
5 |
+
const $root = require('../proto/message.js');
|
6 |
+
|
7 |
+
const regex = /<\|BEGIN_SYSTEM\|>.*?<\|END_SYSTEM\|>.*?<\|BEGIN_USER\|>.*?<\|END_USER\|>/s;
|
8 |
+
|
9 |
+
function generateCursorBody(messages, modelName) {
|
10 |
+
|
11 |
+
const systemMessages = messages
|
12 |
+
.filter((msg) => msg.role === 'system');
|
13 |
+
const instruction = systemMessages.map((msg) => msg.content).join('\n')
|
14 |
+
|
15 |
+
const nonSystemMessages = messages
|
16 |
+
.filter((msg) => msg.role !== 'system');
|
17 |
+
const formattedMessages = nonSystemMessages.map((msg) => ({
|
18 |
+
...msg,
|
19 |
+
role: msg.role === 'user' ? 1 : 2,
|
20 |
+
messageId: uuidv4()
|
21 |
+
}));
|
22 |
+
|
23 |
+
const chatBody = {
|
24 |
+
userMessages: formattedMessages,
|
25 |
+
instructions: {
|
26 |
+
instruction: "Alway respond in 中文.\n" + instruction
|
27 |
+
},
|
28 |
+
model: {
|
29 |
+
name: modelName,
|
30 |
+
empty: '',
|
31 |
+
},
|
32 |
+
unknown13: 1,
|
33 |
+
conversationId: uuidv4(),
|
34 |
+
unknown16: 1,
|
35 |
+
unknown29: 1,
|
36 |
+
unknown31: 0,
|
37 |
+
};
|
38 |
+
|
39 |
+
const errMsg = $root.ChatMessage.verify(chatBody);
|
40 |
+
if (errMsg) throw Error(errMsg);
|
41 |
+
const chatMessageInstance = $root.ChatMessage.create(chatBody);
|
42 |
+
let buffer = $root.ChatMessage.encode(chatMessageInstance).finish();
|
43 |
+
let magicNumber = 0x00
|
44 |
+
if (formattedMessages.length >= 5){
|
45 |
+
buffer = zlib.gzipSync(buffer)
|
46 |
+
magicNumber = 0x01
|
47 |
+
}
|
48 |
+
|
49 |
+
const finalBody = Buffer.concat([
|
50 |
+
Buffer.from([magicNumber]),
|
51 |
+
Buffer.from(buffer.length.toString(16).padStart(8, '0'), 'hex'),
|
52 |
+
buffer
|
53 |
+
])
|
54 |
+
|
55 |
+
return finalBody
|
56 |
+
}
|
57 |
+
|
58 |
+
function chunkToUtf8String(chunk) {
|
59 |
+
const results = []
|
60 |
+
const errorResults = { hasError: false, errorMessage: '' }
|
61 |
+
|
62 |
+
try {
|
63 |
+
if (!chunk || chunk.length === 0) {
|
64 |
+
console.error('收到空的chunk数据');
|
65 |
+
return '';
|
66 |
+
}
|
67 |
+
|
68 |
+
const buffer = Buffer.from(chunk, 'hex');
|
69 |
+
|
70 |
+
for(let i = 0; i < buffer.length; i++){
|
71 |
+
try {
|
72 |
+
if (i + 5 >= buffer.length) {
|
73 |
+
console.error('数据不足以读取魔数和长度');
|
74 |
+
break;
|
75 |
+
}
|
76 |
+
|
77 |
+
const magicNumber = parseInt(buffer.subarray(i, i + 1).toString('hex'), 16)
|
78 |
+
const dataLength = parseInt(buffer.subarray(i + 1, i + 5).toString('hex'), 16)
|
79 |
+
|
80 |
+
if (dataLength <= 0 || dataLength > 1000000) {
|
81 |
+
console.error(`数据长度异常: ${dataLength}`);
|
82 |
+
break;
|
83 |
+
}
|
84 |
+
|
85 |
+
if (i + 5 + dataLength > buffer.length) {
|
86 |
+
console.error('数据不足以读取内容');
|
87 |
+
break;
|
88 |
+
}
|
89 |
+
|
90 |
+
const data = buffer.subarray(i + 5, i + 5 + dataLength)
|
91 |
+
|
92 |
+
// 处理不同类型的消息
|
93 |
+
const processMessage = async (data, isGzip = false) => {
|
94 |
+
try {
|
95 |
+
let processedData = data;
|
96 |
+
if (isGzip) {
|
97 |
+
processedData = zlib.gunzipSync(data);
|
98 |
+
}
|
99 |
+
|
100 |
+
if (magicNumber === 0 || magicNumber === 1) {
|
101 |
+
// 处理 Proto 消息
|
102 |
+
const resMessage = $root.ResMessage.decode(processedData);
|
103 |
+
if (resMessage.content !== undefined) {
|
104 |
+
results.push(resMessage.content);
|
105 |
+
}
|
106 |
+
} else if (magicNumber === 2 || magicNumber === 3) {
|
107 |
+
// 处理 JSON 消息
|
108 |
+
const utf8 = processedData.toString('utf-8');
|
109 |
+
const message = JSON.parse(utf8);
|
110 |
+
|
111 |
+
if (message && message.error) {
|
112 |
+
console.error(`检测到JSON错误对象(${isGzip ? 'Gzip ' : ''}):`, utf8);
|
113 |
+
errorResults.hasError = true;
|
114 |
+
errorResults.errorMessage = utf8;
|
115 |
+
}
|
116 |
+
} else {
|
117 |
+
console.log('未知的魔数:', magicNumber);
|
118 |
+
}
|
119 |
+
} catch (error) {
|
120 |
+
console.error(`处理${isGzip ? 'Gzip ' : ''}消息错误:`, error);
|
121 |
+
}
|
122 |
+
};
|
123 |
+
|
124 |
+
// 根据魔数处理不同类型的消息
|
125 |
+
if (magicNumber === 0 || magicNumber === 2) {
|
126 |
+
processMessage(data, false);
|
127 |
+
} else if (magicNumber === 1 || magicNumber === 3) {
|
128 |
+
processMessage(data, true);
|
129 |
+
}
|
130 |
+
|
131 |
+
i += 5 + dataLength - 1;
|
132 |
+
} catch (chunkParseError) {
|
133 |
+
console.error('解析单个chunk部分错误:', chunkParseError);
|
134 |
+
i += 1;
|
135 |
+
}
|
136 |
+
}
|
137 |
+
} catch (err) {
|
138 |
+
console.error('解析chunk整体错误:', err);
|
139 |
+
}
|
140 |
+
|
141 |
+
if (errorResults.hasError) {
|
142 |
+
return { error: errorResults.errorMessage };
|
143 |
+
}
|
144 |
+
|
145 |
+
return results.join('');
|
146 |
+
}
|
147 |
+
|
148 |
+
function generateUUIDHash(input, salt = '') {
|
149 |
+
const hash = crypto.createHash('sha256').update(input + salt).digest('hex');
|
150 |
+
const hash128 = hash.substring(0, 32);
|
151 |
+
const uuid = `${hash128.substring(0, 8)}-${hash128.substring(8, 12)}-${hash128.substring(12, 16)}-${hash128.substring(16, 20)}-${hash128.substring(20, 32)}`;
|
152 |
+
|
153 |
+
return uuid;
|
154 |
+
}
|
155 |
+
|
156 |
+
function generateHashed64Hex(input, salt = '') {
|
157 |
+
const hash = crypto.createHash('sha256');
|
158 |
+
hash.update(input + salt);
|
159 |
+
return hash.digest('hex');
|
160 |
+
}
|
161 |
+
|
162 |
+
function obfuscateBytes(byteArray) {
|
163 |
+
let t = 165;
|
164 |
+
for (let r = 0; r < byteArray.length; r++) {
|
165 |
+
byteArray[r] = (byteArray[r] ^ t) + (r % 256);
|
166 |
+
t = byteArray[r];
|
167 |
+
}
|
168 |
+
return byteArray;
|
169 |
+
}
|
170 |
+
|
171 |
+
function generateCursorChecksum(token) {
|
172 |
+
// 生成machineId和macMachineId
|
173 |
+
const machineId = generateHashed64Hex(token, 'machineId');
|
174 |
+
const macMachineId = generateHashed64Hex(token, 'macMachineId');
|
175 |
+
|
176 |
+
// 获取时间戳并转换为字节数组
|
177 |
+
const timestamp = Math.floor(Date.now() / 1e6);
|
178 |
+
const byteArray = new Uint8Array([
|
179 |
+
(timestamp >> 40) & 255,
|
180 |
+
(timestamp >> 32) & 255,
|
181 |
+
(timestamp >> 24) & 255,
|
182 |
+
(timestamp >> 16) & 255,
|
183 |
+
(timestamp >> 8) & 255,
|
184 |
+
255 & timestamp,
|
185 |
+
]);
|
186 |
+
|
187 |
+
// 混淆字节数组并进行base64编码
|
188 |
+
const obfuscatedBytes = obfuscateBytes(byteArray);
|
189 |
+
const encodedChecksum = Buffer.from(obfuscatedBytes).toString('base64');
|
190 |
+
|
191 |
+
// 组合最终的checksum
|
192 |
+
return `${encodedChecksum}${machineId}/${macMachineId}`;
|
193 |
+
}
|
194 |
+
|
195 |
+
module.exports = {
|
196 |
+
generateCursorBody,
|
197 |
+
chunkToUtf8String,
|
198 |
+
generateHashed64Hex,
|
199 |
+
generateCursorChecksum
|
200 |
+
};
|