feat(#2415): create Decorator class for modified and bookmarks

Author: alex-courtis

lua/nvim-tree/enum.lua

@@ -12,9 +12,10 @@ M.HL_POSITION = {
 ---Setup options for "*_placement"
 ---@enum ICON_PLACEMENT
 M.ICON_PLACEMENT = {
-  signcolumn = 0,
-  before = 1,
-  after = 2,
+  none = 0,
+  signcolumn = 1,
+  before = 2,
+  after = 3,
 }
 
 return M

lua/nvim-tree/marks/init.lua

@@ -37,7 +37,7 @@ function M.clear_marks()
 end
 
 function M.get_mark(node)
-  return NvimTreeMarks[node.absolute_path]
+  return node and NvimTreeMarks[node.absolute_path]
 end
 
 function M.get_marks()

lua/nvim-tree/modified.lua

@@ -26,7 +26,8 @@ end
 ---@param node table
 ---@return boolean
 function M.is_modified(node)
-  return M.config.enable
+  return node
+    and 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)

lua/nvim-tree/renderer/builder.lua

@@ -4,18 +4,19 @@ local core = require "nvim-tree.core"
 local git = require "nvim-tree.renderer.components.git"
 local pad = require "nvim-tree.renderer.components.padding"
 local icons = require "nvim-tree.renderer.components.icons"
-local modified = require "nvim-tree.renderer.components.modified"
 local diagnostics = require "nvim-tree.renderer.components.diagnostics"
-local bookmarks = require "nvim-tree.renderer.components.bookmarks"
 
 local HL_POSITION = require("nvim-tree.enum").HL_POSITION
+local ICON_PLACEMENT = require("nvim-tree.enum").ICON_PLACEMENT
 
+--- @class Builder
+--- @field decorators Decorator[]
 local Builder = {}
 Builder.__index = Builder
 
 local DEFAULT_ROOT_FOLDER_LABEL = ":~:s?$?/..?"
 
-function Builder.new(root_cwd)
+function Builder.new(root_cwd, decorators)
   return setmetatable({
     index = 0,
     depth = 0,
@@ -24,6 +25,7 @@ function Builder.new(root_cwd)
     markers = {},
     signs = {},
     root_cwd = root_cwd,
+    decorators = decorators,
   }, Builder)
 end
 
@@ -79,22 +81,6 @@ function Builder:configure_diagnostics_icon_placement(where)
   return self
 end
 
-function Builder:configure_bookmark_icon_placement(where)
-  if where ~= "after" and where ~= "before" and where ~= "signcolumn" then
-    where = "before" -- default before
-  end
-  self.bookmarks_placement = where
-  return self
-end
-
-function Builder:configure_modified_placement(where)
-  if where ~= "after" and where ~= "before" and where ~= "signcolumn" then
-    where = "after" -- default after
-  end
-  self.modified_placement = where
-  return self
-end
-
 function Builder:configure_symlink_destination(show)
   self.symlink_destination = show
   return self
@@ -247,8 +233,8 @@ end
 ---@param node table
 ---@return HighlightedString|nil icon
 function Builder:_get_modified_icon(node)
-  local modified_icon = modified.get_icon(node)
-  if modified_icon and self.modified_placement == "signcolumn" then
+  local modified_icon = self.decorators.modified:get_icon(node)
+  if modified_icon and self.decorators.modified.icon_placement == ICON_PLACEMENT.signcolumn then
     table.insert(self.signs, {
       sign = modified_icon.hl[1],
       lnum = self.index + 1,
@@ -262,8 +248,8 @@ end
 ---@param node table
 ---@return HighlightedString[]|nil icon
 function Builder:_get_bookmark_icon(node)
-  local bookmark_icon = bookmarks.get_icon(node)
-  if bookmark_icon and self.bookmarks_placement == "signcolumn" then
+  local bookmark_icon = self.decorators.bookmark:get_icon(node)
+  if bookmark_icon and self.decorators.bookmark.icon_placement == ICON_PLACEMENT.signcolumn then
     table.insert(self.signs, {
       sign = bookmark_icon.hl[1],
       lnum = self.index + 1,
@@ -309,6 +295,23 @@ function Builder:_append_highlight(node, get_hl, icon_hl, name_hl)
   end
 end
 
+---Append optional highlighting to icon or name.
+---@param node table
+---@param decorator Decorator
+---@param icon_hl string[] icons to append to
+---@param name_hl string[] names to append to
+function Builder:_append_dec_highlight(node, decorator, icon_hl, name_hl)
+  local pos, hl = decorator:get_highlight(node)
+  if pos ~= HL_POSITION.none and hl then
+    if pos == HL_POSITION.all or pos == HL_POSITION.icon then
+      table.insert(icon_hl, hl)
+    end
+    if pos == HL_POSITION.all or pos == HL_POSITION.name then
+      table.insert(name_hl, hl)
+    end
+  end
+end
+
 ---@param indent_markers HighlightedString[]
 ---@param arrows HighlightedString[]|nil
 ---@param icon HighlightedString
@@ -341,13 +344,13 @@ function Builder:_format_line(indent_markers, arrows, icon, name, git_icons, dia
   if git_icons and self.git_placement == "before" then
     add_to_end(line, git_icons)
   end
-  if modified_icon and self.modified_placement == "before" then
+  if modified_icon and self.decorators.modified.icon_placement == ICON_PLACEMENT.before then
     add_to_end(line, { modified_icon })
   end
   if diagnostics_icon and self.diagnostics_placement == "before" then
     add_to_end(line, { diagnostics_icon })
   end
-  if bookmark_icon and self.bookmarks_placement == "before" then
+  if bookmark_icon and self.decorators.bookmark.icon_placement == ICON_PLACEMENT.before then
     add_to_end(line, { bookmark_icon })
   end
 
@@ -356,13 +359,13 @@ function Builder:_format_line(indent_markers, arrows, icon, name, git_icons, dia
   if git_icons and self.git_placement == "after" then
     add_to_end(line, git_icons)
   end
-  if modified_icon and self.modified_placement == "after" then
+  if modified_icon and self.decorators.modified.icon_placement == ICON_PLACEMENT.after then
     add_to_end(line, { modified_icon })
   end
   if diagnostics_icon and self.diagnostics_placement == "after" then
     add_to_end(line, { diagnostics_icon })
   end
-  if bookmark_icon and self.bookmarks_placement == "after" then
+  if bookmark_icon and self.decorators.bookmark.icon_placement == ICON_PLACEMENT.after then
     add_to_end(line, { bookmark_icon })
   end
 
@@ -405,9 +408,8 @@ function Builder:_build_line(node, idx, num_children, unloaded_bufnr)
 
   -- extra highighting
   self:_append_highlight(node, git.get_highlight, icon.hl, name.hl)
-  -- TODO opened
-  self:_append_highlight(node, modified.get_highlight, icon.hl, name.hl)
-  self:_append_highlight(node, bookmarks.get_highlight, icon.hl, name.hl)
+  self:_append_dec_highlight(node, self.decorators.modified, icon.hl, name.hl)
+  self:_append_dec_highlight(node, self.decorators.bookmark, icon.hl, name.hl)
   self:_append_highlight(node, diagnostics.get_highlight, icon.hl, name.hl)
   self:_append_highlight(node, copy_paste.get_highlight, icon.hl, name.hl)
 

lua/nvim-tree/renderer/components/bookmarks.lua

@@ -0,0 +1,53 @@
+local marks = require "nvim-tree.marks"
+
+local HL_POSITION = require("nvim-tree.enum").HL_POSITION
+local ICON_PLACEMENT = require("nvim-tree.enum").ICON_PLACEMENT
+
+local Decorator = require "nvim-tree.renderer.decorator"
+
+--- @class DecoratorBookmark: Decorator
+--- @field icon HighlightedString
+local DecoratorBookmark = Decorator:new()
+
+--- @param opts table
+--- @return DecoratorBookmark
+function DecoratorBookmark:new(opts)
+  local o = Decorator.new(self, {
+    hl_pos = HL_POSITION[opts.renderer.highlight_bookmarks] or HL_POSITION.none,
+    icon_placement = ICON_PLACEMENT[opts.renderer.icons.bookmarks_placement] or ICON_PLACEMENT.none,
+  })
+  ---@cast o DecoratorBookmark
+
+  if opts.renderer.icons.show.bookmarks then
+    o.icon = {
+      str = opts.renderer.icons.glyphs.bookmark,
+      hl = { "NvimTreeBookmark" },
+    }
+    o:define_sign(o.icon)
+  end
+
+  return o
+end
+
+--- Bookmark  icon: renderer.icons.show.bookmarks and node is marked
+function DecoratorBookmark:get_icon(node)
+  if marks.get_mark(node) then
+    return self.icon
+  end
+end
+
+--- Bookmark highlight: renderer.highlight_bookmarks and node is marked
+function DecoratorBookmark:get_highlight(node)
+  if self.hl_pos == HL_POSITION.none then
+    return HL_POSITION.none, nil
+  end
+
+  local mark = marks.get_mark(node)
+  if mark then
+    return self.hl_pos, "NvimTreeBookmarkHL"
+  else
+    return HL_POSITION.none, nil
+  end
+end
+
+return DecoratorBookmark

lua/nvim-tree/renderer/components/modified.lua

@@ -0,0 +1,42 @@
+--- @class Decorator
+--- @field hl_pos HL_POSITION
+--- @field icon_placement ICON_PLACEMENT
+local Decorator = {}
+
+--- @param o Decorator|nil
+--- @return Decorator
+function Decorator:new(o)
+  o = o or {}
+  setmetatable(o, self)
+  self.__index = self
+
+  return o
+end
+
+---@diagnostic disable: unused-local
+
+--- Node icon
+--- @param node table
+--- @return HighlightedString|nil modified icon
+function Decorator:get_icon(node) end
+
+--- Node highlight
+--- @param node table
+--- @return HL_POSITION|nil position
+--- @return string|nil group
+function Decorator:get_highlight(node) end
+
+---@diagnostic enable: unused-local
+
+--- Define a sign
+--- @param icon HighlightedString|nil
+function Decorator:define_sign(icon)
+  if icon and #icon.hl > 0 then
+    vim.fn.sign_define(icon.hl[1], {
+      text = icon.str,
+      texthl = icon.hl[1],
+    })
+  end
+end
+
+return Decorator