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 `
已开启
`;
} else {
return `已关闭
`;
}
}
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 });
}