show some icon on opendir even if show_on_open_dir=false

and show all children's status on parent
This commit is contained in:
chomosuke
2022-12-06 21:09:35 +11:00
parent cd1df4f9bf
commit f65dc0d4e3
8 changed files with 141 additions and 51 deletions

View File

@@ -15,7 +15,7 @@ function M.fn(where, what)
for line, node in pairs(nodes_by_line) do for line, node in pairs(nodes_by_line) do
local valid = false local valid = false
if what == "git" then if what == "git" then
valid = explorer_common.shows_git_status(node) valid = explorer_common.get_git_status(node) ~= nil
elseif what == "diag" then elseif what == "diag" then
valid = node.diag_status ~= nil valid = node.diag_status ~= nil
end end

View File

@@ -20,6 +20,8 @@ end
function M.reload_node_status(parent_node, projects) function M.reload_node_status(parent_node, projects)
local project_root = git.get_project_root(parent_node.absolute_path) local project_root = git.get_project_root(parent_node.absolute_path)
local status = projects[project_root] or {} local status = projects[project_root] or {}
require("nvim-tree.log").line("dev", "reloaders")
-- TODO: when is this called
for _, node in ipairs(parent_node.nodes) do for _, node in ipairs(parent_node.nodes) do
if node.nodes then if node.nodes then
node.git_status = status.dirs and status.dirs[node.absolute_path] node.git_status = status.dirs and status.dirs[node.absolute_path]

View File

@@ -1,20 +1,31 @@
local M = {} 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) local function get_dir_git_status(parent_ignored, status, absolute_path)
if parent_ignored then if parent_ignored then
return "!!" return { file = "!!" }
end end
local file_status = status.files and status.files[absolute_path] return {
if file_status then file = status.files and status.files[absolute_path],
return file_status dir = {
end direct = status.dirs.direct[absolute_path],
indirect = status.dirs.indirect[absolute_path],
return status.dirs and status.dirs[absolute_path] },
}
end end
local function get_git_status(parent_ignored, status, absolute_path) local function get_git_status(parent_ignored, status, absolute_path)
return parent_ignored and "!!" or status.files and status.files[absolute_path] local file_status = parent_ignored and "!!" or status.files and status.files[absolute_path]
return { file = file_status }
end end
function M.has_one_child_folder(node) function M.has_one_child_folder(node)
@@ -38,20 +49,71 @@ function M.update_git_status(node, parent_ignored, status)
end end
end end
function M.shows_git_status(node) function M.get_git_status(node)
if not node.git_status then local git_status = node.git_status
if not git_status then
-- status doesn't exist -- status doesn't exist
return false return nil
elseif not node.nodes then
-- status exist and is a file
return true
elseif not node.open then
-- status exist, is a closed dir
return M.config.git.show_on_dirs
else
-- status exist, is a open dir
return M.config.git.show_on_dirs and M.config.git.show_on_open_dirs
end 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,
-- TODO: test if this should be deleted
["DU"] = 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 end
function M.node_destroy(node) function M.node_destroy(node)

View File

@@ -13,7 +13,7 @@ local function get_type_from(type_, cwd)
end end
local function populate_children(handle, cwd, node, git_status) local function populate_children(handle, cwd, node, git_status)
local node_ignored = node.git_status == "!!" local node_ignored = common.is_git_ignored(node)
local nodes_by_path = utils.bool_record(node.nodes, "absolute_path") local nodes_by_path = utils.bool_record(node.nodes, "absolute_path")
local filter_status = filters.prepare(git_status) local filter_status = filters.prepare(git_status)
while true do while true do

View File

@@ -53,7 +53,7 @@ function M.reload(node, git_status, unloaded_bufnr)
local child_names = {} local child_names = {}
local node_ignored = node.git_status == "!!" local node_ignored = common.is_git_ignored(node)
local nodes_by_path = utils.key_by(node.nodes, "absolute_path") local nodes_by_path = utils.key_by(node.nodes, "absolute_path")
while true do while true do
local ok, name, t = pcall(vim.loop.fs_scandir_next, handle) local ok, name, t = pcall(vim.loop.fs_scandir_next, handle)

View File

@@ -4,6 +4,7 @@ local git_utils = require "nvim-tree.git.utils"
local Runner = require "nvim-tree.git.runner" local Runner = require "nvim-tree.git.runner"
local Watcher = require("nvim-tree.watcher").Watcher local Watcher = require("nvim-tree.watcher").Watcher
local Iterator = require "nvim-tree.iterators.node-iterator" local Iterator = require "nvim-tree.iterators.node-iterator"
local explorer_common = require "nvim-tree.explorer.common"
local M = { local M = {
config = {}, config = {},
@@ -103,19 +104,13 @@ local function reload_tree_at(project_root)
end end
M.reload_project(project_root) M.reload_project(project_root)
local project = M.get_project(project_root) local git_status = M.get_project(project_root)
local project_files = project.files and project.files or {}
local project_dirs = project.dirs and project.dirs or {}
Iterator.builder(root_node.nodes) Iterator.builder(root_node.nodes)
:hidden() :hidden()
:applier(function(node) :applier(function(node)
local parent_ignored = node.parent.git_status == "!!" local parent_ignored = explorer_common.is_git_ignored(node.parent)
node.git_status = project_dirs[node.absolute_path] or project_files[node.absolute_path] explorer_common.update_git_status(node, parent_ignored, git_status)
if not node.git_status and parent_ignored then
node.git_status = "!!"
end
end) end)
:recursor(function(node) :recursor(function(node)
return node.nodes and #node.nodes > 0 and node.nodes return node.nodes and #node.nodes > 0 and node.nodes

View File

@@ -55,24 +55,43 @@ function M.should_show_untracked(cwd)
return untracked[cwd] return untracked[cwd]
end end
local function nil_insert(t, k)
t = t or {}
t[k] = true
return t
end
function M.file_status_to_dir_status(status, cwd) function M.file_status_to_dir_status(status, cwd)
local dirs = {} local direct = {}
for p, s in pairs(status) do for p, s in pairs(status) do
if s ~= "!!" then if s ~= "!!" then
local modified = vim.fn.fnamemodify(p, ":h") local modified = vim.fn.fnamemodify(p, ":h")
dirs[modified] = s direct[modified] = nil_insert(direct[modified], s)
end end
end end
for dirname, s in pairs(dirs) do local indirect = {}
local modified = dirname for dirname, statuses in pairs(direct) do
while modified ~= cwd and modified ~= "/" do for s, _ in pairs(statuses) do
modified = vim.fn.fnamemodify(modified, ":h") local modified = dirname
dirs[modified] = s while modified ~= cwd and modified ~= "/" do
modified = vim.fn.fnamemodify(modified, ":h")
indirect[modified] = nil_insert(indirect[modified], s)
end
end end
end end
return dirs local r = { indirect = indirect, direct = direct }
for _, d in pairs(r) do
for dirname, statuses in pairs(d) do
local new_statuses = {}
for s, _ in pairs(statuses) do
table.insert(new_statuses, s)
end
d[dirname] = new_statuses
end
end
return r
end end
return M return M

View File

@@ -59,20 +59,32 @@ local function warn_status(git_status)
end end
local function get_icons_(node) local function get_icons_(node)
if not explorer_common.shows_git_status(node) then local git_statuses = explorer_common.get_git_status(node)
if git_statuses == nil then
return nil return nil
end end
local git_status = node.git_status local inserted = {}
local icons = M.git_icons[git_status] local iconss = {}
if not icons then
if not M.config.highlight_git then for _, git_status in pairs(git_statuses) do
warn_status(git_status) local icons = M.git_icons[git_status]
if not icons then
if not M.config.highlight_git then
warn_status(git_status)
end
return nil
end
for _, icon in pairs(icons) do
if not inserted[icon] then
table.insert(iconss, icon)
inserted[icon] = true
end
end end
return nil
end end
return icons return iconss
end end
local git_hl = { local git_hl = {
@@ -119,12 +131,12 @@ function M.setup_signs(i)
end end
local function get_highlight_(node) local function get_highlight_(node)
local git_status = node.git_status local git_status = explorer_common.get_git_status(node)
if not explorer_common.shows_git_status(node) then if git_status == nil then
return return
end end
return git_hl[git_status] return git_hl[git_status[1]]
end end
function M.setup(opts) function M.setup(opts)