AFFiNE-Mirror/blocksuite/affine/inlines/reference/src/view.ts

import {
  type ViewExtensionContext,
  ViewExtensionProvider,
} from '@blocksuite/affine-ext-loader';

import { effects } from './effects';
import { ReferenceInlineSpecExtension } from './inline-spec';
import { RefNodeSlotsExtension } from './reference-node';
import { referenceNodeToolbar } from './toolbar';

export class ReferenceViewExtension extends ViewExtensionProvider {
  override name = 'affine-reference-inline';

  override effect(): void {
    super.effect();
    effects();
  }

  override setup(context: ViewExtensionContext) {
    super.setup(context);
    context.register(referenceNodeToolbar);
    context.register(ReferenceInlineSpecExtension);
    context.register(RefNodeSlotsExtension);
  }
}