fix: stay loading when refresh token fail (#1194)

This commit is contained in:
Qi
2023-02-24 17:16:21 +08:00
committed by GitHub
parent 28a6c46605
commit da1d00d9bf
5 changed files with 54 additions and 24 deletions

View File

@@ -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}</>;
}