@testim/testim-cli/commons/testimTunnel.js

"use strict";

const Promise = require('bluebird');
const logger = require('./logger').getLogger('tunnel');
const _ = require('lodash');
const ArgError = require('../errors.js').ArgError;
const lazyRequire = require('./lazyRequire');

function connect(options, authData) {
    if (!options.tunnel) {
        return Promise.resolve();
    } else if (options.tunnel && !authData.ngrokToken) {
        throw new ArgError("tunnel feature is not enabled, please contact support - info@testim.io.");
    }

    return new Promise((resolve, reject) => {
        let connectOpt = {
            proto: 'http',
            addr: options.tunnelPort || 80,
            authtoken: authData.ngrokToken
        };

        if (options.tunnelHostHeader) {
            connectOpt.host_header = options.tunnelHostHeader;
        }

        return lazyRequire('ngrok').then(ngrok => {

            return ngrok.connect(connectOpt)
            .then((url) => {
                if (options.tunnelUseHttpAddress && _.startsWith(url, "https://")) {
                    logger.info("replace https to http");
                    const newUrl = url.replace("https://", "http://");
                    return resolve(newUrl);
                }
                return resolve(url);
            }).catch(err => {
                const msg = "Failed to open tunnel error message error - " + err;
                logger.error(msg, {err: err});
                return reject(new Error(msg));
            });
        });

    });
}

function disconnect(url) {
    if (!url) {
        return Promise.resolve();
    }
    return new Promise(function (resolve, reject) {
        return lazyRequire('ngrok').then(ngrok => {
            return ngrok.disconnect(url)
                .then(resolve)
                .catch((err) => {
                    logger.error("catch error - failed to close tunnel", {err: err});
                    reject(err);
                });
            });
    });
}

module.exports = {
    connect: connect,
    disconnect: disconnect
};