piston/api/src/api/v2.js

225 lines
5.7 KiB
JavaScript
Raw Normal View History

2021-04-25 10:55:20 +02:00
const express = require('express');
2021-05-08 02:20:21 +02:00
const router = express.Router();
2021-04-25 10:55:20 +02:00
const config = require('../config');
2021-04-25 10:55:20 +02:00
const runtime = require('../runtime');
const { Job } = require('../job');
2021-05-08 02:20:21 +02:00
const package = require('../package');
2021-05-07 21:35:27 +02:00
const logger = require('logplease').create('api/v2');
2021-04-25 10:55:20 +02:00
router.use((req, res, next) => {
2021-05-08 02:30:40 +02:00
if (['GET', 'HEAD', 'OPTIONS'].includes(req.method)) {
return next();
}
2021-05-08 02:30:40 +02:00
if (!req.headers['content-type'].startsWith('application/json')) {
return res.status(415).send({
message: 'requests must be of type application/json',
});
}
2021-05-08 02:30:40 +02:00
next();
});
router.post('/execute', async (req, res) => {
2021-05-08 02:30:40 +02:00
const {
language,
version,
files,
stdin,
args,
run_timeout,
compile_timeout,
compile_memory_limit,
run_memory_limit,
} = req.body;
if (!language || typeof language !== 'string') {
return res.status(400).send({
message: 'language is required as a string',
});
}
2021-05-08 02:30:40 +02:00
if (!version || typeof version !== 'string') {
return res.status(400).send({
message: 'version is required as a string',
});
}
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
if (!files || !Array.isArray(files)) {
return res.status(400).send({
message: 'files is required as an array',
});
}
2021-05-08 02:30:40 +02:00
for (const [i, file] of files.entries()) {
if (typeof file.content !== 'string') {
return res.status(400).send({
message: `files[${i}].content is required as a string`,
});
}
}
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
if (compile_memory_limit) {
if (typeof compile_memory_limit !== 'number') {
return res.status(400).send({
message: 'if specified, compile_memory_limit must be a number',
});
}
if (
config.compile_memory_limit >= 0 &&
(compile_memory_limit > config.compile_memory_limit ||
compile_memory_limit < 0)
) {
return res.status(400).send({
message:
'compile_memory_limit cannot exceed the configured limit of ' +
config.compile_memory_limit,
});
}
}
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
if (run_memory_limit) {
if (typeof run_memory_limit !== 'number') {
return res.status(400).send({
message: 'if specified, run_memory_limit must be a number',
});
}
if (
config.run_memory_limit >= 0 &&
(run_memory_limit > config.run_memory_limit || run_memory_limit < 0)
) {
return res.status(400).send({
message:
'run_memory_limit cannot exceed the configured limit of ' +
config.run_memory_limit,
});
}
2021-05-08 02:20:21 +02:00
}
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
const rt = runtime.get_latest_runtime_matching_language_version(
language,
version
);
if (rt === undefined) {
return res.status(400).send({
message: `${language}-${version} runtime is unknown`,
});
}
2021-05-08 02:30:40 +02:00
const job = new Job({
runtime: rt,
alias: language,
files: files,
args: args || [],
stdin: stdin || '',
timeouts: {
run: run_timeout || 3000,
compile: compile_timeout || 10000,
},
memory_limits: {
run: run_memory_limit || config.run_memory_limit,
compile: compile_memory_limit || config.compile_memory_limit,
},
2021-05-08 02:20:21 +02:00
});
2021-05-08 02:30:40 +02:00
await job.prime();
const result = await job.execute();
await job.cleanup();
return res.status(200).send(result);
2021-04-25 10:55:20 +02:00
});
router.get('/runtimes', (req, res) => {
2021-05-08 02:30:40 +02:00
const runtimes = runtime.map(rt => {
return {
language: rt.language,
version: rt.version.raw,
aliases: rt.aliases,
runtime: rt.runtime,
};
});
return res.status(200).send(runtimes);
2021-04-25 10:55:20 +02:00
});
router.get('/packages', async (req, res) => {
2021-05-08 02:30:40 +02:00
logger.debug('Request to list packages');
let packages = await package.get_package_list();
packages = packages.map(pkg => {
return {
language: pkg.language,
language_version: pkg.version.raw,
installed: pkg.installed,
};
});
return res.status(200).send(packages);
2021-04-25 10:55:20 +02:00
});
router.post('/packages/:language/:version', async (req, res) => {
2021-05-08 02:30:40 +02:00
logger.debug('Request to install package');
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
const { language, version } = req.params;
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
const pkg = await package.get_package(language, version);
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
if (pkg == null) {
return res.status(404).send({
message: `Requested package ${language}-${version} does not exist`,
});
}
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
try {
const response = await pkg.install();
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
return res.status(200).send(response);
} catch (e) {
logger.error(
`Error while installing package ${pkg.language}-${pkg.version}:`,
e.message
);
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
return res.status(500).send({
message: e.message,
});
}
2021-04-25 10:55:20 +02:00
});
router.delete('/packages/:language/:version', async (req, res) => {
2021-05-08 02:30:40 +02:00
logger.debug('Request to uninstall package');
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
const { language, version } = req.params;
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
const pkg = await package.get_package(language, version);
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
if (pkg == null) {
return res.status(404).send({
message: `Requested package ${language}-${version} does not exist`,
});
}
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
try {
const response = await pkg.uninstall();
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
return res.status(200).send(response);
} catch (e) {
logger.error(
`Error while uninstalling package ${pkg.language}-${pkg.version}:`,
e.message
);
2021-04-25 10:55:20 +02:00
2021-05-08 02:30:40 +02:00
return res.status(500).send({
message: e.message,
});
}
2021-04-25 10:55:20 +02:00
});
module.exports = router;