All files / src/protocols websocket.js

100% Statements 62/62
86.36% Branches 38/44
100% Functions 9/9
100% Lines 62/62

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151      1x 1x         49x   49x 8x       41x 32x   32x   9x 9x         9x     5x       36x 36x             22x 22x   22x   22x 18x 18x     22x   22x 5x 5x     22x     3x   3x 2x     1x   1x 1x       3x 1x       2x 2x 2x   2x       22x 3x   3x 3x       3x   3x 3x       22x 3x     3x 2x       1x   1x 1x                         2x 1x               1x 1x 1x 1x 1x   1x 1x 1x       1x  
'use strict';
 
const
  KuzzleError = require('../KuzzleError'),
  RTWrapper = require('./abstract/realtime');
 
class WSNode extends RTWrapper {
 
  constructor(host, options = {}) {
    super(host, options);
 
    if (typeof host !== 'string' || host === '') {
      throw new Error('host is required');
    }
 
    // Browsers WebSocket API
    if (typeof WebSocket !== 'undefined') {
      this.WebSocketClient = WebSocket;
      // There are no options allowed in the browsers WebSocket API
      this.options = null;
    } else {
      this.WebSocketClient = require('ws');
      this.options = {
        perMessageDeflate: false,
        headers: options.headers || null
      };
 
      if (this.options.headers !== null &&
          (Array.isArray(this.options.headers) ||
          typeof this.options.headers !== 'object')) {
        throw new Error('Invalid "headers" option: expected an object');
      }
    }
 
    this.client = null;
    this.lasturl = null;
  }
 
  /**
   * Connect to the websocket server
   */
  connect () {
    return new Promise((resolve, reject) => {
      const url = `${this.ssl ? 'wss' : 'ws'}://${this.host}:${this.port}`;
 
      super.connect();
 
      if (url !== this.lasturl) {
        this.wasConnected = false;
        this.lasturl = url;
      }
 
      this.client = new this.WebSocketClient(url, this.options);
 
      this.client.onopen = () => {
        this.clientConnected();
        return resolve();
      };
 
      this.client.onclose = (closeEvent, message) => {
        let
          status,
          reason = message;
 
        if (typeof closeEvent === 'number') {
          status = closeEvent;
        }
        else {
          status = closeEvent.code;
 
          Eif (closeEvent.reason) {
            reason = closeEvent.reason;
          }
        }
 
        if (status === 1000) {
          this.clientDisconnected();
        }
        // do not forward a connection close error if no
        // connection has been previously established
        else Eif (this.wasConnected) {
          const error = new Error(reason);
          error.status = status;
 
          this.clientNetworkError(error);
        }
      };
 
      this.client.onerror = error => {
        let err = error;
 
        Eif (!(error instanceof Error)) {
          err = error ?
            new Error(error.message || error) : new Error('Unexpected error');
        }
 
        this.clientNetworkError(err);
 
        Eif ([this.client.CLOSING, this.client.CLOSED].indexOf(this.client.readyState) > -1) {
          return reject(err);
        }
      };
 
      this.client.onmessage = payload => {
        const data = JSON.parse(payload.data || payload);
 
        // for responses, data.room == requestId
        if (data.room) {
          this.emit(data.room, data);
        }
        else {
          // @deprecated
          this.emit('discarded', data);
 
          const error = new KuzzleError(data.error);
          this.emit('queryError', error, data);
        }
      };
 
    });
  }
 
  /**
   * Sends a payload to the connected server
   *
   * @param {Object} payload
   */
  send (payload) {
    if (this.client && this.client.readyState === this.client.OPEN) {
      this.client.send(JSON.stringify(payload));
    }
  }
 
  /**
   * Closes the connection
   */
  close () {
    this.state = 'offline';
    this.removeAllListeners();
    this.wasConnected = false;
    Eif (this.client) {
      this.client.close();
    }
    this.client = null;
    this.stopRetryingToConnect = true;
    super.close();
  }
}
 
module.exports = WSNode;