runtime-node-net/lib/server.js

Wrapper around runtimejs-tcp with node.js

'use strict';

const runtimejs = require('runtimejs');
const TCPServerSocket = runtimejs.net.TCPServerSocket;
const util = require('util');
const EventEmitter = require('events').EventEmitter;
const Socket = require('./socket');

class Server extends EventEmitter {

  constructor(opts, cb) {
    super();

    if (typeof opts === 'function') {
      cb = opts;
      opts = {};
      this.on('connection', cb);
    } else {
      opts = opts || {};

      if (typeof cb === 'function') {
        this.on('connection', cb);
      }
    }

    this._server = new TCPServerSocket();
    this._address = null;
    this._server.onlisten = () => {
      setImmediate(() => {
        // handle if the server has been closed in same tick as it was opened
        if (this._server.listeningSocket.readyState === 'open'){
          this.emit('listening');
        }
      });
    };
    this._server.onclose = () => {
      setImmediate(() => {
        this.emit('close');
      });
    };
    this._server.onconnect = (tcpSocket) => {
      this.emit('connection', new Socket(tcpSocket));
    };
  }

  listen (port, address, cb) {
    const err = _listen(this._server, port);

    if (err) {
      if (err.message === 'E_ADDRESS_IN_USE') {
        err.code = 'EADDRINUSE';
      }
      this.emit('error', err);
      return;
    }

    let family = 'IPv4';
    if (!cb) {
      cb = address;
      address = '::';
      family = 'IPv6';
    }

    this._address = {
      port: this._server.localPort,
      address: address,
      family: family
    }

    if (cb) {
      this.once('listening', cb);
    }
  }

  close (cb) {
    this._address = null;
    if (typeof cb === 'function') {
      this.once('close', cb);
    }
    this._server.close();
  }

  address () {
    return this._address;
  }
}

module.exports = Server;

function _listen (server, port) {
  let err;
  try {
    server.listen(port);
  } catch (e) {
    err = e;
  }
  return err;
}