lodash-decorators/utils/CompositeKeyWeakMap.js

A collection of decorators using lodash at it's core.

'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; }; })();

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 _lodashLangIsUndefined = require('lodash/lang/isUndefined');

var _lodashLangIsUndefined2 = _interopRequireDefault(_lodashLangIsUndefined);

var CompositeKeyWeakMap = (function () {
  function CompositeKeyWeakMap() {
    _classCallCheck(this, CompositeKeyWeakMap);

    this.weakMap = new WeakMap();
  }

  _createClass(CompositeKeyWeakMap, [{
    key: 'set',
    value: function set(keys, value) {
      var map = this.weakMap;

      for (var i = 0, len = keys.length - 1; i < len; i++) {
        var key = keys[i];
        var next = map.get(keys[i]);

        if (!next) {
          next = new Map();
          map.set(keys[i], next);
        }

        map = next;
      }

      map.set(keys[keys.length - 1], value);
    }
  }, {
    key: 'get',
    value: function get(keys) {
      var next = this.weakMap;

      for (var i = 0, len = keys.length; i < len; i++) {
        next = next.get(keys[i]);

        if ((0, _lodashLangIsUndefined2['default'])(next)) {
          break;
        }
      }

      return next;
    }
  }, {
    key: 'has',
    value: function has(keys) {
      return !(0, _lodashLangIsUndefined2['default'])(this.get(keys));
    }
  }]);

  return CompositeKeyWeakMap;
})();

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