feat(overmind): reconfigure the instance on HMR

Author: christianalfoni

packages/node_modules/overmind/src/index.ts

@@ -186,7 +186,7 @@ export class Overmind<ThisConfig extends IConfiguration>
 
     if (IS_DEVELOPMENT && mode.mode === MODE_DEFAULT) {
       if (hotReloadingCache[name]) {
-        return hotReloadingCache[name]
+        return hotReloadingCache[name].reconfigure(configuration)
       } else {
         hotReloadingCache[name] = this
       }
@@ -204,24 +204,7 @@ export class Overmind<ThisConfig extends IConfiguration>
       Create the proxy state tree instance with the state and a wrapper to expose
       the eventHub
     */
-    const proxyStateTree = new ProxyStateTree(
-      this.getState(configuration) as any,
-      {
-        devmode: IS_DEVELOPMENT,
-        dynamicWrapper: (_, path, func) => func(eventHub, proxyStateTree, path),
-        onGetter: IS_DEVELOPMENT
-          ? (path, value) => {
-              // We need to let any initial values be set first
-              setTimeout(() => {
-                this.eventHub.emit(EventType.GETTER, {
-                  path,
-                  value: safeValue(value),
-                })
-              })
-            }
-          : undefined,
-      }
-    )
+    const proxyStateTree = this.createProxyStateTree(configuration, eventHub)
 
     this.state = proxyStateTree.state
     this.effects = configuration.effects || {}
@@ -262,7 +245,12 @@ export class Overmind<ThisConfig extends IConfiguration>
           ? 'NoName'
           : document.title || 'NoName'
 
-        this.initializeDevtools(host, name, eventHub, proxyStateTree)
+        this.initializeDevtools(
+          host,
+          name,
+          eventHub,
+          proxyStateTree.sourceState
+        )
       } else {
         warning +=
           '\n\n - You are not running on localhost. You will have to manually define the devtools option to connect'
@@ -334,6 +322,31 @@ export class Overmind<ThisConfig extends IConfiguration>
       this.initialized = Promise.resolve(null)
     }
   }
+  private createProxyStateTree(
+    configuration: IConfiguration,
+    eventHub: EventEmitter<any> | MockedEventEmitter
+  ) {
+    const proxyStateTree = new ProxyStateTree(
+      this.getState(configuration) as any,
+      {
+        devmode: IS_DEVELOPMENT,
+        dynamicWrapper: (_, path, func) => func(eventHub, proxyStateTree, path),
+        onGetter: IS_DEVELOPMENT
+          ? (path, value) => {
+              // We need to let any initial values be set first
+              setTimeout(() => {
+                this.eventHub.emit(EventType.GETTER, {
+                  path,
+                  value: safeValue(value),
+                })
+              })
+            }
+          : undefined,
+      }
+    )
+
+    return proxyStateTree
+  }
   private createExecution(name, action) {
     if (IS_PRODUCTION) {
       return ({
@@ -584,7 +597,7 @@ export class Overmind<ThisConfig extends IConfiguration>
       return result
     })
   }
-  private initializeDevtools(host, name, eventHub, proxyStateTree) {
+  private initializeDevtools(host, name, eventHub, initialState) {
     const devtools = new Devtools(name)
     devtools.connect(
       host,
@@ -605,7 +618,7 @@ export class Overmind<ThisConfig extends IConfiguration>
     devtools.send({
       type: 'init',
       data: {
-        state: proxyStateTree.sourceState,
+        state: initialState,
       },
     })
     this.devtools = devtools
@@ -717,6 +730,18 @@ export class Overmind<ThisConfig extends IConfiguration>
       }
     })
   }
+  reconfigure(configuration: IConfiguration) {
+    const proxyStateTree = this.proxyStateTree as any
+    this.proxyStateTree.sourceState = this.getState(configuration)
+    proxyStateTree.createTrackStateProxifier()
+    this.state = this.proxyStateTree.state as any
+    this.actions = this.getActions(configuration)
+    this.effects = configuration.effects || {}
+
+    this.proxyStateTree.forceFlush()
+
+    return this
+  }
 }
 
 /*