index.js 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.createSelector = void 0;
  6. exports.createSelectorCreator = createSelectorCreator;
  7. exports.createStructuredSelector = void 0;
  8. Object.defineProperty(exports, "defaultEqualityCheck", {
  9. enumerable: true,
  10. get: function get() {
  11. return _defaultMemoize.defaultEqualityCheck;
  12. }
  13. });
  14. Object.defineProperty(exports, "defaultMemoize", {
  15. enumerable: true,
  16. get: function get() {
  17. return _defaultMemoize.defaultMemoize;
  18. }
  19. });
  20. var _defaultMemoize = require("./defaultMemoize");
  21. function getDependencies(funcs) {
  22. var dependencies = Array.isArray(funcs[0]) ? funcs[0] : funcs;
  23. if (!dependencies.every(function (dep) {
  24. return typeof dep === 'function';
  25. })) {
  26. var dependencyTypes = dependencies.map(function (dep) {
  27. return typeof dep === 'function' ? "function " + (dep.name || 'unnamed') + "()" : typeof dep;
  28. }).join(', ');
  29. throw new Error("createSelector expects all input-selectors to be functions, but received the following types: [" + dependencyTypes + "]");
  30. }
  31. return dependencies;
  32. }
  33. function createSelectorCreator(memoize) {
  34. for (var _len = arguments.length, memoizeOptionsFromArgs = new Array(_len > 1 ? _len - 1 : 0), _key = 1; _key < _len; _key++) {
  35. memoizeOptionsFromArgs[_key - 1] = arguments[_key];
  36. }
  37. var createSelector = function createSelector() {
  38. for (var _len2 = arguments.length, funcs = new Array(_len2), _key2 = 0; _key2 < _len2; _key2++) {
  39. funcs[_key2] = arguments[_key2];
  40. }
  41. var _recomputations = 0;
  42. var _lastResult; // Due to the intricacies of rest params, we can't do an optional arg after `...funcs`.
  43. // So, start by declaring the default value here.
  44. // (And yes, the words 'memoize' and 'options' appear too many times in this next sequence.)
  45. var directlyPassedOptions = {
  46. memoizeOptions: undefined
  47. }; // Normally, the result func or "output selector" is the last arg
  48. var resultFunc = funcs.pop(); // If the result func is actually an _object_, assume it's our options object
  49. if (typeof resultFunc === 'object') {
  50. directlyPassedOptions = resultFunc; // and pop the real result func off
  51. resultFunc = funcs.pop();
  52. }
  53. if (typeof resultFunc !== 'function') {
  54. throw new Error("createSelector expects an output function after the inputs, but received: [" + typeof resultFunc + "]");
  55. } // Determine which set of options we're using. Prefer options passed directly,
  56. // but fall back to options given to createSelectorCreator.
  57. var _directlyPassedOption = directlyPassedOptions,
  58. _directlyPassedOption2 = _directlyPassedOption.memoizeOptions,
  59. memoizeOptions = _directlyPassedOption2 === void 0 ? memoizeOptionsFromArgs : _directlyPassedOption2; // Simplifying assumption: it's unlikely that the first options arg of the provided memoizer
  60. // is an array. In most libs I've looked at, it's an equality function or options object.
  61. // Based on that, if `memoizeOptions` _is_ an array, we assume it's a full
  62. // user-provided array of options. Otherwise, it must be just the _first_ arg, and so
  63. // we wrap it in an array so we can apply it.
  64. var finalMemoizeOptions = Array.isArray(memoizeOptions) ? memoizeOptions : [memoizeOptions];
  65. var dependencies = getDependencies(funcs);
  66. var memoizedResultFunc = memoize.apply(void 0, [function recomputationWrapper() {
  67. _recomputations++; // apply arguments instead of spreading for performance.
  68. return resultFunc.apply(null, arguments);
  69. }].concat(finalMemoizeOptions)); // If a selector is called with the exact same arguments we don't need to traverse our dependencies again.
  70. var selector = memoize(function dependenciesChecker() {
  71. var params = [];
  72. var length = dependencies.length;
  73. for (var i = 0; i < length; i++) {
  74. // apply arguments instead of spreading and mutate a local list of params for performance.
  75. // @ts-ignore
  76. params.push(dependencies[i].apply(null, arguments));
  77. } // apply arguments instead of spreading for performance.
  78. _lastResult = memoizedResultFunc.apply(null, params);
  79. return _lastResult;
  80. });
  81. Object.assign(selector, {
  82. resultFunc: resultFunc,
  83. memoizedResultFunc: memoizedResultFunc,
  84. dependencies: dependencies,
  85. lastResult: function lastResult() {
  86. return _lastResult;
  87. },
  88. recomputations: function recomputations() {
  89. return _recomputations;
  90. },
  91. resetRecomputations: function resetRecomputations() {
  92. return _recomputations = 0;
  93. }
  94. });
  95. return selector;
  96. }; // @ts-ignore
  97. return createSelector;
  98. }
  99. var createSelector = /* #__PURE__ */createSelectorCreator(_defaultMemoize.defaultMemoize);
  100. exports.createSelector = createSelector;
  101. // Manual definition of state and output arguments
  102. var createStructuredSelector = function createStructuredSelector(selectors, selectorCreator) {
  103. if (selectorCreator === void 0) {
  104. selectorCreator = createSelector;
  105. }
  106. if (typeof selectors !== 'object') {
  107. throw new Error('createStructuredSelector expects first argument to be an object ' + ("where each property is a selector, instead received a " + typeof selectors));
  108. }
  109. var objectKeys = Object.keys(selectors);
  110. var resultSelector = selectorCreator( // @ts-ignore
  111. objectKeys.map(function (key) {
  112. return selectors[key];
  113. }), function () {
  114. for (var _len3 = arguments.length, values = new Array(_len3), _key3 = 0; _key3 < _len3; _key3++) {
  115. values[_key3] = arguments[_key3];
  116. }
  117. return values.reduce(function (composition, value, index) {
  118. composition[objectKeys[index]] = value;
  119. return composition;
  120. }, {});
  121. });
  122. return resultSelector;
  123. };
  124. exports.createStructuredSelector = createStructuredSelector;