refactor: workspace loading logic (#1966)

packages/env/src/blocksuite.ts

@@ -0,0 +1,81 @@
+import { DebugLogger } from '@affine/debug';
+import markdown from '@affine/templates/Welcome-to-AFFiNE.md';
+import { ContentParser } from '@blocksuite/blocks/content-parser';
+import type { Page, Workspace } from '@blocksuite/store';
+import { nanoid } from '@blocksuite/store';
+
+declare global {
+  interface Window {
+    lastImportedMarkdown: string;
+  }
+}
+
+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): 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);
+  } else {
+    _initEmptyPage(page);
+  }
+  page.resetHistory();
+}
+
+export function _initEmptyPage(page: Page, title?: string): void {
+  const pageBlockId = page.addBlock('affine:page', {
+    title: new page.Text(title ?? ''),
+  });
+  page.addBlock('affine:surface', {}, null);
+  const frameId = page.addBlock('affine:frame', {}, pageBlockId);
+  page.addBlock('affine:paragraph', {}, frameId);
+}
+
+export function _initPageWithDemoMarkdown(page: Page): void {
+  logger.debug('initPageWithDefaultMarkdown', page.id);
+  const pageBlockId = page.addBlock('affine:page', {
+    title: new page.Text(demoTitle),
+  });
+  page.addBlock('affine:surface', {}, null);
+  const frameId = page.addBlock('affine:frame', {}, pageBlockId);
+  page.addBlock('affine:paragraph', {}, frameId);
+  const contentParser = new ContentParser(page);
+  contentParser.importMarkdown(demoText, frameId);
+  page.workspace.setPageMeta(page.id, { demoTitle });
+}
+
+export function ensureRootPinboard(blockSuiteWorkspace: Workspace) {
+  const metas = blockSuiteWorkspace.meta.pageMetas;
+  const rootMeta = metas.find(m => m.isRootPinboard);
+
+  if (rootMeta) {
+    return rootMeta.id;
+  }
+
+  const rootPinboardPage = blockSuiteWorkspace.createPage(nanoid());
+
+  const title = `${blockSuiteWorkspace.meta.name}'s Pinboard`;
+
+  _initEmptyPage(rootPinboardPage, title);
+
+  blockSuiteWorkspace.meta.setPageMeta(rootPinboardPage.id, {
+    isRootPinboard: true,
+    title,
+  });
+
+  return rootPinboardPage.id;
+}

packages/env/src/constant.ts

@@ -1,5 +1,6 @@
 export const DEFAULT_WORKSPACE_NAME = 'Demo Workspace';
 export const UNTITLED_WORKSPACE_NAME = 'Untitled';
+export const DEFAULT_HELLO_WORLD_PAGE_ID = 'hello-world';
 
 export const enum MessageCode {
   loginError,

packages/env/src/types.d.ts

@@ -0,0 +1,10 @@
+/// <reference types="@webpack/env"" />
+
+// not using import because it will break the declare module line below
+// eslint-disable-next-line @typescript-eslint/triple-slash-reference
+/// <reference path='../../../electron/layers/preload/preload.d.ts' />
+
+declare module '*.md' {
+  const text: string;
+  export default text;
+}