import { definePlugin, type Extension } from '@prosekit/core'
import { createHighlightPlugin, type Parser } from 'prosemirror-highlight'

/**
 * An alias for the `Parser` type from the `prosemirror-highlight` package.
 */
export type HighlightParser = Parser

export type CodeBlockHighlightOptions = {
  /**
   * A parser instance from the `prosemirror-highlight` package.
   */
  parser: HighlightParser
  /**
   * ProseMirror node types to highlight.
   *
   * @default ['codeBlock', 'mathBlock']
   */
  nodeTypes?: string[]
}

/**
 * Adds syntax highlighting to code blocks. This function requires a `Parser`
 * instance from the `prosemirror-highlight` package. See the
 * [documentation](https://github.com/ocavue/prosemirror-highlight) for more
 * information.
 *
 * @param options
 */
export function defineCodeBlockHighlight({
  parser,
  nodeTypes = ['codeBlock', 'mathBlock'],
}: CodeBlockHighlightOptions): Extension {
  return definePlugin(
    createHighlightPlugin({ parser, nodeTypes }),
  )
}