mirror of
https://github.com/toeverything/AFFiNE.git
synced 2026-02-11 20:08:37 +00:00
fix: stay loading when refresh token fail (#1194)
This commit is contained in:
@@ -1,11 +1,22 @@
|
||||
import { toast } from '@affine/component';
|
||||
import { MessageCenter } from '@affine/datacenter';
|
||||
import { DataCenter, MessageCenter } from '@affine/datacenter';
|
||||
import { AffineProvider } from '@affine/datacenter';
|
||||
import { DebugLogger } from '@affine/debug';
|
||||
import { useRouter } from 'next/router';
|
||||
import { ReactNode, useCallback, useEffect } from 'react';
|
||||
|
||||
import { useGlobalState } from '@/store/app';
|
||||
|
||||
const logger = new DebugLogger('messageCenter');
|
||||
|
||||
const clearAuth = (dataCenter: DataCenter, providerName: string) => {
|
||||
const affineProvider = dataCenter.providers.find(p => p.id === providerName);
|
||||
if (affineProvider && affineProvider instanceof AffineProvider) {
|
||||
affineProvider.apis.auth.clear();
|
||||
} else {
|
||||
logger.error('cannot find affine provider, please fix this ASAP');
|
||||
}
|
||||
};
|
||||
export function MessageCenterHandler({ children }: { children?: ReactNode }) {
|
||||
const router = useRouter();
|
||||
const dataCenter = useGlobalState(useCallback(store => store.dataCenter, []));
|
||||
@@ -19,14 +30,7 @@ export function MessageCenterHandler({ children }: { children?: ReactNode }) {
|
||||
// todo: error toast style
|
||||
toast('You have no permission to access this workspace');
|
||||
// todo(himself65): remove dynamic lookup
|
||||
const affineProvider = dataCenter.providers.find(
|
||||
p => p.id === 'affine'
|
||||
);
|
||||
if (affineProvider && affineProvider instanceof AffineProvider) {
|
||||
affineProvider.apis.auth.clear();
|
||||
} else {
|
||||
console.error('cannot find affine provider, please fix this ASAP');
|
||||
}
|
||||
clearAuth(dataCenter, 'affine');
|
||||
// the status of the app right now is unknown, and it won't help if we let
|
||||
// the app continue and let the user auth the app.
|
||||
// that's why so we need to reload the page for now.
|
||||
@@ -36,9 +40,16 @@ export function MessageCenterHandler({ children }: { children?: ReactNode }) {
|
||||
await router.push('/');
|
||||
router.reload();
|
||||
}
|
||||
|
||||
if (message.code === MessageCenter.messageCode.refreshTokenError) {
|
||||
toast('Automatic login fail, please login self later');
|
||||
clearAuth(dataCenter, 'affine');
|
||||
await router.push('/');
|
||||
router.reload();
|
||||
}
|
||||
});
|
||||
}
|
||||
}, [dataCenter?.providers, router]);
|
||||
}, [dataCenter, router]);
|
||||
|
||||
return <>{children}</>;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user