Fix rendering code to be Flow proof

1 files changed, 11 insertions, 3 deletions

diff --git a/app/app.js b/app/app.js
index 40440efa3b..c3d5acbcb8 100644
--- a/app/app.js
+++ b/app/app.js
@@ -75,8 +75,6 @@ backend.on('disconnect', updateTrayIcon);
 // force update tray
 updateTrayIcon();
 
-const rootElement = document.querySelector(document.currentScript.getAttribute('data-container'));
-
 // disable smart pinch.
 webFrame.setZoomLevelLimits(1, 1);
 
@@ -91,6 +89,16 @@ if(navigator.serviceWorker) {
 
 ipcRenderer.send('on-browser-window-ready');
 
+const containerId = document.currentScript.getAttribute('data-container');
+if(!containerId) {
+  throw new Error('Missing data-container attribute.');
+}
+
+const rootElement = document.querySelector(containerId);
+if(!rootElement) {
+  throw new Error('Missing root element.');
+}
+
 ReactDOM.render(
   <Provider store={ store }>
     <ConnectedRouter history={ memoryHistory }>
@@ -98,4 +106,4 @@ ReactDOM.render(
     </ConnectedRouter>
   </Provider>,
   rootElement
-);
+);
\ No newline at end of file