fix(treesitter): remove more double recursion

Do not call `for_each_child` in functions that are already recursive.

(cherry picked from commit e76ae3d232)

runtime/lua/vim/treesitter/languagetree.lua

@@ -800,6 +800,20 @@ function LanguageTree:_edit(
     end
     return true
   end)
+
+  for _, child in pairs(self._children) do
+    child:_edit(
+      start_byte,
+      end_byte_old,
+      end_byte_new,
+      start_row,
+      start_col,
+      end_row_old,
+      end_col_old,
+      end_row_new,
+      end_col_new
+    )
+  end
 end
 
 ---@package
@@ -846,9 +860,7 @@ function LanguageTree:_on_bytes(
   )
 
   -- Edit trees together BEFORE emitting a bytes callback.
-  ---@private
+  self:_edit(
-  self:for_each_child(function(child)
-    child:_edit(
     start_byte,
     start_byte + old_byte,
     start_byte + new_byte,
@@ -859,7 +871,6 @@ function LanguageTree:_on_bytes(
     start_row + new_row,
     new_end_col
   )
-  end, true)
 
   self:_do_callback(
     'bytes',
@@ -913,9 +924,9 @@ function LanguageTree:register_cbs(cbs, recursive)
   end
 
   if recursive then
-    self:for_each_child(function(child)
+    for _, child in pairs(self._children) do
       child:register_cbs(cbs, true)
-    end)
+    end
   end
 end
 

runtime/lua/vim/treesitter/playground.lua

@@ -102,18 +102,16 @@ function TSPlayground:new(bufnr, lang)
   -- the root in the child tree to the {injections} table.
   local root = parser:parse()[1]:root()
   local injections = {} ---@type table<integer,table>
-  parser:for_each_child(function(child, lang_)
+  parser:for_each_tree(function(tree, ltree)
-    child:for_each_tree(function(tree)
     local r = tree:root()
     local node = root:named_descendant_for_range(r:range())
     if node then
       injections[node:id()] = {
-          lang = lang_,
+        lang = ltree:lang(),
         root = r,
       }
     end
   end)
-  end)
 
   local nodes = traverse(root, 0, parser:lang(), injections, {})