add tree-sitter-nix support

Author: GTrunSec

build.rs

@@ -110,6 +110,23 @@ fn main() {
         .file(markdown_dir.join("scanner.cc"))
         .compile("tree_sitter_markdown_scanner");
 
+    // nix
+    let nix_dir: PathBuf = ["vendor", "tree-sitter-nix", "src"].iter().collect();
+
+    println!("cargo:rerun-if-changed=vendor/tree-sitter-nix/src/parser.c");
+    cc::Build::new()
+        .include(&nix_dir)
+        .warnings(false)
+        .file(nix_dir.join("parser.c"))
+        .compile("tree-sitter-nix");
+
+    println!("cargo:rerun-if-changed=vendor/tree-sitter-nix/src/scanner.c");
+    cc::Build::new()
+        .include(&nix_dir)
+        .warnings(false)
+        .file(nix_dir.join("scanner.c"))
+        .compile("tree_sitter_nix_scanner");
+
     // php
     let php_dir: PathBuf = ["vendor", "tree-sitter-php", "src"].iter().collect();
 

flake.lock

@@ -11,6 +11,11 @@
       flake = false;
     };
 
+    tree-sitter-nix = {
+      url = "github:cstrahan/tree-sitter-nix";
+      flake = false;
+    };
+
     tree-sitter-elixir = {
       url = "github:elixir-lang/tree-sitter-elixir/main";
       flake = false;
@@ -81,6 +86,7 @@
           ln -s ${inputs.tree-sitter-ruby} vendor/tree-sitter-ruby
           ln -s ${inputs.tree-sitter-rust} vendor/tree-sitter-rust
           ln -s ${inputs.tree-sitter-typescript} vendor/tree-sitter-typescript
+          ln -s ${inputs.tree-sitter-nix} vendor/tree-sitter-nix
         '';
       in rec {
         # `nix build`

flake.nix

@@ -10,6 +10,7 @@ pub enum Language {
     Haskell,
     JavaScript,
     Markdown,
+    Nix,
     Php,
     Ruby,
     Rust,
@@ -25,6 +26,7 @@ impl Language {
             Language::Haskell,
             Language::JavaScript,
             Language::Markdown,
+            Language::Nix,
             Language::Php,
             Language::Ruby,
             Language::Rust,
@@ -41,6 +43,7 @@ impl Language {
                 Language::Haskell => tree_sitter_haskell(),
                 Language::JavaScript => tree_sitter_javascript(),
                 Language::Markdown => tree_sitter_markdown(),
+                Language::Nix => tree_sitter_nix(),
                 Language::Php => tree_sitter_php(),
                 Language::Ruby => tree_sitter_ruby(),
                 Language::Rust => tree_sitter_rust(),
@@ -61,6 +64,7 @@ impl Language {
             Language::Haskell => "haskell",
             Language::JavaScript => "js",
             Language::Markdown => "markdown",
+            Language::Nix => "nix",
             Language::Php => "php",
             Language::Ruby => "ruby",
             Language::Rust => "rust",
@@ -80,6 +84,7 @@ impl FromStr for Language {
             "haskell" => Ok(Language::Haskell),
             "javascript" => Ok(Language::JavaScript),
             "markdown" => Ok(Language::Markdown),
+            "nix" => Ok(Language::Nix),
             "php" => Ok(Language::Php),
             "ruby" => Ok(Language::Ruby),
             "rust" => Ok(Language::Rust),
@@ -106,6 +111,7 @@ impl Display for Language {
             Language::Haskell => f.write_str("haskell"),
             Language::JavaScript => f.write_str("javascript"),
             Language::Markdown => f.write_str("markdown"),
+            Language::Nix => f.write_str("nix"),
             Language::Php => f.write_str("php"),
             Language::Ruby => f.write_str("ruby"),
             Language::Rust => f.write_str("rust"),
@@ -155,6 +161,7 @@ extern "C" {
     fn tree_sitter_haskell() -> tree_sitter::Language;
     fn tree_sitter_javascript() -> tree_sitter::Language;
     fn tree_sitter_markdown() -> tree_sitter::Language;
+    fn tree_sitter_nix() -> tree_sitter::Language;
     fn tree_sitter_php() -> tree_sitter::Language;
     fn tree_sitter_ruby() -> tree_sitter::Language;
     fn tree_sitter_rust() -> tree_sitter::Language;

src/language.rs

@@ -340,6 +340,20 @@ mod tests {
         ]))
     }
 
+    #[test]
+    fn all_nix() {
+        insta::assert_snapshot!(call(&[
+            "tree-grepper",
+            "-q",
+            "nix",
+            "(_)",
+            "--format=pretty-json",
+            "--sort",
+            "--no-gitignore",
+            "vendor/tree-sitter-nix/test/highlight/basic.nix",
+        ]))
+    }
+
     #[test]
     fn all_php() {
         insta::assert_snapshot!(call(&[