cloudron-box/src/routes/services.js

'use strict';

exports = module.exports = {
    list,
    get,
    configure,
    getLogs,
    getLogStream,
    restart,
    rebuild
};

const assert = require('assert'),
    AuditSource = require('../auditsource.js'),
    BoxError = require('../boxerror.js'),
    HttpError = require('connect-lastmile').HttpError,
    HttpSuccess = require('connect-lastmile').HttpSuccess,
    safe = require('safetydance'),
    services = require('../services.js');

async function list(req, res, next) {
    const [error, result] = await safe(services.listServices());
    if (error) return next(BoxError.toHttpError(error));

    next(new HttpSuccess(200, { services: result }));
}

async function get(req, res, next) {
    assert.strictEqual(typeof req.params.service, 'string');

    req.clearTimeout();

    const [error, result] = await safe(services.getServiceStatus(req.params.service));
    if (error) return next(BoxError.toHttpError(error));

    next(new HttpSuccess(200, { service: result }));
}

async function configure(req, res, next) {
    assert.strictEqual(typeof req.params.service, 'string');

    if (typeof req.body.memoryLimit !== 'number') return next(new HttpError(400, 'memoryLimit must be a number'));
    if ('recoveryMode' in req.body && typeof req.body.recoveryMode !== 'boolean') return next(new HttpError(400, 'recoveryMode must be boolean'));

    const data = {
        memoryLimit: req.body.memoryLimit,
        recoveryMode: req.body.recoveryMode || false
    };

    const [error] = await safe(services.configureService(req.params.service, data, AuditSource.fromRequest(req)));
    if (error) return next(BoxError.toHttpError(error));

    next(new HttpSuccess(202, {}));
}

async function getLogs(req, res, next) {
    assert.strictEqual(typeof req.params.service, 'string');

    const lines = 'lines' in req.query ? parseInt(req.query.lines, 10) : 10; // we ignore last-event-id
    if (isNaN(lines)) return next(new HttpError(400, 'lines must be a number'));

    const options = {
        lines: lines,
        follow: false,
        format: req.query.format || 'json'
    };

    const [error, logStream] = await safe(services.getServiceLogs(req.params.service, options));
    if (error) return next(BoxError.toHttpError(error));

    res.writeHead(200, {
        'Content-Type': 'application/x-logs',
        'Content-Disposition': `attachment; filename="${req.params.service}.log"`,
        'Cache-Control': 'no-cache',
        'X-Accel-Buffering': 'no' // disable nginx buffering
    });
    logStream.pipe(res);
}

// this route is for streaming logs
async function getLogStream(req, res, next) {
    assert.strictEqual(typeof req.params.service, 'string');

    const lines = 'lines' in req.query ? parseInt(req.query.lines, 10) : 10; // we ignore last-event-id
    if (isNaN(lines)) return next(new HttpError(400, 'lines must be a valid number'));

    function sse(id, data) { return 'id: ' + id + '\ndata: ' + data + '\n\n'; }

    if (req.headers.accept !== 'text/event-stream') return next(new HttpError(400, 'This API call requires EventStream'));

    const options = {
        lines: lines,
        follow: true,
        format: 'json'
    };

    const [error, logStream] = await safe(services.getServiceLogs(req.params.service, options));
    if (error) return next(BoxError.toHttpError(error));

    res.writeHead(200, {
        'Content-Type': 'text/event-stream',
        'Cache-Control': 'no-cache',
        'Connection': 'keep-alive',
        'X-Accel-Buffering': 'no', // disable nginx buffering
        'Access-Control-Allow-Origin': '*'
    });
    res.write('retry: 3000\n');
    res.on('close', logStream.close);
    logStream.on('data', function (data) {
        var obj = JSON.parse(data);
        res.write(sse(obj.monotonicTimestamp, JSON.stringify(obj))); // send timestamp as id
    });
    logStream.on('end', res.end.bind(res));
    logStream.on('error', res.end.bind(res, null));
}

async function restart(req, res, next) {
    assert.strictEqual(typeof req.params.service, 'string');

    const [error] = await safe(services.restartService(req.params.service, AuditSource.fromRequest(req)));
    if (error) return next(BoxError.toHttpError(error));

    next(new HttpSuccess(202, {}));
}

async function rebuild(req, res, next) {
    assert.strictEqual(typeof req.params.service, 'string');

    const [error] = await safe(services.rebuildService(req.params.service, AuditSource.fromRequest(req)));
    if (error) return next(BoxError.toHttpError(error));

    next(new HttpSuccess(202, {}));
}