lualine.nvim/lua/lualine/components/diagnostics.lua

175 lines
5.9 KiB
Lua
Raw Normal View History

2021-02-22 00:59:35 +00:00
-- Copyright (c) 2020-2021 hoob3rt
-- MIT license, see LICENSE for more details.
local highlight = require('lualine.highlight')
local utils = require('lualine.utils.utils')
local Diagnostics = require('lualine.component'):new()
-- LuaFormatter off
Diagnostics.default_colors = {
error = '#e32636',
warn = '#ffdf00',
info = '#ffffff',
hint = '#d7afaf',
}
-- LuaFormatter on
2021-02-22 00:59:35 +00:00
-- Initializer
Diagnostics.new = function(self, options, child)
local new_diagnostics = self._parent:new(options, child or Diagnostics)
local default_symbols = new_diagnostics.options.icons_enabled and {
error = '', -- xf659
warn = '', -- xf529
info = '', -- xf7fc
hint = '' -- xf838
} or {error = 'E:', warn = 'W:', info = 'I:', hint = 'H:'}
new_diagnostics.symbols = vim.tbl_extend('force', default_symbols,
new_diagnostics.options.symbols or {})
if new_diagnostics.options.sources == nil then
print('no sources for diagnostics configured')
return ''
end
if new_diagnostics.options.sections == nil then
new_diagnostics.options.sections = {'error', 'warn', 'info', 'hint'}
end
if new_diagnostics.options.colored == nil then
new_diagnostics.options.colored = true
end
if new_diagnostics.options.update_in_insert == nil then
new_diagnostics.options.update_in_insert = false
end
new_diagnostics.last_update = ''
-- apply colors
if not new_diagnostics.options.color_error then
new_diagnostics.options.color_error =
utils.extract_highlight_colors('LspDiagnosticsDefaultError', 'fg') or
utils.extract_highlight_colors('DiffDelete', 'fg') or
Diagnostics.default_colors.error
end
if not new_diagnostics.options.color_warn then
new_diagnostics.options.color_warn =
utils.extract_highlight_colors('LspDiagnosticsDefaultWarning', 'fg') or
utils.extract_highlight_colors('DiffText', 'fg') or
Diagnostics.default_colors.warn
end
if not new_diagnostics.options.color_info then
new_diagnostics.options.color_info =
utils.extract_highlight_colors('LspDiagnosticsDefaultInformation', 'fg') or
utils.extract_highlight_colors('Normal', 'fg') or
Diagnostics.default_colors.info
end
if not new_diagnostics.options.color_hint then
new_diagnostics.options.color_hint =
utils.extract_highlight_colors('LspDiagnosticsDefaultHint', 'fg') or
utils.extract_highlight_colors('DiffChange', 'fg') or
Diagnostics.default_colors.hint
end
if new_diagnostics.options.colored then
new_diagnostics.highlight_groups = {
error = highlight.create_component_highlight_group(
{fg = new_diagnostics.options.color_error}, 'diagnostics_error',
new_diagnostics.options),
warn = highlight.create_component_highlight_group(
{fg = new_diagnostics.options.color_warn}, 'diagnostics_warn',
new_diagnostics.options),
info = highlight.create_component_highlight_group(
{fg = new_diagnostics.options.color_info}, 'diagnostics_info',
new_diagnostics.options),
hint = highlight.create_component_highlight_group(
{fg = new_diagnostics.options.color_hint}, 'diagnostics_hint',
new_diagnostics.options)
}
end
return new_diagnostics
end
Diagnostics.update_status = function(self)
if not self.options.update_in_insert
and vim.api.nvim_get_mode().mode:sub(1,1) == 'i' then
return self.last_update
end
local error_count, warning_count, info_count, hint_count = 0, 0, 0, 0
local diagnostic_data = self.get_diagnostics(self.options.sources)
for _, data in pairs(diagnostic_data) do
error_count = error_count + data.error
warning_count = warning_count + data.warn
info_count = info_count + data.info
hint_count = hint_count + data.hint
end
local result = {}
local data = {
error = error_count,
warn = warning_count,
info = info_count,
hint = hint_count
}
if self.options.colored then
local colors = {}
for name, hl in pairs(self.highlight_groups) do
colors[name] = highlight.component_format_highlight(hl)
end
for _, section in ipairs(self.options.sections) do
if data[section] ~= nil and data[section] > 0 then
table.insert(result,
colors[section] .. self.symbols[section] .. data[section])
end
end
else
for _, section in ipairs(self.options.sections) do
if data[section] ~= nil and data[section] > 0 then
table.insert(result, self.symbols[section] .. data[section])
end
end
end
self.last_update = ''
if result[1] ~= nil then
self.last_update = table.concat(result, ' ')
end
return self.last_update
end
Diagnostics.diagnostic_sources = {
2021-02-22 00:59:35 +00:00
nvim_lsp = function()
local error_count = vim.lsp.diagnostic.get_count(0, 'Error')
local warning_count = vim.lsp.diagnostic.get_count(0, 'Warning')
local info_count = vim.lsp.diagnostic.get_count(0, 'Information')
local hint_count = vim.lsp.diagnostic.get_count(0, 'Hint')
return error_count, warning_count, info_count, hint_count
2021-02-22 00:59:35 +00:00
end,
coc = function()
local data = vim.b.coc_diagnostic_info
if data then
return data.error, data.warning, data.information, data.hint
2021-02-22 00:59:35 +00:00
else
return 0, 0, 0, 0
2021-02-22 00:59:35 +00:00
end
end,
ale = function()
local ok, data = pcall(vim.fn['ale#statusline#Count'], vim.fn.bufnr())
if ok then
return data.error + data.style_error, data.warning + data.style_warning, data.info, 0
2021-02-22 00:59:35 +00:00
else
return 0, 0, 0, 0
2021-02-22 00:59:35 +00:00
end
end
2021-02-22 00:59:35 +00:00
}
Diagnostics.get_diagnostics = function(sources)
2021-02-22 00:59:35 +00:00
local result = {}
for index, source in ipairs(sources) do
local error_count, warning_count, info_count, hint_count =
Diagnostics.diagnostic_sources[source]()
result[index] = {
error = error_count,
warn = warning_count,
info = info_count,
hint = hint_count
}
2021-02-22 00:59:35 +00:00
end
return result
end
return Diagnostics