packages/remirror__react-components/src/buttons/toggle-heading-button.tsx
import React, { FC, useCallback } from 'react';
import { HeadingExtension, HeadingExtensionAttributes } from '@remirror/extension-heading';
import { useActive, useCommands } from '@remirror/react-core';
import { CommandButton, CommandButtonProps } from './command-button';
export interface ToggleHeadingButtonProps
extends Omit<CommandButtonProps, 'commandName' | 'active' | 'enabled' | 'attrs' | 'onSelect'> {
attrs?: Partial<HeadingExtensionAttributes>;
}
export const ToggleHeadingButton: FC<ToggleHeadingButtonProps> = ({ attrs, ...rest }) => {
const { toggleHeading } = useCommands<HeadingExtension>();
const handleSelect = useCallback(() => {
if (toggleHeading.enabled(attrs)) {
toggleHeading(attrs);
}
}, [toggleHeading, attrs]);
const active = useActive<HeadingExtension>().heading(attrs);
const enabled = toggleHeading.enabled(attrs);
return (
<CommandButton
{...rest}
commandName='toggleHeading'
active={active}
enabled={enabled}
attrs={attrs}
onSelect={handleSelect}
/>
);
};