refactor: add workspace events (#1838)

This commit is contained in:
Himself65
2023-04-06 16:14:23 -05:00
committed by GitHub
parent b6bdf257e4
commit 5ac36b6f0a
13 changed files with 105 additions and 71 deletions
+8 -22
View File
@@ -1,30 +1,16 @@
import { currentAffineUserAtom } from '@affine/workspace/affine/atom';
import {
createAffineAuth,
parseIdToken,
setLoginStorage,
SignMethod,
} from '@affine/workspace/affine/login';
import { useSetAtom } from 'jotai';
import { WorkspaceFlavour } from '@affine/workspace/type';
import { useRouter } from 'next/router';
import { useCallback } from 'react';
import { toast } from '../../utils';
export const affineAuth = createAffineAuth();
import { WorkspacePlugins } from '../../plugins';
export function useAffineLogIn() {
const router = useRouter();
const setUser = useSetAtom(currentAffineUserAtom);
return useCallback(async () => {
const response = await affineAuth.generateToken(SignMethod.Google);
if (response) {
setLoginStorage(response);
const user = parseIdToken(response.token);
setUser(user);
router.reload();
} else {
toast('Login failed');
}
}, [router, setUser]);
await WorkspacePlugins[WorkspaceFlavour.AFFINE].Events[
'workspace:access'
]?.();
// todo: remove reload page requirement
router.reload();
}, [router]);
}
@@ -1,26 +1,15 @@
import { currentAffineUserAtom } from '@affine/workspace/affine/atom';
import { clearLoginStorage } from '@affine/workspace/affine/login';
import { jotaiWorkspacesAtom } from '@affine/workspace/atom';
import { WorkspaceFlavour } from '@affine/workspace/type';
import { useSetAtom } from 'jotai';
import { useRouter } from 'next/router';
import { useCallback } from 'react';
import { WorkspacePlugins } from '../../plugins';
export function useAffineLogOut() {
const set = useSetAtom(jotaiWorkspacesAtom);
const router = useRouter();
const setCurrentUser = useSetAtom(currentAffineUserAtom);
return useCallback(() => {
set(workspaces =>
workspaces.filter(
workspace => workspace.flavour !== WorkspaceFlavour.AFFINE
)
);
WorkspacePlugins[WorkspaceFlavour.AFFINE].cleanup?.();
clearLoginStorage();
setCurrentUser(null);
return useCallback(async () => {
await WorkspacePlugins[WorkspaceFlavour.AFFINE].Events[
'workspace:revoke'
]?.();
router.reload();
}, [router, set, setCurrentUser]);
}, [router]);
}
@@ -7,7 +7,7 @@ import {
} from '@affine/workspace/affine/login';
import useSWR from 'swr';
import { affineAuth } from './use-affine-log-in';
import { affineAuth } from '../../plugins/affine';
const logger = new DebugLogger('auth-token');