miao-plugin/apps/admin.js

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.poke",
  "小清新": "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"),
    poke: getStatus("char.poke"),
    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 });

}