feat: store and remember the latest selected workspace and team (#2686)

ever-co · Jul 5, 2024 · 88332dd · 88332dd
1 parent 4f6c3c2
commit 88332dd
Show file tree

Hide file tree

Showing 4 changed files with 55 additions and 3 deletions.
diff --git a/apps/web/app/[locale]/auth/passcode/component.tsx b/apps/web/app/[locale]/auth/passcode/component.tsx
@@ -26,6 +26,7 @@ import stc from 'string-to-color';
 import { ScrollArea, ScrollBar } from '@components/ui/scroll-bar';
 import SocialLogins from '../social-logins-buttons';
 import { useSession } from 'next-auth/react';
+import { LAST_WORSPACE_AND_TEAM, USER_SAW_OUTSTANDING_NOTIFICATION } from '@app/constants';
 
 function AuthPasscode() {
 	const form = useAuthenticationPasscode();
@@ -317,7 +318,8 @@ function WorkSpaceScreen({ form, className }: { form: TAuthenticationPasscode }
 		(e: any) => {
 			if (typeof selectedWorkspace !== 'undefined') {
 				form.handleWorkspaceSubmit(e, form.workspaces[selectedWorkspace].token, selectedTeam);
-				window && window?.localStorage.removeItem('user-saw-notif');
+				window && window?.localStorage.removeItem(USER_SAW_OUTSTANDING_NOTIFICATION);
+				window && window?.localStorage.setItem(LAST_WORSPACE_AND_TEAM, selectedTeam);
 			}
 		},
 		[selectedWorkspace, selectedTeam, form]
@@ -332,6 +334,14 @@ function WorkSpaceScreen({ form, className }: { form: TAuthenticationPasscode }
 
 		if (form.workspaces.length === 1 && currentTeams?.length === 1) {
 			setSelectedTeam(currentTeams[0].team_id);
+		} else {
+			const lastSelectedTeam = window.localStorage.getItem(LAST_WORSPACE_AND_TEAM) || currentTeams[0].team_id;
+			const lastSelectedWorkspace =
+				form.workspaces.findIndex((workspace) =>
+					workspace.current_teams.find((team) => team.team_id === lastSelectedTeam)
+				) || 0;
+			setSelectedTeam(lastSelectedTeam);
+			setSelectedWorkspace(lastSelectedWorkspace);
 		}
 
 		if (form.workspaces.length === 1 && (currentTeams?.length || 0) <= 1) {

diff --git a/apps/web/app/[locale]/auth/password/component.tsx b/apps/web/app/[locale]/auth/password/component.tsx
@@ -12,6 +12,7 @@ import { useRouter } from 'next/navigation';
 import { useCallback, useEffect, useState } from 'react';
 import { WorkSpaceComponent } from '../passcode/component';
 import SocialLogins from '../social-logins-buttons';
+import { LAST_WORSPACE_AND_TEAM, USER_SAW_OUTSTANDING_NOTIFICATION } from '@app/constants';
 
 export default function AuthPassword() {
 	const t = useTranslations();
@@ -108,7 +109,8 @@ function WorkSpaceScreen({ form, className }: { form: TAuthenticationPassword }
 		(e: any) => {
 			if (typeof selectedWorkspace !== 'undefined') {
 				form.handleWorkspaceSubmit(e, form.workspaces[selectedWorkspace].token, selectedTeam);
-				window && window?.localStorage.removeItem('user-saw-notif');
+				window && window?.localStorage.removeItem(USER_SAW_OUTSTANDING_NOTIFICATION);
+				window && window?.localStorage.setItem(LAST_WORSPACE_AND_TEAM, selectedTeam);
 			}
 		},
 		[selectedWorkspace, selectedTeam, form]
@@ -123,6 +125,14 @@ function WorkSpaceScreen({ form, className }: { form: TAuthenticationPassword }
 
 		if (form.workspaces.length === 1 && currentTeams?.length === 1) {
 			setSelectedTeam(currentTeams[0].team_id);
+		} else {
+			const lastSelectedTeam = window.localStorage.getItem(LAST_WORSPACE_AND_TEAM) || currentTeams[0].team_id;
+			const lastSelectedWorkspace =
+				form.workspaces.findIndex((workspace) =>
+					workspace.current_teams.find((team) => team.team_id === lastSelectedTeam)
+				) || 0;
+			setSelectedTeam(lastSelectedTeam);
+			setSelectedWorkspace(lastSelectedWorkspace);
 		}
 
 		if (form.workspaces.length === 1 && (currentTeams?.length || 0) <= 1) {

diff --git a/apps/web/app/[locale]/auth/workspace/component.tsx b/apps/web/app/[locale]/auth/workspace/component.tsx
@@ -10,6 +10,7 @@ import { useAuthenticationSocialLogin } from '@app/hooks/auth/useAuthenticationS
 import { ISigninEmailConfirmWorkspaces } from '@app/interfaces';
 import Cookies from 'js-cookie';
 import { useSession } from 'next-auth/react';
+import { LAST_WORSPACE_AND_TEAM, USER_SAW_OUTSTANDING_NOTIFICATION } from '@app/constants';
 
 export default function SocialLoginChooseWorspace() {
 	const t = useTranslations();
@@ -57,14 +58,41 @@ function WorkSpaceScreen() {
 		loadOAuthSession();
 	}, [session]);
 
+	useEffect(() => {
+		if (workspaces.length === 1) {
+			setSelectedWorkspace(0);
+		}
+
+		const currentTeams = workspaces[0]?.current_teams;
+
+		if (workspaces.length === 1 && currentTeams?.length === 1) {
+			setSelectedTeam(currentTeams[0].team_id);
+		} else {
+			const lastSelectedTeam = window.localStorage.getItem(LAST_WORSPACE_AND_TEAM) || currentTeams[0].team_id;
+			const lastSelectedWorkspace =
+				workspaces.findIndex((workspace) =>
+					workspace.current_teams.find((team) => team.team_id === lastSelectedTeam)
+				) || 0;
+			setSelectedTeam(lastSelectedTeam);
+			setSelectedWorkspace(lastSelectedWorkspace);
+		}
+
+		if (workspaces.length === 1 && (currentTeams?.length || 0) <= 1) {
+			setTimeout(() => {
+				document.getElementById('continue-to-workspace')?.click();
+			}, 100);
+		}
+	}, [workspaces]);
+
 	const signInToWorkspace = (e: any) => {
 		e.preventDefault();
 		updateOAuthSession();
 
 		new Array(3).fill('').forEach((_, i) => {
 			Cookies.remove(`authjs.session-token.${i}`);
 		});
-		window && window?.localStorage.removeItem('user-saw-notif');
+		window && window?.localStorage.removeItem(USER_SAW_OUTSTANDING_NOTIFICATION);
+		window && window?.localStorage.setItem(LAST_WORSPACE_AND_TEAM, selectedTeam);
 	};
 
 	const updateOAuthSession = useCallback(() => {

diff --git a/apps/web/app/constants.ts b/apps/web/app/constants.ts
@@ -264,6 +264,10 @@ export const languagesFlags = [
 	}
 ];
 
+// Local storage keys
+export const LAST_WORSPACE_AND_TEAM = 'last-workspace-and-team';
+export const USER_SAW_OUTSTANDING_NOTIFICATION = 'user-saw-notif';
+
 // OAuth providers keys
 
 export const APPLE_CLIENT_ID = process.env.APPLE_CLIENT_ID;