import { type Highlighter, type BuiltinLanguage as Lang } from 'shikiji'

const highlighter = ref<Highlighter>()

const registeredLang = ref(new Map<string, boolean>())
let shikijiImport: Promise<void> | undefined

export function useHighlighter(lang: Lang) {
  if (!shikijiImport) {
    shikijiImport = import('shikiji')
      .then(async ({ getHighlighter }) => {
        highlighter.value = await getHighlighter({
          themes: [
            'vitesse-dark',
            'vitesse-light',
          ],
          langs: [
            'js',
            'css',
            'html',
          ],
        })
      })
  }

  if (!highlighter.value)
    return undefined

  if (!registeredLang.value.get(lang)) {
    highlighter.value.loadLanguage(lang)
      .then(() => {
        registeredLang.value.set(lang, true)
      })
      .catch(() => {
        const fallbackLang = 'md'
        highlighter.value?.loadLanguage(fallbackLang).then(() => {
          registeredLang.value.set(fallbackLang, true)
        })
      })
    return undefined
  }

  return highlighter.value
}

function useShikijiTheme() {
  return useColorMode().value === 'dark' ? 'vitesse-dark' : 'vitesse-light'
}

const HTML_ENTITIES = {
  '<': '&lt;',
  '>': '&gt;',
  '&': '&amp;',
  '\'': '&apos;',
  '"': '&quot;',
} as Record<string, string>

function escapeHtml(text: string) {
  return text.replace(/[<>&'"]/g, ch => HTML_ENTITIES[ch])
}

export function highlightCode(code: string, lang: Lang) {
  const highlighter = useHighlighter(lang)
  if (!highlighter)
    return escapeHtml(code)

  return highlighter.codeToHtml(code, {
    lang,
    theme: useShikijiTheme(),
  })
}