miao-plugin/apps/admin.js

import fs from 'fs'
import lodash from 'lodash'
import { exec } from 'child_process'
import { Cfg } from '../components/index.js'
import Common from '../components/Common.js'

let cfgMap = {
  角色: 'char.char',
  面板: 'char.profile-data',
  老婆: 'char.wife',
  戳一戳: 'char.poke',
  小清新: 'char.se',
  查他人: 'char.queryOther',
  图鉴: 'wiki.wiki',
  图片: 'wiki.pic',
  统计: 'wiki.stat',
  深渊: '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'),
    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, 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) {
      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/"`
    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 })
}