unifiws-helper.js

const axios = require('axios');
const { CookieJar } = require('tough-cookie');
const { HttpCookieAgent, HttpsCookieAgent } = require('http-cookie-agent/http');
const WebSocket = require('ws');

var ControllerWS = function (hostname, port, unifios, ssl, username, password, site) {

    var _self = this;
    _self._cookieJar = new CookieJar();
    _self._unifios = unifios;
    _self._ssl = ssl;
    _self._baseurl = 'https://127.0.0.1:8443';
    _self._username = username;
    _self._password = password;
    _self._site = site;

    if (typeof (hostname) !== 'undefined' && typeof (port) !== 'undefined') {
        _self._baseurl = 'https://' + hostname + ':' + port;
    }

    const jar = _self._cookieJar;
    const axiosinstance = axios.create({
        httpAgent: new HttpCookieAgent({ cookies: { jar } }),
        httpsAgent: new HttpsCookieAgent({ cookies: { jar }, rejectUnauthorized: _self._ssl, requestCert: true })
    });

    _self.loginws = async function (cb) {

        if (_self._unifios)
            url = _self._baseurl + '/api/auth/login';
        else
            url = _self._baseurl + '/api/login';

        axiosinstance.post(url, {
            username: _self._username,
            password: _self._password
        })
            .then(function (response) {
                if (response.headers['x-csrf-token']) {
                    axiosinstance.defaults.headers.common['x-csrf-token'] = response.headers['x-csrf-token'];
                }
            })
            .catch(function (error) {

            })
            .then(function () {

                jar.getCookieString(_self._baseurl).then(cookies => {
                    const baseurl = _self._baseurl.replace('https://', 'wss://')
                    var eventsUrl = baseurl + '/wss/s/<SITE>/events'.replace('<SITE>', _self._site);
                    if (_self._unifios)
                        eventsUrl = baseurl + '/proxy/network/wss/s/<SITE>/events'.replace('<SITE>', _self._site);

                    _ws = new WebSocket(eventsUrl, {
                        perMessageDeflate: false,
                        rejectUnauthorized: _self._ssl,
                        headers: {
                            Cookie: cookies
                        }
                    });

                    _ws.on('open', function open() {
                        if (typeof (cb) === 'function') {
                            cb(false, 'STATUS_CONNECTED');
                        }
                    });

                    _ws.on('message', function message(data) {
                        try {
                            var obj = JSON.parse(data);
                            var listenTo = ['EVT_WU_Connected', 'EVT_WU_Disconnected', 'EVT_WU_Roam', 'EVT_WU_Roam_Radio', 'EVT_WG_Connected', 'EVT_WG_Disconnected', 'EVT_WG_Roam', 'EVT_WG_Roam_Radio', 'EVT_LU_Disconnected', 'EVT_LU_Connected']
                            if (obj.meta.message == 'events') {
                                if (listenTo.indexOf(obj.data[0].key) == -1) {
                                    //nothing
                                    //console.log('-----------------------NEW EVENT-----------------------');
                                    //console.log('%s', data);
                                    //console.log('-----------------------NEW EVENT-----------------------');
                                } else {
                                    if (typeof (cb) === 'function') {
                                        cb(false, obj);
                                    }
                                }
                            } else if (obj.meta.message == 'ping-test:update') {
                                cb(false, obj);
                            }
                        } catch (error) {
                            //Ignore this message
                        }
                    });

                    _ws.on('error', function error(error) {
                        if (typeof (cb) === 'function') {
                            cb({ message: error });
                        }
                    });

                    _ws.on('close', function close() {
                        if (typeof (cb) === 'function') {
                            cb(false, 'STATUS_DISCONNECTED');
                        }
                    });
                })
            });
    };
};

exports.ControllerWS = ControllerWS;