1 files changed, 16 insertions, 0 deletions

diff --git a/client/src/hmr.ts b/client/src/hmr.ts
new file mode 100644
index 000000000..4d707a250
--- /dev/null
+++ b/client/src/hmr.ts
@@ -0,0 +1,16 @@
+import { NgModuleRef, ApplicationRef } from '@angular/core'
+import { createNewHosts } from '@angularclass/hmr'
+
+export const hmrBootstrap = (module: any, bootstrap: () => Promise<NgModuleRef<any>>) => {
+  let ngModule: NgModuleRef<any>
+  module.hot.accept()
+  bootstrap()
+    .then(mod => ngModule = mod)
+  module.hot.dispose(() => {
+    const appRef: ApplicationRef = ngModule.injector.get(ApplicationRef)
+    const elements = appRef.components.map(c => c.location.nativeElement)
+    const makeVisible = createNewHosts(elements)
+    ngModule.destroy()
+    makeVisible()
+  })
+}