import plugin from '../../lib/plugins/plugin.js' import { createRequire } from 'module' const require = createRequire(import.meta.url) const { exec } = require('child_process') export class Restart extends plugin { constructor (e = '') { super({ name: '重启', dsc: '#重启', event: 'message', priority: 10, rule: [ { reg: '^#重启$', fnc: 'restart', permission: 'master' }, { reg: '^#(停机|关机)$', fnc: 'stop', permission: 'master' } ] }) if (e) this.e = e this.key = 'Yz:restart' } async init () { let restart = await redis.get(this.key) if (restart && process.argv[1].includes('pm2')) { restart = JSON.parse(restart) let time = restart.time || new Date().getTime() time = (new Date().getTime() - time) / 1000 let msg = `重启成功:耗时${time.toFixed(2)}秒` if (restart.isGroup) { Bot.pickGroup(restart.id).sendMsg(msg) } else { Bot.pickUser(restart.id).sendMsg(msg) } redis.del(this.key) } } async restart () { await this.e.reply('开始执行重启,请稍等...') logger.mark(`${this.e.logFnc} 开始执行重启,请稍等...`) let data = JSON.stringify({ isGroup: !!this.e.isGroup, id: this.e.isGroup ? this.e.group_id : this.e.user_id, time: new Date().getTime() }) let npm = await this.checkPnpm() try { await redis.set(this.key, data, { EX: 120 }) let cm = `${npm} start` if (process.argv[1].includes('pm2')) { cm = `${npm} run restart` } else { await this.e.reply('当前为前台运行,重启将转为后台...') } exec(cm, { windowsHide: true }, (error, stdout, stderr) => { if (error) { redis.del(this.key) this.e.reply(`操作失败!\n${error.stack}`) logger.error(`重启失败\n${error.stack}`) } else if (stdout) { logger.mark('重启成功,运行已由前台转为后台') logger.mark(`查看日志请用命令:${npm} run log`) logger.mark(`停止后台运行命令:${npm} stop`) process.exit() } }) } catch (error) { redis.del(this.key) let e = error.stack ?? error this.e.reply(`操作失败!\n${e}`) } return true } async checkPnpm () { let npm = 'npm' let ret = await this.execSync('pnpm -v') if (ret.stdout) npm = 'pnpm' return npm } async execSync (cmd) { return new Promise((resolve, reject) => { exec(cmd, { windowsHide: true }, (error, stdout, stderr) => { resolve({ error, stdout, stderr }) }) }) } async stop () { if (!process.argv[1].includes('pm2')) { logger.mark('关机成功,已停止运行') await this.e.reply('关机成功,已停止运行') process.exit() } logger.mark('关机成功,已停止运行') await this.e.reply('关机成功,已停止运行') let npm = await this.checkPnpm() exec(`${npm} stop`, { windowsHide: true }, (error, stdout, stderr) => { if (error) { this.e.reply(`操作失败!\n${error.stack}`) logger.error(`关机失败\n${error.stack}`) } }) } }