Highlighting multi-line comments

autolab · Apr 22, 2024 · 90bc370 · 90bc370
1 parent 8511d3e
commit 90bc370
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/app/views/submissions/view.html.erb b/app/views/submissions/view.html.erb
@@ -52,6 +52,23 @@
       var headerPositionStr = null;
     <% end %>
     hljs.initHighlightingOnLoad();
+    document.addEventListener('DOMContentLoaded', (event) => {
+        let combinedCode = '';
+        document.querySelectorAll('pre code').forEach((block) => {
+            combinedCode += block.textContent + '\n';
+        });
+
+        let highlightedCode = hljs.highlightAuto(combinedCode).value;
+        let commentSpans = highlightedCode.match(/<span class="hljs-comment">([^<]*)<\/span>/g);
+        let commentContents = commentSpans.map(span => span.replace(/<[^>]+>/g, ''));
+        let splitContent = commentContents.flatMap(str => str.split('\n'));
+
+        document.querySelectorAll('pre code').forEach((block, index) => {
+            if (block.textContent !== null && block.textContent !== "" && splitContent.includes(block.textContent.replace(/\n/g, ''))) {
+                block.innerHTML = `<code class='hljs-comment'>${block.textContent}</code>`;
+            }
+        });
+    });
     PDFJS.workerSrc = "<%= asset_url 'pdf.worker.js' %>";
   </script>
   <%= render partial: "golden-layout" %>