unifi-voucher-site/modules/unifi.js

/**
 * Import vendor modules
 */
const unifi = require('node-unifi');

/**
 * Import own modules
 */
const configProvider = require('./config');

/**
 * Import own modules
 */
const config = {
    unifi: {
        ip: configProvider('unifi_ip') || process.env.UNIFI_IP || '192.168.1.1',
        port: configProvider('unifi_port') || process.env.UNIFI_PORT || 443,
        username: configProvider('unifi_username') || process.env.UNIFI_USERNAME || 'admin',
        password: configProvider('unifi_password') || process.env.UNIFI_PASSWORD || 'password',
        siteID: configProvider('unifi_site_id') || process.env.UNIFI_SITE_ID || 'default'
    }
};

/**
 * Exports the UniFi voucher function
 *
 * @param type
 * @param create
 * @returns {Promise<unknown>}
 */
module.exports = (type, create = true) => {
    if(create) {
        return new Promise((resolve, reject) => {
            /**
             * Create new UniFi controller object
             *
             * @type {Controller}
             */
            const controller = new unifi.Controller({
                host: config.unifi.ip,
                port: config.unifi.port,
                site: config.unifi.siteID,
                sslverify: false
            });

            /**
             * Login and create a voucher
             */
            controller.login(config.unifi.username, config.unifi.password).then(() => {
                controller.getSitesStats().then(() => {
                    controller.createVouchers(type.expiration, 1, parseInt(type.usage) === 1 ? 1 : 0, null, typeof type.upload !== "undefined" ? type.upload : null, typeof type.download !== "undefined" ? type.download : null, typeof type.megabytes !== "undefined" ? type.megabytes : null).then((voucher_data) => {
                        controller.getVouchers(voucher_data[0].create_time).then((voucher_data_complete) => {
                            const voucher = `${[voucher_data_complete[0].code.slice(0, 5), '-', voucher_data_complete[0].code.slice(5)].join('')}`;
                            resolve(voucher);
                        }).catch((e) => {
                            console.log('[UniFi] Error while getting voucher!');
                            console.log(e);
                            reject('[UniFi] Error while getting voucher!');
                        });
                    }).catch((e) => {
                        console.log('[UniFi] Error while creating voucher!');
                        console.log(e);
                        reject('[UniFi] Error while creating voucher!');
                    });
                }).catch((e) => {
                    console.log('[UniFi] Error while getting site stats!');
                    console.log(e);
                    reject('[UniFi] Error while getting site stats!');
                });
            }).catch((e) => {
                console.log('[UniFi] Error while logging in!');
                console.log(e);
                reject('[UniFi] Error while logging in!');
            });
        });
    } else {
        return new Promise((resolve, reject) => {
            /**
             * Create new UniFi controller object
             *
             * @type {Controller}
             */
            const controller = new unifi.Controller({
                host: config.unifi.ip,
                port: config.unifi.port,
                site: config.unifi.siteID,
                sslverify: false
            });

            /**
             * Login and create a voucher
             */
            controller.login(config.unifi.username, config.unifi.password).then(() => {
                controller.getSitesStats().then(() => {
                    controller.getVouchers().then((vouchers) => {
                        resolve(vouchers);
                    }).catch((e) => {
                        console.log('[UniFi] Error while getting voucher!');
                        console.log(e);
                        reject('[UniFi] Error while getting voucher!');
                    });
                }).catch((e) => {
                    console.log('[UniFi] Error while getting site stats!');
                    console.log(e);
                    reject('[UniFi] Error while getting site stats!');
                });
            }).catch((e) => {
                console.log('[UniFi] Error while logging in!');
                console.log(e);
                reject('[UniFi] Error while logging in!');
            });
        });
    }
};