@jeremykross/containership.api-bridge/lib/api-bridge.js

const _ = require('lodash');
const request = require('request');
const net = require('net');

const { ApiImplementation } = require('containership.plugin.v2');

class ContainershipApiBridge extends ApiImplementation {
    constructor(ip, port, core) {
        super();
        this.apiIP = ip || "localhost";
        this.apiPort = port || "8081";
        this.myriadPort = 2666;
        this.apiAddr = `http://${ip}:${port}`;
        this.core = core;
    }

    makeHandler(path, cb) {
        cb = cb || _.identity;

        return (err, resp) => {
            if(err) {
                cb(err);
            } else {
                cb(null, _.get(resp, path));
            }
        };

    }

    apiVERB(verb, endpoint, data, path, cb) {
        request({
            baseUrl: this.apiAddr,
            url: endpoint,
            body: data,
            method: verb,
            json: true
        }, this.makeHandler(path, cb));
    }

    apiGET(endpoint, path, cb) {
        this.apiVERB('GET', endpoint, undefined, path, cb);
    }

    apiPOST(endpoint, data, path, cb) {
        this.apiVERB('POST', endpoint, data, path, cb);
    }

    apiPUT(endpoint, data, path, cb) {
        this.apiVERB('PUT', endpoint, data, path, cb);
    }

    apiDELETE(endpoint, path, cb) {
        this.apiVERB('DELETE', endpoint, null, path, cb);
    }

    getClusterId(cb) {
        this.apiGET('/v1/cluster', ['body', 'id'], cb);
    }

    getApplications(cb) {
        this.apiGET('/v1/applications', 'body', cb);
    }

    createContainers(appId, containerConfig, cb) {
        this.apiPOST(`/v1/applications/${appId}/containers?count=${containerConfig.count}`, {}, 'body', cb);
    }

    createApplication(appDesc, cb) {
        this.apiPOST(`/v1/aplications/${appId}`, appDesc, 'body', cb);
    }

    updateApplication(appId, appDesc, cb) {
        this.apiPUT(`/v1/aplications/${appId}`, appDesc, 'body', cb);
    }

    deleteApplication(appId, cb) {
        this.apiDELETE(`/v1/applications/${appId}`, 'body', cb);
    }

    getHosts(cb) {
        this.apiGET('/v1/hosts', 'body', cb);
    }

    discoverPeers(cidr) {
        if(!this.core) {
            throw new Error("A core reference is required to use this method.");
        }

        this.core.cluster.legiond.options.network.cidr = cidr;
        this.core.cluster.legiond.actions.discover_peers(cidr);
    }

    makeSocketRequest(message, cb) {
        const socket = new net.Socket();

        socket.connect(this.myriadPort, this.apiIP, () => {
            console.log("Connected!");
            socket.write(message);
            socket.write(ContainershipApiBridge.MYRIAD_DELIMETER);
        });

        socket.on('error', (err) => {
            socket.destroy();
            return cb(err);
        });

        let buffer = '';
        socket.on('data', (data) => {
            buffer += data.toString();

            //Wait until we have a complete message.
            if(!_.includes(buffer, ContainershipApiBridge.MYRIAD_DELIMETER)) {
                return;
            }

            socket.end();

            const messageJSON = _.first(_.split(buffer, ContainershipApiBridge.MYRIAD_DELIMIETER));

            let message;

            try {
                message = JSON.parse(messageJSON);
            } catch(err) {
                return cb(null, messageJSON);
            }

            if(!_.isEmpty(message.error)) {
                cb(new Error(message.error));
            } else {
                cb(null, message);
            }

        });
    }

    setDistributedKey(k, v, cb) {
        this.makeSocketRequest(`SET ${k} ${v}`, cb);
    }

    getDistributedKey(k, cb) {
        this.makeSocketRequest(`GET ${k}`, cb);
    }

}

ContainershipApiBridge.MYRIAD_DELIMETER = '\r\n';

module.exports = ContainershipApiBridge;