ewelink-api/mixins/websocket/openWebSocketMixin.js

const W3CWebSocket = require('websocket').w3cwebsocket;
const WebSocketAsPromised = require('websocket-as-promised');

const payloads = require('../../lib/payloads');

const openWebSocketMixin = {
  /**
   * Open a socket connection to eWeLink
   * and execute callback function with server message as argument
   *
   * @param callback
   *
   * @returns {Promise<WebSocketAsPromised>}
   */
  async openWebSocket(callback) {
    const payloadLogin = await payloads.wssLoginPayload({
      at: this.at,
      apiKey: this.apiKey,
    });

    const wsp = new WebSocketAsPromised(this.getApiWebSocket(), {
      createWebSocket: wss => new W3CWebSocket(wss),
    });

    wsp.onMessage.addListener(message => {
      const data = JSON.parse(message);
      callback(data);
    });

    await wsp.open();
    await wsp.send(payloadLogin);

    return wsp;
  },
};

module.exports = openWebSocketMixin;