Improve webview loader

1 files changed, 18 insertions, 10 deletions

diff --git a/src/components/ui/WebviewLoader/index.js b/src/components/ui/WebviewLoader/index.js
index 950312bbb..ab17199e5 100644
--- a/src/components/ui/WebviewLoader/index.js
+++ b/src/components/ui/WebviewLoader/index.js
@@ -1,16 +1,24 @@
-import React from 'react';
+import React, { Component } from 'react';
 import PropTypes from 'prop-types';
+import injectSheet from 'react-jss';
 
 import FullscreenLoader from '../FullscreenLoader';
 
-const WebviewLoader = ({ name }) => (
-  <FullscreenLoader
-    title={`Loading ${name}`}
-  />
-);
+import styles from './styles';
 
-WebviewLoader.propTypes = {
-  name: PropTypes.string.isRequired,
-};
+export default @injectSheet(styles) class WebviewLoader extends Component {
+  static propTypes = {
+    name: PropTypes.string.isRequired,
+    classes: PropTypes.object.isRequired,
+  }
 
-export default WebviewLoader;
+  render() {
+    const { classes, name } = this.props;
+    return (
+      <FullscreenLoader
+        className={classes.component}
+        title={`Loading ${name}`}
+      />
+    );
+  }
+}