fix: Add a new API for how to register the redux store to fix an issu…

…e with dispatching from reactotron into the middleware.
infinitered · Aug 25, 2019 · bdd8702 · bdd8702
1 parent 00de295
commit bdd8702
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/src/enhancer.ts b/src/enhancer.ts
@@ -9,7 +9,7 @@ export default function createEnhancer(
   pluginConfig: PluginConfig,
   handleStoreCreation: () => void
 ) {
-  return () => createStore => (reducer, ...args) => {
+  return (skipSettingStore = false) => createStore => (reducer, ...args) => {
     const originalStore = createStore(
       reactotronReducer(reducer, pluginConfig.restoreActionType),
       ...args
@@ -18,9 +18,11 @@ export default function createEnhancer(
       ...originalStore,
       dispatch: createCustomDispatch(reactotron, originalStore, pluginConfig),
     }
-    reactotron.reduxStore = store
 
-    handleStoreCreation()
+    if (!skipSettingStore) {
+      reactotron.reduxStore = store
+      handleStoreCreation()
+    }
 
     return store
   }

diff --git a/src/index.ts b/src/index.ts
@@ -28,6 +28,10 @@ function reactotronRedux(pluginConfig: PluginConfig = {}) {
       onCommand: createCommandHander(reactotron, mergedPluginConfig, onReduxStoreCreation),
       features: {
         createEnhancer: createEnhancer(reactotron, mergedPluginConfig, handleStoreCreation),
+        setReduxStore: store => {
+          reactotron.reduxStore = store
+          handleStoreCreation()
+        },
         reportReduxAction: createSendAction(reactotron),
       },
     }
@@ -44,6 +48,11 @@ declare module "reactotron-core-client" {
     /**
      * Enhancer creator
      */
-    createEnhancer?: () => StoreEnhancer
+    createEnhancer?: (skipSettingStore?: boolean) => StoreEnhancer
+
+    /**
+     * Store setter
+     */
+    setReduxStore?: (store: any) => void
   }
 }