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 -->
This commit is contained in:
Saul-Mirone
2025-05-07 06:08:43 +00:00
parent 93b1d6c729
commit 267bb3a975
11 changed files with 144 additions and 129 deletions

View File

@@ -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 {