appnexus/sicksync

import _ from 'lodash';
import { EventEmitter } from 'events';
import * as util from '../util';
import { REMOTE as remoteEvents } from '../../conf/events';
import { SERVER_ON_READY as  readyFlag } from '../../conf/text';

const COMMAND_NOT_FOUND = [
  'sicksync not found',
  'no sicksync in',
  'command not found',
];

export class RemoteHelper extends EventEmitter {
  constructor(params) {
    super();

    this._secret = params.secret;
    this._webSocketPort = params.websocketPort;
    this._username = params.username;
    this._hostname = params.hostname;
    this._prefersEncrypted = params.prefersEncrypted || false;
    this._debug = params.debug || false;
  }

  _startRemoteSicksync(ssh) {
    const context = this;

    ssh.stdin.write(
      [
        'sicksync remote',
        '-s', context._secret,
        '-p', context._webSocketPort,
        context._prefersEncrypted ? '-e' : '',
        context._debug ? '-d' : '',
        '\n',
      ].join(' ')
    );
  }

  start() {
    const context = this;
    const bootSicksync = _.once(this._startRemoteSicksync.bind(this));
    const ssh = util.shellIntoRemote(this._username + '@' + this._hostname);

    ssh.stdout.on('data', (data) => {
      const message = data.toString();

      // Boot sicksync (once!)
      bootSicksync(ssh);

      // If we get a 'ready' flag back from the server, emit a ready event
      if (_.includes(message, readyFlag)) {
        return context.emit(remoteEvents.READY);
      }

      // If the message contains the devboxes name, emit the message
      if (_.includes(message, context._secret)) {
        const cleanedMessage = message.replace(context._secret, '').replace('\n', '');

        return context.emit(remoteEvents.MESSAGE, cleanedMessage);
      }

      // Not found/Not installed
      _.each(COMMAND_NOT_FOUND, (notFoundText) => {
        /* istanbul ignore else */
        if (_.includes(message, notFoundText)) {
          context.emit(remoteEvents.NOT_FOUND, message);
        }
      });
    });
  }
}