fix(wrapSelectors): allow chain of wrapping selectors

char0n · May 24, 2021 · 6f70609 · 6f70609
1 parent ea0f2b5
commit 6f70609
Showing 1 changed file with 40 additions and 18 deletions.
diff --git a/src/system/helpers.js b/src/system/helpers.js
@@ -37,33 +37,55 @@ export const systemExtend = (dest = {}, src = {}) => {
   }
 
   /**
-   * Account for wrapActions, make it an array and append to it.
+   * Account for wrapActions/wrapSelectors make it an array and append to it.
    * Modifies `src`.
    * 80% of this code is just safe traversal. We need to address that ( ie: use a lib ).
    */
   const { statePlugins } = dest;
   if (isPlainObject(statePlugins)) {
     keys(statePlugins).forEach((namespace) => {
       const namespaceObj = statePlugins[namespace];
-      if (!isPlainObject(namespaceObj) || !isPlainObject(namespaceObj.wrapActions)) {
+      if (!isPlainObject(namespaceObj)) {
         return;
       }
-      const { wrapActions } = namespaceObj;
-      keys(wrapActions).forEach((actionName) => {
-        let action = wrapActions[actionName];
-
-        // this should only happen if dest is the first plugin, since invocations after that will ensure its an array
-        if (isNotArray(action)) {
-          action = [action];
-          wrapActions[actionName] = action; // put the value inside an array
-        }
-
-        if (hasPath(['statePlugins', namespace, 'wrapActions', actionName], src)) {
-          src.statePlugins[namespace].wrapActions[actionName] = wrapActions[actionName].concat(
-            src.statePlugins[namespace].wrapActions[actionName]
-          );
-        }
-      });
+      const { wrapActions, wrapSelectors } = namespaceObj;
+      // process action wrapping
+      if (isPlainObject(wrapActions)) {
+        keys(wrapActions).forEach((actionName) => {
+          let action = wrapActions[actionName];
+
+          // this should only happen if dest is the first plugin, since invocations after that will ensure its an array
+          if (isNotArray(action)) {
+            action = [action];
+            wrapActions[actionName] = action; // put the value inside an array
+          }
+
+          if (hasPath(['statePlugins', namespace, 'wrapActions', actionName], src)) {
+            src.statePlugins[namespace].wrapActions[actionName] = wrapActions[actionName].concat(
+              src.statePlugins[namespace].wrapActions[actionName]
+            );
+          }
+        });
+      }
+
+      // process selector wrapping
+      if (isPlainObject(wrapSelectors)) {
+        keys(wrapSelectors).forEach((selectorName) => {
+          let selector = wrapSelectors[selectorName];
+
+          // this should only happen if dest is the first plugin, since invocations after that will ensure its an array
+          if (isNotArray(selector)) {
+            selector = [selector];
+            wrapSelectors[selectorName] = selector; // put the value inside an array
+          }
+
+          if (hasPath(['statePlugins', namespace, 'wrapSelectors', selectorName], src)) {
+            src.statePlugins[namespace].wrapSelectors[selectorName] = wrapSelectors[
+              selectorName
+            ].concat(src.statePlugins[namespace].wrapSelectors[selectorName]);
+          }
+        });
+      }
     });
   }
   /* eslint-enable no-param-reassign */