From 306203221a3c17457be11ff336f6976b1789ea59 Mon Sep 17 00:00:00 2001
From: jeffreytse <jeffreytse.mail@gmail.com>
Date: Mon, 24 Aug 2020 11:13:39 +0800
Subject: [PATCH] fix: wrong lang badge of code block

---
 assets/js/main.js | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/assets/js/main.js b/assets/js/main.js
index bb295f4..01c0ca5 100644
--- a/assets/js/main.js
+++ b/assets/js/main.js
@@ -67,16 +67,14 @@ document.addEventListener('DOMContentLoaded', function(event) {
         break;
       }
     }
-    if (lang) {
-      block.setAttribute('class', 'hljs ' + lang);
-    } else {
-      lang = block
-        .getAttribute('class')
-        .replace('hljs ', '');
+    if (!lang) {
+      cls = block.getAttribute('class');
+      lang = cls ? cls.replace('hljs ', '') : '';
     }
     if (lang.startsWith('language-')) {
       lang = lang.substr(9);
     }
+    block.setAttribute('class', 'hljs ' + lang);
     block.parentNode.setAttribute('data-lang', lang);
     hljs.highlightBlock(block);
   });