asarfuckery/appasar/ptb/app_bootstrap/request.js

'use strict';

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

var _extends = Object.assign || function (target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i]; for (var key in source) { if (Object.prototype.hasOwnProperty.call(source, key)) { target[key] = source[key]; } } } return target; };

var _electron = require('electron');

var _querystring = require('querystring');

var _querystring2 = _interopRequireDefault(_querystring);

var _request = require('request');

var _request2 = _interopRequireDefault(_request);

var _events = require('events');

var _events2 = _interopRequireDefault(_events);

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

function _log(_msg) {
  // console.log('[Request] ' + _msg);
}

function requestWithMethod(method, origOpts, origCallback) {
  var _this = this;

  if (typeof origOpts == 'string') {
    origOpts = { url: origOpts };
  }

  var opts = _extends({}, origOpts, { method: method });

  var callback = void 0;
  if (origCallback || opts.callback) {
    var origOptsCallback = opts.callback;
    delete opts.callback;
    callback = function callback() {
      for (var _len = arguments.length, args = Array(_len), _key = 0; _key < _len; _key++) {
        args[_key] = arguments[_key];
      }

      if (origCallback) {
        origCallback.apply(_this, args);
      }
      if (origOptsCallback) {
        origOptsCallback.apply(_this, args);
      }
    };
  }

  var strictOpts = _extends({}, opts, { strictSSL: true });
  var laxOpts = _extends({}, opts, { strictSSL: false });

  var rv = new _events2.default();

  if (callback) {
    _log('have callback, so wrapping');
    rv.on('response', function (response) {
      var chunks = [];
      response.on('data', function (chunk) {
        return chunks.push(chunk);
      });
      response.on('end', function () {
        callback(null, response, Buffer.concat(chunks));
      });
    });
    rv.on('error', function (error) {
      return callback(error);
    });
  }

  var requestTypes = [{
    factory: function factory() {
      return (0, _request2.default)(strictOpts);
    },
    method: 'node_request_strict'
  }, {
    factory: function factory() {
      var qs = _querystring2.default.stringify(strictOpts.qs);
      var nr = _electron.net.request(_extends({}, strictOpts, { url: strictOpts.url + '?' + qs }));
      nr.end();
      return nr;
    },
    method: 'electron_net_request_strict'
  }, {
    factory: function factory() {
      return (0, _request2.default)(laxOpts);
    },
    method: 'node_request_lax'
  }];

  function attempt(index) {
    var _requestTypes$index = requestTypes[index],
        factory = _requestTypes$index.factory,
        method = _requestTypes$index.method;

    _log('Attempt #' + (index + 1) + ': ' + method);
    factory().on('response', function (response) {
      _log(method + ' success! emitting response ' + response);
      rv.emit('response', response);
    }).on('error', function (error) {
      if (index + 1 < requestTypes.length) {
        _log(method + ' failure, trying next option');
        attempt(index + 1);
      } else {
        _log(method + ' failure, out of options');
        rv.emit('error', error);
      }
    });
  }

  attempt(0);

  return rv;
}

// only supports get for now, since retrying is non-idempotent and
// we'd want to grovel the errors to make sure it's safe to retry
var _arr = ['get'];
for (var _i = 0; _i < _arr.length; _i++) {
  var method = _arr[_i];
  requestWithMethod[method] = requestWithMethod.bind(null, method);
}

exports.default = requestWithMethod;
module.exports = exports['default'];