green-light/lib/browser.js

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});

var _createClass = function () { function defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } } return function (Constructor, protoProps, staticProps) { if (protoProps) defineProperties(Constructor.prototype, protoProps); if (staticProps) defineProperties(Constructor, staticProps); return Constructor; }; }();

var _jsdom = require('jsdom');

var _jsdom2 = _interopRequireDefault(_jsdom);

var _path = require('path');

var _path2 = _interopRequireDefault(_path);

var _fs = require('fs');

var _fs2 = _interopRequireDefault(_fs);

var _promisePoller = require('promise-poller');

var _promisePoller2 = _interopRequireDefault(_promisePoller);

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }

var Browser = function () {
  function Browser(config) {
    _classCallCheck(this, Browser);

    this.config = {
      verbose: config.verbose,
      baseUrl: config.baseUrl,
      jQuery: config.jQuery,
      setupWindow: config.setupWindow || function (w) {
        return w;
      },
      readyWhen: config.readyWhen || function (w) {
        return true;
      },
      readyWhenInterval: config.readyWhenInterval || 10,
      readyWhenTimeout: config.readyWhenTimeout || 2000
    };

    if (!this.config.baseUrl) {
      throw new Error('No baseUrl set for Browser');
    }
  }

  _createClass(Browser, [{
    key: 'start',
    value: function start() {
      // Setup jQuery
      if (this.config.jQuery) {
        // TODO: Use readFile-not-sync
        // TODO: Use jquery from package
        var jQueryPath = _path2.default.resolve(__dirname, './jquery/jquery.js');
        this.jQuery = _fs2.default.readFileSync(jQueryPath).toString();
      }

      // Setup console
      this.virtualConsole = _jsdom2.default.createVirtualConsole();
      var logError = function logError() {
        var _console;

        for (var _len = arguments.length, output = Array(_len), _key = 0; _key < _len; _key++) {
          output[_key] = arguments[_key];
        }

        return (_console = console).log.apply(_console, ['\x1b[31m'].concat(output, ['\x1b[0m']));
      };
      if (this.config.verbose) {
        this.virtualConsole.sendTo(Object.assign({}, console, { error: logError }));
      } else {
        this.virtualConsole.sendTo({ error: logError });
      }

      return Promise.resolve();
    }
  }, {
    key: 'go',
    value: function go(path) {
      var _this = this;

      var url = this.config.baseUrl + path;
      this.url = url;

      if (this.config.verbose) {
        console.log('Going to url:', url);
      }

      return this._getDOM(url).then(function (window) {
        return _this.window = window;
      }).then(function () {
        return _this._waitForReady(_this.window);
      }).then(function () {
        return _this.config.setupWindow(_this.window);
      }).then(function () {
        return _this.window;
      }).catch(function (err) {
        console.error('Something went wrong at: ' + url);
        console.trace(err);
        throw err;
      });
    }
  }, {
    key: 'reset',
    value: function reset() {
      if (this.window) {
        this.window.close();
        delete this.window;
      }
    }
  }, {
    key: '_getDOM',
    value: function _getDOM(url) {
      var _this2 = this;

      return new Promise(function (resolve, reject) {
        _jsdom2.default.env({
          url: url,
          src: [_this2.jQuery],
          features: {
            FetchExternalResources: ['script'],
            ProcessExternalResources: ['script']
          },
          virtualConsole: _this2.virtualConsole,
          done: function done(err, window) {
            if (err) {
              return reject(err);
            }
            if (!window) {
              return reject(new Error('no window'));
            }
            resolve(window);
          }
        });
      });
    }
  }, {
    key: '_waitForReady',
    value: function _waitForReady(window) {
      var _this3 = this;

      return (0, _promisePoller2.default)({
        taskFn: function taskFn() {
          return _this3.config.readyWhen(window) ? Promise.resolve() : Promise.reject();
        },
        interval: this.config.readyWhenInterval,
        // timeout: this.config.readyWhenTimeout,
        retries: this.config.readyWhenTimeout / this.config.readyWhenInterval }).catch(function (err) {
        return Promise.reject(new Error('browser timeout'));
      });
    }
  }]);

  return Browser;
}();

exports.default = Browser;