test: add test case for blocksuite editor (#1528)

apps/web/src/utils/blocksuite.ts

@@ -0,0 +1,55 @@
+import { DebugLogger } from '@affine/debug';
+import { EditorContainer } from '@blocksuite/editor';
+import { Page } from '@blocksuite/store';
+
+import markdown from '../templates/Welcome-to-AFFiNE.md';
+
+const demoTitle = markdown
+  .split('\n')
+  .splice(0, 1)
+  .join('')
+  .replaceAll('#', '')
+  .trim();
+
+const demoText = markdown.split('\n').slice(1).join('\n');
+
+const logger = new DebugLogger('init-page');
+
+export function initPage(page: Page, editor: Readonly<EditorContainer>): void {
+  logger.debug('initEmptyPage', page.id);
+  // Add page block and surface block at root level
+  const isFirstPage = page.meta.init === true;
+  if (isFirstPage) {
+    page.workspace.setPageMeta(page.id, {
+      init: false,
+    });
+    _initPageWithDemoMarkdown(page, editor);
+  } else {
+    _initEmptyPage(page, editor);
+  }
+  page.resetHistory();
+}
+
+export function _initEmptyPage(page: Page, _: Readonly<EditorContainer>) {
+  const pageBlockId = page.addBlockByFlavour('affine:page', {
+    title: new page.Text(''),
+  });
+  page.addBlockByFlavour('affine:surface', {}, null);
+  const frameId = page.addBlockByFlavour('affine:frame', {}, pageBlockId);
+  page.addBlockByFlavour('affine:paragraph', {}, frameId);
+}
+
+export function _initPageWithDemoMarkdown(
+  page: Page,
+  editor: Readonly<EditorContainer>
+): void {
+  logger.debug('initPageWithDefaultMarkdown', page.id);
+  const pageBlockId = page.addBlockByFlavour('affine:page', {
+    title: new page.Text(demoTitle),
+  });
+  page.addBlockByFlavour('affine:surface', {}, null);
+  const frameId = page.addBlockByFlavour('affine:frame', {}, pageBlockId);
+  page.addBlockByFlavour('affine:paragraph', {}, frameId);
+  editor.clipboard.importMarkdown(demoText, frameId);
+  page.workspace.setPageMeta(page.id, { demoTitle });
+}

apps/web/src/utils/index.ts

@@ -1,5 +1,5 @@
 import { __unstableSchemas, builtInSchemas } from '@blocksuite/blocks/models';
-import type { BlobOptionsGetter } from '@blocksuite/store';
+import type { BlobOptionsGetter, Generator } from '@blocksuite/store';
 
 import { BlockSuiteWorkspace } from '../shared';
 
@@ -27,7 +27,8 @@ export function stringToColour(str: string) {
 const hashMap = new Map<string, BlockSuiteWorkspace>();
 export const createEmptyBlockSuiteWorkspace = (
   id: string,
-  blobOptionsGetter?: BlobOptionsGetter
+  blobOptionsGetter?: BlobOptionsGetter,
+  idGenerator?: Generator
 ): BlockSuiteWorkspace => {
   if (hashMap.has(id)) {
     return hashMap.get(id) as BlockSuiteWorkspace;
@@ -36,6 +37,7 @@ export const createEmptyBlockSuiteWorkspace = (
     id,
     isSSR: typeof window === 'undefined',
     blobOptionsGetter,
+    idGenerator,
   })
     .register(builtInSchemas)
     .register(__unstableSchemas);