47 lines
1.1 KiB
TypeScript
47 lines
1.1 KiB
TypeScript
import chalk from 'chalk'
|
|
import prism from 'prismjs'
|
|
import loadLanguages from 'prismjs/components/index'
|
|
import escapeHtml from 'escape-html'
|
|
|
|
// required to make embedded highlighting work...
|
|
loadLanguages(['markup', 'css', 'javascript'])
|
|
|
|
function wrap(code: string, lang: string): string {
|
|
if (lang === 'text') {
|
|
code = escapeHtml(code)
|
|
}
|
|
return `<pre v-pre><code>${code}</code></pre>`
|
|
}
|
|
|
|
export const highlight = (str: string, lang: string) => {
|
|
if (!lang) {
|
|
return wrap(str, 'text')
|
|
}
|
|
lang = lang.toLowerCase()
|
|
const rawLang = lang
|
|
if (lang === 'vue' || lang === 'html') {
|
|
lang = 'markup'
|
|
}
|
|
if (lang === 'md') {
|
|
lang = 'markdown'
|
|
}
|
|
if (lang === 'ts') {
|
|
lang = 'typescript'
|
|
}
|
|
if (lang === 'py') {
|
|
lang = 'python'
|
|
}
|
|
if (!prism.languages[lang]) {
|
|
try {
|
|
loadLanguages([lang])
|
|
} catch (e) {
|
|
console.warn(chalk.yellow(`[vitepress] Syntax highlight for language "${lang}" is not supported.`))
|
|
}
|
|
}
|
|
if (prism.languages[lang]) {
|
|
const code = prism.highlight(str, prism.languages[lang], lang)
|
|
return wrap(code, rawLang)
|
|
}
|
|
return wrap(str, 'text')
|
|
}
|