import { ComponentType } from 'react';
import { addFilter } from '@wordpress/hooks';
import { ImageToolbar } from './imageToolbar';
import { BlockEditProps } from '@/types';
import { TextToolbar } from './textToolbar';

const BlockToolbar = (BlockEdit: ComponentType<BlockEditProps>) =>
  function (props: BlockEditProps) {
    const { name } = props;

    return (
      <>
        <BlockEdit {...props} />
        {['core/image', 'core/cover', 'core/media-text'].includes(name) && <ImageToolbar {...props} />}
        {['core/paragraph', 'core/heading', 'core/list-item'].includes(name) && <TextToolbar {...props} />}
      </>
    );
  };

addFilter('editor.BlockEdit', 'filter-ai/block-toolbar', BlockToolbar);