use NotificationCenter

SeedCompany · Oct 10, 2024 · 12a134c · 12a134c
1 parent 305d9bf
commit 12a134c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/scenes/Root/Header/ProfileToolbar/ProfileToolbar.tsx b/src/scenes/Root/Header/ProfileToolbar/ProfileToolbar.tsx
@@ -2,7 +2,9 @@ import { AccountCircle, SupervisedUserCircle } from '@mui/icons-material';
 import { IconButton, MenuProps, Stack, Typography } from '@mui/material';
 import { useContext, useState } from 'react';
 import { ImpersonationContext } from '~/api/client/ImpersonationContext';
-import { useSession } from '../../../../components/Session';
+import { Feature } from '~/components/Feature';
+import { NotificationCenter } from '~/components/NotificationsCenter/NotificationsCenter';
+import { useSession } from '~/components/Session';
 import { ProfileMenu } from '../ProfileMenu';
 
 export const ProfileToolbar = () => {
@@ -13,6 +15,10 @@ export const ProfileToolbar = () => {
   return (
     <>
       <Stack direction="row" spacing={1} alignItems="center">
+        {/* Wrapping with FF here to avoid API call */}
+        <Feature flag="notifications" match>
+          <NotificationCenter />
+        </Feature>
         <Typography color="primary" sx={{ fontWeight: 'medium' }}>
           {session?.fullName}
         </Typography>