87 lines
2.0 KiB
Lua
87 lines
2.0 KiB
Lua
local git_utils = require'nvim-tree.git.utils'
|
|
local Runner = require'nvim-tree.git.runner'
|
|
|
|
local M = {
|
|
config = nil,
|
|
projects = {},
|
|
cwd_to_project_root = {}
|
|
}
|
|
|
|
function M.reload(callback)
|
|
local num_projects = vim.tbl_count(M.projects)
|
|
if not M.config.enable or num_projects == 0 then
|
|
return callback({})
|
|
end
|
|
|
|
local done = 0
|
|
for project_root in pairs(M.projects) do
|
|
M.projects[project_root] = {}
|
|
Runner.run {
|
|
project_root = project_root,
|
|
list_untracked = git_utils.should_show_untracked(project_root),
|
|
list_ignored = true,
|
|
timeout = M.config.timeout,
|
|
on_end = function(git_status)
|
|
M.projects[project_root] = {
|
|
files = git_status,
|
|
dirs = git_utils.file_status_to_dir_status(git_status, project_root)
|
|
}
|
|
done = done + 1
|
|
if done == num_projects then
|
|
callback(M.projects)
|
|
end
|
|
end
|
|
}
|
|
end
|
|
end
|
|
|
|
function M.get_project_root(cwd)
|
|
if M.cwd_to_project_root[cwd] then
|
|
return M.cwd_to_project_root[cwd]
|
|
end
|
|
|
|
if M.cwd_to_project_root[cwd] == false then
|
|
return nil
|
|
end
|
|
|
|
local project_root = git_utils.get_toplevel(cwd)
|
|
return project_root
|
|
end
|
|
|
|
function M.load_project_status(cwd, callback)
|
|
if not M.config.enable then
|
|
return callback({})
|
|
end
|
|
|
|
local project_root = M.get_project_root(cwd)
|
|
if not project_root then
|
|
M.cwd_to_project_root[cwd] = false
|
|
return callback({})
|
|
end
|
|
|
|
local status = M.projects[project_root]
|
|
if status then
|
|
return callback(status)
|
|
end
|
|
|
|
Runner.run {
|
|
project_root = project_root,
|
|
list_untracked = git_utils.should_show_untracked(project_root),
|
|
list_ignored = true,
|
|
timeout = M.config.timeout,
|
|
on_end = function(git_status)
|
|
M.projects[project_root] = {
|
|
files = git_status,
|
|
dirs = git_utils.file_status_to_dir_status(git_status, project_root)
|
|
}
|
|
callback(M.projects[project_root])
|
|
end
|
|
}
|
|
end
|
|
|
|
function M.setup(opts)
|
|
M.config = opts.git
|
|
end
|
|
|
|
return M
|