feat(editor): add editor store (#9584)

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

@@ -0,0 +1,17 @@
+import type { Container } from '@blocksuite/global/di';
+
+/**
+ * Generic extension.
+ * Extensions are used to set up the dependency injection container.
+ * In most cases, you won't need to use this class directly.
+ * We provide helper classes like `CommandExtension` and `BlockViewExtension` to make it easier to create extensions.
+ */
+export abstract class Extension {
+  static setup(_di: Container): void {
+    // do nothing
+  }
+}
+
+export interface ExtensionType {
+  setup(di: Container): void;
+}

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

@@ -1,17 +1,2 @@
-import type { Container } from '@blocksuite/global/di';
-
-/**
- * Generic extension.
- * Extensions are used to set up the dependency injection container.
- * In most cases, you won't need to use this class directly.
- * We provide helper classes like `CommandExtension` and `BlockViewExtension` to make it easier to create extensions.
- */
-export abstract class Extension {
-  static setup(_di: Container): void {
-    // do nothing
-  }
-}
-
-export interface ExtensionType {
-  setup(di: Container): void;
-}
+export * from './extension';
+export * from './store-extension';

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

@@ -0,0 +1,41 @@
+import { type Container, createIdentifier } from '@blocksuite/global/di';
+import { BlockSuiteError, ErrorCode } from '@blocksuite/global/exceptions';
+
+import { StoreIdentifier } from '../store/identifier';
+import type { Store } from '../store/store';
+import { Extension } from './extension';
+
+export const StoreExtensionIdentifier =
+  createIdentifier<StoreExtension>('StoreExtension');
+
+export const storeExtensionSymbol = Symbol('StoreExtension');
+
+export class StoreExtension extends Extension {
+  constructor(readonly store: Store) {
+    super();
+  }
+
+  static readonly [storeExtensionSymbol] = true;
+
+  static override setup(di: Container) {
+    if (!this.key) {
+      throw new BlockSuiteError(
+        ErrorCode.ValueNotExists,
+        'Key is not defined in the StoreExtension'
+      );
+    }
+
+    di.add(this, [StoreIdentifier]);
+    di.addImpl(StoreExtensionIdentifier(this.key), provider =>
+      provider.get(this)
+    );
+  }
+
+  static readonly key: string;
+}
+
+export function isStoreExtensionConstructor(
+  extension: object
+): extension is typeof StoreExtension {
+  return storeExtensionSymbol in extension;
+}