Fix bug with highlighting code

tc39 · Aug 19, 2015 · 27bd1fc · 27bd1fc
1 parent 323a97e
commit 27bd1fc
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/Spec.js b/lib/Spec.js
@@ -185,7 +185,11 @@ module.exports = class Spec {
     this._log('Highlighting syntax...');
     const codes = this.doc.querySelectorAll('pre code');
     for (let i = 0; i < codes.length; i++) {
-      hljs.highlightBlock(codes[i]);
+      const classAttr = codes[i].getAttribute('class');
+      const lang = classAttr.replace(/lang(uage)?\-/, '');
+      const result = hljs.highlight(lang, codes[i].textContent);
+      codes[i].innerHTML = result.value;
+      codes[i].setAttribute('class', classAttr + ' hljs');
     }
   }