import { isPlainObject } from 'vuepress/shared'
import type { Plugin } from 'vuepress/core'
import { copyCodeButtonPlugin } from './copy-code-button/index.js'
import { highlight } from './highlight/index.js'
import {
  collapsedLinesPlugin,
  highlightLinesPlugin,
  lineNumberPlugin,
  preWrapperPlugin,
} from './markdown/index.js'
import { prepareClientConfigFile } from './prepareClientConfigFile.js'
import type {
  CopyCodeOptions,
  HighlighterOptions,
  LineNumberOptions,
  PreWrapperOptions,
} from './types.js'

export interface ShikiPluginOptions
  extends HighlighterOptions, LineNumberOptions, PreWrapperOptions {
  /**
   * Add copy code button
   *
   * @default true
   */
  copyCode?: boolean | CopyCodeOptions
}

export function shikiPlugin({
  preWrapper = true,
  lineNumbers = true,
  copyCode = true,
  collapsedLines = false,
  ...options
}: ShikiPluginOptions = {}): Plugin {
  const copyCodeOptions: CopyCodeOptions = isPlainObject(copyCode) ? copyCode : {}

  return {
    name: '@vuepress-plume/plugin-shikiji',

    define: {
      __CC_DURATION__: copyCodeOptions.duration ?? 2000,
      __CC_SELECTOR__: `div[class*="language-"] > button.${copyCodeOptions.className || 'copy'}`,
    },

    clientConfigFile: app => prepareClientConfigFile(app, {
      copyCode: copyCode !== false,
      twoslash: options.twoslash ?? false,
    }),

    extendsMarkdown: async (md, app) => {
      const theme = options.theme ?? { light: 'github-light', dark: 'github-dark' }

      md.options.highlight = await highlight(theme, options)

      md.use(highlightLinesPlugin)
      md.use(preWrapperPlugin, { preWrapper })

      if (preWrapper) {
        copyCodeButtonPlugin(md, app, copyCode)
        md.use(lineNumberPlugin, { lineNumbers })
        md.use(collapsedLinesPlugin, { collapsedLines })
      }
    },

    extendsMarkdownOptions: (options) => {
      // 注入 floating-vue 后，需要关闭 代码块 的 v-pre 配置
      if ((options as any).vPre !== false) {
        const vPre = isPlainObject((options as any).vPre) ? (options as any).vPre : { block: true }
        if (vPre.block) {
          (options as any).vPre ??= {}
          ;(options as any).vPre.block = false
        }
      }
    },
  }
}