import type { NodeSpec, NodeType } from 'prosemirror-model';
import { type CoreEditor, Node } from '@kerebron/editor';
import type {
  CommandFactories,
  CommandShortcuts,
} from '@kerebron/editor/commands';

export class NodeHorizontalRule extends Node {
  override name = 'hr';
  requires = ['doc'];

  override getNodeSpec(): NodeSpec {
    return {
      inline: true,
      group: 'inline',
      parseDOM: [{ tag: 'hr' }],
      toDOM() {
        return ['hr'];
      },
    };
  }

  override getCommandFactories(
    editor: CoreEditor,
    type: NodeType,
  ): Partial<CommandFactories> {
    return {
      'setHorizontalRule': () => (state, dispatch) => {
        if (dispatch) {
          dispatch(
            state.tr.replaceSelectionWith(type.create()).scrollIntoView(),
          );
        }
        return true;
      },
    };
  }

  override getKeyboardShortcuts(): Partial<CommandShortcuts> {
    return {
      'Mod-_': 'setHorizontalRule',
    };
  }
}