mirror of
https://github.com/ikechan8370/chatgpt-plugin.git
synced 2025-12-16 21:37:11 +00:00
103 lines
4.1 KiB
JavaScript
103 lines
4.1 KiB
JavaScript
import ChatGPTConfig from '../config/config.js'
|
||
import { createCRUDCommandRules, createSwitchCommandRules } from '../utils/command.js'
|
||
import { Chaite } from 'chaite'
|
||
|
||
export class ChatGPTManagement extends plugin {
|
||
constructor () {
|
||
const cmdPrefix = ChatGPTConfig.basic.commandPrefix
|
||
super({
|
||
name: 'ChatGPT-Plugin管理',
|
||
dsc: 'ChatGPT-Plugin管理',
|
||
event: 'message',
|
||
priority: 20,
|
||
rule: [
|
||
{
|
||
reg: `^${cmdPrefix}管理面板$`,
|
||
fnc: 'managementPanel',
|
||
permission: 'master'
|
||
},
|
||
{
|
||
reg: `^${cmdPrefix}结束(全部)?对话$`,
|
||
fnc: 'destroyConversation'
|
||
},
|
||
{
|
||
reg: `^${cmdPrefix}(bym|伪人)设置默认预设`,
|
||
fnc: 'setDefaultBymPreset',
|
||
permission: 'master'
|
||
}
|
||
]
|
||
})
|
||
this.initCommand(cmdPrefix)
|
||
}
|
||
|
||
async initCommand (cmdPrefix) {
|
||
const waitForChaite = async () => {
|
||
while (!Chaite.getInstance()) {
|
||
await new Promise(resolve => setTimeout(resolve, 1000))
|
||
}
|
||
return Chaite.getInstance()
|
||
}
|
||
|
||
await waitForChaite()
|
||
this.rule.push(...[
|
||
...createCRUDCommandRules.bind(this)(cmdPrefix, '渠道', 'channels'),
|
||
...createCRUDCommandRules.bind(this)(cmdPrefix, '预设', 'presets'),
|
||
...createCRUDCommandRules.bind(this)(cmdPrefix, '工具', 'tools'),
|
||
...createCRUDCommandRules.bind(this)(cmdPrefix, '处理器', 'processors'),
|
||
createSwitchCommandRules.bind(this)(cmdPrefix, '(预设切换|其他人切换预设)', 'customPreset', 1),
|
||
createSwitchCommandRules.bind(this)(cmdPrefix, '(调试|debug)(模式)?', 'debug'),
|
||
...createCRUDCommandRules.bind(this)(cmdPrefix, '预设切换黑名单', 'customPresetUserBlackList', false),
|
||
...createCRUDCommandRules.bind(this)(cmdPrefix, '预设切换白名单', 'customPresetUserWhiteList', false),
|
||
...createCRUDCommandRules.bind(this)(cmdPrefix, '输入屏蔽词', 'promptBlockWords', false),
|
||
...createCRUDCommandRules.bind(this)(cmdPrefix, '输出屏蔽词', 'responseBlockWords', false),
|
||
...createCRUDCommandRules.bind(this)(cmdPrefix, '黑名单群', 'blackGroups', false),
|
||
...createCRUDCommandRules.bind(this)(cmdPrefix, '白名单群', 'whiteGroups', false),
|
||
...createCRUDCommandRules.bind(this)(cmdPrefix, '黑名单用户', 'blackUsers', false),
|
||
...createCRUDCommandRules.bind(this)(cmdPrefix, '白名单用户', 'whiteUsers', false),
|
||
createSwitchCommandRules(cmdPrefix, '(伪人|bym)', 'bym')
|
||
])
|
||
}
|
||
|
||
managementPanel (e) {
|
||
// todo
|
||
// this.reply(`(todo)管理面板地址：http://${ChatGPTConfig.chaite.host}:${ChatGPTConfig.chaite.host}`)
|
||
const token = Chaite.getInstance().getFrontendAuthHandler().generateToken(300)
|
||
this.reply(`token: ${token}, 有效期300秒`, true)
|
||
}
|
||
|
||
async setDefaultBymPreset (e) {
|
||
const presetId = e.msg.replace(`${ChatGPTConfig.basic.commandPrefix}伪人设置默认预设`, '')
|
||
const preset = await Chaite.getInstance().getChatPresetManager().getInstance(presetId)
|
||
if (preset) {
|
||
ChatGPTConfig.bym.defaultPreset = presetId
|
||
this.reply(`伪人模式默认预设已切换为${presetId}(${preset.name})`)
|
||
} else {
|
||
this.reply(`未找到预设${presetId}`)
|
||
}
|
||
}
|
||
|
||
async destroyConversation (e) {
|
||
if (e.msg.includes('全部')) {
|
||
if (!e.isMaster) {
|
||
this.reply('仅限主人使用')
|
||
}
|
||
const userStates = await Chaite.getInstance().getUserStateStorage().listItems()
|
||
let num = 0
|
||
for (const userState of userStates) {
|
||
if (userState.current.conversationId) {
|
||
num++
|
||
}
|
||
userState.current.conversationId = ''
|
||
userState.current.messageId = ''
|
||
await Chaite.getInstance().getUserStateStorage().setItem(userState.userId + '', userState)
|
||
}
|
||
this.reply(`已结束${num}个用户的对话`)
|
||
} else {
|
||
const state = await Chaite.getInstance().getUserStateStorage().getItem(e.sender.user_id + '')
|
||
state.current.conversationId = ''
|
||
state.current.messageId = ''
|
||
await Chaite.getInstance().getUserStateStorage().setItem(e.sender.user_id + '', state)
|
||
this.reply('已结束当前对话')
|
||
}
|
||
}
|
||
}
|