feat: support revert changes (#1782)

apps/web/src/hooks/__tests__/use-blocksuite-workspace-helper.spec.ts

@@ -14,21 +14,21 @@ import { usePageMeta } from '../use-page-meta';
 
 let blockSuiteWorkspace: BlockSuiteWorkspace;
 
+function handleNewPage(page: Page) {
+  const pageBlockId = page.addBlock('affine:page', { title: '' });
+  const frameId = page.addBlock('affine:frame', {}, pageBlockId);
+  page.addBlock('affine:paragraph', {}, frameId);
+}
+
 beforeEach(() => {
   blockSuiteWorkspace = new BlockSuiteWorkspace({
     id: 'test',
   })
     .register(AffineSchemas)
     .register(__unstableSchemas);
-  blockSuiteWorkspace.slots.pageAdded.on(pageId => {
-    const page = blockSuiteWorkspace.getPage(pageId) as Page;
-    const pageBlockId = page.addBlock('affine:page', { title: '' });
-    const frameId = page.addBlock('affine:frame', {}, pageBlockId);
-    page.addBlock('affine:paragraph', {}, frameId);
-  });
-  blockSuiteWorkspace.createPage('page0');
-  blockSuiteWorkspace.createPage('page1');
-  blockSuiteWorkspace.createPage('page2');
+  handleNewPage(blockSuiteWorkspace.createPage('page0'));
+  handleNewPage(blockSuiteWorkspace.createPage('page1'));
+  handleNewPage(blockSuiteWorkspace.createPage('page2'));
 });
 
 describe('useBlockSuiteWorkspaceHelper', () => {
@@ -46,4 +46,20 @@ describe('useBlockSuiteWorkspaceHelper', () => {
     pageMetaHook.rerender();
     expect(pageMetaHook.result.current.length).toBe(4);
   });
+
+  test('milestone', async () => {
+    expect(blockSuiteWorkspace.meta.pageMetas.length).toBe(3);
+    const helperHook = renderHook(() =>
+      useBlockSuiteWorkspaceHelper(blockSuiteWorkspace)
+    );
+    await helperHook.result.current.markMilestone('test');
+    expect(blockSuiteWorkspace.meta.pageMetas.length).toBe(3);
+    handleNewPage(helperHook.result.current.createPage('page4'));
+    expect(blockSuiteWorkspace.meta.pageMetas.length).toBe(4);
+    expect(await helperHook.result.current.listMilestone()).toHaveProperty(
+      'test'
+    );
+    await helperHook.result.current.revertMilestone('test');
+    expect(blockSuiteWorkspace.meta.pageMetas.length).toBe(3);
+  });
 });

apps/web/src/hooks/use-blocksuite-workspace-helper.ts

@@ -1,5 +1,10 @@
 import type { Page } from '@blocksuite/store';
 import { assertExists } from '@blocksuite/store';
+import {
+  getMilestones,
+  markMilestone,
+  revertUpdate,
+} from '@toeverything/y-indexeddb';
 import { useMemo } from 'react';
 
 import type { BlockSuiteWorkspace } from '../shared';
@@ -13,6 +18,27 @@ export function useBlockSuiteWorkspaceHelper(
         assertExists(blockSuiteWorkspace);
         return blockSuiteWorkspace.createPage(pageId, parentId);
       },
+      markMilestone: async (name: string) => {
+        assertExists(blockSuiteWorkspace);
+        const doc = blockSuiteWorkspace.doc;
+        await markMilestone(blockSuiteWorkspace.id, doc, name);
+      },
+      revertMilestone: async (name: string) => {
+        assertExists(blockSuiteWorkspace);
+        const doc = blockSuiteWorkspace.doc;
+        const list = await getMilestones(blockSuiteWorkspace.id);
+        if (!list) {
+          throw new Error('no milestone');
+        }
+        const milestone = list[name];
+        if (milestone) {
+          revertUpdate(doc, milestone, () => 'Map');
+        }
+      },
+      listMilestone: async () => {
+        assertExists(blockSuiteWorkspace);
+        return await getMilestones(blockSuiteWorkspace.id);
+      },
     }),
     [blockSuiteWorkspace]
   );

apps/web/src/shared/apis.ts

@@ -1,3 +1,4 @@
+import { DebugLogger } from '@affine/debug';
 import { config } from '@affine/env';
 import {
   createUserApis,
@@ -30,6 +31,8 @@ const affineApis = {} as ReturnType<typeof createUserApis> &
 Object.assign(affineApis, createUserApis(prefixUrl));
 Object.assign(affineApis, createWorkspaceApis(prefixUrl));
 
+const debugLogger = new DebugLogger('affine-debug-apis');
+
 if (!globalThis.AFFINE_APIS) {
   globalThis.AFFINE_APIS = affineApis;
   globalThis.setLogin = (response: LoginResponse) => {