elk/composables/shiki.ts

77 lines
1.6 KiB
TypeScript
Raw Normal View History

2022-11-30 05:45:42 +00:00
import type { Highlighter, Lang } from 'shiki-es'
2022-11-24 03:42:03 +00:00
2022-12-13 13:02:43 +00:00
const shiki = ref<Highlighter>()
2022-11-24 03:42:03 +00:00
const registeredLang = ref(new Map<string, boolean>())
let shikiImport: Promise<void> | undefined
export function useHighlighter(lang: Lang) {
2022-11-24 03:42:03 +00:00
if (!shikiImport) {
2022-11-30 05:45:42 +00:00
shikiImport = import('shiki-es')
2022-11-24 03:42:03 +00:00
.then(async (r) => {
r.setCDN('/shiki/')
shiki.value = await r.getHighlighter({
themes: [
'vitesse-dark',
'vitesse-light',
],
2022-11-30 07:15:26 +00:00
langs: [
'js',
'css',
2022-11-30 07:15:26 +00:00
'html',
],
2022-11-24 03:42:03 +00:00
})
})
}
if (!shiki.value)
2022-12-13 13:02:43 +00:00
return undefined
2022-11-24 03:42:03 +00:00
if (!registeredLang.value.get(lang)) {
shiki.value.loadLanguage(lang)
.then(() => {
registeredLang.value.set(lang, true)
})
2023-01-13 15:08:08 +00:00
.catch(() => {
const fallbackLang = 'md'
shiki.value?.loadLanguage(fallbackLang).then(() => {
registeredLang.value.set(fallbackLang, true)
})
2022-11-24 08:57:24 +00:00
})
2022-12-13 13:02:43 +00:00
return undefined
2022-11-24 03:42:03 +00:00
}
2022-12-13 13:02:43 +00:00
return shiki.value
}
export function useShikiTheme() {
return useColorMode().value === 'dark' ? 'vitesse-dark' : 'vitesse-light'
2022-12-13 13:02:43 +00:00
}
const HTML_ENTITIES = {
'<': '&lt;',
'>': '&gt;',
'&': '&amp;',
'\'': '&apos;',
'"': '&quot;',
} as Record<string, string>
function escapeHtml(text: string) {
return text.replace(/[<>&'"]/g, ch => HTML_ENTITIES[ch])
}
2022-12-13 13:02:43 +00:00
export function highlightCode(code: string, lang: Lang) {
2023-01-17 06:25:57 +00:00
const shiki = useHighlighter(lang)
2022-12-13 13:02:43 +00:00
if (!shiki)
return escapeHtml(code)
2022-12-13 13:02:43 +00:00
return shiki.codeToHtml(code, {
2022-11-24 03:42:03 +00:00
lang,
2022-12-13 13:02:43 +00:00
theme: useShikiTheme(),
2022-11-24 03:42:03 +00:00
})
}
2022-12-13 13:02:43 +00:00
export function useShiki() {
return shiki
}