Support comment.foo highlighting queries

This should improve highlighting in cases like Elm that use `@comment.elm`.
Wilfred · May 11, 2024 · e51834f · e51834f
1 parent 835c5eb
commit e51834f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sample_files/compare.expected b/sample_files/compare.expected
@@ -257,7 +257,7 @@ sample_files/text_1.txt sample_files/text_2.txt
 87914b8ef4a6b042054633ddb182ce02  -
 
 sample_files/todomvc_1.gleam sample_files/todomvc_2.gleam
-070fbdea2a503392f0f0c37f1b6c04bb  -
+0a6f8335944adbd779c93aa52b7a30c5  -
 
 sample_files/toml_1.toml sample_files/toml_2.toml
 df36cf07bb1b56fbdbcea7f651b4344d  -

diff --git a/src/parse/tree_sitter_parser.rs b/src/parse/tree_sitter_parser.rs
@@ -1302,7 +1302,7 @@ fn tree_highlights(
             type_capture_ids.push(idx as u32);
         }
 
-        if name == "comment" {
+        if name == "comment" || name.starts_with("comment.") {
             comment_capture_ids.push(idx as u32);
         }
     }