Merge branch 'dgmcdona/tree-sitter-powershell'

README.md

@@ -103,6 +103,7 @@ javascript
 markdown
 nix
 php
+powershell
 python
 ruby
 rust

build.rs

@@ -284,4 +284,21 @@ fn main() {
         .warnings(false)
         .file(typescript_dir.join("scanner.c"))
         .compile("tree_sitter_typescript_scanner");
+
+    // powershell
+    let powershell_dir: PathBuf = ["vendor", "tree-sitter-powershell", "src"].iter().collect();
+
+    println!("cargo:rerun-if-changed=vendor/tree-sitter-powershell/src/parser.c");
+    cc::Build::new()
+        .include(&powershell_dir)
+        .warnings(false)
+        .file(&powershell_dir.join("parser.c"))
+        .compile("tree-sitter-powershell");
+
+    println!("cargo:rerun-if-changed=vendor/tree-sitter-powershell/src/scanner.c");
+    cc::Build::new()
+        .include(&powershell_dir)
+        .warnings(false)
+        .file(&powershell_dir.join("scanner.c"))
+        .compile("tree_sitter_powershell_scanner");
 }

flake.nix

@@ -90,6 +90,11 @@
       url = "github:theHamsta/tree-sitter-cuda/v0.20.3";
       flake = false;
     };
+
+    tree-sitter-powershell = {
+      url = "github:airbus-cert/tree-sitter-powershell";
+      flake = false;
+    };
   };
 
   outputs = inputs:
@@ -123,6 +128,7 @@
           ln -s ${inputs.tree-sitter-typescript} vendor/tree-sitter-typescript
           ln -s ${inputs.tree-sitter-nix} vendor/tree-sitter-nix
           ln -s ${inputs.tree-sitter-cuda} vendor/tree-sitter-cuda
+          ln -s ${inputs.tree-sitter-powershell} vendor/tree-sitter-powershell
         '';
       in rec {
         # `nix build`

src/extractor_chooser.rs

@@ -15,6 +15,7 @@ impl<'extractor> ExtractorChooser<'extractor> {
         types_builder.add_defaults();
         types_builder.add_def("cuda:*.cu,*.cuh,*.hpp")?;
         types_builder.add_def("cuda:include:cpp")?;
+        types_builder.add_def("powershell:*.ps1")?;
 
         let mut names_to_extractors = HashMap::with_capacity(extractors.len());
 

src/language.rs

@@ -18,6 +18,7 @@ pub enum Language {
     Markdown,
     Nix,
     Php,
+    PowerShell,
     Python,
     Ruby,
     Rust,
@@ -45,6 +46,7 @@ impl Language {
                 Language::Markdown => tree_sitter_markdown(),
                 Language::Nix => tree_sitter_nix(),
                 Language::Php => tree_sitter_php(),
+                Language::PowerShell => tree_sitter_powershell(),
                 Language::Python => tree_sitter_python(),
                 Language::Ruby => tree_sitter_ruby(),
                 Language::Rust => tree_sitter_rust(),
@@ -77,6 +79,7 @@ impl Language {
             Language::Rust => "rust",
             Language::Sass => "sass",
             Language::TypeScript => "ts",
+            Language::PowerShell => "powershell",
         }
     }
 }
@@ -112,6 +115,7 @@ extern "C" {
     fn tree_sitter_markdown() -> tree_sitter::Language;
     fn tree_sitter_nix() -> tree_sitter::Language;
     fn tree_sitter_php() -> tree_sitter::Language;
+    fn tree_sitter_powershell() -> tree_sitter::Language;
     fn tree_sitter_python() -> tree_sitter::Language;
     fn tree_sitter_ruby() -> tree_sitter::Language;
     fn tree_sitter_rust() -> tree_sitter::Language;

tests/cmd/languages.trycmd

@@ -12,6 +12,7 @@ javascript
 markdown
 nix
 php
+powershell
 python
 ruby
 rust