mirror of
https://github.com/yoimiya-kokomi/miao-plugin.git
synced 2024-11-25 09:27:34 +00:00
267 lines
7.2 KiB
JavaScript
267 lines
7.2 KiB
JavaScript
import { segment } from "oicq";
|
||
import fs from "fs";
|
||
import lodash from "lodash";
|
||
import { createRequire } from "module";
|
||
import { exec } from "child_process";
|
||
import { Cfg } from "../components/index.js";
|
||
import Common from "../components/Common.js";
|
||
|
||
|
||
const require = createRequire(import.meta.url);
|
||
|
||
let cfgMap = {
|
||
"角色": "char.char",
|
||
"面板": "char.profile-data",
|
||
"老婆": "char.wife",
|
||
"小清新": "char.se",
|
||
"查他人": "char.queryOther",
|
||
"图鉴": "wiki.wiki",
|
||
"图片": "wiki.pic",
|
||
"深渊": "wiki.abyss",
|
||
"渲染": "sys.scale",
|
||
"帮助": "sys.help",
|
||
};
|
||
let sysCfgReg = `^#喵喵设置\s*(${lodash.keys(cfgMap).join("|")})?\s*(.*)$`;
|
||
export const rule = {
|
||
updateRes: {
|
||
hashMark: true,
|
||
reg: "^#喵喵更新图像$",
|
||
describe: "【#管理】更新素材",
|
||
},
|
||
updateMiaoPlugin: {
|
||
hashMark: true,
|
||
reg: "^#喵喵(强制)?更新",
|
||
describe: "【#管理】喵喵更新",
|
||
},
|
||
sysCfg: {
|
||
hashMark: true,
|
||
reg: sysCfgReg,
|
||
describe: "【#管理】系统设置"
|
||
},
|
||
profileCfg: {
|
||
hashMark: true,
|
||
reg: "^#喵喵面板(?:设置)?.*",
|
||
describe: "【#管理】面板设置"
|
||
}
|
||
};
|
||
|
||
|
||
const _path = process.cwd();
|
||
const resPath = `${_path}/plugins/miao-plugin/resources/`;
|
||
const plusPath = `${resPath}/miao-res-plus/`;
|
||
|
||
const checkAuth = async function (e) {
|
||
return await e.checkAuth({
|
||
auth: "master",
|
||
replyMsg: `只有主人才能命令喵喵哦~
|
||
(*/ω＼*)`
|
||
});
|
||
}
|
||
|
||
export async function sysCfg(e, { render }) {
|
||
if (!await checkAuth(e)) {
|
||
return true;
|
||
}
|
||
|
||
let cfgReg = new RegExp(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-data"),
|
||
wife: getStatus("char.wife"),
|
||
se: getStatus("char.se", false),
|
||
other: getStatus("char.queryOther"),
|
||
wiki: getStatus("wiki.wiki"),
|
||
pic: getStatus("wiki.pic"),
|
||
abyss: getStatus("wiki.abyss"),
|
||
imgPlus: fs.existsSync(plusPath),
|
||
help: getStatus("sys.help", false),
|
||
scale: Cfg.get("sys.scale", 100)
|
||
}
|
||
|
||
//渲染图像
|
||
return await Common.render("admin/index", {
|
||
...cfg,
|
||
}, { e, render, 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>`;
|
||
}
|
||
}
|
||
|
||
export async function updateRes(e) {
|
||
if (!await checkAuth(e)) {
|
||
return true;
|
||
}
|
||
let command = "";
|
||
if (fs.existsSync(`${resPath}/miao-res-plus/`)) {
|
||
e.reply("开始尝试更新，请耐心等待~");
|
||
command = `git pull`;
|
||
exec(command, { cwd: `${resPath}/miao-res-plus/` }, function (error, stdout, stderr) {
|
||
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/"`;
|
||
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;
|
||
|
||
export 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);
|
||
redis.set("miao:restart-msg", JSON.stringify({
|
||
msg: "重启成功，新版喵喵已经生效",
|
||
qq: e.user_id
|
||
}), { EX: 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;
|
||
}
|
||
|
||
export async function profileCfg(e, { render }) {
|
||
if (!await checkAuth(e)) {
|
||
return true;
|
||
}
|
||
|
||
let keyMap = {
|
||
"好友": "friend",
|
||
"群": "group",
|
||
"陌生人": "stranger"
|
||
}
|
||
|
||
let regRet = /喵喵面板(?:设置)?\s*(好友|群|群聊|陌生人)?\s*(\d*)\s*(开启|关闭|删除)?\s*$/.exec(e.msg);
|
||
|
||
if (!regRet) {
|
||
return;
|
||
}
|
||
|
||
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-data.groups', {});
|
||
lodash.forEach(lodash.keys(groups), (group, idx) => {
|
||
|
||
if (lodash.isUndefined(groups[group])) {
|
||
return;
|
||
}
|
||
cfg.groups.push({
|
||
group,
|
||
idx: idx + 1,
|
||
status: getStatus(`profile.groups.${group}.status`, true)
|
||
})
|
||
})
|
||
|
||
//渲染图像
|
||
return await Common.render("admin/profile", {
|
||
...cfg,
|
||
}, { e, render, scale: 1.4 });
|
||
|
||
}
|