fix: render stacked 'statuscolumn' highlights

nvim-treesitter · Jan 23, 2025 · 02ae16d · 02ae16d
1 parent bece284
commit 02ae16d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lua/treesitter-context/render.lua b/lua/treesitter-context/render.lua
@@ -225,6 +225,7 @@ end
 
 --- @class StatusLineHighlight
 --- @field group string
+--- @field groups? string[]
 --- @field start integer
 
 --- @param ctx_node_line_num integer
@@ -291,9 +292,13 @@ local function highlight_lno_str(buf, text, highlights)
       local col = hl.start
       local endcol = hlidx < #linehl and linehl[hlidx + 1].start or #text[line]
       if col ~= endcol then
+        local hl_groups = hl.groups or { hl.group }
+        for i, shl in ipairs(hl_groups) do
+          hl_groups[i] = shl:find('LineNr') and 'TreesitterContextLineNumber' or shl
+        end
         add_extmark(buf, line - 1, col, {
           end_col = endcol,
-          hl_group = hl.group:find('LineNr') and 'TreesitterContextLineNumber' or hl.group,
+          hl_group = hl.groups and hl_groups or hl_groups[1]
         })
       end
     end