AFFiNE-Mirror/blocksuite/affine/block-code/src/effects.ts

import type * as CommandsType from '@blocksuite/affine-shared/commands';

import { CodeBlockComponent } from './code-block';
import type { CodeBlockConfig } from './code-block-config';
import {
  AFFINE_CODE_TOOLBAR_WIDGET,
  AffineCodeToolbarWidget,
} from './code-toolbar';
import { AffineCodeToolbar } from './code-toolbar/components/code-toolbar';
import { LanguageListButton } from './code-toolbar/components/lang-button';
import { AffineCodeUnit } from './highlight/affine-code-unit';

export function effects() {
  customElements.define('language-list-button', LanguageListButton);
  customElements.define('affine-code-toolbar', AffineCodeToolbar);
  customElements.define(AFFINE_CODE_TOOLBAR_WIDGET, AffineCodeToolbarWidget);
  customElements.define('affine-code-unit', AffineCodeUnit);
  customElements.define('affine-code', CodeBlockComponent);
}

declare type _GLOBAL_ = typeof CommandsType;

declare global {
  namespace BlockSuite {
    interface BlockConfigs {
      'affine:code': CodeBlockConfig;
    }
  }

  interface HTMLElementTagNameMap {
    'language-list-button': LanguageListButton;
    'affine-code-toolbar': AffineCodeToolbar;
    [AFFINE_CODE_TOOLBAR_WIDGET]: AffineCodeToolbarWidget;
  }
}