135 lines
3.1 KiB
Lua
135 lines
3.1 KiB
Lua
local M = {}
|
|
|
|
-- node.git_status structure:
|
|
-- {
|
|
-- file = string | nil,
|
|
-- dir = {
|
|
-- direct = { string } | nil,
|
|
-- indirect = { string } | nil,
|
|
-- } | nil,
|
|
-- }
|
|
|
|
local function get_dir_git_status(parent_ignored, status, absolute_path)
|
|
if parent_ignored then
|
|
return { file = "!!" }
|
|
end
|
|
|
|
return {
|
|
file = status.files and status.files[absolute_path],
|
|
dir = status.dirs and {
|
|
direct = status.dirs.direct[absolute_path],
|
|
indirect = status.dirs.indirect[absolute_path],
|
|
},
|
|
}
|
|
end
|
|
|
|
local function get_git_status(parent_ignored, status, absolute_path)
|
|
local file_status = parent_ignored and "!!" or status.files and status.files[absolute_path]
|
|
return { file = file_status }
|
|
end
|
|
|
|
function M.has_one_child_folder(node)
|
|
return #node.nodes == 1 and node.nodes[1].nodes and vim.loop.fs_access(node.nodes[1].absolute_path, "R")
|
|
end
|
|
|
|
function M.update_git_status(node, parent_ignored, status)
|
|
local get_status
|
|
if node.nodes then
|
|
get_status = get_dir_git_status
|
|
else
|
|
get_status = get_git_status
|
|
end
|
|
|
|
-- status of the node's absolute path
|
|
node.git_status = get_status(parent_ignored, status, node.absolute_path)
|
|
|
|
-- status of the link target, if the link itself is not dirty
|
|
if node.link_to and not node.git_status then
|
|
node.git_status = get_status(parent_ignored, status, node.link_to)
|
|
end
|
|
end
|
|
|
|
function M.get_git_status(node)
|
|
local git_status = node.git_status
|
|
if not git_status then
|
|
-- status doesn't exist
|
|
return nil
|
|
end
|
|
|
|
if not node.nodes then
|
|
-- file
|
|
return git_status.file and { git_status.file }
|
|
end
|
|
|
|
-- dir
|
|
if not M.config.git.show_on_dirs then
|
|
return nil
|
|
end
|
|
|
|
local status = {}
|
|
if not node.open or M.config.git.show_on_open_dirs then
|
|
-- dir is closed or we should show on open_dirs
|
|
if git_status.file ~= nil then
|
|
table.insert(status, git_status.file)
|
|
end
|
|
if git_status.dir ~= nil then
|
|
if git_status.dir.direct ~= nil then
|
|
for _, s in pairs(node.git_status.dir.direct) do
|
|
table.insert(status, s)
|
|
end
|
|
end
|
|
if git_status.dir.indirect ~= nil then
|
|
for _, s in pairs(node.git_status.dir.indirect) do
|
|
table.insert(status, s)
|
|
end
|
|
end
|
|
end
|
|
else
|
|
-- dir is open and we shouldn't show on open_dirs
|
|
if git_status.file ~= nil then
|
|
table.insert(status, git_status.file)
|
|
end
|
|
if git_status.dir ~= nil and git_status.dir.direct ~= nil then
|
|
local deleted = {
|
|
[" D"] = true,
|
|
["D "] = true,
|
|
["RD"] = true,
|
|
["DD"] = true,
|
|
}
|
|
for _, s in pairs(node.git_status.dir.direct) do
|
|
if deleted[s] then
|
|
table.insert(status, s)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
if #status == 0 then
|
|
return nil
|
|
else
|
|
return status
|
|
end
|
|
end
|
|
|
|
function M.is_git_ignored(node)
|
|
return node.git_status and node.git_status.file == "!!"
|
|
end
|
|
|
|
function M.node_destroy(node)
|
|
if not node then
|
|
return
|
|
end
|
|
|
|
if node.watcher then
|
|
node.watcher:destroy()
|
|
node.watcher = nil
|
|
end
|
|
end
|
|
|
|
function M.setup(opts)
|
|
M.config = {
|
|
git = opts.git,
|
|
}
|
|
end
|
|
|
|
return M
|