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

function _possibleConstructorReturn(self, call) { if (!self) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return call && (typeof call === "object" || typeof call === "function") ? call : self; }

function _inherits(subClass, superClass) { if (typeof superClass !== "function" && superClass !== null) { throw new TypeError("Super expression must either be null or a function, not " + typeof superClass); } subClass.prototype = Object.create(superClass && superClass.prototype, { constructor: { value: subClass, enumerable: false, writable: true, configurable: true } }); if (superClass) Object.setPrototypeOf ? Object.setPrototypeOf(subClass, superClass) : subClass.__proto__ = superClass; }

import Map from './map';

var MapWithDefault = function (_Map) {
  _inherits(MapWithDefault, _Map);

  function MapWithDefault(options) {
    _classCallCheck(this, MapWithDefault);

    var _this = _possibleConstructorReturn(this, _Map.call(this));

    _this.defaultValue = options.defaultValue;
    return _this;
  }

  MapWithDefault.prototype.get = function get(key) {
    var hasValue = this.has(key);

    if (hasValue) {
      return _Map.prototype.get.call(this, key);
    } else {
      var defaultValue = this.defaultValue(key);
      this.set(key, defaultValue);
      return defaultValue;
    }
  };

  return MapWithDefault;
}(Map);

export default MapWithDefault;