snapshot of https://github.com/toeverything/AFFiNE.git (canary, 2026-09-16; blobs > 131072B omitted)

blocksuite/affine/blocks/surface/src/extensions/element-renderer.ts

@@ -0,0 +1,31 @@
+import {
+  createIdentifier,
+  type ServiceIdentifier,
+} from '@blocksuite/global/di';
+import type {
+  GfxLocalElementModel,
+  GfxPrimitiveElementModel,
+} from '@blocksuite/std/gfx';
+import type { ExtensionType } from '@blocksuite/store';
+
+import type { ElementRenderer } from '../renderer/elements';
+
+export const ElementRendererIdentifier =
+  createIdentifier<unknown>('element-renderer');
+
+export const ElementRendererExtension = <
+  T extends GfxPrimitiveElementModel | GfxLocalElementModel,
+>(
+  id: string,
+  renderer: ElementRenderer<T>
+): ExtensionType & {
+  identifier: ServiceIdentifier<ElementRenderer<T>>;
+} => {
+  const identifier = ElementRendererIdentifier<ElementRenderer<T>>(id);
+  return {
+    setup: di => {
+      di.addImpl(identifier, () => renderer);
+    },
+    identifier,
+  };
+};