feat(editor): add loaded and disposed hook for store extension (#10669)

blocksuite/framework/store/src/extension/selection/selection-extension.ts

@@ -2,7 +2,6 @@ import { BlockSuiteError, ErrorCode } from '@blocksuite/global/exceptions';
 import { Slot } from '@blocksuite/global/slot';
 import { computed, signal } from '@preact/signals-core';
 
-import type { Store } from '../../model';
 import { nanoid } from '../../utils/id-generator';
 import type { StackItem } from '../../yjs';
 import { StoreExtension } from '../store-extension';
@@ -47,9 +46,7 @@ export class StoreSelectionExtension extends StoreExtension {
     remoteChanged: new Slot<Map<number, BaseSelection[]>>(),
   };
 
-  constructor(store: Store) {
-    super(store);
-
+  override loaded() {
     this.store.provider.getAll(SelectionIdentifier).forEach(ctor => {
       [ctor].flat().forEach(ctor => {
         this._selectionConstructors[ctor.type] = ctor;

blocksuite/framework/store/src/extension/store-extension.ts

@@ -17,6 +17,10 @@ export class StoreExtension extends Extension {
     super();
   }
 
+  loaded() {}
+
+  disposed() {}
+
   static readonly [storeExtensionSymbol] = true;
 
   static override setup(di: Container) {

blocksuite/framework/store/src/model/store/store.ts

@@ -594,6 +594,10 @@ export class Store {
   }
 
   dispose() {
+    this._provider.getAll(StoreExtensionIdentifier).forEach(ext => {
+      ext.disposed();
+    });
+
     this._disposeBlockUpdated.dispose();
     this.slots.ready.dispose();
     this.slots.blockUpdated.dispose();
@@ -699,7 +703,9 @@ export class Store {
 
   load(initFn?: () => void) {
     this._doc.load(initFn);
-    this._provider.getAll(StoreExtensionIdentifier);
+    this._provider.getAll(StoreExtensionIdentifier).forEach(ext => {
+      ext.loaded();
+    });
     this.slots.ready.emit();
     return this;
   }