Update login form for federated logout (#1644)

<!--
Ensure the title clearly reflects what was changed.
Provide a clear and concise description of the changes made. The PR
should only contain the changes related to the issue, and no other
unrelated changes.
-->

Fixes OPS-2947

Need it for federated login

---------

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: alexandrudanpop

packages/react-ui/src/app/common/auth/lib/utils.ts

@@ -0,0 +1,6 @@
+import { FlagsMap } from '@/app/lib/flags-api';
+
+export const getFederatedUrlBasedOnFlags = (flags: FlagsMap) =>
+  flags.FEDERATED_LOGIN_ENABLED
+    ? (flags.FRONTEGG_URL as string | undefined)
+    : undefined;

packages/react-ui/src/app/common/guards/allow-logged-in-user-only-guard.tsx

@@ -4,6 +4,7 @@ import { jwtDecode } from 'jwt-decode';
 import { Suspense, useEffect } from 'react';
 import { Navigate, useLocation, useNavigate } from 'react-router-dom';
 
+import { getFederatedUrlBasedOnFlags } from '@/app/common/auth/lib/utils';
 import { flagsHooks } from '@/app/common/hooks/flags-hooks';
 import { platformHooks } from '@/app/common/hooks/platform-hooks';
 import { projectHooks } from '@/app/common/hooks/project-hooks';
@@ -61,7 +62,7 @@ export const AllowOnlyLoggedInUserOnlyGuard = ({
         await authenticationSession.logOut({
           userInitiated: false,
           navigate,
-          federatedLoginUrl: flags?.FRONTEGG_URL as string | undefined,
+          federatedLoginUrl: getFederatedUrlBasedOnFlags(flags),
         });
       } catch (e) {
         if (isMounted) {

packages/react-ui/src/app/features/authentication/components/sign-in-form.tsx

@@ -15,6 +15,7 @@ import { t } from 'i18next';
 import { SubmitHandler, useForm } from 'react-hook-form';
 import { Link, useNavigate } from 'react-router-dom';
 
+import { getFederatedUrlBasedOnFlags } from '@/app/common/auth/lib/utils';
 import { flagsHooks } from '@/app/common/hooks/flags-hooks';
 import { HttpError, api } from '@/app/lib/api';
 import { authenticationApi } from '@/app/lib/authentication-api';
@@ -26,7 +27,7 @@ import {
   SignInRequest,
   emailRegex,
 } from '@openops/shared';
-import { useEffectOnce } from 'react-use';
+import { useEffect } from 'react';
 import { navigationUtil } from '../../../lib/navigation-util';
 
 const SignInSchema = Type.Object({
@@ -52,11 +53,17 @@ const SignInForm: React.FC = () => {
     mode: 'onChange',
   });
 
-  useEffectOnce(() => {
+  const { data: flags } = flagsHooks.useFlags();
+
+  useEffect(() => {
+    if (!flags) {
+      return;
+    }
     authenticationSession.logOut({
       userInitiated: false,
+      federatedLoginUrl: getFederatedUrlBasedOnFlags(flags),
     });
-  });
+  }, [flags]);
 
   const { data: edition } = flagsHooks.useFlag(FlagId.EDITION);
   const { data: showSignUpLink } = flagsHooks.useFlag(FlagId.SHOW_SIGN_UP_LINK);

packages/react-ui/src/app/interceptors.ts

@@ -1,3 +1,4 @@
+import { getFederatedUrlBasedOnFlags } from '@/app/common/auth/lib/utils';
 import { API_URL, isUrlRelative } from '@/app/lib/api';
 import { authenticationSession } from '@/app/lib/authentication-session';
 import axios, { AxiosError, HttpStatusCode } from 'axios';
@@ -60,12 +61,13 @@ axios.interceptors.response.use(
       if (url !== OPENOPS_CLOUD_USER_INFO_API_URL && !isSignInRoute) {
         console.warn('JWT expired logging out');
 
-        const flags = queryClient.getQueryData<FlagsMap>([QueryKeys.flags]);
+        // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+        const flags = queryClient.getQueryData<FlagsMap>([QueryKeys.flags])!;
         authenticationSession.logOut({
           userInitiated: false,
-          federatedLoginUrl: flags?.FRONTEGG_URL as string | undefined,
+          federatedLoginUrl: getFederatedUrlBasedOnFlags(flags),
         });
-        if (!flags?.FRONTEGG_URL) {
+        if (!flags?.FEDERATED_LOGIN_ENABLED) {
           window.location.reload();
         }
       }