feat(filter): add show_diags_only_under_cursor option. Fixes #156

Author: rachartier

.beads/issues.jsonl

@@ -198,6 +198,9 @@ require("tiny-inline-diagnostic").setup({
         -- Show all diagnostics on the current cursor line, not just those under the cursor
         show_all_diags_on_cursorline = false,
 
+        -- Only show diagnostics when the cursor is directly over them, no fallback to line diagnostics
+        show_diags_only_under_cursor = false,
+
         -- Display related diagnostics from LSP relatedInformation
         show_related = {
             enabled = true,           -- Enable displaying related diagnostics

README.md

@@ -37,7 +37,11 @@ function M.at_position(opts, diagnostics, line, col)
     return diag.lnum == line and col >= diag.col and col <= diag.end_col
   end, diagnostics)
 
-  return #current_pos_diags > 0 and current_pos_diags or diags_on_line
+  if opts.options.show_diags_only_under_cursor then
+    return current_pos_diags
+  else
+    return #current_pos_diags > 0 and current_pos_diags or diags_on_line
+  end
 end
 
 ---@param related_info table
@@ -82,6 +86,28 @@ local function extract_related_diagnostics(diag, max_count)
   return related
 end
 
+---@param opts table
+---@param diagnostics table
+---@return table
+local function add_related_diagnostics(opts, diagnostics)
+  if not opts.options.show_related or not opts.options.show_related.enabled then
+    return diagnostics
+  end
+
+  local result = {}
+  local max_count = opts.options.show_related.max_count or 3
+
+  for _, diag in ipairs(diagnostics) do
+    table.insert(result, diag)
+    if has_related_info(diag) then
+      local related = extract_related_diagnostics(diag, max_count)
+      vim.list_extend(result, related)
+    end
+  end
+
+  return result
+end
+
 ---@param opts table
 ---@param buf number
 ---@param diagnostics table
@@ -101,22 +127,7 @@ function M.under_cursor(opts, buf, diagnostics)
 
   filtered_diags = M.by_severity(opts, filtered_diags)
 
-  if not opts.options.show_related or not opts.options.show_related.enabled then
-    return filtered_diags
-  end
-
-  local result = {}
-  local max_count = opts.options.show_related.max_count or 3
-
-  for _, diag in ipairs(filtered_diags) do
-    table.insert(result, diag)
-    if has_related_info(diag) then
-      local related = extract_related_diagnostics(diag, max_count)
-      vim.list_extend(result, related)
-    end
-  end
-
-  return result
+  return add_related_diagnostics(opts, filtered_diags)
 end
 
 ---@param opts table
@@ -128,6 +139,22 @@ function M.for_display(opts, bufnr, diagnostics)
     return M.under_cursor(opts, bufnr, diagnostics)
   end
 
+  if opts.options.show_diags_only_under_cursor then
+    local cursor_pos = vim.api.nvim_win_get_cursor(0)
+    local current_line = cursor_pos[1] - 1
+
+    local under_cursor_on_line = M.at_position(opts, diagnostics, current_line, cursor_pos[2])
+
+    local other_diags = vim.tbl_filter(function(diag)
+      return diag.lnum ~= current_line
+    end, diagnostics)
+
+    local result = vim.list_extend({}, under_cursor_on_line)
+    vim.list_extend(result, other_diags)
+    result = M.by_severity(opts, result)
+    return add_related_diagnostics(opts, result)
+  end
+
   if opts.options.multilines.always_show then
     local under_cursor = M.under_cursor(opts, bufnr, diagnostics)
     local multiline_diags = diagnostics

lua/tiny-inline-diagnostic/filter.lua

@@ -50,6 +50,7 @@ local default_config = {
       severity = nil,
     },
     show_all_diags_on_cursorline = false,
+    show_diags_only_under_cursor = false,
     enable_on_insert = false,
     enable_on_select = false,
     format = nil,

lua/tiny-inline-diagnostic/init.lua

@@ -56,6 +56,16 @@ T["at_position"]["returns line diagnostics when cursor not in range"] = function
   MiniTest.expect.equality(#result, 2)
 end
 
+T["at_position"]["returns empty when show_diags_only_under_cursor enabled and cursor not in range"] = function()
+  local diagnostics = {
+    H.make_diagnostic({ lnum = 5, col = 10, end_col = 20 }),
+    H.make_diagnostic({ lnum = 5, col = 30, end_col = 40 }),
+  }
+
+  local result = filter.at_position({ options = { show_diags_only_under_cursor = true } }, diagnostics, 5, 0)
+  MiniTest.expect.equality(#result, 0)
+end
+
 T["under_cursor"] = MiniTest.new_set()
 
 T["under_cursor"]["returns empty for invalid buffer"] = function()
@@ -276,4 +286,33 @@ T["for_display"]["filters by multilines.severity when always_show is false"] = f
   end)
 end
 
+T["for_display"]["show_diags_only_under_cursor with multilines shows under cursor on current line and all on other lines"] = function()
+  H.with_win_buf({ "line1", "line2", "line3" }, { 2, 0 }, nil, function(buf)
+    local diagnostics = {
+      H.make_diagnostic({ lnum = 0, col = 0, end_col = 5 }), -- line 1
+      H.make_diagnostic({ lnum = 0, col = 10, end_col = 15 }), -- line 1, not under cursor
+      H.make_diagnostic({ lnum = 1, col = 0, end_col = 5 }), -- line 2 (current line)
+      H.make_diagnostic({ lnum = 2, col = 0, end_col = 5 }), -- line 3
+    }
+
+    local result = filter.for_display({
+      options = {
+        multilines = { enabled = true },
+        show_diags_only_under_cursor = true,
+      },
+    }, buf, diagnostics)
+
+    -- Should show: all from line 1 (other line), 1 diag from line 2 (under cursor), 1 from line 3
+    MiniTest.expect.equality(#result, 4)
+    -- Check that we have diagnostics from all lines
+    local lines = {}
+    for _, diag in ipairs(result) do
+      lines[diag.lnum] = (lines[diag.lnum] or 0) + 1
+    end
+    MiniTest.expect.equality(lines[0], 2) -- line 1
+    MiniTest.expect.equality(lines[1], 1) -- line 2 (current)
+    MiniTest.expect.equality(lines[2], 1) -- line 3
+  end)
+end
+
 return T