textbringer-tree-sitter 1.2.2 → 1.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 28756ea2a173ddd14d86a892280b4bb9aef8b18c092fc2f406a447bf45d409da
-  data.tar.gz: e81b17e1b08b48f2ce5220d750d0431311a7ecd6463c37007f928fd515258c0d
+  metadata.gz: 1546bddc8666286698d030db8e1a5e37dd436b4b6a3d83061f93c55a2a19a94d
+  data.tar.gz: bd1f3ca1dc7104c04a5ee2d8d45ae99c63b91ec721789aece097e4e8d38a78a1
 SHA512:
-  metadata.gz: 75175a3f5529504d9318c2667c5bfbc0cfa368dc723da8b3ef3384b51db0d65b56eb6308d761ec4461df3caef9e89465af28d29fd622c94d6fa011206d98c8e4
-  data.tar.gz: 58e279671e32eba8b409daf1c7def3662de72abe3dadf5ecf39333d71679a6ac41b4e757ecb1fa93e60e7cfad236e5facb8591203a1514a34bfd8360fb222976
+  metadata.gz: d1e6d76abe0b42b1fb05aa3423eff9aa62ea0f084e1d2bce563d123e60149fc177648d379ff8163cbe49e14129afad1bc66cb608736e84da88bae8db5b038167
+  data.tar.gz: 13d4fda0fe03e09372cf538f82421848c70ba49ea4c356b5649dd1adef6b73bc8880ba1691cbc0fb3105a5070cc6017e4b66fc943d784d7a020686a00931074a

data/lib/textbringer/tree_sitter/version.rb

@@ -2,6 +2,6 @@
 
 module Textbringer
   module TreeSitter
-    VERSION = "1.2.2"
+    VERSION = "1.2.3"
   end
 end

data/lib/textbringer/tree_sitter_adapter.rb

@@ -66,7 +66,8 @@ module Textbringer
         highlight_on = {}
         highlight_off = {}
 
-        visit_node(tree.root_node) do |node, start_byte, end_byte|
+        node_map = TreeSitter::NodeMaps.for(tree_sitter_language)
+        visit_node(tree.root_node, node_map) do |node, start_byte, end_byte|
           face = node_type_to_face(node.type.to_sym)
           next unless face
 
@@ -165,13 +166,16 @@ module Textbringer
         end
       end
 
-      def visit_node(node, &block)
+      def visit_node(node, node_map = nil, &block)
         if node.child_count == 0
           block.call(node, node.start_byte, node.end_byte)
         else
+          if node_map&.key?(node.type.to_sym)
+            block.call(node, node.start_byte, node.end_byte)
+          end
           node.child_count.times do |i|
             child = node.child(i)
-            visit_node(child, &block) if child
+            visit_node(child, node_map, &block) if child
           end
         end
       end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: textbringer-tree-sitter
 version: !ruby/object:Gem::Version
-  version: 1.2.2
+  version: 1.2.3
 platform: ruby
 authors:
 - yancya