feat: add enableProdMode() in RenderNgAppService when NODE_ENV !== 'd…

…evelopment'
storybookjs · Nov 23, 2020 · 6f0bd1e · 6f0bd1e
1 parent ead9581
commit 6f0bd1e
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/app/angular/src/client/preview/angular/RenderNgAppService.ts b/app/angular/src/client/preview/angular/RenderNgAppService.ts
@@ -1,5 +1,5 @@
 /* eslint-disable no-undef */
-import { NgModule, NO_ERRORS_SCHEMA, Type } from '@angular/core';
+import { enableProdMode, NgModule, NO_ERRORS_SCHEMA, PlatformRef, Type } from '@angular/core';
 import { BrowserModule } from '@angular/platform-browser';
 import { platformBrowserDynamic } from '@angular/platform-browser-dynamic';
 
@@ -28,7 +28,7 @@ export class RenderNgAppService {
 
   public static SELECTOR_STORYBOOK_WRAPPER = 'storybook-wrapper';
 
-  private platform = platformBrowserDynamic();
+  private platform: PlatformRef;
 
   private staticRoot = document.getElementById('root');
 
@@ -77,6 +77,16 @@ export class RenderNgAppService {
     }
     this.storyProps$ = new BehaviorSubject<ICollection>(storyObj.props);
 
+    if (typeof NODE_ENV === 'string' && NODE_ENV !== 'development') {
+      try {
+        enableProdMode();
+      } catch (e) {
+        // eslint-disable-next-line no-console
+        console.error(e);
+      }
+    }
+    // platform should be set after enableProdMode()
+    this.platform = platformBrowserDynamic();
     await this.platform.bootstrapModule(
       createModuleFromMetadata(getNgModuleMetadata(storyObj, this.storyProps$))
     );