115 lines
3.6 KiB
JavaScript
115 lines
3.6 KiB
JavaScript
const logger = require('./logger.js');
|
|
const { spawn } = require('child_process');
|
|
|
|
const cmds = new Map();
|
|
|
|
async function execute(endpoint) {
|
|
if (endpoint === undefined) {
|
|
return;
|
|
}
|
|
let unique = endpoint.options?.unique;
|
|
if (unique !== undefined && isCommandActive(endpoint)) {
|
|
unique = unique.toLowerCase();
|
|
switch (unique.toLowerCase()) {
|
|
case 'true':
|
|
logger.info('not executing unique command \'' + endpoint.command + '\' (args: \'' + endpoint.args + '\') because it is already active');
|
|
throw new Error('command is already active');
|
|
case 'restart':
|
|
logger.info('killing and restarting unique command \'' + endpoint.command + '\' (args: \'' + endpoint.args + '\')');
|
|
await killCommand(endpoint);
|
|
break;
|
|
case 'toggle':
|
|
logger.info('stopping unique command \'' + endpoint.command + '\' (args: \'' + endpoint.args + '\')');
|
|
return await killCommand(endpoint);
|
|
}
|
|
}
|
|
return new Promise((resolve, reject) => {
|
|
logger.info('executing command \'' + endpoint.command + '\' (args: \'' + endpoint.args + '\')...');
|
|
var cmd = spawn(endpoint.command, endpoint.args);
|
|
cmd.timestamp = new Date().getTime();
|
|
let result = '';
|
|
let error = '';
|
|
cmd.stdout.on('data', (data) => {
|
|
result += data;
|
|
});
|
|
cmd.stderr.on('data', (data) => {
|
|
error += data;
|
|
});
|
|
cmd.on('spawn', () => {
|
|
logger.info('spawned command \'' + endpoint.command + '\' (args: \'' + endpoint.args + '\')');
|
|
addCommand(cmd, endpoint);
|
|
if (endpoint.options?.detach) {
|
|
resolve();
|
|
}
|
|
});
|
|
cmd.on('error', (err) => {
|
|
error += err;
|
|
removeCommand(endpoint);
|
|
if (endpoint.options?.detach) {
|
|
reject();
|
|
}
|
|
});
|
|
cmd.on('exit', (code) => {
|
|
if (code === null) {
|
|
code = 0;
|
|
}
|
|
removeCommand(endpoint);
|
|
let fn = logger.info;
|
|
let msg = 'command \'' + endpoint.command + '\' (args: \'' + endpoint.args + '\') finished with exit code ' + code + ' after ' + (new Date().getTime() - cmd.timestamp) + 'ms';
|
|
if (error !== undefined && error.length > 0) {
|
|
error = error.trim();
|
|
msg += ' > error: ' + error;
|
|
fn = logger.error;
|
|
reject(error);
|
|
}
|
|
if (result !== undefined && result.length > 0) {
|
|
result = result.trim();
|
|
msg += ' > data: ' + result;
|
|
}
|
|
fn(msg);
|
|
resolve(result);
|
|
});
|
|
});
|
|
}
|
|
|
|
function addCommand(command, endpoint) {
|
|
if (command === undefined || endpoint === undefined) {
|
|
return;
|
|
}
|
|
cmds.set(endpoint, command);
|
|
}
|
|
|
|
function removeCommand(endpoint) {
|
|
if (endpoint === undefined) {
|
|
return;
|
|
}
|
|
cmds.delete(endpoint);
|
|
}
|
|
|
|
function isCommandActive(endpoint) {
|
|
return endpoint !== undefined && cmds.has(endpoint);
|
|
}
|
|
|
|
async function killCommand(endpoint) {
|
|
if (endpoint === undefined) {
|
|
return;
|
|
}
|
|
const command = cmds.get(endpoint);
|
|
if (command === undefined) {
|
|
return;
|
|
}
|
|
process.kill(command.pid, 'SIGINT');
|
|
while (isCommandActive(endpoint)) {
|
|
await sleep(1);
|
|
}
|
|
}
|
|
|
|
async function sleep(milliseconds) {
|
|
return new Promise((resolve, reject) => {
|
|
setTimeout(resolve, milliseconds);
|
|
});
|
|
}
|
|
|
|
module.exports = {
|
|
execute
|
|
} |