Allow returning arbitrary data from the handleAction function

knpwrs · Feb 5, 2019 · 79f945e · 79f945e
1 parent 8167bba
commit 79f945e
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/handle-action.ts b/src/handle-action.ts
@@ -23,8 +23,14 @@ export default function handleAction<S, AC extends TsActionCreator<any> = any>(
   return (state: S | undefined, action: ReturnType<AC>) => {
     if (action.type === ac.type && state) {
       const draft = createDraft(state);
-      re(draft, action);
-      return finishDraft(draft);
+      const reResult = re(draft, action);
+      const finishedDraft = finishDraft(draft);
+
+      if (finishedDraft === state && reResult !== undefined) {
+        return reResult;
+      } else {
+        return finishedDraft;
+      }
     }
     return (state || s) as any;
   };