5ad87620ec
* Reapply "refactor(#2871, #2886): multi instance: node classes created (#2916)"
This reverts commit 50e919426a.
* fix(#2945): stack overflow on api.git.reload or fugitive event
115 lines
3.4 KiB
Lua
115 lines
3.4 KiB
Lua
local diagnostics = require("nvim-tree.diagnostics")
|
|
|
|
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")
|
|
|
|
-- highlight groups by severity
|
|
local HG_ICON = {
|
|
[vim.diagnostic.severity.ERROR] = "NvimTreeDiagnosticErrorIcon",
|
|
[vim.diagnostic.severity.WARN] = "NvimTreeDiagnosticWarnIcon",
|
|
[vim.diagnostic.severity.INFO] = "NvimTreeDiagnosticInfoIcon",
|
|
[vim.diagnostic.severity.HINT] = "NvimTreeDiagnosticHintIcon",
|
|
}
|
|
local HG_FILE = {
|
|
[vim.diagnostic.severity.ERROR] = "NvimTreeDiagnosticErrorFileHL",
|
|
[vim.diagnostic.severity.WARN] = "NvimTreeDiagnosticWarnFileHL",
|
|
[vim.diagnostic.severity.INFO] = "NvimTreeDiagnosticInfoFileHL",
|
|
[vim.diagnostic.severity.HINT] = "NvimTreeDiagnosticHintFileHL",
|
|
}
|
|
local HG_FOLDER = {
|
|
[vim.diagnostic.severity.ERROR] = "NvimTreeDiagnosticErrorFolderHL",
|
|
[vim.diagnostic.severity.WARN] = "NvimTreeDiagnosticWarnFolderHL",
|
|
[vim.diagnostic.severity.INFO] = "NvimTreeDiagnosticInfoFolderHL",
|
|
[vim.diagnostic.severity.HINT] = "NvimTreeDiagnosticHintFolderHL",
|
|
}
|
|
-- opts.diagnostics.icons.
|
|
local ICON_KEYS = {
|
|
["error"] = vim.diagnostic.severity.ERROR,
|
|
["warning"] = vim.diagnostic.severity.WARN,
|
|
["info"] = vim.diagnostic.severity.INFO,
|
|
["hint"] = vim.diagnostic.severity.HINT,
|
|
}
|
|
|
|
---@class (exact) DecoratorDiagnostics: Decorator
|
|
---@field icons HighlightedString[]?
|
|
local DecoratorDiagnostics = Decorator:new()
|
|
|
|
---Static factory method
|
|
---@param opts table
|
|
---@param explorer Explorer
|
|
---@return DecoratorDiagnostics
|
|
function DecoratorDiagnostics:create(opts, explorer)
|
|
---@type DecoratorDiagnostics
|
|
local o = {
|
|
explorer = explorer,
|
|
enabled = opts.diagnostics.enable,
|
|
hl_pos = HL_POSITION[opts.renderer.highlight_diagnostics] or HL_POSITION.none,
|
|
icon_placement = ICON_PLACEMENT[opts.renderer.icons.diagnostics_placement] or ICON_PLACEMENT.none,
|
|
}
|
|
o = self:new(o) --[[@as DecoratorDiagnostics]]
|
|
|
|
if not o.enabled then
|
|
return o
|
|
end
|
|
|
|
if opts.renderer.icons.show.diagnostics then
|
|
o.icons = {}
|
|
for name, sev in pairs(ICON_KEYS) do
|
|
o.icons[sev] = {
|
|
str = opts.diagnostics.icons[name],
|
|
hl = { HG_ICON[sev] },
|
|
}
|
|
o:define_sign(o.icons[sev])
|
|
end
|
|
end
|
|
|
|
return o
|
|
end
|
|
|
|
---Diagnostic icon: diagnostics.enable, renderer.icons.show.diagnostics and node has status
|
|
---@param node Node
|
|
---@return HighlightedString[]|nil icons
|
|
function DecoratorDiagnostics:calculate_icons(node)
|
|
if node and self.enabled and self.icons then
|
|
local diag_status = diagnostics.get_diag_status(node)
|
|
local diag_value = diag_status and diag_status.value
|
|
|
|
if diag_value then
|
|
return { self.icons[diag_value] }
|
|
end
|
|
end
|
|
end
|
|
|
|
---Diagnostic highlight: diagnostics.enable, renderer.highlight_diagnostics and node has status
|
|
---@param node Node
|
|
---@return string|nil group
|
|
function DecoratorDiagnostics:calculate_highlight(node)
|
|
if not node or not self.enabled or self.hl_pos == HL_POSITION.none then
|
|
return nil
|
|
end
|
|
|
|
local diag_status = diagnostics.get_diag_status(node)
|
|
local diag_value = diag_status and diag_status.value
|
|
|
|
if not diag_value then
|
|
return nil
|
|
end
|
|
|
|
local group
|
|
if node.nodes then
|
|
group = HG_FOLDER[diag_value]
|
|
else
|
|
group = HG_FILE[diag_value]
|
|
end
|
|
|
|
if group then
|
|
return group
|
|
else
|
|
return nil
|
|
end
|
|
end
|
|
|
|
return DecoratorDiagnostics
|