refactor(actions): move actions into semantic modules (#1410)

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

@@ -0,0 +1,39 @@
+local renderer = require "nvim-tree.renderer"
+local utils = require "nvim-tree.utils"
+local core = require "nvim-tree.core"
+local Iterator = require "nvim-tree.iterators.node-iterator"
+
+local M = {}
+
+function M.fn(keep_buffers)
+  if not core.get_explorer() then
+    return
+  end
+
+  local buffer_paths = vim.tbl_map(function(buffer)
+    return vim.api.nvim_buf_get_name(buffer)
+  end, vim.api.nvim_list_bufs())
+
+  Iterator.builder(core.get_explorer().nodes)
+    :hidden()
+    :applier(function(node)
+      node.open = false
+      if keep_buffers == true then
+        for _, buffer_path in ipairs(buffer_paths) do
+          local matches = utils.str_find(buffer_path, node.absolute_path)
+          if matches then
+            node.open = true
+            return
+          end
+        end
+      end
+    end)
+    :recursor(function(n)
+      return n.nodes
+    end)
+    :iterate()
+
+  renderer.draw()
+end
+
+return M

lua/nvim-tree/actions/tree-modifiers/expand-all.lua

@@ -0,0 +1,71 @@
+local core = require "nvim-tree.core"
+local renderer = require "nvim-tree.renderer"
+local utils = require "nvim-tree.utils"
+local Iterator = require "nvim-tree.iterators.node-iterator"
+
+local M = {}
+
+local function to_lookup_table(list)
+  local table = {}
+  for _, element in ipairs(list) do
+    table[element] = true
+  end
+
+  return table
+end
+
+local function expand(node)
+  node.open = true
+  if #node.nodes == 0 then
+    core.get_explorer():expand(node)
+  end
+end
+
+local function should_expand(expansion_count, node)
+  local should_halt = expansion_count >= M.MAX_FOLDER_DISCOVERY
+  local should_exclude = M.EXCLUDE[node.name]
+  return not should_halt and node.nodes and not node.open and not should_exclude
+end
+
+local function gen_iterator()
+  local expansion_count = 0
+
+  return function(parent)
+    if parent.parent and parent.nodes and not parent.open then
+      expansion_count = expansion_count + 1
+      expand(parent)
+    end
+
+    Iterator.builder(parent.nodes)
+      :hidden()
+      :applier(function(node)
+        if should_expand(expansion_count, node) then
+          expansion_count = expansion_count + 1
+          expand(node)
+        end
+      end)
+      :recursor(function(node)
+        return expansion_count < M.MAX_FOLDER_DISCOVERY and node.open and node.nodes
+      end)
+      :iterate()
+
+    if expansion_count >= M.MAX_FOLDER_DISCOVERY then
+      return true
+    end
+  end
+end
+
+function M.fn(base_node)
+  local node = base_node.nodes and base_node or core.get_explorer()
+  if gen_iterator()(node) then
+    utils.warn("expansion iteration was halted after " .. M.MAX_FOLDER_DISCOVERY .. " discovered folders")
+  end
+  renderer.draw()
+end
+
+function M.setup(opts)
+  M.MAX_FOLDER_DISCOVERY = opts.actions.expand_all.max_folder_discovery
+  M.EXCLUDE = to_lookup_table(opts.actions.expand_all.exclude)
+end
+
+return M

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

@@ -0,0 +1,28 @@
+local view = require "nvim-tree.view"
+local filters = require "nvim-tree.explorer.filters"
+local renderer = require "nvim-tree.renderer"
+local reloaders = require "nvim-tree.actions.reloaders.reloaders"
+
+local M = {}
+
+function M.custom()
+  filters.config.filter_custom = not filters.config.filter_custom
+  return reloaders.reload_explorer()
+end
+
+function M.git_ignored()
+  filters.config.filter_git_ignored = not filters.config.filter_git_ignored
+  return reloaders.reload_explorer()
+end
+
+function M.dotfiles()
+  filters.config.filter_dotfiles = not filters.config.filter_dotfiles
+  return reloaders.reload_explorer()
+end
+
+function M.help()
+  view.toggle_help()
+  renderer.draw()
+end
+
+return M