e962e97cab
* refactor(#2830): multi instance marks * refactor(#2830): multi instance marks * refactor(#2830): multi instance marks
52 lines
1.5 KiB
Lua
52 lines
1.5 KiB
Lua
local core = require "nvim-tree.core"
|
|
|
|
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 DecoratorBookmarks: Decorator
|
|
---@field icon HighlightedString
|
|
local DecoratorBookmarks = Decorator:new()
|
|
|
|
---@param opts table
|
|
---@return DecoratorBookmarks
|
|
function DecoratorBookmarks:new(opts)
|
|
local o = Decorator.new(self, {
|
|
enabled = true,
|
|
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 DecoratorBookmarks
|
|
|
|
if opts.renderer.icons.show.bookmarks then
|
|
o.icon = {
|
|
str = opts.renderer.icons.glyphs.bookmark,
|
|
hl = { "NvimTreeBookmarkIcon" },
|
|
}
|
|
o:define_sign(o.icon)
|
|
end
|
|
|
|
return o
|
|
end
|
|
|
|
---Bookmark icon: renderer.icons.show.bookmarks and node is marked
|
|
---@param node Node
|
|
---@return HighlightedString[]|nil icons
|
|
function DecoratorBookmarks:calculate_icons(node)
|
|
if core.get_explorer() and core.get_explorer().marks:get(node) then
|
|
return { self.icon }
|
|
end
|
|
end
|
|
|
|
---Bookmark highlight: renderer.highlight_bookmarks and node is marked
|
|
---@param node Node
|
|
---@return string|nil group
|
|
function DecoratorBookmarks:calculate_highlight(node)
|
|
if self.hl_pos ~= HL_POSITION.none and core.get_explorer() and core.get_explorer().marks:get(node) then
|
|
return "NvimTreeBookmarkHL"
|
|
end
|
|
end
|
|
|
|
return DecoratorBookmarks
|