mirror of
https://gitee.com/element-plus/element-plus.git
synced 2024-12-05 20:58:22 +08:00
bb939c6a58
* refactor(build): move to internal package * ci: fix * chore: fix * fix: docs * chore: add fast-glob to root package * chore: fix PR comments * build: add tsx * refactor: re-organize * refactor: improve log
57 lines
1.4 KiB
TypeScript
57 lines
1.4 KiB
TypeScript
// ref https://github.com/vuejs/vitepress/blob/main/src/node/markdown/plugins/highlight.ts
|
|
import chalk from 'chalk'
|
|
import escapeHtml from 'escape-html'
|
|
import prism from 'prismjs'
|
|
import consola from 'consola'
|
|
|
|
// prism is listed as actual dep so it's ok to require
|
|
// eslint-disable-next-line @typescript-eslint/no-var-requires
|
|
const loadLanguages = require('prismjs/components/index')
|
|
|
|
// 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 {
|
|
// eslint-disable-next-line no-console
|
|
consola.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')
|
|
}
|