chore: merge blocksuite source code (#9213)

blocksuite/framework/block-std/src/gfx/extension.ts

@@ -0,0 +1,53 @@
+import { type Container, createIdentifier } from '@blocksuite/global/di';
+import { BlockSuiteError, ErrorCode } from '@blocksuite/global/exceptions';
+
+import { Extension } from '../extension/extension.js';
+import type { GfxController } from './controller.js';
+import { GfxControllerIdentifier } from './identifiers.js';
+
+export const GfxExtensionIdentifier =
+  createIdentifier<GfxExtension>('GfxExtension');
+
+export const GfxClassExtenderIdentifier = createIdentifier<{
+  extendFn: (gfx: GfxController) => void;
+}>('GfxClassExtender');
+
+export abstract class GfxExtension extends Extension {
+  static key: string;
+
+  get std() {
+    return this.gfx.std;
+  }
+
+  constructor(protected readonly gfx: GfxController) {
+    super();
+  }
+
+  // This method is used to extend the GfxController
+  static extendGfx(_: GfxController) {}
+
+  static override setup(di: Container) {
+    if (!this.key) {
+      throw new BlockSuiteError(
+        ErrorCode.ValueNotExists,
+        'key is not defined in the GfxExtension'
+      );
+    }
+
+    di.addImpl(GfxClassExtenderIdentifier(this.key), {
+      extendFn: this.extendGfx,
+    });
+
+    di.add(this as unknown as { new (gfx: GfxController): GfxExtension }, [
+      GfxControllerIdentifier,
+    ]);
+
+    di.addImpl(GfxExtensionIdentifier(this.key), provider =>
+      provider.get(this)
+    );
+  }
+
+  mounted() {}
+
+  unmounted() {}
+}