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

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,
  };
};