@lskjs/proxy/lib/Proxy.js

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
Object.defineProperty(exports, "__esModule", {
  value: true
});
exports["default"] = exports.Proxy = void 0;
var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
var _classCallCheck2 = _interopRequireDefault(require("@babel/runtime/helpers/classCallCheck"));
var _createClass2 = _interopRequireDefault(require("@babel/runtime/helpers/createClass"));
var _avg = _interopRequireDefault(require("@lskjs/utils/avg"));
var _inc = _interopRequireDefault(require("@lskjs/utils/inc"));
var _events = _interopRequireDefault(require("events"));
var _pick = _interopRequireDefault(require("lodash/pick"));
var _createKey = require("./utils/createKey");
var _createUri = require("./utils/createUri");
var _getIpv = require("./utils/getIpv");
var _getProxyAgent = require("./utils/getProxyAgent");
function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); enumerableOnly && (symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; })), keys.push.apply(keys, symbols); } return keys; }
function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = null != arguments[i] ? arguments[i] : {}; i % 2 ? ownKeys(Object(source), !0).forEach(function (key) { (0, _defineProperty2["default"])(target, key, source[key]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)) : ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } return target; }
var Proxy = /*#__PURE__*/function () {
  function Proxy() {
    var _this$tags;
    var props = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
    (0, _classCallCheck2["default"])(this, Proxy);
    this.stats = {};
    Object.assign(this, props);
    if (!this.key) this.key = this.getKey();
    if (!this.uri) this.uri = this.getUri();
    if (!((_this$tags = this.tags) !== null && _this$tags !== void 0 && _this$tags.ipv) && this.ip) {
      if (!this.tags) this.tags = {};
      this.tags.ipv = (0, _getIpv.getIpv)(this.ip);
    }
    this.ee = new _events["default"]();
  }
  (0, _createClass2["default"])(Proxy, [{
    key: "getState",
    value: function getState() {
      // const ipv = getIpv(ip);
      return _objectSpread(_objectSpread({}, (0, _pick["default"])(this, ['type', 'host', 'port', 'user', 'password', 'provider', 'ip', 'tests', 'targets'])), {}, {
        // , 'uri', 'key'
        tags: _objectSpread({}, this.tags),
        key: this.getKey(),
        uri: this.getUri()
      });
    }
  }, {
    key: "toObject",
    value: function toObject() {
      return this.getState();
    }
  }, {
    key: "toJSON",
    value: function toJSON() {
      return this.getState();
    }
  }, {
    key: "getJson",
    value: function getJson() {
      return this.getState();
    }
  }, {
    key: "getUrl",
    value: function getUrl() {
      return this.getUri();
    }
  }, {
    key: "getUri",
    value: function getUri() {
      if (this.provider === 'localhost') return null;
      if (this.uri) return this.uri;
      return (0, _createUri.createUri)(this);
    }
  }, {
    key: "getKey",
    value: function getKey() {
      if (this.provider === 'localhost') return 'localhost';
      return (0, _createKey.createKey)(this);
    }
  }, {
    key: "getOptions",
    value: function getOptions() {
      (0, _inc["default"])(this.stats, 'get');
      var proxy = {
        host: this.host,
        port: this.port
      };
      if (this.user && this.password) {
        proxy.auth = {
          username: this.user,
          password: this.password
        };
      }
      return proxy;
    }
  }, {
    key: "getAgent",
    value: function getAgent() {
      (0, _inc["default"])(this.stats, 'get');
      if (this.provider === 'localhost') return null;
      return (0, _getProxyAgent.getProxyAgent)(this);
    }
  }, {
    key: "getProviderOptions",
    value: function getProviderOptions() {
      (0, _inc["default"])(this.stats, 'get');
      // lib
      if (this.provider === 'localhost') return {};
      var httpsAgent = (0, _getProxyAgent.getProxyAgent)(this);
      return {
        httpsAgent: httpsAgent,
        httpAgent: httpsAgent
      };
    }
  }, {
    key: "emit",
    value: function emit() {
      var _this$ee;
      (_this$ee = this.ee).emit.apply(_this$ee, arguments);
    }
  }, {
    key: "on",
    value: function on() {
      var _this$ee2;
      (_this$ee2 = this.ee).on.apply(_this$ee2, arguments);
    }
  }, {
    key: "feedback",
    value: function feedback() {
      var props = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
      var _props$status = props.status,
        status = _props$status === void 0 ? 'unknown' : _props$status,
        err = props.err,
        time = props.time,
        fatal = props.fatal;
      (0, _inc["default"])(this.stats, 'count');
      (0, _inc["default"])(this.stats, 'time', time);
      (0, _avg["default"])(this.stats, "statuses.".concat(status), time);
      if (fatal) (0, _inc["default"])(this.stats, 'fatal');
      if (err) (0, _avg["default"])(this.stats, "errors.".concat(err), time);
      this.emit('feedback', props, this);
    }
  }]);
  return Proxy;
}();
exports.Proxy = Proxy;
var _default = Proxy;
exports["default"] = _default;
//# sourceMappingURL=Proxy.js.map