From 19d71dd596cadad21c91cdb6cc6f171dbe42db74 Mon Sep 17 00:00:00 2001
From: Kyren223 <kyren223@proton.me>
Date: Mon, 4 Aug 2025 13:24:57 +0300
Subject: Fixed incorrect highlighting of TODOs

---
 .config/nvim/lua/autocmds.lua | 44 ++++++++++++++++++++++++++++---------------
 1 file changed, 29 insertions(+), 15 deletions(-)

(limited to '.config/nvim/lua/autocmds.lua')

diff --git a/.config/nvim/lua/autocmds.lua b/.config/nvim/lua/autocmds.lua
index f7440ef..34bcbea 100644
--- a/.config/nvim/lua/autocmds.lua
+++ b/.config/nvim/lua/autocmds.lua
@@ -220,24 +220,38 @@ function RenderTodoHighlights(bufnr)
                     -- 1) highlight keyword
                     vim.api.nvim_buf_add_highlight(bufnr, ns, cfg.hl, i - 1, s - 1, e - 1)
 
-                    -- 2) highlight (...) parts only if a keyword was found
-                    local start = 1
-                    local s, e = line:find('%b()', start)
-                    if not s then
-                        break
+                    local next_char = line:sub(e, e)
+
+                    -- 2) highlight (...) if it directly follows keyword
+                    if next_char == '(' then
+                        local ps = e
+                        local pe = nil
+                        for j = e + 1, math.min(#line, e + 100) do
+                            if line:sub(j, j) == ')' then
+                                pe = j
+                                break
+                            end
+                        end
+
+                        if pe then
+                            vim.api.nvim_buf_add_highlight(bufnr, ns, scopes.bracket, i - 1, ps - 1, ps)
+                            vim.api.nvim_buf_add_highlight(bufnr, ns, scopes.bracket, i - 1, pe - 1, pe)
+                            if pe - ps > 1 then
+                                vim.api.nvim_buf_add_highlight(bufnr, ns, scopes.constant, i - 1, ps, pe - 1)
+                            end
+
+                            e = pe
+                        end
                     end
-                    -- "(" and ")"
-                    vim.api.nvim_buf_add_highlight(bufnr, ns, scopes.bracket, i - 1, s - 1, s)
-                    vim.api.nvim_buf_add_highlight(bufnr, ns, scopes.bracket, i - 1, e - 1, e)
-                    -- inner constant
-                    if e - s > 1 then
-                        vim.api.nvim_buf_add_highlight(bufnr, ns, scopes.constant, i - 1, s, e - 1)
+
+                    -- -- 3) highlight ":" if directly after keyword or after parens
+                    local colon_pos = nil
+                    if line:sub(e, e) == ':' then
+                        colon_pos = e
                     end
-                    start = e + 1
 
-                    -- 3) highlight any ":" as delimiter only if a group keyword was found
-                    for s, _ in line:gmatch('():') do
-                        vim.api.nvim_buf_add_highlight(bufnr, ns, scopes.delimiter, i - 1, s - 1, s)
+                    if colon_pos then
+                        vim.api.nvim_buf_add_highlight(bufnr, ns, scopes.delimiter, i - 1, colon_pos - 1, colon_pos)
                     end
                 end
             end
-- 
cgit v1.3.1