Skip to content
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ import {
spacing,
useDarkMode,
} from '@mongodb-js/compass-components';
import { useAtlasLoginActions, useAtlasSignedInUser } from '../provider';
import { useAtlasLoginActions, useAtlasSignInStatus } from '../provider';

const containerStyles = css({
display: 'flex',
Expand Down Expand Up @@ -57,7 +57,7 @@ export const AtlasConnectionStatus: React.FunctionComponent<
AtlasConnectionStatusProps
> = ({ 'data-testid': dataTestId = 'atlas-connection-status' }) => {
const darkMode = useDarkMode();
const userInfo = useAtlasSignedInUser();
const signInStatus = useAtlasSignInStatus();
const { signOut } = useAtlasLoginActions();

const handleDisconnect = useCallback(() => {
Expand All @@ -76,7 +76,7 @@ export const AtlasConnectionStatus: React.FunctionComponent<
})();
}, [signOut]);

if (!userInfo) {
if (!signInStatus.user) {
return null;
}

Expand Down
3 changes: 1 addition & 2 deletions packages/atlas-service/src/provider.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -68,9 +68,8 @@ export { AtlasAuthService } from './atlas-auth-service';
export type { AtlasService } from './atlas-service';
export type { AtlasUserInfo } from './renderer';
export {
useAtlasSignedInUser,
useAtlasSignInStatus,
useAtlasLoginActions,
useIsAtlasSignInStateResolved,
} from './store/atlas-signin-store-context';
export type { AtlasLoginActions } from './store/atlas-signin-store-context';
export { AtlasConnectionStatus } from './components/atlas-connection-status';
Expand Down
Loading
Loading