Optimize code

src/node/markdown/plugins/highlight.ts

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