MurasameNoa/chatgpt-plugin
1
0
Fork 0
mirror of https://github.com/ikechan8370/chatgpt-plugin.git synced 2025-12-16 13:27:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

添加锅巴用户数据

Browse source
This commit is contained in:
zyc404 2023-10-15 17:33:57 +08:00
parent ae14be4cf5
commit 471da4f3b1
1 changed files with 17 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

19
server/modules/user.js
View file

@ -6,12 +6,26 @@ async function User(fastify, options) {
// 登录
fastify.post('/login', async (request, reply) => {
const body = request.body || {}
let guobaLoginService
let guobaAPI = ''
try {
let { LoginService } = await import('../../../../guoba-plugin/server/service/both/LoginService.js')
let { getAllWebAddress } = await import('../../../../guoba-plugin/utils/common.js')
guobaLoginService = new LoginService()
guobaAPI = await getAllWebAddress().remote
}
catch (err) {
guobaLoginService = {
signToken: () => {return null}
}
}
if (body.qq && body.passwd) {
const token = randomString(32)
if (body.qq == getUin() && await redis.get('CHATGPT:ADMIN_PASSWD') == body.passwd) {
const guobaToken = await guobaLoginService.signToken(body.qq)
AddUser({ user: body.qq, token: token, autho: 'admin' })
reply.setCookie('token', token, { path: '/' })
reply.send({ login: true, autho: 'admin', token: token })
reply.send({ login: true, autho: 'admin', token: token, guobaToken: guobaToken, guoba: guobaAPI })
} else {
const user = await getUserData(body.qq)
if (user.passwd != '' && user.passwd === body.passwd) {
@ -26,9 +40,10 @@ async function User(fastify, options) {
const token = randomString(32)
const opt = await redis.get(`CHATGPT:SERVER_QUICK`)
if (opt && body.otp == opt) {
const guobaToken = await guobaLoginService.signToken(getUin())
AddUser({ user: getUin(), token: token, autho: 'admin' })
reply.setCookie('token', token, { path: '/' })
reply.send({ login: true, autho: 'admin', token: token, user: getUin() })
reply.send({ login: true, autho: 'admin', token: token, user: getUin(), guobaToken: guobaToken, guoba: guobaAPI })
} else {
reply.send({ login: false, err: `快捷登录代码错误，请检查后重试` })
}