import { wrappingInputRule, type InputRule } from 'prosemirror-inputrules'
import { type NodeType } from 'prosemirror-model'

// Given a list node type, returns an input rule that turns a number
// followed by a dot at the start of a textblock into an ordered list.
// Copied from https://github.com/ProseMirror/prosemirror-example-setup/blob/master/src/inputrules.js
export const orderedListRule = (nodeType: NodeType): InputRule =>
  wrappingInputRule(
    /^(\d+)\.\s$/,
    nodeType,
    (match) => ({ order: +match[1] }),
    (match, node) => node.childCount + node.attrs.order == +match[1],
  )

// Given a list node type, returns an input rule that turns a bullet
// (dash, plush, or asterisk) at the start of a textblock into a
// bullet list.
// Copied from https://github.com/ProseMirror/prosemirror-example-setup/blob/master/src/inputrules.js
export const bulletListRule = (nodeType: NodeType): InputRule =>
  wrappingInputRule(/^\s*([-+*])\s$/, nodeType)