refactor(editor): introduce store container to make implement doc easier (#12146)

<!-- This is an auto-generated comment: release notes by coderabbit.ai -->
## Summary by CodeRabbit

- **New Features**
  - Introduced a new store management system for handling document stores, improving efficiency and flexibility when working with document data.

- **Refactor**
  - Updated internal store handling to use a centralized store container, simplifying store retrieval and removal across various components.
  - Renamed and updated several store-related method signatures for consistency and clarity.
  - Replaced editor extension loading logic with a new local implementation for better modularity.

- **Chores**
  - Improved and streamlined the export of store-related modules for better maintainability.
  - Removed obsolete and redundant code related to previous store management approaches.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

packages/frontend/core/src/blocksuite/ai/components/text-renderer.ts

@@ -268,7 +268,7 @@ export class TextRenderer extends WithDisposable(ShadowlessElement) {
               query: this._query,
             });
             this.disposables.add(() => {
-              doc.doc.clearQuery(this._query);
+              doc.doc.removeStore({ query: this._query });
             });
             this._doc.readonly = true;
             this.requestUpdate();

packages/frontend/core/src/blocksuite/block-suite-editor/lit-adaper.tsx

@@ -7,6 +7,7 @@ import {
   type PageEditor,
 } from '@affine/core/blocksuite/editors';
 import type { AffineEditorViewOptions } from '@affine/core/blocksuite/manager/editor-view';
+import { getViewManager } from '@affine/core/blocksuite/manager/migrating-view';
 import { useEnableAI } from '@affine/core/components/hooks/affine/use-enable-ai';
 import type { DocCustomPropertyInfo } from '@affine/core/modules/db';
 import type {
@@ -21,8 +22,9 @@ import { WorkspaceService } from '@affine/core/modules/workspace';
 import track from '@affine/track';
 import type { DocTitle } from '@blocksuite/affine/fragments/doc-title';
 import type { DocMode } from '@blocksuite/affine/model';
-import type { Store } from '@blocksuite/affine/store';
+import type { ExtensionType, Store } from '@blocksuite/affine/store';
 import {
+  type FrameworkProvider,
   useFramework,
   useLiveData,
   useService,
@@ -42,7 +44,6 @@ import {
   type DefaultOpenProperty,
   DocPropertiesTable,
 } from '../../components/doc-properties';
-import { enableEditorExtension } from '../extensions/entry/enable-editor';
 import { BiDirectionalLinkPanel } from './bi-directional-link-panel';
 import { BlocksuiteEditorJournalDocTitle } from './journal-doc-title';
 import { StarterBar } from './starter-bar';
@@ -301,3 +302,20 @@ export const BlocksuiteEdgelessEditor = forwardRef<
     </div>
   );
 });
+
+function enableEditorExtension(
+  framework: FrameworkProvider,
+  mode: 'edgeless' | 'page',
+  enableAI: boolean,
+  options: AffineEditorViewOptions
+): ExtensionType[] {
+  const manager = getViewManager(framework, enableAI, options);
+  if (BUILD_CONFIG.isMobileEdition) {
+    if (mode === 'page') {
+      return manager.get('mobile-page');
+    }
+
+    return manager.get('mobile-edgeless');
+  }
+  return manager.get(mode);
+}

packages/frontend/core/src/blocksuite/extensions/entry/enable-editor.ts

@@ -1,22 +0,0 @@
-import type { AffineEditorViewOptions } from '@affine/core/blocksuite/manager/editor-view';
-import type { ExtensionType } from '@blocksuite/affine/store';
-import { type FrameworkProvider } from '@toeverything/infra';
-
-import { getViewManager } from '../../manager/migrating-view';
-
-export function enableEditorExtension(
-  framework: FrameworkProvider,
-  mode: 'edgeless' | 'page',
-  enableAI: boolean,
-  options: AffineEditorViewOptions
-): ExtensionType[] {
-  const manager = getViewManager(framework, enableAI, options);
-  if (BUILD_CONFIG.isMobileEdition) {
-    if (mode === 'page') {
-      return manager.get('mobile-page');
-    }
-
-    return manager.get('mobile-edgeless');
-  }
-  return manager.get(mode);
-}

packages/frontend/core/src/modules/workspace/impls/doc.ts

@@ -3,9 +3,8 @@ import {
   AwarenessStore,
   type Doc,
   type ExtensionType,
-  type GetBlocksOptions,
-  type Query,
-  Store,
+  type GetStoreOptions,
+  StoreContainer,
   type Workspace,
   type YBlock,
 } from '@blocksuite/affine/store';
@@ -21,7 +20,7 @@ type DocOptions = {
 export class DocImpl implements Doc {
   private readonly _collection: Workspace;
 
-  private readonly _storeMap = new Map<string, Store>();
+  private readonly _storeContainer: StoreContainer;
 
   private readonly _initSpaceDoc = () => {
     {
@@ -105,19 +104,15 @@ export class DocImpl implements Doc {
 
     this._yBlocks = this._ySpaceDoc.getMap('blocks');
     this._collection = collection;
-  }
-
-  private _getReadonlyKey(readonly?: boolean): 'true' | 'false' {
-    return (readonly?.toString() as 'true' | 'false') ?? 'false';
+    this._storeContainer = new StoreContainer(this);
   }
 
   clear() {
     this._yBlocks.clear();
   }
 
-  clearQuery(query: Query, readonly?: boolean) {
-    const key = this._getQueryKey({ readonly, query });
-    this._storeMap.delete(key);
+  get removeStore() {
+    return this._storeContainer.removeStore;
   }
 
   private _destroy() {
@@ -134,58 +129,26 @@ export class DocImpl implements Doc {
     }
   }
 
-  private readonly _getQueryKey = (
-    idOrOptions: string | { readonly?: boolean; query?: Query }
-  ) => {
-    if (typeof idOrOptions === 'string') {
-      return idOrOptions;
-    }
-    const { readonly, query } = idOrOptions;
-    const readonlyKey = this._getReadonlyKey(readonly);
-    const key = JSON.stringify({
-      readonlyKey,
-      query,
-    });
-    return key;
-  };
-
   getStore({
     readonly,
     query,
     provider,
     extensions,
     id,
-  }: GetBlocksOptions = {}) {
-    let idOrOptions: string | { readonly?: boolean; query?: Query };
-    if (readonly || query) {
-      idOrOptions = { readonly, query };
-    } else if (!id) {
-      idOrOptions = this.workspace.idGenerator();
-    } else {
-      idOrOptions = id;
-    }
-    const key = this._getQueryKey(idOrOptions);
-
-    if (this._storeMap.has(key)) {
-      return this._storeMap.get(key) as Store;
-    }
-
+  }: GetStoreOptions = {}) {
     const storeExtensions = getStoreManager().get('store');
-    const extensionSet = new Set(
-      storeExtensions.concat(extensions ?? []).concat(this.storeExtensions)
-    );
+    const exts = storeExtensions
+      .concat(extensions ?? [])
+      .concat(this.storeExtensions);
+    const extensionSet = new Set(exts);
 
-    const doc = new Store({
-      doc: this,
+    return this._storeContainer.getStore({
+      id,
       readonly,
       query,
       provider,
       extensions: Array.from(extensionSet),
     });
-
-    this._storeMap.set(key, doc);
-
-    return doc;
   }
 
   load(initFn?: () => void): this {