refactor: add workspace events (#1838)

apps/web/src/plugins/affine/index.tsx

@@ -1,4 +1,13 @@
-import { getLoginStorage } from '@affine/workspace/affine/login';
+import { currentAffineUserAtom } from '@affine/workspace/affine/atom';
+import {
+  clearLoginStorage,
+  createAffineAuth,
+  getLoginStorage,
+  parseIdToken,
+  setLoginStorage,
+  SignMethod,
+} from '@affine/workspace/affine/login';
+import { jotaiStore, jotaiWorkspacesAtom } from '@affine/workspace/atom';
 import type { AffineWorkspace } from '@affine/workspace/type';
 import { LoadPriority, WorkspaceFlavour } from '@affine/workspace/type';
 import { createEmptyBlockSuiteWorkspace } from '@affine/workspace/utils';
@@ -15,7 +24,7 @@ import { PageDetailEditor } from '../../components/page-detail-editor';
 import { AffineSWRConfigProvider } from '../../providers/AffineSWRConfigProvider';
 import { BlockSuiteWorkspace } from '../../shared';
 import { affineApis } from '../../shared/apis';
-import { initPage } from '../../utils';
+import { initPage, toast } from '../../utils';
 import type { WorkspacePlugin } from '..';
 import { QueryKey } from './fetcher';
 
@@ -56,11 +65,32 @@ const getPersistenceAllWorkspace = () => {
   return allWorkspaces;
 };
 
+export const affineAuth = createAffineAuth();
+
 export const AffinePlugin: WorkspacePlugin<WorkspaceFlavour.AFFINE> = {
   flavour: WorkspaceFlavour.AFFINE,
   loadPriority: LoadPriority.HIGH,
-  cleanup: () => {
-    storage.removeItem(kAffineLocal);
+  Events: {
+    'workspace:access': async () => {
+      const response = await affineAuth.generateToken(SignMethod.Google);
+      if (response) {
+        setLoginStorage(response);
+        const user = parseIdToken(response.token);
+        jotaiStore.set(currentAffineUserAtom, user);
+      } else {
+        toast('Login failed');
+      }
+    },
+    'workspace:revoke': async () => {
+      jotaiStore.set(jotaiWorkspacesAtom, workspaces =>
+        workspaces.filter(
+          workspace => workspace.flavour !== WorkspaceFlavour.AFFINE
+        )
+      );
+      storage.removeItem(kAffineLocal);
+      clearLoginStorage();
+      jotaiStore.set(currentAffineUserAtom, null);
+    },
   },
   CRUD: {
     create: async blockSuiteWorkspace => {

apps/web/src/plugins/index.tsx

@@ -1,4 +1,5 @@
 import type {
+  AppEvents,
   LoadPriority,
   WorkspaceCRUD,
   WorkspaceUISchema,
@@ -12,8 +13,7 @@ export interface WorkspacePlugin<Flavour extends WorkspaceFlavour> {
   flavour: Flavour;
   // Plugin will be loaded according to the priority
   loadPriority: LoadPriority;
-  // fixme: this is a hack
-  cleanup?: () => void;
+  Events: Partial<AppEvents>;
   // Fetch necessary data for the first render
   CRUD: WorkspaceCRUD<Flavour>;
   UI: WorkspaceUISchema<Flavour>;

apps/web/src/plugins/local/index.tsx

@@ -1,5 +1,10 @@
+import { DebugLogger } from '@affine/debug';
+import { DEFAULT_WORKSPACE_NAME } from '@affine/env';
+import { jotaiStore, jotaiWorkspacesAtom } from '@affine/workspace/atom';
 import { CRUD } from '@affine/workspace/local/crud';
 import { LoadPriority, WorkspaceFlavour } from '@affine/workspace/type';
+import { createEmptyBlockSuiteWorkspace } from '@affine/workspace/utils';
+import { assertEquals, assertExists, nanoid } from '@blocksuite/store';
 import React from 'react';
 
 import { PageNotFoundError } from '../../components/affine/affine-error-eoundary';
@@ -9,9 +14,33 @@ import { PageDetailEditor } from '../../components/page-detail-editor';
 import { initPage } from '../../utils';
 import type { WorkspacePlugin } from '..';
 
+const logger = new DebugLogger('use-create-first-workspace');
+
 export const LocalPlugin: WorkspacePlugin<WorkspaceFlavour.LOCAL> = {
   flavour: WorkspaceFlavour.LOCAL,
   loadPriority: LoadPriority.LOW,
+  Events: {
+    'app:first-init': async () => {
+      const blockSuiteWorkspace = createEmptyBlockSuiteWorkspace(
+        nanoid(),
+        (_: string) => undefined
+      );
+      blockSuiteWorkspace.meta.setName(DEFAULT_WORKSPACE_NAME);
+      const id = await LocalPlugin.CRUD.create(blockSuiteWorkspace);
+      const workspace = await LocalPlugin.CRUD.get(id);
+      assertExists(workspace);
+      assertEquals(workspace.id, id);
+      // todo: use a better way to set initial workspace
+      jotaiStore.set(jotaiWorkspacesAtom, ws => [
+        ...ws,
+        {
+          id: workspace.id,
+          flavour: WorkspaceFlavour.LOCAL,
+        },
+      ]);
+      logger.debug('create first workspace', workspace);
+    },
+  },
   CRUD,
   UI: {
     Provider: ({ children }) => {