From 040205045fe27723e756e766310ab516aeca6370 Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Thu, 24 Mar 2022 20:04:08 +0800
Subject: [PATCH] core: implement postRestart

---
 packages/hydrooj/bin/commands.ts       | 9 +++++----
 packages/hydrooj/package.json          | 2 +-
 packages/hydrooj/src/handler/manage.ts | 9 ++++++++-
 3 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/packages/hydrooj/bin/commands.ts b/packages/hydrooj/bin/commands.ts
index 8e4160282..6caf16626 100644
--- a/packages/hydrooj/bin/commands.ts
+++ b/packages/hydrooj/bin/commands.ts
@@ -40,8 +40,9 @@ if (argv._[0] === 'backup') {
     const url = buildUrl(JSON.parse(dbConfig));
     const dir = `${os.tmpdir()}/${Math.random().toString(36).substring(2)}`;
     exec('mongodump', [url, `--out=${dir}/dump`], { stdio: 'inherit' });
-    const env = `${os.homedir()}/.hydro/env`;
-    if (fs.existsSync(env)) fs.copySync(env, `${dir}/env`);
+    // Do not backup .env & config.json as they are all related to local installation
+    const env = `${os.homedir()}/.hydro/addon.json`;
+    if (fs.existsSync(env)) fs.copySync(env, `${dir}/addon.json`);
     const target = `${process.cwd()}/backup-${new Date().toISOString().replace(':', '-').split(':')[0]}.zip`;
     exec('zip', ['-r', target, 'dump'], { cwd: dir, stdio: 'inherit' });
     if (!argv.dbOnly) {
@@ -66,8 +67,8 @@ if (argv._[0] === 'restore') {
         exec('rm', ['-rf', '/data/file/*'], { stdio: 'inherit' });
         exec('bash', ['-c', `mv ${dir}/file/* /data/file`], { stdio: 'inherit' });
     }
-    if (fs.existsSync(`${dir}/env`)) {
-        fs.copySync(`${dir}/env`, `${os.homedir()}/.hydro/env`, { overwrite: true });
+    if (fs.existsSync(`${dir}/addon.json`)) {
+        fs.copySync(`${dir}/addon.json`, `${os.homedir()}/.hydro/addon.json`, { overwrite: true });
     }
     fs.removeSync(dir);
     console.log('Successfully restored.');
diff --git a/packages/hydrooj/package.json b/packages/hydrooj/package.json
index 46ab89ea7..44d41ef57 100644
--- a/packages/hydrooj/package.json
+++ b/packages/hydrooj/package.json
@@ -1,6 +1,6 @@
 {
     "name": "hydrooj",
-    "version": "3.10.5",
+    "version": "3.10.6",
     "bin": "bin/hydrooj.js",
     "main": "src/loader",
     "module": "src/loader",
diff --git a/packages/hydrooj/src/handler/manage.ts b/packages/hydrooj/src/handler/manage.ts
index fcc37d615..fc3df258d 100644
--- a/packages/hydrooj/src/handler/manage.ts
+++ b/packages/hydrooj/src/handler/manage.ts
@@ -1,8 +1,9 @@
+import { exec } from 'child_process';
 import { inspect } from 'util';
 import Ajv from 'ajv';
 import * as yaml from 'js-yaml';
 import * as check from '../check';
-import { ValidationError } from '../error';
+import { BadRequestError, ValidationError } from '../error';
 import {
     isEmail, isPassword, isUname, validate,
 } from '../lib/validator';
@@ -86,6 +87,12 @@ class SystemDashboardHandler extends SystemHandler {
     async get() {
         this.response.template = 'manage_dashboard.html';
     }
+
+    async postRestart() {
+        if (!process.env.pm_cwd) throw new BadRequestError('Not launched by pm2');
+        exec(`pm2 reload "${process.env.name}"`);
+        this.back();
+    }
 }
 
 class SystemScriptHandler extends SystemHandler {