adapt to feedback

Author: lorentzlasson

lua/nvim-tree/actions/tree/modifiers/collapse-all.lua renamed to lua/nvim-tree/actions/tree/modifiers/collapse.lua

@@ -23,12 +23,8 @@ local function buf_match()
   end
 end
 
----@param opts ApiTreeCollapseOpts|nil
-function M.fn(opts)
-  opts = opts or {}
-  local keep_buffers = opts.keep_buffers or false
-  local under_cursor = opts.under_cursor or false
-
+---@param opts ApiTreeCollapseOpts
+function M.collapse(opts)
   local explorer = core.get_explorer()
   if not explorer then
     return
@@ -42,7 +38,7 @@ function M.fn(opts)
   local matches = buf_match()
 
   local selected_nodes
-  if under_cursor then
+  if opts.under_cursor then
     if not node or not node.nodes then
       return
     end
@@ -56,7 +52,7 @@ function M.fn(opts)
     :applier(function(n)
       local dir = n:as(DirectoryNode)
       if dir then
-        dir.open = keep_buffers and matches(dir.absolute_path)
+        dir.open = opts.keep_buffers and matches(dir.absolute_path)
       end
     end)
     :recursor(function(n)
@@ -68,4 +64,9 @@ function M.fn(opts)
   utils.focus_node_or_parent(node)
 end
 
+---@param keep_buffers boolean
+function M.collapse_all(keep_buffers)
+  M.collapse({ keep_buffers = keep_buffers, under_cursor = false })
+end
+
 return M

lua/nvim-tree/actions/tree/modifiers/init.lua

@@ -1,6 +1,6 @@
 local M = {}
 
-M.collapse_all = require("nvim-tree.actions.tree.modifiers.collapse-all")
+M.collapse = require("nvim-tree.actions.tree.modifiers.collapse")
 M.expand_all = require("nvim-tree.actions.tree.modifiers.expand-all")
 
 function M.setup(opts)

lua/nvim-tree/api.lua

@@ -187,7 +187,8 @@ Api.tree.search_node = wrap(actions.finders.search_node.fn)
 ---@field under_cursor boolean
 ---@field keep_buffers boolean
 
-Api.tree.collapse_all = wrap(actions.tree.modifiers.collapse_all.fn)
+Api.tree.collapse = wrap(actions.tree.modifiers.collapse.collapse)
+Api.tree.collapse_all = wrap(actions.tree.modifiers.collapse.collapse_all)
 Api.tree.expand_all = wrap_node(actions.tree.modifiers.expand_all.fn)
 Api.tree.toggle_enable_filters = wrap_explorer_member("filters", "toggle")
 Api.tree.toggle_gitignore_filter = wrap_explorer_member_args("filters", "toggle", "git_ignored")

lua/nvim-tree/commands.lua

@@ -121,7 +121,7 @@ local CMDS = {
       bar = true,
     },
     command = function()
-      api.tree.collapse_all({ under_cursor = false, keep_buffers = false })
+      api.tree.collapse({ under_cursor = false, keep_buffers = false })
     end,
   },
   {
@@ -131,7 +131,7 @@ local CMDS = {
       bar = true,
     },
     command = function()
-      api.tree.collapse_all({ under_cursor = true, keep_buffers = false })
+      api.tree.collapse({ under_cursor = true, keep_buffers = false })
     end,
   },
   {