From 091ae954417b7895062691e3713ebf406c986f29 Mon Sep 17 00:00:00 2001
From: A Farzat <a@farzat.xyz>
Date: Mon, 8 Jun 2026 04:05:14 +0300
Subject: Support YAML, JS, and TOML

---
 src/util/language.rs | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

(limited to 'src/util')

diff --git a/src/util/language.rs b/src/util/language.rs
index f0870c8..080b26e 100644
--- a/src/util/language.rs
+++ b/src/util/language.rs
@@ -16,6 +16,9 @@ fn detect_from_extension(filename: &Path) -> Option<&'static str> {
         Some("rs") => Some("rust"),
         Some("py") => Some("python"),
         Some("json") => Some("json"),
+        Some("toml") => Some("toml"),
+        Some("yaml") | Some("yml") => Some("yaml"),
+        Some("js") | Some("jsx") => Some("javascript"),
         _ => None,
     }
 }
@@ -69,4 +72,29 @@ mod tests {
         let contents = "#!/bin/bash\nprint('hello')";
         assert_eq!(detect_language(Path::new("main.py"), contents), "python");
     }
+
+    #[test]
+    fn yaml_extension_is_detected() {
+        assert_eq!(detect_language(Path::new("config.yaml"), ""), "yaml");
+    }
+
+    #[test]
+    fn yml_alias_maps_to_yaml() {
+        assert_eq!(detect_language(Path::new("config.yml"), ""), "yaml");
+    }
+
+    #[test]
+    fn js_extension_is_detected() {
+        assert_eq!(detect_language(Path::new("file.js"), ""), "javascript");
+    }
+
+    #[test]
+    fn jsx_maps_to_javascript() {
+        assert_eq!(detect_language(Path::new("file.jsx"), ""), "javascript");
+    }
+
+    #[test]
+    fn toml_extension_is_detected() {
+        assert_eq!(detect_language(Path::new("Cargo.toml"), ""), "toml");
+    }
 }
-- 
cgit v1.3.1