Fix SatelliteToken org admin check to handle if user info is not yet …

…loaded
RedHatInsights · Sep 13, 2024 · 0bf8ad8 · 0bf8ad8
1 parent 3e365d1
commit 0bf8ad8
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/layouts/SatelliteToken.tsx b/src/layouts/SatelliteToken.tsx
@@ -9,19 +9,24 @@ import { Page, PageSection } from '@patternfly/react-core/dist/dynamic/component
 import SatelliteTable from '../components/Satellite/SatelliteTable';
 import IPWhitelistTable from '../components/Satellite/IPWhitelistTable';
 import { getEnv } from '../utils/common';
-import ChromeAuthContext from '../auth/ChromeAuthContext';
+import useChrome from '@redhat-cloud-services/frontend-components/useChrome';
 import NotFoundRoute from '../components/NotFoundRoute';
 
 const SatelliteToken: React.FC = () => {
   const [token, setToken] = useState('');
   const [error, setError] = useState(null);
-  const { user } = useContext(ChromeAuthContext);
+  const [isOrgAdmin, setIsOrgAdmin] = useState<boolean>(false);
+  const { auth } = useChrome();
   const isITLess = useFlag('platform.chrome.itless');
 
   if (!isITLess) {
     return <NotFoundRoute />;
   }
 
+  useEffect(() => {
+    auth.getUser().then((user) => user && setIsOrgAdmin(!!user?.identity?.user?.is_org_admin));
+  }, []);
+
   const generateToken = () => {
     axios
       .get('/api/identity/certificate/token')
@@ -90,7 +95,7 @@ const SatelliteToken: React.FC = () => {
             </CardBody>
           </Card>
         </PageSection>
-        {user.identity.user?.is_org_admin ? (
+        {isOrgAdmin ? (
           <PageSection>
             <Card>
               <CardTitle>IP Address Allow List</CardTitle>