mirror of https://github.com/vuejs/vitepress
close #627 Co-authored-by: Kia Ishii <kia.king.08@gmail.com>pull/660/head
Ryo_gk
3 years ago
committed by
GitHub
parent
8bc14e6366
commit
1fcf4a4462
@ -1,52 +1,14 @@
|
||||
import chalk from 'chalk'
|
||||
import escapeHtml from 'escape-html'
|
||||
import prism from 'prismjs'
|
||||
import { getHighlighter } from 'shiki'
|
||||
|
||||
// prism is listed as actual dep so it's ok to require
|
||||
const loadLanguages = require('prismjs/components/index')
|
||||
export const highlight = async (theme = 'material-palenight') => {
|
||||
const highlighter = await getHighlighter({ theme })
|
||||
|
||||
// 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.`
|
||||
)
|
||||
)
|
||||
return (str: string, lang: string) => {
|
||||
if (!lang || lang === 'text') {
|
||||
return `<pre v-pre><code>${escapeHtml(str)}</code></pre>`
|
||||
}
|
||||
|
||||
return highlighter.codeToHtml(str, lang).replace(/^<pre.*?>/, '<pre v-pre>')
|
||||
}
|
||||
if (prism.languages[lang]) {
|
||||
const code = prism.highlight(str, prism.languages[lang], lang)
|
||||
return wrap(code, rawLang)
|
||||
}
|
||||
return wrap(str, 'text')
|
||||
}
|
||||
|
||||
Loading…
Reference in new issue