diff options
-rw-r--r-- | assets/js/main.js | 10 |
1 files 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); }); |