hide jwt page if special key is used (#232)

* hide jwt page if special key is used * fmt
qdrant · Sep 9, 2024 · 1b76c4f · 1b76c4f
1 parent 54f9e8c
commit 1b76c4f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/src/components/Sidebar/Sidebar.jsx b/src/components/Sidebar/Sidebar.jsx
@@ -57,7 +57,7 @@ const Drawer = styled(MuiDrawer, {
   }),
 }));
 
-export default function Sidebar({ open, version, jwtEnabled }) {
+export default function Sidebar({ open, version, jwtEnabled, jwtVisible }) {
   return (
     <Drawer variant="permanent" open={open}>
       <DrawerHeader />
@@ -69,7 +69,8 @@ export default function Sidebar({ open, version, jwtEnabled }) {
           <SidebarTutorialSection isSidebarOpen={open} />
         </ListItem>
         {sidebarItem('Datasets', <Animation />, '/datasets', open)}
-        {sidebarItem('Access Tokens', <Key />, '/jwt', open, jwtEnabled)}
+
+        {jwtVisible && sidebarItem('Access Tokens', <Key />, '/jwt', open, jwtEnabled)}
       </List>
       <List style={{ marginTop: `auto` }}>
         <ListItem>
@@ -116,4 +117,5 @@ Sidebar.propTypes = {
   open: PropTypes.bool,
   version: PropTypes.string,
   jwtEnabled: PropTypes.bool,
+  jwtVisible: PropTypes.bool,
 };
diff --git a/src/pages/Home.jsx b/src/pages/Home.jsx
@@ -28,6 +28,7 @@ export default function MiniDrawer() {
   const [open, setOpen] = React.useState(false);
   const [version, setVersion] = useState('???');
   const [jwtEnabled, setJwtEnabled] = useState(false);
+  const [jwtVisible, setJwtVisible] = useState(true);
   const colorMode = React.useContext(ColorModeContext);
 
   const [apiKeyDialogOpen, setApiKeyDialogOpen] = useState(false);
@@ -38,6 +39,10 @@ export default function MiniDrawer() {
       const telemetry = await qdrantClient.api('service').telemetry();
       setVersion(telemetry.data.result.app.version);
       setJwtEnabled(telemetry.data.result.app?.jwt_rbac || false);
+
+      if (telemetry.data.result.app?.hide_jwt_dashboard) {
+        setJwtVisible(false);
+      }
     } catch (error) {
       if (error.status === 403 || error.status === 401) {
         setApiKeyDialogOpen(true);
@@ -98,7 +103,7 @@ export default function MiniDrawer() {
           </Tooltip>
         </Toolbar>
       </AppBar>
-      <Sidebar open={open} version={version} jwtEnabled={jwtEnabled} />
+      <Sidebar open={open} version={version} jwtEnabled={jwtEnabled} jwtVisible={jwtVisible} />
       <Box component="main" sx={{ flexGrow: 1, overflow: 'hidden' }}>
         <DrawerHeader />
         <Outlet />