refactor: improve API readability and tidy actions submodules (#2593)
* refactor: improve API readability, tidy actions modules * Apply requested changes * `actions/reloaders/reloaders.lua` -> `actions/reloaders.lua` --------- Co-authored-by: Alexander Courtis <alex@courtis.org>
This commit is contained in:
53
lua/nvim-tree/actions/tree/modifiers/collapse-all.lua
Normal file
53
lua/nvim-tree/actions/tree/modifiers/collapse-all.lua
Normal file
@@ -0,0 +1,53 @@
|
||||
local renderer = require "nvim-tree.renderer"
|
||||
local utils = require "nvim-tree.utils"
|
||||
local core = require "nvim-tree.core"
|
||||
local lib = require "nvim-tree.lib"
|
||||
local Iterator = require "nvim-tree.iterators.node-iterator"
|
||||
|
||||
local M = {}
|
||||
|
||||
---@return fun(path: string): boolean
|
||||
local function buf_match()
|
||||
local buffer_paths = vim.tbl_map(function(buffer)
|
||||
return vim.api.nvim_buf_get_name(buffer)
|
||||
end, vim.api.nvim_list_bufs())
|
||||
|
||||
return function(path)
|
||||
for _, buffer_path in ipairs(buffer_paths) do
|
||||
local matches = utils.str_find(buffer_path, path)
|
||||
if matches then
|
||||
return true
|
||||
end
|
||||
end
|
||||
return false
|
||||
end
|
||||
end
|
||||
|
||||
---@param keep_buffers boolean
|
||||
function M.fn(keep_buffers)
|
||||
local node = lib.get_node_at_cursor()
|
||||
local explorer = core.get_explorer()
|
||||
|
||||
if explorer == nil then
|
||||
return
|
||||
end
|
||||
|
||||
local matches = buf_match()
|
||||
|
||||
Iterator.builder(explorer.nodes)
|
||||
:hidden()
|
||||
:applier(function(n)
|
||||
if n.nodes ~= nil then
|
||||
n.open = keep_buffers == true and matches(n.absolute_path)
|
||||
end
|
||||
end)
|
||||
:recursor(function(n)
|
||||
return n.group_next and { n.group_next } or n.nodes
|
||||
end)
|
||||
:iterate()
|
||||
|
||||
renderer.draw()
|
||||
utils.focus_node_or_parent(node)
|
||||
end
|
||||
|
||||
return M
|
||||
80
lua/nvim-tree/actions/tree/modifiers/expand-all.lua
Normal file
80
lua/nvim-tree/actions/tree/modifiers/expand-all.lua
Normal file
@@ -0,0 +1,80 @@
|
||||
local core = require "nvim-tree.core"
|
||||
local renderer = require "nvim-tree.renderer"
|
||||
local Iterator = require "nvim-tree.iterators.node-iterator"
|
||||
local notify = require "nvim-tree.notify"
|
||||
local lib = require "nvim-tree.lib"
|
||||
|
||||
local M = {}
|
||||
|
||||
---@param list string[]
|
||||
---@return table
|
||||
local function to_lookup_table(list)
|
||||
local table = {}
|
||||
for _, element in ipairs(list) do
|
||||
table[element] = true
|
||||
end
|
||||
|
||||
return table
|
||||
end
|
||||
|
||||
---@param node Node
|
||||
local function expand(node)
|
||||
node = lib.get_last_group_node(node)
|
||||
node.open = true
|
||||
if #node.nodes == 0 then
|
||||
core.get_explorer():expand(node)
|
||||
end
|
||||
end
|
||||
|
||||
---@param expansion_count integer
|
||||
---@param node Node
|
||||
---@return boolean
|
||||
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.group_next and { node.group_next } or (node.open and node.nodes))
|
||||
end)
|
||||
:iterate()
|
||||
|
||||
if expansion_count >= M.MAX_FOLDER_DISCOVERY then
|
||||
return true
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
---@param base_node table
|
||||
function M.fn(base_node)
|
||||
local node = base_node.nodes and base_node or core.get_explorer()
|
||||
if gen_iterator()(node) then
|
||||
notify.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
|
||||
11
lua/nvim-tree/actions/tree/modifiers/init.lua
Normal file
11
lua/nvim-tree/actions/tree/modifiers/init.lua
Normal file
@@ -0,0 +1,11 @@
|
||||
local M = {}
|
||||
|
||||
M.collapse_all = require "nvim-tree.actions.tree.modifiers.collapse-all"
|
||||
M.expand_all = require "nvim-tree.actions.tree.modifiers.expand-all"
|
||||
M.toggles = require "nvim-tree.actions.tree.modifiers.toggles"
|
||||
|
||||
function M.setup(opts)
|
||||
M.expand_all.setup(opts)
|
||||
end
|
||||
|
||||
return M
|
||||
44
lua/nvim-tree/actions/tree/modifiers/toggles.lua
Normal file
44
lua/nvim-tree/actions/tree/modifiers/toggles.lua
Normal file
@@ -0,0 +1,44 @@
|
||||
local lib = require "nvim-tree.lib"
|
||||
local utils = require "nvim-tree.utils"
|
||||
local filters = require "nvim-tree.explorer.filters"
|
||||
local reloaders = require "nvim-tree.actions.reloaders"
|
||||
|
||||
local M = {}
|
||||
|
||||
local function reload()
|
||||
local node = lib.get_node_at_cursor()
|
||||
reloaders.reload_explorer()
|
||||
utils.focus_node_or_parent(node)
|
||||
end
|
||||
|
||||
function M.custom()
|
||||
filters.config.filter_custom = not filters.config.filter_custom
|
||||
reload()
|
||||
end
|
||||
|
||||
function M.git_ignored()
|
||||
filters.config.filter_git_ignored = not filters.config.filter_git_ignored
|
||||
reload()
|
||||
end
|
||||
|
||||
function M.git_clean()
|
||||
filters.config.filter_git_clean = not filters.config.filter_git_clean
|
||||
reload()
|
||||
end
|
||||
|
||||
function M.no_buffer()
|
||||
filters.config.filter_no_buffer = not filters.config.filter_no_buffer
|
||||
reload()
|
||||
end
|
||||
|
||||
function M.no_bookmark()
|
||||
filters.config.filter_no_bookmark = not filters.config.filter_no_bookmark
|
||||
reload()
|
||||
end
|
||||
|
||||
function M.dotfiles()
|
||||
filters.config.filter_dotfiles = not filters.config.filter_dotfiles
|
||||
reload()
|
||||
end
|
||||
|
||||
return M
|
||||
Reference in New Issue
Block a user