From a7f9808cfb1cbb24e5b0fca7437b29a9390a219b Mon Sep 17 00:00:00 2001
From: Nikhil Nandgopal <nikhil.nandagopal@curefit.com>
Date: Sun, 10 Feb 2019 18:51:19 +0530
Subject: [PATCH] moved widget building into registry

---
 app/client/src/App.tsx                  |  2 ++
 app/client/src/utils/WidgetRegistry.tsx | 18 ++++++++++++------
 2 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/app/client/src/App.tsx b/app/client/src/App.tsx
index 05d4e7f0ad..a3205f3091 100755
--- a/app/client/src/App.tsx
+++ b/app/client/src/App.tsx
@@ -3,7 +3,9 @@ import logo from './assets/images/logo.svg';
 import './App.css';
 import { createStore } from 'redux'
 import appReducer from './reducers';
+import WidgetBuilderRegistry from "./utils/WidgetRegistry"
 
+WidgetBuilderRegistry.registerWidgetBuilders()
 const store = createStore(appReducer)
 
 class App extends Component {
diff --git a/app/client/src/utils/WidgetRegistry.tsx b/app/client/src/utils/WidgetRegistry.tsx
index b660d94801..bc1d758601 100644
--- a/app/client/src/utils/WidgetRegistry.tsx
+++ b/app/client/src/utils/WidgetRegistry.tsx
@@ -5,10 +5,16 @@ import ContainerWidget, {
 import { IContainerProps } from "../editorComponents/ContainerComponent"
 import WidgetFactory from "./WidgetFactory"
 
-WidgetFactory.registerWidgetBuilder("CONTAINER_WIDGET", {
-  buildWidget(
-    widgetData: IContainerWidgetProps
-  ): BaseWidget<IContainerWidgetProps, IContainerProps> {
-    return new ContainerWidget(widgetData)
+class WidgetBuilderRegistry {
+  static registerWidgetBuilders() {
+    WidgetFactory.registerWidgetBuilder("CONTAINER_WIDGET", {
+      buildWidget(
+        widgetData: IContainerWidgetProps
+      ): BaseWidget<IContainerWidgetProps, IContainerProps> {
+        return new ContainerWidget(widgetData)
+      }
+    })
   }
-})
+}
+
+export default WidgetBuilderRegistry