67 lines
1.5 KiB
Lua
67 lines
1.5 KiB
Lua
local utils = require("nvim-tree.utils")
|
|
local core = require("nvim-tree.core")
|
|
local Iterator = require("nvim-tree.iterators.node-iterator")
|
|
|
|
local DirectoryNode = require("nvim-tree.node.directory")
|
|
|
|
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 node Node
|
|
---@param keep_buffers boolean
|
|
function M.fn(node, keep_buffers)
|
|
local explorer = core.get_explorer()
|
|
if not explorer then
|
|
return
|
|
end
|
|
|
|
local nodeAtCursor = explorer:get_node_at_cursor()
|
|
if not nodeAtCursor then
|
|
return
|
|
end
|
|
|
|
local matches = buf_match()
|
|
|
|
local nodesToIterate = explorer.nodes
|
|
if node then
|
|
local dir = node:as(DirectoryNode)
|
|
if dir then
|
|
nodesToIterate = { dir }
|
|
end
|
|
end
|
|
|
|
Iterator.builder(nodesToIterate)
|
|
:hidden()
|
|
:applier(function(n)
|
|
local dir = n:as(DirectoryNode)
|
|
if dir then
|
|
dir.open = keep_buffers and matches(dir.absolute_path)
|
|
end
|
|
end)
|
|
:recursor(function(n)
|
|
return n.group_next and { n.group_next } or n.nodes
|
|
end)
|
|
:iterate()
|
|
|
|
explorer.renderer:draw()
|
|
utils.focus_node_or_parent(nodeAtCursor)
|
|
end
|
|
|
|
return M
|