Restore highlight.ts

pull/267/head
Anthony Fu 4 years ago committed by GitHub
parent de550047d7
commit 0091f62b75
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -17,30 +17,21 @@ export const highlight = (str: string, lang: string) => {
if (!lang) { if (!lang) {
return wrap(str, 'text') return wrap(str, 'text')
} }
lang = lang.toLowerCase() lang = lang.toLowerCase()
const rawLang = lang const rawLang = lang
if (lang === 'vue' || lang === 'html') { if (lang === 'vue' || lang === 'html') {
lang = 'markup' lang = 'markup'
} }
if (lang === 'md') { if (lang === 'md') {
lang = 'markdown' lang = 'markdown'
} }
if (lang === 'ts') { if (lang === 'ts') {
lang = 'typescript' lang = 'typescript'
} }
if (lang === 'py') { if (lang === 'py') {
lang = 'python' lang = 'python'
} }
if (!prism.languages[lang]) {
if (prism.languages[lang]) {
const code = prism.highlight(str, prism.languages[lang], lang)
return wrap(code, rawLang)
} else {
try { try {
loadLanguages([lang]) loadLanguages([lang])
} catch (e) { } catch (e) {
@ -50,6 +41,10 @@ export const highlight = (str: string, lang: string) => {
) )
) )
} }
return wrap(str, 'text')
} }
if (prism.languages[lang]) {
const code = prism.highlight(str, prism.languages[lang], lang)
return wrap(code, rawLang)
}
return wrap(str, 'text')
} }

Loading…
Cancel
Save