From 2dcd722021a1577dbe41d172795e4b27972dff4d Mon Sep 17 00:00:00 2001
From: unknown <jmguoyucheng@126.com>
Date: Sat, 27 Mar 2021 16:55:12 +0800
Subject: [PATCH] Optimize code

---
 src/node/markdown/plugins/highlight.ts | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/src/node/markdown/plugins/highlight.ts b/src/node/markdown/plugins/highlight.ts
index b9dbe0c9..ba966e58 100644
--- a/src/node/markdown/plugins/highlight.ts
+++ b/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')
 }