fix: wrong lang badge of code block

jeffreytse · Aug 24, 2020 · 3062032 · 3062032
1 parent 83eac82
commit 3062032
Showing 1 changed file with 4 additions and 6 deletions.
diff --git 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);
   });