multiwii-msp/lib/TcpServer.js

NodeJS MultiWii Serial Protocol to send or receive commands from MultiWii via TCP/IP

var Util = require('util');
var EventEmitter = require('events').EventEmitter;
var Net = require('net');
var Fiber = require('fibers');

var TcpPackageManager = require('./TcpPackageManager');
var Device = require('./Device');

/**
 *
 * @param {int}     port        - port to listen
 * @param {boolean} [log=false] - if TRUE enable logging
 * @constructor
 */
function TcpServer(port, log) {
    var self;

    self = this;

    function logMsg() {
        if (log) {
            console.log.apply(this, arguments);
        }
    }

    self._devices = {};
    self.server = Net.createServer(function (socket) {
        Fiber(function () {
            var packageManager, address, key;

            logMsg('SERVER: Client connecting...');

            socket.on('close', function () {
                self._devices[key].disconnect();
                logMsg('SERVER: Client disconnected');
            });

            socket.on('error', function (error) {
                console.log('SERVER: ERROR');
                console.log(error.stack);
            });

            packageManager = new TcpPackageManager(socket);

            address = socket.address();
            key = address.address + ':' + address.port;

            if (!self._devices.hasOwnProperty(key)) {
                self._devices[key] = new Device();
                /**
                 * Register event
                 *
                 * Fires on first device connection
                 *
                 * @event TcpServer#register
                 * @property {string} key    - registered device key
                 * @property {Device} device - device object
                 */
                self.emit('register', key, self._devices[key]);
            }

            self._devices[key].connect(packageManager);

            logMsg('SERVER: Client connected');
        }).run();
    });

    self.server.on('error', function (error) {
        console.log('SERVER: ERROR');
        console.log(error.stack);
    });

    self.server.listen(port, function () {
        logMsg('SERVER: Server listening...');
    });
}

Util.inherits(TcpServer, EventEmitter);

/**
 *
 * Get device by key
 *
 * @param {string} key - device key
 * @returns {null|Device}
 */
TcpServer.prototype.getDevice = function (key) {
    if (this._devices.hasOwnProperty(key)) {
        return this._devices[key];
    }

    return null;
};

/**
 *
 * Check is server has device at key
 *
 * @param {string} key
 * @returns {boolean}
 */
TcpServer.prototype.hasDevice = function (key) {
    return this._devices.hasOwnProperty(key);
};

/**
 *
 * Get list of devices
 *
 * @returns {object} - object with key-object pairs
 */
TcpServer.prototype.listDevices = function () {
    return this._devices;
};

module.exports = TcpServer;