mirror of
https://github.com/ikechan8370/chatgpt-plugin.git
synced 2025-12-16 21:37:11 +00:00
新增管理面板，重写必应token管理，错误修复 (#340)
* 修复引用转发，默认bing模式并发 * 开启stream增加稳定性 * fix: remove queue element only in non-bing mode * 使用chatgpt-api自带的超时逻辑，文字过多时启动切换到图片输出防止被吞 * Update chat.js * 添加Bing专用的图片输出样式 * 添加chatgpt的新图片模式，临时处理切换api导致的对话异常 * 修改bing样式表 * 为图片添加外部页面缓存 * 为图片模式添加MathJax * feat: add switch for qrcode * 防止script攻击 * 修复网页模板错误 * 修复bing页面引用错误 * 缓存服务器异常时处理 * 添加默认配置加载 * 修复配置文件路径错误 * 删除重复的模板文件，修复二维码地址错误 * 修正图片渲染错误 * 修复引用渲染错误 * 二维码网址统一改为使用本地配置 * 添加关闭思考提示的配置项 * 修复在Windows上无法载入配置文件的问题 * 修复关闭qr的情况下渲染错误 * 改为使用base64传递返回数据 * 当异常过多时使用图片输出 * 添加锅巴面板配置支持 * 补充遗漏的默认配置 * 修复qr模式下引用未被传递的问题 * 修复未将引用数据传输给缓存服务器的问题 * 删除无用的bingTimeoutMs配置项 * 添加消息队列超时弹出 * 优化图片模式处理，解决对话队列卡住的问题 * 添加对图片ocr的支持 * 添加图片识别配置项 * 添加黑名单配置项 * 修复一些bug * 修改锅巴配置格式和描述 * 传入数据也使用markdown * 图片识别换行改为marked兼容 * 添加绘图CD配置项 * 独立render模块，添加图片回复引用 * 添加必应风格 * 修复上下文，修改bing样式 * 修复上下文 * 添加Sydney上下文支持 * 调整不同模式下的bing渲染颜色 * 修复样式 * 修复无法结束会话的问题 * fix: 更新版本号 * 修复无法结束对话的问题 * 向缓存服务器传送样式 * 为网址格式的配置添加验证 * 去除重复的Keyv删除，取消锅巴配置格式检查 * 闭合中断的代码块 * 试添加Sydney图片模式的情感显示 * 修复at不兼容 * 处理意外的markdown包裹和结构解析修复 * 修复markdown处理的顺序错误 * 兼容json换行 * 重写completeJSON和使用 * 修复换行格式异常 * 均衡BingToken使用 * 修复删除token的数组处理错误 * 修改token文字描述 * 创建本地缓存服务 * 修复首次使用无法添加bingtoken的问题 * 修复意外的删除格式问题，添加查看token功能 * 修复路由错误，暂时固定ip测试 * 恢复引用功能 * 更新渲染页面 * 更换缓存目录 * 清除调试用消息 * 调整屏幕分辨率 * 使用服务器生成的访问地址 * 改为使用api获取公网ip * 修复引用显示 * 添加依赖需求 * 更新渲染页面和渲染api * 修复渲染页面错误 * 修复建议字符串切割，添加帮助路由 * 添加内容中图片数据获取功能 * 试修复suggestbug * 修复图片导致服务器卡死的问题 * 暂时禁用图片 * 尝试恢复图片 * 添加链接图片识别 * 替换掉request * 修复可能的responseUrls空值 * 优化格式 * 更新渲染页面 * 尝试新的引用索引 * 取消渲染时旧的策略 * 更新帮助页面 * 修复帮助路由 * 修复渲染页面错误 * 修复错误的正则 * 修改系统api服务 * 添加配置项 * 将新渲染方式加入配置并还原原渲染方式，进行并存 * 暂时取消端口设置功能 * 重新开启端口设置 * 修复旧渲染引用 * 更新帮助样式 * 更新帮助，增强功能 * 有cacheHost的情况下不再附带端口号 * 添加渲染图片的宽度设置 * 添加渲染页面宽度调整，修bug * 修复二维码不显示 * 添加第三方渲染支持 * 修复一些渲染页面问题 * 更新渲染页面 * 修正错误的变量调用 * 添加新渲染模式bot命名 * 修复空消息问题 * 撤销之前的修复，使用新方法修复 * 修复返回空页面问题 * 尝试不依赖网络获取外网地址 * 修bug，初步创建管理系统 * 依赖名写错了 * 修复错误的异步 * 修正错误的配置调用 * 放弃本机设置的获取方案，对服务器获取多半失效 * 添加配置页面接口 * 更新渲染页面 * 添加依赖 * 修复bug * 移除windows性能显示，更换依赖 * 添加依赖 * 修复图片异常时不反回文字而是直接报错的问题 * 修改必应token记录和均衡方法，更新渲染页面 * 修复错误 * 修复bug，更新渲染页面 * 更新渲染 * 修复ip错误 * 完善配置页面 * 渲染页面错误修复 * 更新版本号 * 只获取一次有效ip * 修复渲染页面bug --------- Co-authored-by: ikechan8370 <geyinchibuaa@gmail.com> Co-authored-by: Err0rCM <68117733+Err0rCM@users.noreply.github.com>
This commit is contained in:
HalcyonAlcedo
GitHub
parent
c0a596a608
commit
458b04c666
30 changed files with 767 additions and 236 deletions
269
server/index.js
269
server/index.js
|
|
@ -1,46 +1,65 @@
|
|||
import fastify from 'fastify'
|
||||
import fastifyCookie from '@fastify/cookie'
|
||||
import cors from '@fastify/cors'
|
||||
import fstatic from '@fastify/static'
|
||||
|
||||
import fs from 'fs'
|
||||
import path from 'path'
|
||||
import http from 'http'
|
||||
import os from 'os'
|
||||
import schedule from 'node-schedule'
|
||||
|
||||
import { Config } from '../utils/config.js'
|
||||
|
||||
function getPublicIP() {
|
||||
return new Promise((resolve, reject) => {
|
||||
http.get('http://ipinfo.io/json', (res) => {
|
||||
let data = ''
|
||||
res.on('data', (chunk) => {
|
||||
data += chunk
|
||||
});
|
||||
res.on('end', () => {
|
||||
try {
|
||||
const ip = JSON.parse(data).ip
|
||||
resolve(ip)
|
||||
} catch (e) {
|
||||
reject(e)
|
||||
}
|
||||
})
|
||||
}).on('error', (err) => {
|
||||
reject(err)
|
||||
})
|
||||
})
|
||||
}
|
||||
import { randomString, getPublicIP } from '../utils/common.js'
|
||||
|
||||
const __dirname = path.resolve()
|
||||
const server = fastify({
|
||||
logger: Config.debug
|
||||
})
|
||||
|
||||
let usertoken = ''
|
||||
let Statistics = {
|
||||
SystemAccess: {
|
||||
count: 0,
|
||||
oldCount: 0
|
||||
},
|
||||
CacheFile: {
|
||||
count: 0,
|
||||
oldCount: 0
|
||||
},
|
||||
WebAccess: {
|
||||
count: 0,
|
||||
oldCount: 0
|
||||
},
|
||||
SystemLoad: {
|
||||
count: 0,
|
||||
oldCount: 0
|
||||
}
|
||||
}
|
||||
|
||||
async function getLoad() {
|
||||
// 获取当前操作系统平台
|
||||
const platform = os.platform();
|
||||
// 判断平台是Linux还是Windows
|
||||
if (platform === 'linux') {
|
||||
// 如果是Linux，使用os.loadavg()方法获取负载平均值
|
||||
const loadAvg = os.loadavg();
|
||||
return loadAvg[0] * 100
|
||||
} else if (platform === 'win32') {
|
||||
// 如果是Windows不获取性能
|
||||
return 0
|
||||
} else {
|
||||
return 0
|
||||
}
|
||||
}
|
||||
|
||||
export async function createServer() {
|
||||
await server.register(cors, {
|
||||
origin: '*',
|
||||
origin: '*',
|
||||
})
|
||||
await server.register(fstatic, {
|
||||
root: path.join(__dirname, 'plugins/chatgpt-plugin/server/static/'),
|
||||
root: path.join(__dirname, 'plugins/chatgpt-plugin/server/static/'),
|
||||
})
|
||||
await server.register(fastifyCookie)
|
||||
await server.get('/page/*', (request, reply) => {
|
||||
const stream = fs.createReadStream('plugins/chatgpt-plugin/server/static/index.html')
|
||||
reply.type('text/html').send(stream)
|
||||
|
|
@ -49,69 +68,171 @@ export async function createServer() {
|
|||
const stream = fs.createReadStream('plugins/chatgpt-plugin/server/static/index.html')
|
||||
reply.type('text/html').send(stream)
|
||||
})
|
||||
await server.get('/auth/*', (request, reply) => {
|
||||
const stream = fs.createReadStream('plugins/chatgpt-plugin/server/static/index.html')
|
||||
reply.type('text/html').send(stream)
|
||||
})
|
||||
await server.get('/admin/*', (request, reply) => {
|
||||
const token = request.cookies.token || 'unknown'
|
||||
if (token != usertoken) {
|
||||
reply.redirect(301, '/auth/login')
|
||||
}
|
||||
const stream = fs.createReadStream('plugins/chatgpt-plugin/server/static/index.html')
|
||||
reply.type('text/html').send(stream)
|
||||
})
|
||||
// 登录
|
||||
server.post('/login', async (request, reply) => {
|
||||
const body = request.body || {}
|
||||
if (body.qq && body.passwd) {
|
||||
if (body.qq == Bot.uin && await redis.get('CHATGPT:ADMIN_PASSWD') == body.passwd) {
|
||||
usertoken = randomString(32)
|
||||
reply.setCookie('token', usertoken, {path: '/'})
|
||||
reply.send({login:true})
|
||||
} else {
|
||||
reply.send({login:false,err:'用户名密码错误'})
|
||||
}
|
||||
} else {
|
||||
reply.send({login:false,err:'未输入用户名或密码'})
|
||||
}
|
||||
})
|
||||
// 页面数据获取
|
||||
server.post('/page', async (request, reply) => {
|
||||
const body = request.body || {}
|
||||
if (body.code) {
|
||||
const dir = 'resources/ChatGPTCache/page'
|
||||
const filename = body.code + '.json'
|
||||
const filepath = path.join(dir, filename)
|
||||
|
||||
let data = fs.readFileSync(filepath, 'utf8')
|
||||
reply.send(data)
|
||||
}
|
||||
const body = request.body || {}
|
||||
if (body.code) {
|
||||
const dir = 'resources/ChatGPTCache/page'
|
||||
const filename = body.code + '.json'
|
||||
const filepath = path.join(dir, filename)
|
||||
let data = fs.readFileSync(filepath, 'utf8')
|
||||
reply.send(data)
|
||||
}
|
||||
})
|
||||
// 帮助内容获取
|
||||
server.post('/help', async (request, reply) => {
|
||||
const body = request.body || {}
|
||||
if (body.use) {
|
||||
const dir = 'plugins/chatgpt-plugin/resources'
|
||||
const filename = 'help.json'
|
||||
const filepath = path.join(dir, filename)
|
||||
let data = fs.readFileSync(filepath, 'utf8')
|
||||
data = JSON.parse(data)
|
||||
reply.send(data[body.use])
|
||||
const dir = 'plugins/chatgpt-plugin/resources'
|
||||
const filename = 'help.json'
|
||||
const filepath = path.join(dir, filename)
|
||||
let data = fs.readFileSync(filepath, 'utf8')
|
||||
data = JSON.parse(data)
|
||||
reply.send(data[body.use])
|
||||
}
|
||||
})
|
||||
// 创建页面缓存内容
|
||||
server.post('/cache', async (request, reply) => {
|
||||
const body = request.body || {}
|
||||
if (body.content) {
|
||||
const dir = 'resources/ChatGPTCache/page'
|
||||
const filename = body.entry + '.json'
|
||||
const filepath = path.join(dir, filename)
|
||||
const regexUrl = /\b((?:https?|ftp|file):\/\/[-a-zA-Z0-9+&@#\/%?=~_|!:,.;]*[-a-zA-Z0-9+&@#\/%=~_|])/g
|
||||
const ip = await getPublicIP()
|
||||
try {
|
||||
fs.mkdirSync(dir, { recursive: true });
|
||||
fs.writeFileSync(filepath, JSON.stringify({
|
||||
user: body.content.senderName,
|
||||
bot: Config.chatViewBotName || (body.bing ? 'Bing' : 'ChatGPT'),
|
||||
userImg: body.userImg || '',
|
||||
botImg: body.botImg || '',
|
||||
question: body.content.prompt,
|
||||
message: body.content.content,
|
||||
group: body.content.group,
|
||||
herf: `http://${body.cacheHost || (ip + ':' + Config.serverPort || 3321)}/page/${body.entry}`,
|
||||
quote: body.content.quote,
|
||||
images: body.content.images || [],
|
||||
suggest: body.content.suggest || [],
|
||||
time: new Date()
|
||||
}))
|
||||
reply.send({ file: body.entry, cacheUrl: `http://${ip}:${Config.serverPort || 3321}/page/${body.entry}` })
|
||||
} catch (err) {
|
||||
console.error(err)
|
||||
reply.send({ file: body.entry, cacheUrl: `http://${ip}/page/${body.entry}`, error: '生成失败' })
|
||||
}
|
||||
const body = request.body || {}
|
||||
if (body.content) {
|
||||
const dir = 'resources/ChatGPTCache/page'
|
||||
const filename = body.entry + '.json'
|
||||
const filepath = path.join(dir, filename)
|
||||
const regexUrl = /\b((?:https?|ftp|file):\/\/[-a-zA-Z0-9+&@#\/%?=~_|!:,.;]*[-a-zA-Z0-9+&@#\/%=~_|])/g
|
||||
const ip = await getPublicIP()
|
||||
try {
|
||||
fs.mkdirSync(dir, { recursive: true });
|
||||
fs.writeFileSync(filepath, JSON.stringify({
|
||||
user: body.content.senderName,
|
||||
bot: Config.chatViewBotName || (body.bing ? 'Bing' : 'ChatGPT'),
|
||||
userImg: body.userImg || '',
|
||||
botImg: body.botImg || '',
|
||||
question: body.content.prompt,
|
||||
message: body.content.content,
|
||||
group: body.content.group,
|
||||
herf: `http://${body.cacheHost || (ip + ':' + Config.serverPort || 3321)}/page/${body.entry}`,
|
||||
quote: body.content.quote,
|
||||
images: body.content.images || [],
|
||||
suggest: body.content.suggest || [],
|
||||
time: new Date()
|
||||
}))
|
||||
Statistics.CacheFile.count += 1
|
||||
reply.send({ file: body.entry, cacheUrl: `http://${ip}:${Config.serverPort || 3321}/page/${body.entry}` })
|
||||
} catch (err) {
|
||||
console.error(err)
|
||||
reply.send({ file: body.entry, cacheUrl: `http://${ip}:${Config.serverPort || 3321}/page/${body.entry}`, error: '生成失败' })
|
||||
}
|
||||
}
|
||||
})
|
||||
server.listen({
|
||||
port: Config.serverPort || 3321,
|
||||
host: '0.0.0.0'
|
||||
}, (error) => {
|
||||
if (error) {
|
||||
console.error(error);
|
||||
// 获取系统状态
|
||||
server.post('/system-statistics', async (request, reply) => {
|
||||
Statistics.SystemLoad.count = await getLoad()
|
||||
reply.send(Statistics)
|
||||
})
|
||||
|
||||
server.post('/sysconfig', async (request, reply) => {
|
||||
const token = request.cookies.token || 'unknown'
|
||||
if (token != usertoken) {
|
||||
reply.send({err: '未登录'})
|
||||
} else {
|
||||
let redisConfig = {}
|
||||
if (await redis.exists('CHATGPT:BING_TOKENS') != 0) {
|
||||
let bingTokens = await redis.get('CHATGPT:BING_TOKENS')
|
||||
if (bingTokens)
|
||||
bingTokens = JSON.parse(bingTokens)
|
||||
else bingTokens = []
|
||||
redisConfig.bingTokens = bingTokens
|
||||
} else {
|
||||
redisConfig.bingTokens = []
|
||||
}
|
||||
server.log.info(`server listening on ${server.server.address().port}`)
|
||||
if (await redis.exists('CHATGPT:CONFIRM') != 0) {
|
||||
redisConfig.turnConfirm = await redis.get('CHATGPT:CONFIRM') === 'on'
|
||||
}
|
||||
reply.send({
|
||||
chatConfig: Config,
|
||||
redisConfig: redisConfig
|
||||
})
|
||||
}
|
||||
})
|
||||
server.post('/saveconfig', async (request, reply) => {
|
||||
const token = request.cookies.token || 'unknown'
|
||||
if (token != usertoken) {
|
||||
reply.send({err: '未登录'})
|
||||
} else {
|
||||
const body = request.body || {}
|
||||
const chatdata = body.chatConfig || {}
|
||||
for (let [keyPath, value] of Object.entries(chatdata)) {
|
||||
if (keyPath === 'blockWords' || keyPath === 'promptBlockWords' || keyPath === 'initiativeChatGroups') { value = value.toString().split(/[,，;；\|]/) }
|
||||
if (Config[keyPath] != value) { Config[keyPath] = value }
|
||||
}
|
||||
const redisConfig = body.redisConfig || {}
|
||||
if (redisConfig.bingTokens != null) {
|
||||
await redis.set('CHATGPT:BING_TOKENS', JSON.stringify(redisConfig.bingTokens))
|
||||
}
|
||||
if (redisConfig.turnConfirm != null) {
|
||||
await redis.set('CHATGPT:CONFIRM', redisConfig.turnConfirm ? 'on' : 'off')
|
||||
}
|
||||
}
|
||||
})
|
||||
|
||||
server.addHook('onRequest', (request, reply, done) => {
|
||||
if(request.method == 'POST')
|
||||
Statistics.SystemAccess.count += 1
|
||||
if(request.method == 'GET')
|
||||
Statistics.WebAccess.count += 1
|
||||
done()
|
||||
})
|
||||
//定时任务
|
||||
var rule = new schedule.RecurrenceRule();
|
||||
rule.hour = 0;
|
||||
rule.minute = 0;
|
||||
let job_Statistics = schedule.scheduleJob(rule, function() {
|
||||
Statistics.SystemAccess.oldCount = Statistics.SystemAccess.count
|
||||
Statistics.CacheFile.oldCount = Statistics.CacheFile.count
|
||||
Statistics.WebAccess.oldCount = Statistics.WebAccess.count
|
||||
Statistics.SystemAccess.count = 0
|
||||
Statistics.CacheFile.count = 0
|
||||
Statistics.WebAccess.count = 0
|
||||
});
|
||||
let job_Statistics_SystemLoad = schedule.scheduleJob('0 * * * *', async function(){
|
||||
Statistics.SystemLoad.count = await getLoad()
|
||||
Statistics.SystemLoad.oldCount = Statistics.SystemLoad.count
|
||||
});
|
||||
|
||||
server.listen({
|
||||
port: Config.serverPort || 3321,
|
||||
host: '0.0.0.0'
|
||||
}, (error) => {
|
||||
if (error) {
|
||||
console.error(error)
|
||||
}
|
||||
server.log.info(`server listening on ${server.server.address().port}`)
|
||||
})
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue