refactor: setup files (#8201)

packages/frontend/core/src/bootstrap/app.ts

@@ -0,0 +1,4 @@
+import '@affine/env/constant';
+import '../types/types.d.ts';
+
+export { setupGlobal as setupEnvironment } from '@affine/env/global';

packages/frontend/core/src/bootstrap/blocksuite.ts

@@ -3,9 +3,11 @@ import { builtInTemplates as builtInStickersTemplates } from '@affine/templates/
 import type { TemplateManager } from '@blocksuite/blocks';
 import { EdgelessTemplatePanel } from '@blocksuite/blocks';
 
-EdgelessTemplatePanel.templates.extend(
-  builtInStickersTemplates as TemplateManager
-);
-EdgelessTemplatePanel.templates.extend(
-  builtInEdgelessTemplates as TemplateManager
-);
+export function setupBlocksuite() {
+  EdgelessTemplatePanel.templates.extend(
+    builtInStickersTemplates as TemplateManager
+  );
+  EdgelessTemplatePanel.templates.extend(
+    builtInEdgelessTemplates as TemplateManager
+  );
+}

packages/frontend/core/src/bootstrap/index.ts

@@ -0,0 +1,16 @@
+import '../types/types.d.ts';
+
+import { setupEnvironment } from './app';
+import { polyfillBrowser, polyfillElectron } from './polyfill';
+
+export function setupElectron() {
+  polyfillElectron();
+  setupEnvironment();
+}
+
+export async function setupBrowser() {
+  await polyfillBrowser();
+  setupEnvironment();
+}
+
+export { setupBlocksuite } from './blocksuite';

packages/frontend/core/src/bootstrap/polyfill/dispose.ts

@@ -0,0 +1,8 @@
+export async function polyfillDispose() {
+  if (typeof Symbol.dispose !== 'symbol') {
+    // @ts-expect-error ignore
+    await import('core-js/modules/esnext.symbol.async-dispose');
+    // @ts-expect-error ignore
+    await import('core-js/modules/esnext.symbol.dispose');
+  }
+}

packages/frontend/core/src/bootstrap/polyfill/index.ts

@@ -0,0 +1,17 @@
+import { polyfillDispose } from './dispose';
+import { polyfillI18n } from './intl-segmenter';
+import { polyfillPromise } from './promise-with-resolvers';
+import { polyfillEventLoop } from './request-idle-callback';
+import { polyfillResizeObserver } from './resize-observer';
+
+export function polyfillElectron() {
+  polyfillResizeObserver();
+}
+
+export async function polyfillBrowser() {
+  polyfillResizeObserver();
+  polyfillEventLoop();
+  await polyfillI18n();
+  await polyfillPromise();
+  await polyfillDispose();
+}

packages/frontend/core/src/bootstrap/polyfill/intl-segmenter.ts

@@ -0,0 +1,11 @@
+export async function polyfillI18n() {
+  if (Intl.Segmenter === undefined) {
+    await import('intl-segmenter-polyfill-rs').then(({ Segmenter }) => {
+      Object.defineProperty(Intl, 'Segmenter', {
+        value: Segmenter,
+        configurable: true,
+        writable: true,
+      });
+    });
+  }
+}

packages/frontend/core/src/bootstrap/polyfill/promise-with-resolvers.ts

@@ -0,0 +1,6 @@
+export async function polyfillPromise() {
+  if (typeof Promise.withResolvers !== 'function') {
+    // @ts-expect-error ignore
+    await import('core-js/features/promise/with-resolvers');
+  }
+}

packages/frontend/core/src/bootstrap/polyfill/request-idle-callback.ts

@@ -0,0 +1,21 @@
+export function polyfillEventLoop() {
+  window.requestIdleCallback =
+    window.requestIdleCallback ||
+    function (cb) {
+      const start = Date.now();
+      return setTimeout(function () {
+        cb({
+          didTimeout: false,
+          timeRemaining: function () {
+            return Math.max(0, 50 - (Date.now() - start));
+          },
+        });
+      }, 1);
+    };
+
+  window.cancelIdleCallback =
+    window.cancelIdleCallback ||
+    function (id) {
+      clearTimeout(id);
+    };
+}

packages/frontend/core/src/bootstrap/polyfill/resize-observer.ts

@@ -0,0 +1,5 @@
+import { ResizeObserver } from '@juggle/resize-observer';
+
+export function polyfillResizeObserver() {
+  window.ResizeObserver = ResizeObserver;
+}

packages/frontend/core/src/bootstrap/preload.ts

@@ -1,11 +0,0 @@
-import { ResizeObserver } from '@juggle/resize-observer';
-window.ResizeObserver = ResizeObserver;
-
-import '@affine/env/constant';
-import './edgeless-template';
-
-import { setupGlobal } from '@affine/env/global';
-
-setupGlobal();
-
-import '../types/types.d.ts';

packages/frontend/core/src/components/affine/page-history-modal/data.ts

@@ -6,7 +6,7 @@ import { listHistoryQuery, recoverDocMutation } from '@affine/graphql';
 import { i18nTime } from '@affine/i18n';
 import { assertEquals } from '@blocksuite/global/utils';
 import { DocCollection } from '@blocksuite/store';
-import { globalBlockSuiteSchema } from '@toeverything/infra';
+import { getAFFiNEWorkspaceSchema } from '@toeverything/infra';
 import { useEffect, useMemo } from 'react';
 import useSWRImmutable from 'swr/immutable';
 import {
@@ -108,7 +108,7 @@ const getOrCreateShellWorkspace = (workspaceId: string) => {
       blobSources: {
         main: blobStorage,
       },
-      schema: globalBlockSuiteSchema,
+      schema: getAFFiNEWorkspaceSchema(),
     });
     docCollectionMap.set(workspaceId, docCollection);
     docCollection.doc.emit('sync', [true, docCollection.doc]);

packages/frontend/core/src/components/blocksuite/block-suite-header/journal/date-picker.tsx

@@ -4,8 +4,7 @@ import {
   useJournalInfoHelper,
   useJournalRouteHelper,
 } from '@affine/core/hooks/use-journal';
-import type { DocCollection } from '@affine/core/shared';
-import type { Doc } from '@blocksuite/store';
+import type { Doc, DocCollection } from '@blocksuite/store';
 import dayjs from 'dayjs';
 import { useEffect, useRef, useState } from 'react';
 

packages/frontend/core/src/components/blocksuite/block-suite-header/journal/today-button.tsx

@@ -1,7 +1,7 @@
 import { Button } from '@affine/component';
 import { useJournalRouteHelper } from '@affine/core/hooks/use-journal';
-import type { DocCollection } from '@affine/core/shared';
 import { useI18n } from '@affine/i18n';
+import type { DocCollection } from '@blocksuite/store';
 import { useCallback } from 'react';
 
 export interface JournalTodayButtonProps {

packages/frontend/core/src/components/blocksuite/block-suite-page-list/utils.tsx

@@ -1,11 +1,10 @@
 import { toast } from '@affine/component';
 import { WorkbenchService } from '@affine/core/modules/workbench';
 import type { DocMode } from '@blocksuite/blocks';
+import type { DocCollection } from '@blocksuite/store';
 import { DocsService, useServices } from '@toeverything/infra';
 import { useCallback, useMemo } from 'react';
 
-import type { DocCollection } from '../../../shared';
-
 export const usePageHelper = (docCollection: DocCollection) => {
   const { docsService, workbenchService } = useServices({
     DocsService,

packages/frontend/core/src/components/root-app-sidebar/import-page.tsx

@@ -2,8 +2,8 @@ import { useAsyncCallback } from '@affine/core/hooks/affine-async-hooks';
 import { track } from '@affine/core/mixpanel';
 import { useI18n } from '@affine/i18n';
 import { ImportIcon } from '@blocksuite/icons/rc';
+import type { DocCollection } from '@blocksuite/store';
 
-import type { DocCollection } from '../../shared';
 import { MenuItem } from '../app-sidebar';
 import { usePageHelper } from '../blocksuite/block-suite-page-list/utils';
 

packages/frontend/core/src/components/root-app-sidebar/journal-button.tsx

@@ -4,10 +4,10 @@ import {
   useJournalRouteHelper,
 } from '@affine/core/hooks/use-journal';
 import { WorkbenchService } from '@affine/core/modules/workbench';
-import type { DocCollection } from '@affine/core/shared';
 import { isNewTabTrigger } from '@affine/core/utils';
 import { useI18n } from '@affine/i18n';
 import { TodayIcon, TomorrowIcon, YesterdayIcon } from '@blocksuite/icons/rc';
+import type { DocCollection } from '@blocksuite/store';
 import { useLiveData, useService } from '@toeverything/infra';
 import { type MouseEvent } from 'react';
 

packages/frontend/core/src/hooks/affine/use-reference-link-helper.ts

@@ -1,7 +1,6 @@
+import type { DocCollection } from '@blocksuite/store';
 import { useCallback } from 'react';
 
-import type { DocCollection } from '../../shared';
-
 export function useReferenceLinkHelper(docCollection: DocCollection) {
   const addReferenceLink = useCallback(
     (pageId: string, referenceId: string) => {

packages/frontend/core/src/hooks/use-journal.ts

@@ -1,11 +1,11 @@
 import { i18nTime } from '@affine/i18n';
+import type { DocCollection } from '@blocksuite/store';
 import { DocsService, initEmptyPage, useService } from '@toeverything/infra';
 import dayjs from 'dayjs';
 import { useCallback, useMemo } from 'react';
 
 import { track } from '../mixpanel';
 import { WorkbenchService } from '../modules/workbench';
-import type { DocCollection } from '../shared';
 import { useCurrentWorkspacePropertiesAdapter } from './use-affine-adapter';
 import { useDocCollectionHelper } from './use-block-suite-workspace-helper';
 

packages/frontend/core/src/modules/workspace-engine/impls/cloud.ts

@@ -24,7 +24,7 @@ import {
   type WorkspaceMetadata,
   type WorkspaceProfileInfo,
 } from '@toeverything/infra';
-import { effect, globalBlockSuiteSchema, Service } from '@toeverything/infra';
+import { effect, getAFFiNEWorkspaceSchema, Service } from '@toeverything/infra';
 import { isEqual } from 'lodash-es';
 import { nanoid } from 'nanoid';
 import { EMPTY, map, mergeMap } from 'rxjs';
@@ -96,7 +96,7 @@ export class CloudWorkspaceFlavourProviderService
     const docCollection = new DocCollection({
       id: workspaceId,
       idGenerator: () => nanoid(),
-      schema: globalBlockSuiteSchema,
+      schema: getAFFiNEWorkspaceSchema(),
       blobSources: {
         main: blobStorage,
       },
@@ -216,7 +216,7 @@ export class CloudWorkspaceFlavourProviderService
 
     const bs = new DocCollection({
       id,
-      schema: globalBlockSuiteSchema,
+      schema: getAFFiNEWorkspaceSchema(),
     });
 
     if (localData) applyUpdate(bs.doc, localData);

packages/frontend/core/src/modules/workspace-engine/impls/local.ts

@@ -9,7 +9,11 @@ import type {
   WorkspaceMetadata,
   WorkspaceProfileInfo,
 } from '@toeverything/infra';
-import { globalBlockSuiteSchema, LiveData, Service } from '@toeverything/infra';
+import {
+  getAFFiNEWorkspaceSchema,
+  LiveData,
+  Service,
+} from '@toeverything/infra';
 import { isEqual } from 'lodash-es';
 import { nanoid } from 'nanoid';
 import { Observable } from 'rxjs';
@@ -70,7 +74,7 @@ export class LocalWorkspaceFlavourProvider
     const docCollection = new DocCollection({
       id: id,
       idGenerator: () => nanoid(),
-      schema: globalBlockSuiteSchema,
+      schema: getAFFiNEWorkspaceSchema(),
       blobSources: { main: blobStorage },
     });
 
@@ -143,7 +147,7 @@ export class LocalWorkspaceFlavourProvider
 
     const bs = new DocCollection({
       id,
-      schema: globalBlockSuiteSchema,
+      schema: getAFFiNEWorkspaceSchema(),
     });
 
     if (localData) applyUpdate(bs.doc, localData);

packages/frontend/core/src/shared/index.ts

@@ -1,7 +1,4 @@
 import { DebugLogger } from '@affine/debug';
-import { DocCollection } from '@blocksuite/store';
-
-export { DocCollection };
 
 export enum WorkspaceSubPath {
   ALL = 'all',