refactor: login method (#1676)

apps/web/src/hooks/affine/use-affine-log-in.ts

@@ -0,0 +1,26 @@
+import { toast } from '@affine/component';
+import {
+  createAffineAuth,
+  setLoginStorage,
+  SignMethod,
+} from '@affine/workspace/affine/login';
+import { useRouter } from 'next/router';
+import { useCallback } from 'react';
+
+import { apis } from '../../shared/apis';
+
+export const affineAuth = createAffineAuth();
+
+export function useAffineLogIn() {
+  const router = useRouter();
+  return useCallback(async () => {
+    const response = await affineAuth.generateToken(SignMethod.Google);
+    if (response) {
+      setLoginStorage(response);
+      apis.auth.setLogin(response);
+      router.reload();
+    } else {
+      toast('Login failed');
+    }
+  }, [router]);
+}

apps/web/src/hooks/affine/use-affine-log-out.ts

@@ -1,13 +1,14 @@
+import { clearLoginStorage } from '@affine/workspace/affine/login';
 import { WorkspaceFlavour } from '@affine/workspace/type';
 import { useSetAtom } from 'jotai';
 import { useRouter } from 'next/router';
 import { useCallback } from 'react';
 
-import { jotaiWorkspacesAtom } from '../atoms';
-import { WorkspacePlugins } from '../plugins';
-import { apis } from '../shared/apis';
+import { jotaiWorkspacesAtom } from '../../atoms';
+import { WorkspacePlugins } from '../../plugins';
+import { apis } from '../../shared/apis';
 
-export function useOnGoogleLogout() {
+export function useAffineLogOut() {
   const set = useSetAtom(jotaiWorkspacesAtom);
   const router = useRouter();
   return useCallback(() => {
@@ -18,6 +19,7 @@ export function useOnGoogleLogout() {
       )
     );
     WorkspacePlugins[WorkspaceFlavour.AFFINE].cleanup?.();
+    clearLoginStorage();
     router.reload();
   }, [router, set]);
 }

apps/web/src/hooks/affine/use-affine-refresh-auth-token.ts

@@ -0,0 +1,36 @@
+import { DebugLogger } from '@affine/debug';
+import {
+  getLoginStorage,
+  isExpired,
+  parseIdToken,
+  setLoginStorage,
+} from '@affine/workspace/affine/login';
+import useSWR from 'swr';
+
+import { apis } from '../../shared/apis';
+import { affineAuth } from './use-affine-log-in';
+
+const logger = new DebugLogger('auth-token');
+
+const revalidate = async () => {
+  const storage = getLoginStorage();
+  if (storage) {
+    const tokenMessage = parseIdToken(storage.token);
+    logger.debug('revalidate affine user');
+    if (isExpired(tokenMessage)) {
+      logger.debug('need to refresh token');
+      const response = await affineAuth.refreshToken(storage);
+      if (response) {
+        setLoginStorage(response);
+        apis.auth.setLogin(response);
+      }
+    }
+  }
+  return true;
+};
+
+export function useAffineRefreshAuthToken() {
+  useSWR('autoRefreshToken', {
+    fetcher: revalidate,
+  });
+}