From 6f0bd1e62ee5823753ef8bac37dbb4a969339288 Mon Sep 17 00:00:00 2001
From: ThibaudAv <thibaud.avenier@gmail.com>
Date: Mon, 23 Nov 2020 16:13:45 +0100
Subject: [PATCH] feat: add enableProdMode() in RenderNgAppService when
 NODE_ENV !== 'development'

---
 .../client/preview/angular/RenderNgAppService.ts   | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/app/angular/src/client/preview/angular/RenderNgAppService.ts b/app/angular/src/client/preview/angular/RenderNgAppService.ts
index cf20663e2f3..6d1b5b752f8 100644
--- 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$))
     );