erl_docgen: Make code type attribute influence highlight mode

1 files changed, 14 insertions, 3 deletions

diff --git a/lib/erl_docgen/priv/js/highlight.js b/lib/erl_docgen/priv/js/highlight.js
index 0594b42aa3..b176a60928 100644
--- a/lib/erl_docgen/priv/js/highlight.js
+++ b/lib/erl_docgen/priv/js/highlight.js
@@ -28,11 +28,22 @@ window.addEventListener("load", function () {
 
         window.clearInterval(intervalId);
 
-        var i, len, nodes = document.querySelectorAll('.example');
-        for (i = 0, len = nodes.length; i < len; i += 1) {
-            window.hljs.highlightBlock(nodes[i]);
+        function highlightLanguage(css, languages) {
+            var i, len, nodes = document.querySelectorAll(css);
+            for (i = 0, len = nodes.length; i < len; i += 1) {
+                hljs.configure({languages: languages});
+                window.hljs.highlightBlock(nodes[i]);
+            }
         }
 
+        highlightLanguage('.example-erl',["erlang"]);
+        highlightLanguage('.example-erl-repl',["erlang-repl"]);
+        highlightLanguage('.example-c',["cpp"]);
+        highlightLanguage('.example-cpp',["cpp"]);
+        highlightLanguage('.example-sh',["bash"]);
+        highlightLanguage('.example-diff',["diff"]);
+        highlightLanguage('.example-none',["erlang","erlang-repl","bash","cpp","diff"]);
+
     }
 
     intervalId = window.setInterval(doHighlight, 50);