fix: patch refresh token logic (#1665)

2026-07-31 17:19:56 +08:00 · 2023-03-23 00:12:20 -05:00
parent 059d9e5de2
commit 2551785451
3 changed files with 58 additions and 21 deletions
@@ -0,0 +1,23 @@
+import { useSetAtom } from 'jotai';
+import { useRouter } from 'next/router';
+import { useCallback } from 'react';
+
+import { jotaiWorkspacesAtom } from '../atoms';
+import { WorkspacePlugins } from '../plugins';
+import { RemWorkspaceFlavour } from '../shared';
+import { apis } from '../shared/apis';
+
+export function useOnGoogleLogout() {
+  const set = useSetAtom(jotaiWorkspacesAtom);
+  const router = useRouter();
+  return useCallback(() => {
+    apis.auth.clear();
+    set(workspaces =>
+      workspaces.filter(
+        workspace => workspace.flavour !== RemWorkspaceFlavour.AFFINE
+      )
+    );
+    WorkspacePlugins[RemWorkspaceFlavour.AFFINE].cleanup?.();
+    router.reload();
+  }, [router, set]);
+}