106 lines
3.2 KiB
Lua
106 lines
3.2 KiB
Lua
local diagnostics = require("nvim-tree.diagnostics")
|
|
|
|
local Decorator = require("nvim-tree.renderer.decorator")
|
|
local DirectoryNode = require("nvim-tree.node.directory")
|
|
|
|
-- 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 private explorer Explorer
|
|
---@field private diag_icons HighlightedString[]?
|
|
local DecoratorDiagnostics = Decorator:extend()
|
|
|
|
---@class DecoratorDiagnostics
|
|
---@overload fun(args: DecoratorArgs): DecoratorDiagnostics
|
|
|
|
---@protected
|
|
---@param args DecoratorArgs
|
|
function DecoratorDiagnostics:new(args)
|
|
self.explorer = args.explorer
|
|
|
|
self.enabled = true
|
|
self.highlight_range = self.explorer.opts.renderer.highlight_diagnostics or "none"
|
|
self.icon_placement = self.explorer.opts.renderer.icons.diagnostics_placement or "none"
|
|
|
|
if self.explorer.opts.renderer.icons.show.diagnostics then
|
|
self.diag_icons = {}
|
|
for name, sev in pairs(ICON_KEYS) do
|
|
self.diag_icons[sev] = {
|
|
str = self.explorer.opts.diagnostics.icons[name],
|
|
hl = { HG_ICON[sev] },
|
|
}
|
|
self:define_sign(self.diag_icons[sev])
|
|
end
|
|
end
|
|
end
|
|
|
|
---Diagnostic icon: diagnostics.enable, renderer.icons.show.diagnostics and node has status
|
|
---@param node Node
|
|
---@return HighlightedString[]? icons
|
|
function DecoratorDiagnostics:icons(node)
|
|
if node and self.diag_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.diag_icons[diag_value] }
|
|
end
|
|
end
|
|
end
|
|
|
|
---Diagnostic highlight: diagnostics.enable, renderer.highlight_diagnostics and node has status
|
|
---@param node Node
|
|
---@return string? highlight_group
|
|
function DecoratorDiagnostics:highlight_group(node)
|
|
if self.highlight_range == "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:is(DirectoryNode) 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
|