1 files changed, 56 insertions, 0 deletions
diff --git a/docs/code.js b/docs/code.js
new file mode 100644
index 0000000..8440591
--- /dev/null
+++ b/docs/code.js
@@ -0,0 +1,56 @@
+// high-quality syntax highlighter
+// TODO: Implement actual highlighter (or fix many regex bugs)
+
+const term = (t) =>
+  t
+    .replaceAll(/'(.)'/g, "<span class='string'>'$1'</span>")
+    .replaceAll(/"([^\"]*)"/g, "<span class='string'>\"$1\"</span>")
+    .replaceAll(/(\([+-][0-9]+[ubt]?\))/g, "<span class='number'>$1</span>")
+    .replaceAll(/(?<!\>)(\()/g, "<span class='left-app'>(</span>")
+    .replaceAll(/(\))(?!\<)/g, "<span class='right-app'>)</span>")
+    .replaceAll("[", "<span class='left-abs'>[</span>")
+    .replaceAll("]", "<span class='right-abs'>]</span>")
+    .replaceAll(/(?<![+-\d])([0-9])/g, "<span class='index'>$1</span>");
+
+const highlightTerm = (elem) => {
+  elem.innerHTML = term(elem.innerHTML);
+};
+
+const highlight = (elem) => {
+  const fixPath = (p) => p.replace("/", "_");
+
+  elem.innerHTML = elem.innerHTML
+    .replaceAll(
+      /^:import std\/(.*) (.*)$/gm,
+      (_, p, s) =>
+        `<span class="com">:import</span> <a href='${fixPath(
+          p,
+        )}.bruijn.html'>std/${p}</a> ${s}`,
+    )
+    .replaceAll(
+      /^:input std\/(.*)$/gm,
+      (_, p) =>
+        `<span class="com">:input</span> <a href='${fixPath(
+          p,
+        )}.bruijn.html'>std/${p}</a>`,
+    )
+    .replaceAll(
+      /^:test (\(.*\)) (\(.*\))$/gm,
+      (_, t1, t2) => `<span class='com'>:test</span> ${term(t1)} ${term(t2)}`,
+    )
+    .replaceAll(
+      /^:time (.*)$/gm,
+      (_, t) => `<span class='com'>:time</span> ${term(t)}`,
+    )
+    .replaceAll(
+      /^([^:\n<#][^ ]*) (.*)$/gm,
+      (_, d, t) => `<span class='def'>${d}</span> ${term(t)}`,
+    )
+    .replaceAll(/^# (.*)$/gm, "<span class='comment'># $1</span>")
+    .replaceAll(/ ⧗ (.*)\n/g, " ⧗ <span class='type'>$1</span>\n");
+};
+
+window.onload = () => {
+  [...document.querySelectorAll("code.language-bruijn")].forEach(highlight);
+  [...document.querySelectorAll("code.bruijn")].forEach(highlightTerm);
+};