From bd510e00a2425a76afd6a4f87f7f2c53ec09159f Mon Sep 17 00:00:00 2001
From: OFCL <fcarrillo@arkusnexus.com>
Date: Fri, 30 Jul 2021 17:43:03 -0700
Subject: [PATCH] Added custom loading prop

---
 src/components/file-viewer.jsx | 8 +++++---
 src/components/loading.jsx     | 4 +++-
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/components/file-viewer.jsx b/src/components/file-viewer.jsx
index f1a524e8..35c4e713 100644
--- a/src/components/file-viewer.jsx
+++ b/src/components/file-viewer.jsx
@@ -83,16 +83,18 @@ class FileViewer extends Component {
 }
 
 FileViewer.propTypes = {
-  fileType: PropTypes.string.isRequired,
+  errorComponent: PropTypes.element,
   filePath: PropTypes.string.isRequired,
+  fileType: PropTypes.string.isRequired,
+  loadingComponent: PropTypes.element,
   onError: PropTypes.func,
-  errorComponent: PropTypes.element,
   unsupportedComponent: PropTypes.element,
 };
 
 FileViewer.defaultProps = {
-  onError: () => null,
   errorComponent: null,
+  loadingComponent: null,
+  onError: () => null,
   unsupportedComponent: null,
 };
 
diff --git a/src/components/loading.jsx b/src/components/loading.jsx
index f51433b7..da78175c 100644
--- a/src/components/loading.jsx
+++ b/src/components/loading.jsx
@@ -6,7 +6,9 @@ import 'styles/loading.scss';
 
 const Loading = () => (
   <div className="loading-container">
-    <span className="loading" />
+    {props.loadingComponent
+      ? <props.loadingComponent {...props} />
+      : <span className="loading" />}
   </div>
 );