mirror of
https://github.com/yoimiya-kokomi/miao-plugin.git
synced 2024-11-23 08:00:57 +00:00
271 lines
7.3 KiB
JavaScript
271 lines
7.3 KiB
JavaScript
import fs from 'fs'
|
||
import lodash from 'lodash'
|
||
import { exec } from 'child_process'
|
||
import { Cfg, Common, Data, App } from '../components/index.js'
|
||
import { MysApi } from '../models/index.js'
|
||
|
||
let cfgMap = {
|
||
角色: 'char.char',
|
||
面板: 'char.profile',
|
||
老婆: 'char.wife',
|
||
戳一戳: 'char.poke',
|
||
小清新: 'char.se',
|
||
查他人: 'char.queryOther',
|
||
图鉴: 'wiki.wiki',
|
||
图片: 'wiki.pic',
|
||
统计: 'wiki.stat',
|
||
深渊: 'wiki.abyss',
|
||
渲染: 'sys.scale',
|
||
帮助: 'sys.help'
|
||
}
|
||
let app = App.init({
|
||
id: 'admin',
|
||
name: '喵喵设置',
|
||
desc: '喵喵设置'
|
||
})
|
||
|
||
let sysCfgReg = new RegExp(`^#喵喵设置\\s*(${lodash.keys(cfgMap).join('|')})?\\s*(.*)$`)
|
||
|
||
app.reg('update-res', updateRes, {
|
||
rule: /^#喵喵(强制)?(更新图像|图像更新)$/,
|
||
desc: '【#管理】更新素材'
|
||
})
|
||
app.reg('update', updateMiaoPlugin, {
|
||
rule: /^#喵喵(强制)?更新/,
|
||
desc: '【#管理】喵喵更新'
|
||
})
|
||
app.reg('sys-cfg', sysCfg, {
|
||
rule: sysCfgReg,
|
||
desc: '【#管理】系统设置'
|
||
})
|
||
app.reg('profile-cfg', profileCfg, {
|
||
rule: /^#喵喵面板(?:设置)?.*/,
|
||
desc: '面板设置'
|
||
})
|
||
|
||
export default app
|
||
|
||
const _path = process.cwd()
|
||
const resPath = `${_path}/plugins/miao-plugin/resources/`
|
||
const plusPath = `${resPath}/miao-res-plus/`
|
||
|
||
const checkAuth = async function (e) {
|
||
if (!e.isMaster) {
|
||
e.reply(`只有主人才能命令喵喵哦~
|
||
(*/ω＼*)`)
|
||
return false
|
||
}
|
||
return await MysApi.initUser(e)
|
||
}
|
||
|
||
async function sysCfg (e) {
|
||
if (!await checkAuth(e)) {
|
||
return true
|
||
}
|
||
|
||
let cfgReg = sysCfgReg
|
||
let regRet = cfgReg.exec(e.msg)
|
||
|
||
if (!regRet) {
|
||
return true
|
||
}
|
||
|
||
if (regRet[1]) {
|
||
// 设置模式
|
||
let val = regRet[2] || ''
|
||
|
||
let cfgKey = cfgMap[regRet[1]]
|
||
if (cfgKey === 'sys.scale') {
|
||
val = Math.min(200, Math.max(50, val * 1 || 100))
|
||
} else {
|
||
val = !/关闭/.test(val)
|
||
}
|
||
|
||
if (cfgKey) {
|
||
Cfg.set(cfgKey, val)
|
||
}
|
||
}
|
||
|
||
let cfg = {
|
||
chars: getStatus('char.char'),
|
||
profile: getStatus('char.profile'),
|
||
wife: getStatus('char.wife'),
|
||
poke: getStatus('char.poke'),
|
||
se: getStatus('char.se', false),
|
||
other: getStatus('char.queryOther'),
|
||
wiki: getStatus('wiki.wiki'),
|
||
pic: getStatus('wiki.pic'),
|
||
stat: getStatus('wiki.stat'),
|
||
abyss: getStatus('wiki.abyss', false),
|
||
imgPlus: fs.existsSync(plusPath),
|
||
help: getStatus('sys.help', false),
|
||
scale: Cfg.get('sys.scale', 100)
|
||
}
|
||
|
||
// 渲染图像
|
||
return await Common.render('admin/index', {
|
||
...cfg
|
||
}, { e, scale: 1.4 })
|
||
}
|
||
|
||
const getStatus = function (rote, def = true) {
|
||
if (Cfg.get(rote, def)) {
|
||
return '<div class="cfg-status" >已开启</div>'
|
||
} else {
|
||
return '<div class="cfg-status status-off">已关闭</div>'
|
||
}
|
||
}
|
||
|
||
async function updateRes (e) {
|
||
if (!await checkAuth(e)) {
|
||
return true
|
||
}
|
||
let isForce = e.msg.includes('强制')
|
||
let command = ''
|
||
if (fs.existsSync(`${resPath}/miao-res-plus/`)) {
|
||
e.reply('开始尝试更新，请耐心等待~')
|
||
command = 'git pull'
|
||
if (isForce) {
|
||
command = 'git checkout . && git pull'
|
||
}
|
||
exec(command, { cwd: `${resPath}/miao-res-plus/` }, function (error, stdout, stderr) {
|
||
console.log(stdout)
|
||
if (/(Already up[ -]to[ -]date|已经是最新的)/.test(stdout)) {
|
||
e.reply('目前所有图片都已经是最新了~')
|
||
return true
|
||
}
|
||
let numRet = /(\d*) files changed,/.exec(stdout)
|
||
if (numRet && numRet[1]) {
|
||
e.reply(`报告主人，更新成功，此次更新了${numRet[1]}个图片~`)
|
||
return true
|
||
}
|
||
if (error) {
|
||
e.reply('更新失败！\nError code: ' + error.code + '\n' + error.stack + '\n 请稍后重试。')
|
||
} else {
|
||
e.reply('图片加量包更新成功~')
|
||
}
|
||
})
|
||
} else {
|
||
command = `git clone https://gitee.com/yoimiya-kokomi/miao-res-plus.git "${resPath}/miao-res-plus/" --depth=1`
|
||
e.reply('开始尝试安装图片加量包，可能会需要一段时间，请耐心等待~')
|
||
exec(command, function (error, stdout, stderr) {
|
||
if (error) {
|
||
e.reply('角色图片加量包安装失败！\nError code: ' + error.code + '\n' + error.stack + '\n 请稍后重试。')
|
||
} else {
|
||
e.reply('角色图片加量包安装成功！您后续也可以通过 #喵喵更新图像 命令来更新图像')
|
||
}
|
||
})
|
||
}
|
||
return true
|
||
}
|
||
|
||
let timer
|
||
|
||
async function updateMiaoPlugin (e) {
|
||
if (!await checkAuth(e)) {
|
||
return true
|
||
}
|
||
let isForce = e.msg.includes('强制')
|
||
let command = 'git pull'
|
||
if (isForce) {
|
||
command = 'git checkout . && git pull'
|
||
e.reply('正在执行强制更新操作，请稍等')
|
||
} else {
|
||
e.reply('正在执行更新操作，请稍等')
|
||
}
|
||
exec(command, { cwd: `${_path}/plugins/miao-plugin/` }, function (error, stdout, stderr) {
|
||
if (/(Already up[ -]to[ -]date|已经是最新的)/.test(stdout)) {
|
||
e.reply('目前已经是最新版喵喵了~')
|
||
return true
|
||
}
|
||
if (error) {
|
||
e.reply('喵喵更新失败！\nError code: ' + error.code + '\n' + error.stack + '\n 请稍后重试。')
|
||
return true
|
||
}
|
||
e.reply('喵喵更新成功，正在尝试重新启动Yunzai以应用更新...')
|
||
timer && clearTimeout(timer)
|
||
Data.setCacheJSON('miao:restart-msg', {
|
||
msg: '重启成功，新版喵喵已经生效',
|
||
qq: e.user_id
|
||
}, 30)
|
||
timer = setTimeout(function () {
|
||
let command = 'npm run start'
|
||
if (process.argv[1].includes('pm2')) {
|
||
command = 'npm run restart'
|
||
}
|
||
exec(command, function (error, stdout, stderr) {
|
||
if (error) {
|
||
e.reply('自动重启失败，请手动重启以应用新版喵喵。\nError code: ' + error.code + '\n' + error.stack + '\n')
|
||
Bot.logger.error(`重启失败\n${error.stack}`)
|
||
return true
|
||
} else if (stdout) {
|
||
Bot.logger.mark('重启成功，运行已转为后台，查看日志请用命令：npm run log')
|
||
Bot.logger.mark('停止后台运行命令：npm stop')
|
||
process.exit()
|
||
}
|
||
})
|
||
}, 1000)
|
||
})
|
||
return true
|
||
}
|
||
|
||
async function profileCfg (e) {
|
||
if (!await checkAuth(e)) {
|
||
return true
|
||
}
|
||
|
||
let keyMap = {
|
||
好友: 'friend',
|
||
群: 'group',
|
||
陌生人: 'stranger'
|
||
}
|
||
|
||
let regRet = /喵喵面板(?:设置)?\s*(好友|群|群聊|陌生人)?\s*(\d*)\s*(开启|关闭|删除)?\s*$/.exec(e.msg)
|
||
|
||
if (!regRet) {
|
||
return false
|
||
}
|
||
|
||
let [, target, groupId, actionType] = regRet
|
||
if (target === '群聊') {
|
||
target = '群'
|
||
}
|
||
|
||
if (target) {
|
||
if (groupId && (target === '群' || !target)) {
|
||
if (actionType === '删除') {
|
||
Cfg.del(`profile.groups.群${groupId}`)
|
||
} else {
|
||
Cfg.set(`profile.groups.群${groupId}.status`, actionType !== '关闭')
|
||
}
|
||
} else {
|
||
Cfg.set(`profile.${keyMap[target]}.status`, actionType !== '关闭')
|
||
}
|
||
}
|
||
|
||
let cfg = {
|
||
groups: []
|
||
}
|
||
|
||
lodash.forEach(['friend', 'group', 'stranger'], (key) => {
|
||
cfg[key] = getStatus(`profile.${key}.status`, true)
|
||
})
|
||
|
||
let groups = Cfg.get('profile.groups', {})
|
||
lodash.forEach(lodash.keys(groups), (group, idx) => {
|
||
if (lodash.isUndefined(groups[group])) {
|
||
return true
|
||
}
|
||
cfg.groups.push({
|
||
group,
|
||
idx: idx + 1,
|
||
status: getStatus(`profile.groups.${group}.status`, true)
|
||
})
|
||
})
|
||
|
||
// 渲染图像
|
||
return await Common.render('admin/profile', {
|
||
...cfg
|
||
}, { e, scale: 1.4 })
|
||
}
|