1 files changed, 14 insertions, 2 deletions

diff --git a/client/src/main.ts b/client/src/main.ts
index f3825fe50..b02b6830f 100644
--- a/client/src/main.ts
+++ b/client/src/main.ts
@@ -4,10 +4,22 @@ import { platformBrowserDynamic } from '@angular/platform-browser-dynamic'
 import { AppModule } from './app/app.module'
 import { environment } from './environments/environment'
 
+import { hmrBootstrap } from './hmr'
+
 if (environment.production) {
   enableProdMode()
 }
 
-platformBrowserDynamic()
+const bootstrap = () => platformBrowserDynamic()
   .bootstrapModule(AppModule)
-  .catch(err => console.log(err))
+
+if (environment.hmr) {
+  if (module[ 'hot' ]) {
+    hmrBootstrap(module, bootstrap)
+  } else {
+    console.error('HMR is not enabled for webpack-dev-server!')
+    console.log('Are you using the --hmr flag for ng serve?')
+  }
+} else {
+  bootstrap()
+}