import type { Page, Theme } from '@vuepress/core'
import { templateRenderer } from '@vuepress/utils'
import type { PlumeThemeOptions, PlumeThemePageData } from '../shared/index.js'
import { mergeLocaleOptions } from './defaultOptions.js'
import { setupPlugins } from './plugins.js'
import { extendsPageData, setupPage } from './setupPages.js'
import { getThemePackage, resolve, templates } from './utils.js'

const THEME_NAME = 'vuepress-theme-plume'

export function plumeTheme({
  themePlugins,
  plugins,
  ...localeOptions
}: PlumeThemeOptions = {}): Theme {
  localeOptions = mergeLocaleOptions(localeOptions)
  const pluginsOptions = plugins ?? themePlugins ?? {}
  const pkg = getThemePackage()

  return app => ({
    name: THEME_NAME,
    templateBuild: templates('build.html'),
    clientConfigFile: resolve('client/config.js'),
    plugins: setupPlugins(app, pluginsOptions, localeOptions),
    onInitialized: app => setupPage(app, localeOptions),
    extendsPage: page => extendsPageData(app, page as Page<PlumeThemePageData>, localeOptions),
    templateBuildRenderer(template, context) {
      template = template
        .replace('{{ themeVersion }}', pkg.version || '')
        .replace(/^\s+|\s+$/gm, '')
        .replace(/\n/g, '')
      return templateRenderer(template, context)
    },
  })
}