45 lines
1.0 KiB
Lua
45 lines
1.0 KiB
Lua
local view = require "nvim-tree.view"
|
|
local utils = require "nvim-tree.utils"
|
|
local core = require "nvim-tree.core"
|
|
local lib = require "nvim-tree.lib"
|
|
|
|
local M = {}
|
|
|
|
---@param should_close boolean|nil
|
|
---@return fun(node: Node): boolean|nil
|
|
function M.fn(should_close)
|
|
should_close = should_close or false
|
|
|
|
return function(node)
|
|
local explorer = core.get_explorer()
|
|
node = lib.get_last_group_node(node)
|
|
if should_close and node.open then
|
|
node.open = false
|
|
if explorer then
|
|
explorer.renderer:draw()
|
|
end
|
|
return
|
|
end
|
|
|
|
local parent = utils.get_parent_of_group(node).parent
|
|
|
|
if not parent or not parent.parent then
|
|
return view.set_cursor { 1, 0 }
|
|
end
|
|
|
|
local _, line = utils.find_node(core.get_explorer().nodes, function(n)
|
|
return n.absolute_path == parent.absolute_path
|
|
end)
|
|
|
|
view.set_cursor { line + 1, 0 }
|
|
if should_close then
|
|
parent.open = false
|
|
if explorer then
|
|
explorer.renderer:draw()
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
return M
|