update counts

epwalsh · Jul 30, 2024 · 3e9d89a · 3e9d89a
1 parent d4cf161
commit 3e9d89a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lua/obsidian/ui.lua b/lua/obsidian/ui.lua
@@ -727,7 +727,8 @@ local function update_extmarks(bufnr, ns_id, ui_opts)
     elseif not inside_code_block then
       -- Mark that we have a highlight group now
       if string.match(line, search.Patterns.Callout) then
-        local count = #(line:match "^(>+)" or ">")
+        local arrows, _ = line:gsub(">", ""):find "%[!%]"
+        local count = arrows - 1
         local highlight = get_callout_hl_group(line, ui_opts)
         callout_block_highlights[count] = highlight
         log.debug("Inserting callout highlight: " .. highlight .. " into idx: " .. count)