fefa335f1c
* feat(#1826): allow using config from vim.diagnostic for icons + severity * update help default options, add index --------- Co-authored-by: Alexander Courtis <alex@courtis.org>
118 lines
3.6 KiB
Lua
118 lines
3.6 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) DiagnosticsDecorator: Decorator
|
|
---@field private explorer Explorer
|
|
---@field private diag_icons HighlightedString[]?
|
|
local DiagnosticsDecorator = Decorator:extend()
|
|
|
|
---@class DiagnosticsDecorator
|
|
---@overload fun(args: DecoratorArgs): DiagnosticsDecorator
|
|
|
|
---@protected
|
|
---@param args DecoratorArgs
|
|
function DiagnosticsDecorator: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"
|
|
|
|
local vim_diagnostic_icons = {}
|
|
|
|
if self.explorer.opts.diagnostics.diagnostic_opts then
|
|
local vim_diagnostic_config = vim.diagnostic.config() or {}
|
|
local signs = vim_diagnostic_config.signs or {}
|
|
if type(signs) == "function" then
|
|
signs = signs(0, 0)
|
|
end
|
|
|
|
vim_diagnostic_icons = (type(signs) == "table" and signs.text) or {}
|
|
end
|
|
|
|
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 = vim_diagnostic_icons[sev] or 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 DiagnosticsDecorator: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 DiagnosticsDecorator: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 DiagnosticsDecorator
|