yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-08-23 16:38:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ui): Update flags from AuthProvider only (#6566)

Browse Source
This commit is contained in:
Chirag Madlani 2022-08-04 17:29:08 +05:30 committed by GitHub
parent 45e6c2c7d7
commit c6d35f7c4c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
openmetadata-ui/src/main/resources/ui/src/authentication/auth-provider/AuthProvider.tsx
View File

@ -109,11 +109,13 @@ export const AuthProvider = ({
let silentSignInRetries = 0;
const onLoginHandler = () => {
setLoading(true);
authenticatorRef.current?.invokeLogin();
};
const onLogoutHandler = () => {
authenticatorRef.current?.invokeLogout();
setLoading(false);
};
const onRenewIdTokenHandler = () => {
@ -349,11 +351,13 @@ export const AuthProvider = ({
const handleFailedLogin = () => {
setIsSigningIn(false);
setIsUserAuthenticated(false);
setLoading(false);
history.push(ROUTES.SIGNIN);
};
const handleSuccessfulLogin = (user: OidcUser) => {
setLoading(true);
setIsUserAuthenticated(true);
getUserByName(getNameFromEmail(user.profile.email), userAPIQueryFields)
.then((res: AxiosResponse) => {
if (res.data) {

25
openmetadata-ui/src/main/resources/ui/src/authentication/authenticators/MsalAuthenticator.tsx
View File

@ -16,7 +16,7 @@ import {
InteractionRequiredAuthError,
InteractionStatus,
} from '@azure/msal-browser';
import { useAccount, useIsAuthenticated, useMsal } from '@azure/msal-react';
import { useAccount, useMsal } from '@azure/msal-react';
import { AxiosError } from 'axios';
import React, {
forwardRef,
@ -28,7 +28,6 @@ import React, {
import { useMutex } from 'react-context-mutex';
import { oidcTokenKey } from '../../constants/constants';
import { msalLoginRequest } from '../../utils/AuthProvider.util';
import { useAuthContext } from '../auth-provider/AuthProvider';
import {
AuthenticatorRef,
OidcUser,
@ -46,10 +45,7 @@ const MsalAuthenticator = forwardRef<AuthenticatorRef, Props>(
{ children, onLoginSuccess, onLogoutSuccess, onLoginFailure }: Props,
ref
) => {
const { setIsAuthenticated, loading, setLoadingIndicator } =
useAuthContext();
const { instance, accounts, inProgress } = useMsal();
const isMsalAuthenticated = useIsAuthenticated();
const account = useAccount(accounts[0] || {});
const MutexRunner = useMutex();
const mutex = new MutexRunner('fetchIdToken');
@ -64,12 +60,16 @@ const MsalAuthenticator = forwardRef<AuthenticatorRef, Props>(
};
const login = () => {
setLoadingIndicator(true);
instance.loginPopup(msalLoginRequest);
try {
instance.loginPopup(msalLoginRequest);
} catch (error) {
// eslint-disable-next-line no-console
console.error(error);
onLoginFailure(error as AxiosError);
}
};
const logout = () => {
setLoadingIndicator(false);
handleOnLogoutSuccess();
};
@ -87,14 +87,13 @@ const MsalAuthenticator = forwardRef<AuthenticatorRef, Props>(
},
};
setIsAuthenticated(isMsalAuthenticated);
localStorage.setItem(oidcTokenKey, idToken);
return user;
};
const fetchIdToken = async (
shouldFaullbackToPopup = false
shouldFallbackToPopup = false
): Promise<OidcUser> => {
const tokenRequest = {
account: account || accounts[0], // This is an example - Select account based on your app's requirements
@ -107,7 +106,7 @@ const MsalAuthenticator = forwardRef<AuthenticatorRef, Props>(
} catch (error) {
if (
error instanceof InteractionRequiredAuthError &&
shouldFaullbackToPopup
shouldFallbackToPopup
) {
const response = await instance
.loginPopup(tokenRequest)
@ -140,15 +139,11 @@ const MsalAuthenticator = forwardRef<AuthenticatorRef, Props>(
fetchIdToken(true)
.then((user) => {
if ((user as OidcUser).id_token) {
setLoadingIndicator(false);
onLoginSuccess(user as OidcUser);
}
})
.catch(onLoginFailure)
.finally(() => {
if (loading) {
setLoadingIndicator(false);
}
mutex.unlock();
});
});