-
Notifications
You must be signed in to change notification settings - Fork 145
/
index.js
79 lines (66 loc) · 2.13 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
import yargs from 'yargs';
import cfonts from 'cfonts';
import { fileURLToPath } from 'url';
import { join, dirname } from 'path';
import { createRequire } from 'module';
import { createInterface } from 'readline';
import { setupMaster, fork } from 'cluster';
import { watchFile, unwatchFile } from 'fs';
// Setup console output
const { say } = cfonts;
const rl = createInterface(process.stdin, process.stdout);
const __dirname = dirname(fileURLToPath(import.meta.url));
const require = createRequire(__dirname);
const { name, author } = require(join(__dirname, './package.json'));
say('Lightweight\nWhatsApp Bot', { font: 'chrome', align: 'center', gradient: ['red', 'magenta'] });
say(`'${name}' By @${author.name || author}`, { font: 'console', align: 'center', gradient: ['red', 'magenta'] });
console.log('🐾 Starting...');
var isRunning = false;
/**
* Start a js file
* @param {String} file `path/to/file`
*/
function start(file) {
if (isRunning) return;
isRunning = true;
let args = [join(__dirname, file), ...process.argv.slice(2)];
say([process.argv[0], ...args].join(' '), { font: 'console', align: 'center', gradient: ['red', 'magenta'] });
setupMaster({ exec: args[0], args: args.slice(1) });
let p = fork();
p.on('message', data => {
console.log('[✅RECEIVED]', data);
switch (data) {
case 'reset':
p.kill(); // Change here
isRunning = false;
start(file);
break;
case 'uptime':
p.send(process.uptime());
break;
default:
console.warn('[⚠️ UNRECOGNIZED MESSAGE]', data);
}
});
p.on('exit', (_, code) => {
isRunning = false;
console.error('[❗] Exited with code:', code);
if (code !== 0) {
console.log('[🔄 Restarting worker due to non-zero exit code...');
return start(file);
}
watchFile(args[0], () => {
unwatchFile(args[0]);
start(file);
});
});
let opts = yargs(process.argv.slice(2)).exitProcess(false).parse();
if (!opts['test']) {
if (!rl.listenerCount()) {
rl.on('line', line => {
p.emit('message', line.trim());
});
}
}
}
start('main.js');