feat: add option to set arrow color to diagnostic severity. Fixes #104

Author: rachartier

README.md

@@ -80,6 +80,9 @@ require("tiny-inline-diagnostic").setup({
         -- Use icons defined in the diagnostic configuration
         use_icons_from_diagnostic = false,
 
+        -- Set the arrow icon to the same color as the first diagnostic severity
+        set_arrow_to_diag_color = false,
+
         -- Add messages to diagnostics when multiline diagnostics are enabled
         -- If set to false, only signs will be displayed
         add_messages = true,

lua/tiny-inline-diagnostic/chunk.lua

@@ -181,16 +181,25 @@ end
 ---@param opts table: The options table.
 ---@param diagnostic_line number: The line number of the diagnostic message.
 ---@param ret table: The return table containing diagnostic information.
+---@param hl_diag_hi string: The highlight group for the diagnostic message.
 ---@return table: A table representing the virtual text array for the arrow.
-function M.get_arrow_from_chunk(opts, diagnostic_line, ret)
+function M.get_arrow_from_chunk(opts, diagnostic_line, ret, hl_diag_hi)
 	local arrow = opts.signs.arrow
 	local need_to_be_under = ret.need_to_be_under
 
 	local chunk = {}
+
 	local hi = "TinyInlineDiagnosticVirtualTextArrow"
+	if opts.options.set_arrow_to_diag_color then
+		hi = hl_diag_hi .. "CursorLine"
+	end
 
 	if diagnostic_line ~= ret.line or ret.need_to_be_under then
 		hi = "TinyInlineDiagnosticVirtualTextArrowNoBg"
+
+		if opts.options.set_arrow_to_diag_color then
+			hi = hl_diag_hi .. "NoBg"
+		end
 	end
 
 	if need_to_be_under then

lua/tiny-inline-diagnostic/highlights.lua

@@ -96,6 +96,12 @@ local function create_highlight_groups(colors, blends, transparent)
 			italic = colors[string.lower(name)].italic,
 		}
 
+		hi[HIGHLIGHT_PREFIX .. name .. "NoBg"] = {
+			fg = colors[string.lower(name)].fg,
+			bg = "None",
+			italic = colors[string.lower(name)].italic,
+		}
+
 		-- Inverse highlights with and without background
 		hi[INV_HIGHLIGHT_PREFIX .. name] = {
 			fg = blends[string.lower(name)],

lua/tiny-inline-diagnostic/init.lua

@@ -25,6 +25,7 @@ local default_config = {
 	options = {
 		show_source = false,
 		add_messages = true,
+		set_arrow_to_diag_color = true,
 		use_icons_from_diagnostic = false,
 		throttle = 20,
 		softwrap = 30,

lua/tiny-inline-diagnostic/virtual_text.lua

@@ -49,7 +49,7 @@ local function build_first_chunk(opts, chunk_info, message, hl, index_diag, tota
 
 	if index_diag == 1 then
 		local cursor_line = vim.api.nvim_win_get_cursor(0)[1] - 1
-		local chunk_arrow = chunk_utils.get_arrow_from_chunk(opts, cursor_line, chunk_info)
+		local chunk_arrow = chunk_utils.get_arrow_from_chunk(opts, cursor_line, chunk_info, hl.diag_hi)
 
 		if type(chunk_arrow[1]) == "table" then
 			return { chunk_arrow, chunk_header }