From 0091f62b759c97f7b80b744434f5b657b109e1da Mon Sep 17 00:00:00 2001
From: Anthony Fu <anthonyfu117@hotmail.com>
Date: Thu, 27 May 2021 16:21:15 +0800
Subject: [PATCH] Restore highlight.ts

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

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