node-ipc/node-ipc.js

'use strict';

const os = require('os'),
    util = require('util'),
    colors = require('colors'),
    Client = require('./lib/client.js'),
    Server = require('./lib/socketServer.js');

colors.setTheme(
    {
        good    : 'green',
        notice  : 'yellow',
        warn    : 'red',
        error   : 'redBG',
        debug   : 'magenta',
        variable: 'cyan',
        data    : 'blue'
    }
);

const IPType=os.networkInterfaces()[
    Object.keys(os.networkInterfaces())[0]
][0].family;

let defaults={
    appspace        : 'app.',
    socketRoot      : '/tmp/',
    networkHost     : (IPType=='IPv6')? '::1' : '127.0.0.1',
    networkPort     : 8000,
    id              : os.hostname(),
    encoding        : 'utf8',
    rawBuffer       : false,
    sync            : false,
    silent          : false,
    maxConnections  : 100,
    retry           : 500,
    maxRetries      : Infinity,
    stopRetrying    : false,
    IPType          : IPType,
    tls             : false
};

let ipc = {
    config      : defaults,
    connectTo   : connect,
    connectToNet: connectNet,
    disconnect  : disconnect,
    serve       : serve,
    serveNet    : serveNet,
    of          : {},
    server      : false,
    log         : log
};

function log(){
    if(ipc.config.silent){
        return;
    }

    let args=Array.prototype.slice.call(arguments);

    for(let i=0, count=args.length; i<count; i++){
        if(typeof args[i] != 'object'){
            continue;
        }

        args[i]=util.inspect(args[i],{colors:true});
    }

    console.log(
        args.join(' ')
    );
}

function disconnect(id){
    if(!ipc.of[id]){
        return;
    }

    ipc.of[id].config.stopRetrying=true;

    ipc.of[id].off('*');
    if(ipc.of[id].socket){
        if(ipc.of[id].socket.destroy){
            ipc.of[id].socket.destroy();
        }
    }

    delete ipc.of[id];
}

function serve(path,callback){
    if(typeof path=='function'){
        callback=path;
        path=false;
    }
    if(!path){
        ipc.log(
            'Server path not specified, so defaulting to'.notice,
            'ipc.config.socketRoot + ipc.config.appspace + ipc.config.id'.variable,
            (ipc.config.socketRoot+ipc.config.appspace+ipc.config.id).data
        );
        path=ipc.config.socketRoot+ipc.config.appspace+ipc.config.id;
    }

    if(!callback){
        callback=emptyCallback;
    }

    ipc.server=new Server(
        path,
        ipc.config,
        log
    );

    ipc.server.on(
        'start',
        callback
    );
}

function emptyCallback(){
    //Do Nothing
}

function serveNet(host,port,UDPType,callback){
    if(typeof host=='number'){
        callback=UDPType;
        UDPType=port;
        port=host;
        host=false;
    }
    if(typeof host=='function'){
        callback=host;
        UDPType=false;
        host=false;
        port=false;
    }
    if(!host){
        ipc.log(
            'Server host not specified, so defaulting to'.notice,
            'ipc.config.networkHost'.variable,
            ipc.config.networkHost.data
        );
        host=ipc.config.networkHost;
    }
    if(host.toLowerCase()=='udp4' || host.toLowerCase()=='udp6'){
        callback=port;
        UDPType=host.toLowerCase();
        port=false;
        host=ipc.config.networkHost;
    }

    if(typeof port=='string'){
        callback=UDPType;
        UDPType=port;
        port=false;
    }
    if(typeof port=='function'){
        callback=port;
        UDPType=false;
        port=false;
    }
    if(!port){
        ipc.log(
            'Server port not specified, so defaulting to'.notice,
            'ipc.config.networkPort'.variable,
            ipc.config.networkPort
        );
        port=ipc.config.networkPort;
    }

    if(typeof UDPType=='function'){
        callback=UDPType;
        UDPType=false;
    }

    if(!callback){
        callback=emptyCallback;
    }

    ipc.server=new Server(
        host,
        ipc.config,
        log,
        port
    );

    if(UDPType){
        ipc.server[UDPType]=true;
    }

    ipc.server.on(
        'start',
        callback
    );
}

function connect(id,path,callback){
    if(typeof path == 'function'){
        callback=path;
        path=false;
    }

    if(!callback){
        callback=emptyCallback;
    }

    if(!id){
        ipc.log(
            'Service id required'.warn,
            'Requested service connection without specifying service id. Aborting connection attempt'.notice
        );
        return;
    }

    if(!path){
        ipc.log(
            'Service path not specified, so defaulting to'.notice,
            'ipc.config.socketRoot + ipc.config.appspace + id'.variable,
            (ipc.config.socketRoot+ipc.config.appspace+id).data
        );
        path=ipc.config.socketRoot+ipc.config.appspace+id;
    }

    if(ipc.of[id]){
        if(!ipc.of[id].socket.destroyed){
            ipc.log(
                'Already Connected to'.notice,
                id.variable,
                '- So executing success without connection'.notice
            );
            callback();
            return;
        }
        ipc.of[id].socket.destroy();
    }

    ipc.of[id] = new Client(ipc.config,ipc.log);
    ipc.of[id].id = id;
    ipc.of[id].path = path;

    ipc.of[id].connect();

    callback(ipc);
}

function connectNet(id,host,port,callback){
    if(!id){
        ipc.log(
            'Service id required'.warn,
            'Requested service connection without specifying service id. Aborting connection attempt'.notice
        );
        return;
    }
    if(typeof host=='number'){
        callback=port;
        port=host;
        host=false;
    }
    if(typeof host=='function'){
        callback=host;
        host=false;
        port=false;
    }
    if(!host){
        ipc.log(
            'Server host not specified, so defaulting to'.notice,
            'ipc.config.networkHost'.variable,
            ipc.config.networkHost.data
        );
        host=ipc.config.networkHost;
    }

    if(typeof port=='function'){
        callback=port;
        port=false;
    }
    if(!port){
        ipc.log(
            'Server port not specified, so defaulting to'.notice,
            'ipc.config.networkPort'.variable,
            ipc.config.networkPort
        );
        port=ipc.config.networkPort;
    }

    if(typeof callback == 'string'){
        UDPType=callback;
        callback=false;
    }
    if(!callback){
        callback=emptyCallback;
    }

    if(ipc.of[id]){
        if(!ipc.of[id].socket.destroyed){
            ipc.log(
                'Already Connected to'.notice,
                id.variable,
                '- So executing success without connection'.notice
            );
            callback();
            return;
        }
        ipc.of[id].socket.destroy();
    }

    ipc.of[id] = new Client(ipc.config,ipc.log);
    ipc.of[id].id = id;
    ipc.of[id].path = host;
    ipc.of[id].port = port;

    ipc.of[id].connect();

    callback(ipc);
}

module.exports=ipc;