From 624da203cef4caa820861a0facfb0b1f63a12845 Mon Sep 17 00:00:00 2001
From: Ryan Mathis <rmathis@unicon.net>
Date: Thu, 10 Jun 2021 09:36:20 -0700
Subject: [PATCH] SHIBUI-1943/1945 user role fixes

---
 ui/src/app/core/components/Header.js             | 16 +++++++++++-----
 .../metadata/wizard/MetadataProviderWizard.js    |  1 +
 .../app/metadata/wizard/MetadataSourceWizard.js  |  1 +
 3 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/ui/src/app/core/components/Header.js b/ui/src/app/core/components/Header.js
index 69b3c7020..54d0c402c 100644
--- a/ui/src/app/core/components/Header.js
+++ b/ui/src/app/core/components/Header.js
@@ -9,12 +9,16 @@ import { FontAwesomeIcon } from '@fortawesome/react-fontawesome';
 import { faTh, faSignOutAlt, faPlusCircle, faCube, faCubes } from '@fortawesome/free-solid-svg-icons';
 
 import Translate from '../../i18n/components/translate';
-import { useTranslation } from '../../i18n/hooks';
+import { useTranslator } from '../../i18n/hooks';
 
 import { brand } from '../../app.brand';
+import { useIsAdmin } from '../user/UserContext';
 
 export function Header () {
-    const logoutLabel = useTranslation('action.logout');
+
+    const translator = useTranslator();
+
+    const isAdmin = useIsAdmin();
 
     return (
         <Navbar expand="md" fixed="top" bg="">
@@ -26,6 +30,7 @@ export function Header () {
             <Navbar.Toggle aria-controls="basic-navbar-nav" />
             <Navbar.Collapse id="basic-navbar-nav">
                 <Nav className="ml-auto align-items-center" navbar>
+                    {isAdmin &&
                     <Dropdown className="mr-2" id="basic-nav-dropdown">
                         <Dropdown.Toggle variant="outline-primary" id="dropdown-basic" size="sm">
                             <FontAwesomeIcon icon={faPlusCircle} className="mr-2" />
@@ -38,6 +43,7 @@ export function Header () {
                             </Dropdown.Item>
                         </Dropdown.Menu>
                     </Dropdown>
+                    }
                     <Dropdown className="" id="basic-nav-dropdown">
                         <Dropdown.Toggle variant="outline-primary" id="dropdown-basic" size="sm">
                             <FontAwesomeIcon icon={faPlusCircle} className="mr-2" />
@@ -48,10 +54,10 @@ export function Header () {
                                 <FontAwesomeIcon icon={faCube} className="mr-2" />
                                 <Translate value="action.add-new-source" />
                             </Dropdown.Item>
-                            <Dropdown.Item as={Link} to="/metadata/provider/new" className="text-primary py-2">
+                            {isAdmin && <Dropdown.Item as={Link} to="/metadata/provider/new" className="text-primary py-2">
                                 <FontAwesomeIcon icon={faCubes} className="mr-2" />
                                 <Translate value="action.add-new-provider" />
-                            </Dropdown.Item>
+                            </Dropdown.Item> }
                         </Dropdown.Menu>
                     </Dropdown>
                     <Link to="/dashboard" className="nav-link" aria-label="Metadata Dashboard">
@@ -59,7 +65,7 @@ export function Header () {
                         <FontAwesomeIcon icon={faTh} className="mr-2" />
                         <Translate value="action.dashboard">Dashboard</Translate>
                     </Link>
-                    <Nav.Link href="/logout" target="_self" aria-label={logoutLabel}>
+                    <Nav.Link href="/logout" target="_self" aria-label={translator('action.logout')}>
                         <FontAwesomeIcon icon={faSignOutAlt} className="mr-2" />
                         <Translate value="action.logout">Logout</Translate>
                     </Nav.Link>
diff --git a/ui/src/app/metadata/wizard/MetadataProviderWizard.js b/ui/src/app/metadata/wizard/MetadataProviderWizard.js
index 752a9be67..b9695a02c 100644
--- a/ui/src/app/metadata/wizard/MetadataProviderWizard.js
+++ b/ui/src/app/metadata/wizard/MetadataProviderWizard.js
@@ -57,6 +57,7 @@ export function MetadataProviderWizard({onRestart}) {
         const body = removeNull(definition.parser(metadata), true);
         await post('', body);
         if (response.ok) {
+            setBlocking(false);
             history.push('/dashboard/metadata/manager/providers');
         } else {
             const { errorCode, errorMessage, cause } = response.data;
diff --git a/ui/src/app/metadata/wizard/MetadataSourceWizard.js b/ui/src/app/metadata/wizard/MetadataSourceWizard.js
index b329f1e36..5173d02fe 100644
--- a/ui/src/app/metadata/wizard/MetadataSourceWizard.js
+++ b/ui/src/app/metadata/wizard/MetadataSourceWizard.js
@@ -61,6 +61,7 @@ export function MetadataSourceWizard ({ onShowNav }) {
         const body = removeNull(metadata, true);
         await post('', body);
         if (response.ok) {
+            setBlocking(false);
             history.push('/');
         } else {
             console.log(response.body);