refactor(infra): directory structure (#4615)

2026-02-13 21:05:19 +00:00 · 2023-10-18 23:30:08 +08:00
parent 814d552be8
commit bed9310519
1150 changed files with 539 additions and 584 deletions
--- a/packages/frontend/workspace/src/manager/index.ts
+++ b/packages/frontend/workspace/src/manager/index.ts
@@ -0,0 +1,145 @@
+import { isBrowser } from '@affine/env/constant';
+import type { BlockSuiteFeatureFlags } from '@affine/env/global';
+import { WorkspaceFlavour } from '@affine/env/workspace';
+import { createAffinePublicProviders } from '@affine/workspace/providers';
+import { __unstableSchemas, AffineSchemas } from '@blocksuite/blocks/models';
+import type { DocProviderCreator, StoreOptions } from '@blocksuite/store';
+import {
+  createIndexeddbStorage,
+  Generator,
+  Schema,
+  Workspace,
+} from '@blocksuite/store';
+import { INTERNAL_BLOCKSUITE_HASH_MAP } from '@toeverything/infra/__internal__/workspace';
+import type { Doc } from 'yjs';
+import type { Transaction } from 'yjs';
+
+import { createCloudBlobStorage } from '../blob/cloud-blob-storage';
+import { createStaticStorage } from '../blob/local-static-storage';
+import { createSQLiteStorage } from '../blob/sqlite-blob-storage';
+import { createAffineProviders, createLocalProviders } from '../providers';
+
+function setEditorFlags(workspace: Workspace) {
+  Object.entries(runtimeConfig.editorFlags).forEach(([key, value]) => {
+    workspace.awarenessStore.setFlag(
+      key as keyof BlockSuiteFeatureFlags,
+      value
+    );
+  });
+  workspace.awarenessStore.setFlag(
+    'enable_bookmark_operation',
+    environment.isDesktop
+  );
+}
+
+type UpdateCallback = (
+  update: Uint8Array,
+  origin: string | number | null,
+  doc: Doc,
+  transaction: Transaction
+) => void;
+
+type SubdocEvent = {
+  loaded: Set<Doc>;
+  removed: Set<Doc>;
+  added: Set<Doc>;
+};
+
+const docUpdateCallbackWeakMap = new WeakMap<Doc, UpdateCallback>();
+
+export const globalBlockSuiteSchema = new Schema();
+
+globalBlockSuiteSchema.register(AffineSchemas).register(__unstableSchemas);
+
+const createMonitor = (doc: Doc) => {
+  const onUpdate: UpdateCallback = (_, origin) => {
+    if (process.env.NODE_ENV === 'development') {
+      if (typeof origin !== 'string' && typeof origin !== 'number') {
+        console.warn(
+          'origin is not a string or number, this will cause problems in the future',
+          origin
+        );
+      }
+    } else {
+      // todo: add monitor in the future
+    }
+  };
+  docUpdateCallbackWeakMap.set(doc, onUpdate);
+  doc.on('update', onUpdate);
+  const onSubdocs = (event: SubdocEvent) => {
+    event.added.forEach(subdoc => {
+      if (!docUpdateCallbackWeakMap.has(subdoc)) {
+        createMonitor(subdoc);
+      }
+    });
+    event.removed.forEach(subdoc => {
+      if (docUpdateCallbackWeakMap.has(subdoc)) {
+        docUpdateCallbackWeakMap.delete(subdoc);
+      }
+    });
+  };
+  doc.on('subdocs', onSubdocs);
+  doc.on('destroy', () => {
+    docUpdateCallbackWeakMap.delete(doc);
+    doc.off('update', onSubdocs);
+  });
+};
+
+// if not exist, create a new workspace
+export function getOrCreateWorkspace(
+  id: string,
+  flavour: WorkspaceFlavour
+): Workspace {
+  const providerCreators: DocProviderCreator[] = [];
+  if (INTERNAL_BLOCKSUITE_HASH_MAP.has(id)) {
+    return INTERNAL_BLOCKSUITE_HASH_MAP.get(id) as Workspace;
+  }
+  const idGenerator = Generator.NanoID;
+
+  const blobStorages: StoreOptions['blobStorages'] = [];
+  if (flavour === WorkspaceFlavour.AFFINE_CLOUD) {
+    if (isBrowser) {
+      blobStorages.push(createIndexeddbStorage);
+      blobStorages.push(createCloudBlobStorage);
+      if (environment.isDesktop && runtimeConfig.enableSQLiteProvider) {
+        blobStorages.push(createSQLiteStorage);
+      }
+      providerCreators.push(...createAffineProviders());
+
+      // todo(JimmFly): add support for cloud storage
+    }
+  } else if (flavour === WorkspaceFlavour.LOCAL) {
+    if (isBrowser) {
+      blobStorages.push(createIndexeddbStorage);
+      if (environment.isDesktop && runtimeConfig.enableSQLiteProvider) {
+        blobStorages.push(createSQLiteStorage);
+      }
+    }
+    providerCreators.push(...createLocalProviders());
+  } else if (flavour === WorkspaceFlavour.AFFINE_PUBLIC) {
+    if (isBrowser) {
+      blobStorages.push(createIndexeddbStorage);
+      if (environment.isDesktop && runtimeConfig.enableSQLiteProvider) {
+        blobStorages.push(createSQLiteStorage);
+      }
+    }
+    blobStorages.push(createCloudBlobStorage);
+    providerCreators.push(...createAffinePublicProviders());
+  } else {
+    throw new Error('unsupported flavour');
+  }
+  blobStorages.push(createStaticStorage);
+
+  const workspace = new Workspace({
+    id,
+    isSSR: !isBrowser,
+    providerCreators: typeof window === 'undefined' ? [] : providerCreators,
+    blobStorages: blobStorages,
+    idGenerator,
+    schema: globalBlockSuiteSchema,
+  });
+  createMonitor(workspace.doc);
+  setEditorFlags(workspace);
+  INTERNAL_BLOCKSUITE_HASH_MAP.set(id, workspace);
+  return workspace;
+}