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 escapeHtml from 'escape-html'
|
||||||
import prism from 'prismjs'
|
import { getHighlighter } from 'shiki'
|
||||||
|
|
||||||
// prism is listed as actual dep so it's ok to require
|
export const highlight = async (theme = 'material-palenight') => {
|
||||||
const loadLanguages = require('prismjs/components/index')
|
const highlighter = await getHighlighter({ theme })
|
||||||
|
|
||||||
// required to make embedded highlighting work...
|
return (str: string, lang: string) => {
|
||||||
loadLanguages(['markup', 'css', 'javascript'])
|
if (!lang || lang === 'text') {
|
||||||
|
return `<pre v-pre><code>${escapeHtml(str)}</code></pre>`
|
||||||
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 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