core: pm2 cluster management

packages/hydrooj/bin/hydrooj.js

@@ -5,12 +5,12 @@
 require('source-map-support').install();
 const os = require('os');
 const path = require('path');
-const cluster = require('cluster');
 const fs = require('fs-extra');
 const argv = require('cac')().parse();
 const child = require('child_process');
 const esbuild = require('esbuild');
 
+if (!process.env.NODE_APP_INSTANCE) process.env.NODE_APP_INSTANCE = '0';
 const major = +process.version.split('.')[0].split('v')[1];
 const minor = +process.version.split('.')[1];
 
@@ -55,75 +55,66 @@ function buildUrl(opts) {
     return mongourl;
 }
 
-if (!cluster.isMaster) {
+const hydroPath = path.resolve(os.homedir(), '.hydro');
+fs.ensureDirSync(hydroPath);
+const addonPath = path.resolve(hydroPath, 'addon.json');
+if (!fs.existsSync(addonPath)) fs.writeFileSync(addonPath, '[]');
+let addons = JSON.parse(fs.readFileSync(addonPath).toString());
+
+if (argv.args[0] === 'db') {
+    const dbConfig = fs.readFileSync(path.resolve(hydroPath, 'config.json'), 'utf-8');
+    const url = buildUrl(JSON.parse(dbConfig));
+    return child.spawn('mongo', [url], { stdio: 'inherit' });
+}
+
+if (!addons.includes('@hydrooj/ui-default')) {
+    try {
+        const ui = argv.options.ui || '@hydrooj/ui-default';
+        require.resolve(ui);
+        addons.push(ui);
+    } catch (e) {
+        console.error('Please also install @hydrooj/ui-default');
+    }
+}
+
+if (argv.args[0] && argv.args[0] !== 'cli') {
+    const operation = argv.args[0];
+    const arg1 = argv.args[1];
+    const arg2 = argv.args[2];
+    if (operation === 'addon') {
+        if (arg1 === 'create') {
+            fs.mkdirSync('/root/addon');
+            child.execSync('yarn init -y', { cwd: '/root/addon' });
+            fs.mkdirSync('/root/addon/templates');
+            fs.mkdirSync('/root/addon/locales');
+            fs.mkdirSync('/root/addon/public');
+            addons.push('/root/addon');
+        } else if (arg1 === 'add') {
+            for (let i = 0; i < addons.length; i++) {
+                if (addons[i] === arg2) {
+                    addons.splice(i, 1);
+                    break;
+                }
+            }
+            addons.push(arg2);
+        } else if (arg1 === 'remove') {
+            for (let i = 0; i < addons.length; i++) {
+                if (addons[i] === arg2) {
+                    addons.splice(i, 1);
+                    break;
+                }
+            }
+        }
+        addons = Array.from(new Set(addons));
+        console.log('Current Addons: ', addons);
+        fs.writeFileSync(addonPath, JSON.stringify(addons, null, 2));
+    }
+} else {
     const hydro = require('../src/loader');
-    // Forked by hydro
-    hydro.load().catch((e) => {
+    addons = Array.from(new Set(addons));
+    for (const addon of addons) hydro.addon(addon);
+    (argv.args[0] === 'cli' ? hydro.loadCli : hydro.load)().catch((e) => {
         console.error(e);
         process.exit(1);
     });
-} else {
-    const hydroPath = path.resolve(os.homedir(), '.hydro');
-    fs.ensureDirSync(hydroPath);
-    const addonPath = path.resolve(hydroPath, 'addon.json');
-    if (!fs.existsSync(addonPath)) fs.writeFileSync(addonPath, '[]');
-    let addons = JSON.parse(fs.readFileSync(addonPath).toString());
-
-    if (argv.args[0] === 'db') {
-        const dbConfig = fs.readFileSync(path.resolve(hydroPath, 'config.json'), 'utf-8');
-        const url = buildUrl(JSON.parse(dbConfig));
-        return child.spawn('mongo', [url], { stdio: 'inherit' });
-    }
-
-    if (!addons.includes('@hydrooj/ui-default')) {
-        try {
-            const ui = argv.options.ui || '@hydrooj/ui-default';
-            require.resolve(ui);
-            addons.push(ui);
-        } catch (e) {
-            console.error('Please also install @hydrooj/ui-default');
-        }
-    }
-
-    if (argv.args[0] && argv.args[0] !== 'cli') {
-        const operation = argv.args[0];
-        const arg1 = argv.args[1];
-        const arg2 = argv.args[2];
-        if (operation === 'addon') {
-            if (arg1 === 'create') {
-                fs.mkdirSync('/root/addon');
-                child.execSync('yarn init -y', { cwd: '/root/addon' });
-                fs.mkdirSync('/root/addon/templates');
-                fs.mkdirSync('/root/addon/locales');
-                fs.mkdirSync('/root/addon/public');
-                addons.push('/root/addon');
-            } else if (arg1 === 'add') {
-                for (let i = 0; i < addons.length; i++) {
-                    if (addons[i] === arg2) {
-                        addons.splice(i, 1);
-                        break;
-                    }
-                }
-                addons.push(arg2);
-            } else if (arg1 === 'remove') {
-                for (let i = 0; i < addons.length; i++) {
-                    if (addons[i] === arg2) {
-                        addons.splice(i, 1);
-                        break;
-                    }
-                }
-            }
-            addons = Array.from(new Set(addons));
-            console.log('Current Addons: ', addons);
-            fs.writeFileSync(addonPath, JSON.stringify(addons, null, 2));
-        }
-    } else {
-        const hydro = require('../src/loader');
-        addons = Array.from(new Set(addons));
-        for (const addon of addons) hydro.addon(addon);
-        (argv.args[0] === 'cli' ? hydro.loadCli : hydro.load)().catch((e) => {
-            console.error(e);
-            process.exit(1);
-        });
-    }
 }