Restore highlight.ts

src/node/markdown/plugins/highlight.ts

@@ -17,30 +17,21 @@ 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]) {
-    const code = prism.highlight(str, prism.languages[lang], lang)
-    return wrap(code, rawLang)
-  } else {
+  if (!prism.languages[lang]) {
     try {
       loadLanguages([lang])
     } 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')
 }