90 lines
2.0 KiB
JavaScript
90 lines
2.0 KiB
JavaScript
const fs = require('fs');
|
|
const path = require('path');
|
|
const { spawn } = require('child_process');
|
|
const express = require('express');
|
|
const app = express();
|
|
|
|
const shPath = './sh.d';
|
|
|
|
class Runner {
|
|
constructor() {
|
|
this.files = fs
|
|
.readdirSync(shPath)
|
|
.filter(v => /.sh$/.test(v))
|
|
.map(v => path.join(shPath, v));
|
|
console.log('loaded scripts:', this.files);
|
|
this.currentTarget = null;
|
|
}
|
|
psleep(ms) {
|
|
return new Promise(resolve => {
|
|
setTimeout(resolve, ms);
|
|
});
|
|
}
|
|
kill(sig = 'SIGINT') {
|
|
try {
|
|
process.kill(-this.currentTarget.pid, sig || 'SIGINT');
|
|
} catch (e) {
|
|
console.error(e);
|
|
}
|
|
}
|
|
spawn(target) {
|
|
return new Promise(resolve => {
|
|
const child = spawn(target, {
|
|
sid: true,
|
|
detached: true,
|
|
});
|
|
const tag = `${target}:`;
|
|
child.stdout.on('data', data => {
|
|
console.log(tag, data.toString().trim());
|
|
});
|
|
child.stderr.on('data', data => {
|
|
console.log(tag, data.toString().trim());
|
|
});
|
|
child.on('close', code => {
|
|
console.log(tag, 'close');
|
|
resolve(code);
|
|
});
|
|
child.on('exit', code => {
|
|
console.log(tag, 'exit');
|
|
this.kill();
|
|
resolve(code);
|
|
});
|
|
this.currentTarget = child;
|
|
});
|
|
}
|
|
async runner() {
|
|
let counter = 0;
|
|
while (true) {
|
|
const target = this.files[counter++ % this.files.length];
|
|
console.log('running', target);
|
|
await this.spawn(target).catch(err => {
|
|
console.error(err);
|
|
});
|
|
}
|
|
}
|
|
start() {
|
|
this.runner();
|
|
}
|
|
}
|
|
|
|
const runner = new Runner();
|
|
runner.start();
|
|
|
|
const onExit = () => {
|
|
console.log('exitting');
|
|
runner.kill();
|
|
process.exit(0);
|
|
};
|
|
|
|
process.on('SIGTERM', onExit);
|
|
process.on('SIGINT', onExit);
|
|
|
|
app.get('/', (req, res) => {
|
|
const sig = req.query.force ? 'SIGKILL' : null;
|
|
runner.kill(sig);
|
|
res.sendStatus(200);
|
|
});
|
|
|
|
const port = process.env.PORT || 5501;
|
|
app.listen(port, 'localhost', () => console.log('listening on', port));
|