revoked company

telehire-client/src/components/Profile/Profile.tsx

@@ -18,24 +18,32 @@ import { User } from '@/models/User';
 import { ResumeAction } from '@/store/resume/ResumeActions';
 import { history } from '@/utils/history';
 import { FC, useMemo } from 'react';
-import { useDispatch, useSelector } from 'react-redux';
+import { useDispatch } from 'react-redux';
 import { CVWrapper, InactiveWrapper } from './Profile.styles';
-import { selectCurrentCompany } from '@/store/company/CompanySelectors';
 
 type Props = {
   user: User;
 };
 
 const Profile: FC<Props> = ({ user }) => {
   const dispatch = useDispatch();
-  const company = useSelector(selectCurrentCompany);
 
   useMainButton({
     onClick: () => history.push('/create-resume'),
     text: 'Create new resume',
     condition: user.loggedInAs === 'applicant',
   });
 
+  const company = useMemo(() => {
+    const currentCompanyId = localStorage.getItem(CURRENT_COMPANY_ID);
+
+    if (currentCompanyId && user.loggedInAs === 'company') {
+      return user?.companyList?.find(
+        (innerCompany: Company) => innerCompany.id === Number(currentCompanyId),
+      );
+    }
+  }, [user]);
+
   const activeResume = useMemo(() => {
     return user.resumes!.find(
       (innerResume) => innerResume.id === user.activeResumeId,
@@ -54,7 +62,7 @@ const Profile: FC<Props> = ({ user }) => {
           <SmallWrapper>
             <CompanyItem
               onClick={() => history.push('/companies/' + company?.id)}
-              company={company ?? undefined}
+              company={company}
             />
           </SmallWrapper>
         </BigWrapper>