piston/cli/commands/execute.js

98 lines
2.5 KiB
JavaScript
Raw Normal View History

2021-03-13 04:44:33 +01:00
//const fetch = require('node-fetch');
2021-02-22 11:38:11 +01:00
const fs = require('fs');
const path = require('path');
const chalk = require('chalk');
2021-04-23 01:59:54 +02:00
exports.command = ['execute <language> <file> [args..]']
2021-02-22 11:38:11 +01:00
exports.aliases = ['run']
exports.describe = 'Executes file with the specified runner'
exports.builder = {
2021-04-23 01:59:54 +02:00
languageVersion: {
string: true,
desc: 'Set the version of the language to use',
alias: ['l'],
default: '*'
},
2021-02-22 11:38:11 +01:00
stdin: {
boolean: true,
desc: 'Read input from stdin and pass to executor',
alias: ['i']
},
run_timeout: {
alias: ['rt', 'r'],
number: true,
desc: 'Milliseconds before killing run process',
default: 3000
},
compile_timeout: {
alias: ['ct', 'c'],
number: true,
desc: 'Milliseconds before killing compile process',
default: 10000,
},
files: {
alias: ['f'],
array: true,
desc: 'Additional files to add',
}
}
exports.handler = async function(argv){
const files = [...(argv.files || []),argv.file]
.map(file_path => ({
name: path.basename(file_path),
content: fs.readFileSync(file_path).toString()
}));
const stdin = (argv.stdin && await new Promise((resolve, _)=>{
var data = "";
process.stdin.on('data', d=> data += d)
process.stdin.on('end', _ => resolve(data))
})) || "";
2021-03-13 04:08:35 +01:00
const request = {
2021-03-05 07:40:47 +01:00
language: argv.language,
version: argv['language-version'],
files: files,
2021-03-05 12:44:35 +01:00
args: argv.args,
2021-02-22 11:38:11 +01:00
stdin,
2021-03-05 07:40:47 +01:00
compile_timeout: argv.ct,
run_timeout: argv.rt
2021-03-13 04:08:35 +01:00
};
2021-03-13 04:44:33 +01:00
let response = await argv.axios.post('/jobs', request);
2021-03-13 11:09:30 +01:00
response = response.data
2021-02-22 11:38:11 +01:00
function step(name, ctx){
console.log(chalk.bold(`== ${name} ==`))
if(ctx.stdout){
console.log(" ",chalk.bold(`STDOUT`))
console.log(" ",ctx.stdout.replace(/\n/g,'\n '))
}
if(ctx.stderr){
console.log(chalk.bold(`STDERR`))
console.log(" ",ctx.stderr.replace(/\n/g,'\n '))
}
if(ctx.code)
console.log(
chalk.bold(`Exit Code:`),
chalk.bold[ctx.code > 0 ? 'red' : 'green'](ctx.code)
)
if(ctx.signal)
console.log(
chalk.bold(`Signal:`),
chalk.bold.yellow(ctx.signal)
)
}
2021-03-14 07:36:12 +01:00
2021-02-22 11:38:11 +01:00
if(response.compile) step('Compile', response.compile)
step('Run', response.run)
}