forked from smogon/pokemon-showdown
-
Notifications
You must be signed in to change notification settings - Fork 0
/
repl.js
89 lines (77 loc) · 2.19 KB
/
repl.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
80
81
82
83
84
85
86
87
88
89
'use strict';
const REPL_ENABLED = false;
const fs = require('fs');
const path = require('path');
const net = require('net');
let sockets = [];
function cleanup() {
for (let s = 0; s < sockets.length; ++s) {
try {
fs.unlinkSync(sockets[s]);
} catch (e) {}
}
}
process.on("exit", cleanup);
// exit handlers aren't called by the default handler
if (process.listeners('SIGHUP').length === 0) {
process.on('SIGHUP', () => process.exit(128 + 1));
}
if (process.listeners('SIGINT').length === 0) {
process.on('SIGINT', () => process.exit(128 + 2));
}
// The eval function is passed in because there is no other way to access a file's non-global context
exports.start = function (prefix, suffix, evalFunction) {
if (!REPL_ENABLED) return;
if (process.platform === 'win32') return; // Windows doesn't support sockets mounted in the filesystem
let resolvedPrefix = path.resolve(__dirname, Config.replsocketprefix || 'logs/repl', prefix);
if (!evalFunction) {
evalFunction = suffix;
suffix = "";
}
let name = resolvedPrefix + suffix;
if (prefix === 'app') {
// Clear out any old sockets
let directory = path.dirname(resolvedPrefix);
for (let file of fs.readdirSync(directory)) {
let stat = fs.statSync(directory + '/' + file);
if (!stat.isSocket()) {
continue;
}
let socket = net.connect(directory + '/' + file, () => {
socket.end();
socket.destroy();
}).on('error', () => {
fs.unlink(directory + '/' + file, () => {});
});
}
}
net.createServer(socket => {
require('repl').start({
input: socket,
output: socket,
eval: (cmd, context, filename, callback) => {
try {
callback(null, evalFunction(cmd));
} catch (e) {
callback(e);
}
},
}).on('exit', () => socket.end());
socket.on('error', () => socket.destroy());
}).listen(name, () => {
fs.chmodSync(name, Config.replsocketmode || 0o600);
sockets.push(name);
}).on('error', e => {
if (e.code === "EADDRINUSE") {
fs.unlink(name, e => {
if (e && e.code !== "ENOENT") {
require('./crashlogger.js')(e, 'REPL: ' + name);
return;
}
exports.start(prefix, suffix, evalFunction);
});
return;
}
require('./crashlogger.js')(e, 'REPL: ' + name);
});
};