Pre Merge pull request !163 from 千奈千祁/N/A

ksr.js

@@ -0,0 +1,51 @@
+import { spawn } from 'child_process';
+import log4js from 'log4js';
+import http from 'http'
+
+/* keep ssh run */
+
+log4js.configure({
+  appenders: { console: { type: 'console' } },
+  categories: { default: { appenders: ['console'], level: 'debug' } }
+});
+const logger = log4js.getLogger('app');
+
+let serverProcess;
+const startServer = async () => {
+  logger.info('Starting Bot...');
+  serverProcess = spawn('node', ['app.js'], { stdio: 'inherit' });
+  serverProcess.on('close', (code) => {
+    logger.info(`Bot process exited with code ${code}`);
+    if (code == null) return
+    process.exit()
+  });
+};
+startServer();
+
+const serverHttpexit = http.createServer(async (req, res) => {
+  let remoteIP = req.socket.remoteAddress;
+  if (remoteIP.startsWith('::ffff:')) {
+    remoteIP = remoteIP.slice(7);
+  }
+  if (remoteIP !== `::1` && remoteIP !== `127.0.0.1`) {
+    console.log(remoteIP)
+    res.writeHead(403, { 'Content-Type': 'text/plain' });
+    res.end('Access Forbidden\n');
+    return
+  }
+  if (req.url === `/restart`) {
+    await serverProcess.kill();
+    await startServer();
+    res.writeHead(200, { 'Content-Type': 'text/plain' });
+    res.end('OK\n');
+  } else if (req.url === `/exit`) {
+    res.writeHead(200, { 'Content-Type': 'text/plain' });
+    res.end('OK\n');
+    process.exit()
+  } else {
+    res.writeHead(404, { 'Content-Type': 'text/plain' });
+    res.end('Not Found\n');
+  }
+})
+
+serverHttpexit.listen(27881, () => { });

plugins/other/restart.js

@@ -1,9 +1,20 @@
 import plugin from '../../lib/plugins/plugin.js'
 import { createRequire } from 'module'
+import fetch from 'node-fetch'
+import net from 'net'
 
 const require = createRequire(import.meta.url)
 const { exec } = require('child_process')
 
+const isPortTaken = async (port) => {
+  return new Promise((resolve) => {
+    const tester = net.createServer()
+      .once('error', () => resolve(true))
+      .once('listening', () => tester.once('close', () => resolve(false)).close())
+      .listen(port);
+  });
+};
+
 export class Restart extends plugin {
   constructor (e = '') {
     super({
@@ -62,9 +73,22 @@ export class Restart extends plugin {
     })
 
     let npm = await this.checkPnpm()
-
-    try {
     await redis.set(this.key, data, { EX: 120 })
+    if(await isPortTaken(27881)) {
+      try {
+        let result = await fetch(`http://localhost:27881/restart`)
+        result = await result.text()
+        if(result !== `OK`) {
+          redis.del(this.key)
+          this.e.reply(`操作失败！`)
+          logger.error(`重启失败`)
+        }
+      } catch(error) {
+        redis.del(this.key)
+        this.e.reply(`操作失败！\n${error}`)
+      }
+    } else {
+      try {
         let cm = `${npm} start`
         if (process.argv[1].includes('pm2')) {
           cm = `${npm} run restart`
@@ -87,6 +111,7 @@ export class Restart extends plugin {
         let e = error.stack ?? error
         this.e.reply(`操作失败！\n${e}`)
       }
+    }
 
     return true
   }
@@ -107,6 +132,17 @@ export class Restart extends plugin {
   }
 
   async stop () {
+    if(await isPortTaken(27881)) {
+      try {
+        logger.mark('关机成功，已停止运行')
+        await this.e.reply(`关机成功，已停止运行`)
+        await fetch(`http://localhost:27881/exit`)
+      } catch(error) {
+        this.e.reply(`操作失败！\n${error}`)
+        logger.error(`关机失败\n${error}`)
+      }
+    }
+
     if (!process.argv[1].includes('pm2')) {
       logger.mark('关机成功，已停止运行')
       await this.e.reply('关机成功，已停止运行')