show_on_dirs and show_on_open_dirs

Author: chomosuke

lua/nvim-tree.lua

@@ -12,6 +12,7 @@ local copy_paste = require "nvim-tree.actions.fs.copy-paste"
 local collapse_all = require "nvim-tree.actions.tree-modifiers.collapse-all"
 local git = require "nvim-tree.git"
 local filters = require "nvim-tree.explorer.filters"
+local modified = require "nvim-tree.modified"
 
 local _config = {}
 
@@ -481,6 +482,7 @@ local function setup_autocommands(opts)
     create_nvim_tree_autocmd("BufModifiedSet", {
       callback = function()
         utils.debounce("BufModifiedSet:modified_files", opts.modified.debounce_delay, function()
+          modified.reload()
           reloaders.reload_explorer()
         end)
       end,
@@ -828,6 +830,7 @@ function M.setup(conf)
   require("nvim-tree.renderer").setup(opts)
   require("nvim-tree.live-filter").setup(opts)
   require("nvim-tree.marks").setup(opts)
+  require("nvim-tree.modified").setup(opts)
   if M.config.renderer.icons.show.file and pcall(require, "nvim-web-devicons") then
     require("nvim-web-devicons").setup()
   end

lua/nvim-tree/modified.lua

@@ -0,0 +1,34 @@
+local M = {}
+
+---@type table<string, boolean> record of which file is modified
+M.record = {}
+
+---refresh M.record
+function M.reload()
+  M.record = {}
+  local bufs = vim.fn.getbufinfo { bufmodified = true, buflisted = true }
+  for _, buf in pairs(bufs) do
+    local path = buf.name
+    M.record[path] = true
+    while path ~= vim.fn.getcwd() and path ~= "/" do
+      path = vim.fn.fnamemodify(path, ":h")
+      M.record[path] = true
+    end
+  end
+end
+
+---@param node table TODO: we should probably have a Node type
+---@return boolean
+function M.is_modified(node)
+  return M.config.enable
+    and M.record[node.absolute_path]
+    and (not node.nodes or M.config.show_on_dirs)
+    and (not node.open or M.config.show_on_open_dirs)
+end
+
+---@param opts table
+function M.setup(opts)
+  M.config = opts.modified
+end
+
+return M

lua/nvim-tree/renderer/builder.lua

@@ -1,5 +1,6 @@
 local utils = require "nvim-tree.utils"
 local core = require "nvim-tree.core"
+local modified = require "nvim-tree.modified"
 
 local git = require "nvim-tree.renderer.components.git"
 local pad = require "nvim-tree.renderer.components.padding"
@@ -68,10 +69,11 @@ function Builder:configure_git_icons_placement(where)
   return self
 end
 
-function Builder:configure_modified(modified_icon, modified_placement, modified)
-  self.modified = modified
-  self.modified.icon = modified_icon
-  self.modified.placement = modified_placement
+function Builder:configure_modified(modified_icon, modified_placement)
+  self.modified = {
+    icon = modified_icon,
+    placement = modified_placement,
+  }
   return self
 end
 
@@ -118,6 +120,9 @@ function Builder:_build_folder(node, padding, git_hl, git_icons_tbl)
   local offset = string.len(padding)
 
   local name = get_folder_name(node)
+  if modified.is_modified(node) then
+    name = name .. self.modified.icon
+  end
   local has_children = #node.nodes ~= 0 or node.has_children
   local icon = icons.get_folder_icon(node.open, node.link_to ~= nil, has_children)
 
@@ -213,7 +218,7 @@ function Builder:_highlight_opened_files(node, offset, icon_length, git_icons_le
   self:_insert_highlight("NvimTreeOpenedFile", from, to)
 end
 
-function Builder:_build_file(node, padding, git_highlight, git_icons_tbl, unloaded_bufnr, modified)
+function Builder:_build_file(node, padding, git_highlight, git_icons_tbl, unloaded_bufnr)
   local offset = string.len(padding)
 
   local icon = self:_build_file_icon(node, offset)
@@ -222,7 +227,7 @@ function Builder:_build_file(node, padding, git_highlight, git_icons_tbl, unload
   local git_icons = self:_unwrap_git_data(git_icons_tbl, git_icons_starts_at)
 
   local name = node.name
-  if modified then
+  if modified.is_modified(node) then
     name = name .. self.modified.icon
   end
   self:_insert_line(self:_format_line(padding .. icon, name, git_icons))
@@ -274,8 +279,6 @@ function Builder:_build_line(node, idx, num_children, unloaded_bufnr)
     end
   end
 
-  local modified = vim.fn.bufloaded(node.absolute_path) > 0 and vim.fn.getbufinfo(node.absolute_path)[1].changed == 1
-
   local is_folder = node.nodes ~= nil
   local is_symlink = node.link_to ~= nil
 
@@ -284,7 +287,7 @@ function Builder:_build_line(node, idx, num_children, unloaded_bufnr)
   elseif is_symlink then
     self:_build_symlink(node, padding, git_highlight, git_icons_tbl)
   else
-    self:_build_file(node, padding, git_highlight, git_icons_tbl, unloaded_bufnr, modified)
+    self:_build_file(node, padding, git_highlight, git_icons_tbl, unloaded_bufnr)
   end
   self.index = self.index + 1
 

lua/nvim-tree/renderer/init.lua

@@ -70,7 +70,7 @@ function M.draw(unloaded_bufnr)
       :configure_opened_file_highlighting(M.config.highlight_opened_files)
       :configure_git_icons_padding(M.config.icons.padding)
       :configure_git_icons_placement(M.config.icons.git_placement)
-      :configure_modified(M.config.icons.glyphs.modified, M.config.icons.modified_placement, M.config.modified)
+      :configure_modified(M.config.icons.glyphs.modified, M.config.icons.modified_placement)
       :configure_symlink_destination(M.config.symlink_destination)
       :configure_filter(live_filter.filter, live_filter.prefix)
       :build_header(view.is_root_folder_visible(core.get_cwd()))