refactor: migrate from shiki to shikiji (#2520)

This commit is contained in:
ocavue 2023-12-21 02:54:40 +08:00 committed by GitHub
parent e63473a5f8
commit 74138a9a58
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
17 changed files with 124 additions and 223 deletions

71
composables/shikiji.ts Normal file
View file

@ -0,0 +1,71 @@
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(),
})
}