bytedance-flowgram/packages/plugins/select-box-plugin/src/create-select-box-plugin.ts

import {
  FlowSelectorBoundsLayer,
  FlowSelectorBoundsLayerOptions,
  FlowSelectorBoxLayer,
  FlowSelectorBoxOptions,
  type SelectorBoxPopoverProps,
} from '@flowgram.ai/renderer';
import { definePluginCreator } from '@flowgram.ai/core';

// import { SelectorBounds } from './selector-bounds';

export { type SelectorBoxPopoverProps };
export interface SelectBoxPluginOptions
  extends FlowSelectorBoundsLayerOptions,
    FlowSelectorBoxOptions {
  enable?: boolean;
}

export const createSelectBoxPlugin = definePluginCreator<SelectBoxPluginOptions>({
  onInit(ctx, opts): void {
    // 默认可用，所以强制判断 false
    if (opts.enable !== false) {
      ctx.playground.registerLayer<FlowSelectorBoundsLayer>(FlowSelectorBoundsLayer, opts);
      ctx.playground.registerLayer<FlowSelectorBoxLayer>(FlowSelectorBoxLayer, {
        canSelect: opts.canSelect,
      });
    }
  },
});