import type { Ctx } from '@milkdown/ctx'

import { editorViewOptionsCtx } from '@milkdown/core'
import clsx from 'clsx'

import './style.css'

export function nord(ctx: Ctx): void {
  ctx.update(editorViewOptionsCtx, (prev) => {
    const prevClass = prev.attributes

    return {
      ...prev,
      attributes: (state) => {
        const attrs =
          typeof prevClass === 'function' ? prevClass(state) : prevClass

        return {
          ...attrs,
          class: clsx(
            'prose dark:prose-invert',
            attrs?.class || '',
            'milkdown-theme-nord'
          ),
        }
      },
    }
  })
}