es6.weak-map.js 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. 'use strict';
  2. var each = require('./_array-methods')(0)
  3. , redefine = require('./_redefine')
  4. , meta = require('./_meta')
  5. , assign = require('./_object-assign')
  6. , weak = require('./_collection-weak')
  7. , isObject = require('./_is-object')
  8. , has = require('./_has')
  9. , getWeak = meta.getWeak
  10. , isExtensible = Object.isExtensible
  11. , uncaughtFrozenStore = weak.ufstore
  12. , tmp = {}
  13. , InternalMap;
  14. var wrapper = function(get){
  15. return function WeakMap(){
  16. return get(this, arguments.length > 0 ? arguments[0] : undefined);
  17. };
  18. };
  19. var methods = {
  20. // 23.3.3.3 WeakMap.prototype.get(key)
  21. get: function get(key){
  22. if(isObject(key)){
  23. var data = getWeak(key);
  24. if(data === true)return uncaughtFrozenStore(this).get(key);
  25. return data ? data[this._i] : undefined;
  26. }
  27. },
  28. // 23.3.3.5 WeakMap.prototype.set(key, value)
  29. set: function set(key, value){
  30. return weak.def(this, key, value);
  31. }
  32. };
  33. // 23.3 WeakMap Objects
  34. var $WeakMap = module.exports = require('./_collection')('WeakMap', wrapper, methods, weak, true, true);
  35. // IE11 WeakMap frozen keys fix
  36. if(new $WeakMap().set((Object.freeze || Object)(tmp), 7).get(tmp) != 7){
  37. InternalMap = weak.getConstructor(wrapper);
  38. assign(InternalMap.prototype, methods);
  39. meta.NEED = true;
  40. each(['delete', 'has', 'get', 'set'], function(key){
  41. var proto = $WeakMap.prototype
  42. , method = proto[key];
  43. redefine(proto, key, function(a, b){
  44. // store frozen objects on internal weakmap shim
  45. if(isObject(a) && !isExtensible(a)){
  46. if(!this._f)this._f = new InternalMap;
  47. var result = this._f[key](a, b);
  48. return key == 'set' ? this : result;
  49. // store all the rest on native weakmap
  50. } return method.call(this, a, b);
  51. });
  52. });
  53. }