collar-dev-server/indexES5.js

'use strict';

var path = require('path');
var express = require('express');
var app = express();
var http = require('http');

var bodyParser = require('body-parser');

var argv = require('minimist')(process.argv.slice(2));

var port = argv.p || 7500;

if (argv.v) {
  var pack = require('./package.json');
  console.log('collar-dev-server version ' + pack.version);
  process.exit();
}

// const io = require('socket.io')(server);

app.use(bodyParser.json());
app.use(bodyParser.urlencoded({ extended: false }));
app.use(express.static(path.join(__dirname, 'public')));

var elementBuffer = [];
var signalBuffer = [];

var apps = {}; // app id

var DEFAULT_ELEMENT_BLOCK = 10;
var DEFAULT_SIGNAL_BLOCK = 10;

var record = false;
var count = 0;

function getProcessId(data) {
  var processId = '__anonymous__';
  if (data && data.process) {
    processId = data.process;
  }
  return processId;
}

var WebSocketServer = require('collar-websocket').Server;

var server = http.createServer(app);
var wss = new WebSocketServer(server);
wss.onAuth(function (id, secret, done) {
  // todo: authenticate the requests
  // accept all request
  done(null, { id: id });
});
wss.on('connect', function (socket) {
  var ns = socket.ns;

  if (ns === '/client') {
    console.log('client [' + socket.clientId + '] connected');
    /*
     * Clients
     */
    socket.on('start record', function (data) {
      console.log('start recording');
      record = true;
      count = new Date().getTime();
      signalBuffer = [];
    });

    socket.on('stop record', function (data) {
      console.log('stop recording');
      record = false;
      count = new Date().getTime();
      signalBuffer = [];
    });

    socket.on('push', function (data) {
      var id = data.nodeId;
      var signal = data.signal;

      wss.of('/app').emit('push', data);
    });

    socket.on('send', function (data) {
      var id = data.nodeId;
      var signal = data.signal;

      wss.of('/app').emit('send', data);
    });

    socket.on('clear', function (data) {
      elementBuffer = [];
      signalBuffer = [];
    });

    socket.on('ready to receive elements', function (blockSize) {
      if (!blockSize || typeof blockSize !== 'number') {
        blockSize = DEFAULT_ELEMENT_BLOCK;
      }

      var tobesent = [];
      for (var i = 0; i < blockSize; i++) {
        if (elementBuffer.length > i) {
          tobesent.push(elementBuffer.shift());
        } else {
          break;
        }
      }

      if (tobesent.length > 0) {
        wss.of('/client').emit('append elements', {
          elements: tobesent
        });
      }
    });

    socket.on('ready to receive signals', function (blockSize) {
      if (!record) return;

      if (!blockSize || typeof blockSize !== 'number') {
        blockSize = DEFAULT_SIGNAL_BLOCK;
      }

      var tobesent = [];
      for (var i = 0; i < blockSize; i++) {
        if (signalBuffer.length > i) {
          tobesent.push(signalBuffer.shift());
        } else {
          break;
        }
      }

      if (tobesent.length > 0 && record) {
        wss.of('/client').emit('append signals', {
          signals: tobesent
        });
      }
    });
  } else if (ns === '/app') {
    console.log('application [' + socket.clientId + '] connected');

    /*
     *  Applications
     */
    socket.on('new model', function (data) {
      var processId = getProcessId(data);
      wss.of('/client').emit('new model', {
        processId: processId
      });
    });

    socket.on('append elements', function (data) {
      if (typeof data === 'string') {
        try {
          data = JSON.parse(data);
        } catch (err) {
          console.warn('error parsing received "append elements" data');
          return;
        }
      }

      var _iteratorNormalCompletion = true;
      var _didIteratorError = false;
      var _iteratorError = undefined;

      try {
        for (var _iterator = data.elements[Symbol.iterator](), _step; !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = true) {
          var elem = _step.value;

          if (!elem.data) elem.data = {};
          elementBuffer.push(elem);
        }
      } catch (err) {
        _didIteratorError = true;
        _iteratorError = err;
      } finally {
        try {
          if (!_iteratorNormalCompletion && _iterator.return) {
            _iterator.return();
          }
        } finally {
          if (_didIteratorError) {
            throw _iteratorError;
          }
        }
      }
    });

    socket.on('append signals', function (data) {
      if (typeof data === 'string') {
        try {
          data = JSON.parse(data);
        } catch (err) {
          console.warn('error parsing received "append elements" data');
          return;
        }
      }

      var signals = data.signals.filter(function (n) {
        return n != undefined && n != null;
      });

      if (record) {
        for (var i = 0; i < signals.length; i++) {
          if (!signals[i]) {
            continue;
          }
          signals[i].time = count;
          count++;

          signalBuffer.push(signals[i]);
        }
      }
    });
  }
});

server.listen(port, function () {
  console.log('collar dev server listening on port', port);
});