fix: issue with Super Admin logging out after completing a quote BUN-…

…1172
bigcommerce · Jun 27, 2023 · b3cdf86 · b3cdf86
1 parent 93ee9ee
commit b3cdf86
Showing 1 changed file with 27 additions and 2 deletions.
diff --git a/apps/storefront/src/components/layout/B3AccountInfo.tsx b/apps/storefront/src/components/layout/B3AccountInfo.tsx
@@ -4,6 +4,8 @@ import { Box } from '@mui/material'
 
 import { useMobile } from '@/hooks'
 import { GlobaledContext } from '@/shared/global'
+import { superAdminEndMasquerade } from '@/shared/service/b2b'
+import { B3SStorage } from '@/utils'
 
 import B3DropDown from '../B3DropDown'
 
@@ -29,14 +31,37 @@ export default function B3AccountInfo({ closeSidebar }: B3AccountInfoProps) {
   const {
     state: {
       customer: { firstName = '', lastName = '' },
+      B3UserId,
+      salesRepCompanyId = 0,
+      isAgenting,
     },
+    dispatch,
   } = useContext(GlobaledContext)
 
   const navigate = useNavigate()
 
-  const handleItemClick = (item: ListProps) => {
+  const handleItemClick = async (item: ListProps) => {
     if (item.key === 'logout') {
-      navigate('/login?loginFlag=3')
+      try {
+        if (isAgenting) {
+          await superAdminEndMasquerade(+salesRepCompanyId, +B3UserId)
+
+          B3SStorage.delete('isAgenting')
+          B3SStorage.delete('salesRepCompanyId')
+          B3SStorage.delete('salesRepCompanyName')
+
+          dispatch({
+            type: 'common',
+            payload: {
+              salesRepCompanyId: '',
+              salesRepCompanyName: '',
+              isAgenting: false,
+            },
+          })
+        }
+      } finally {
+        navigate('/login?loginFlag=3')
+      }
     } else if (item.type === 'path') {
       navigate(item.key)
     }