From e7f729c61b4b693945ad195b7db35d61c7533fc8 Mon Sep 17 00:00:00 2001
From: KtorZ <matthias.benkort@gmail.com>
Date: Fri, 16 Dec 2022 23:55:09 +0100
Subject: [PATCH] Define HighlightJS definition for Aiken.

---
 crates/project/src/docs.rs                    |  4 ++--
 crates/project/templates/_layout.html         |  4 ++--
 ...hlightjs-gleam.js => highlightjs-aiken.js} | 22 ++++++++++++++-----
 crates/project/templates/module.html          |  8 +++----
 4 files changed, 25 insertions(+), 13 deletions(-)
 rename crates/project/templates/js/{highlightjs-gleam.js => highlightjs-aiken.js} (82%)

diff --git a/crates/project/src/docs.rs b/crates/project/src/docs.rs
index 7bec74ca..c99e97a1 100644
--- a/crates/project/src/docs.rs
+++ b/crates/project/src/docs.rs
@@ -185,8 +185,8 @@ fn generate_static_assets(search_indexes: Vec<SearchIndex>) -> Vec<DocFile> {
     });
 
     assets.push(DocFile {
-        path: PathBuf::from("js/highlightjs-gleam.js"),
-        content: std::include_str!("../templates/js/highlightjs-gleam.js").to_string(),
+        path: PathBuf::from("js/highlightjs-aiken.js"),
+        content: std::include_str!("../templates/js/highlightjs-aiken.js").to_string(),
     });
 
     assets.push(DocFile {
diff --git a/crates/project/templates/_layout.html b/crates/project/templates/_layout.html
index fd23a59a..875704b3 100644
--- a/crates/project/templates/_layout.html
+++ b/crates/project/templates/_layout.html
@@ -252,11 +252,11 @@
     </svg>
 
     <script src="{{ breadcrumbs }}/js/highlight.min.js?v={{ aiken_version }}"></script>
-    <script src="{{ breadcrumbs }}/js/highlightjs-gleam.js?v={{ aiken_version }}"></script>
+    <script src="{{ breadcrumbs }}/js/highlightjs-aiken.js?v={{ aiken_version }}"></script>
     <script>
       document.querySelectorAll("pre code").forEach((elem) => {
         if (elem.className === "") {
-          elem.classList.add("gleam");
+          elem.classList.add("aiken");
         }
       });
       hljs.highlightAll();
diff --git a/crates/project/templates/js/highlightjs-gleam.js b/crates/project/templates/js/highlightjs-aiken.js
similarity index 82%
rename from crates/project/templates/js/highlightjs-gleam.js
rename to crates/project/templates/js/highlightjs-aiken.js
index 7d4423cf..66525204 100644
--- a/crates/project/templates/js/highlightjs-gleam.js
+++ b/crates/project/templates/js/highlightjs-aiken.js
@@ -1,7 +1,6 @@
-hljs.registerLanguage("gleam", function (hljs) {
+hljs.registerLanguage("aiken", function (hljs) {
   const KEYWORDS =
-    "as assert case const external fn if import let " +
-    "use opaque pub todo try tuple type";
+    "as assert when is const fn if let use opaque pub assert check todo type";
   const STRING = {
     className: "string",
     variants: [{ begin: /"/, end: /"/ }],
@@ -42,8 +41,8 @@ hljs.registerLanguage("gleam", function (hljs) {
   };
 
   return {
-    name: "Gleam",
-    aliases: ["gleam"],
+    name: "Aiken",
+    aliases: ["aiken"],
     contains: [
       hljs.C_LINE_COMMENT_MODE,
       STRING,
@@ -80,6 +79,19 @@ hljs.registerLanguage("gleam", function (hljs) {
           },
         ],
       },
+      {
+        className: "keyword",
+        beginKeywords: "use",
+        end: "\n",
+        excludeEnd: true,
+        contains: [
+          {
+            className: "title",
+            begin: "[a-z][a-z0-9_/]*\\w*",
+            relevance: 0,
+          },
+        ],
+      },
       {
         className: "keyword",
         beginKeywords: KEYWORDS,
diff --git a/crates/project/templates/module.html b/crates/project/templates/module.html
index ad5bf69f..9a651cf2 100644
--- a/crates/project/templates/module.html
+++ b/crates/project/templates/module.html
@@ -57,7 +57,7 @@
     </div>
     <div class="custom-type-constructors">
       <div class="rendered-markdown">{{ type_info.documentation|safe }}</div>
-      <pre><code class="hljs gleam">{{ type_info.definition }}</code></pre>
+      <pre><code class="hljs aiken">{{ type_info.definition }}</code></pre>
       {% if !type_info.constructors.is_empty() %}
       <h3>
         Constructors
@@ -67,7 +67,7 @@
         <li class="constructor-item">
           <div class="constructor-row">
             <svg class="icon icon-star"><use xlink:href="#icon-star"></use></svg>
-            <pre class="constructor-name"><code class="hljs gleam">{{ constructor.definition }}</code></pre>
+            <pre class="constructor-name"><code class="hljs aiken">{{ constructor.definition }}</code></pre>
           </div>
 
           <div class="constructor-item-docs">
@@ -124,7 +124,7 @@
       </a>
       {% endif %}
     </div>
-    <pre><code class="hljs gleam">{{ constant.definition }}</code></pre>
+    <pre><code class="hljs aiken">{{ constant.definition }}</code></pre>
     <div class="rendered-markdown">{{ constant.documentation|safe }}</div>
   </div>
   {% endfor %}
@@ -150,7 +150,7 @@
       </a>
       {% endif %}
     </div>
-    <pre><code class="hljs gleam">{{ function.signature }}</code></pre>
+    <pre><code class="hljs aiken">{{ function.signature }}</code></pre>
     <div class="rendered-markdown">{{ function.documentation|safe }}</div>
   </div>
   {% endfor %}