perf: store components in context to avoid duplicate queries

Author: MeanderingProgrammer

Diff for: benches/medium_spec.lua

@@ -5,7 +5,7 @@ local util = require('benches.util')
 describe('medium.md', function()
     it('default', function()
         local base_marks = 46
-        util.less_than(util.setup('temp/medium.md'), 30)
+        util.less_than(util.setup('temp/medium.md'), 35)
         util.num_marks(base_marks)
 
         util.less_than(util.move_down(3), 0.5)

Diff for: benches/medium_table_spec.lua

@@ -5,7 +5,7 @@ local util = require('benches.util')
 describe('medium-table.md', function()
     it('default', function()
         local base_marks = 180
-        util.less_than(util.setup('temp/medium-table.md'), 105)
+        util.less_than(util.setup('temp/medium-table.md'), 110)
         util.num_marks(base_marks)
 
         util.less_than(util.move_down(1), 0.5)

Diff for: lua/render-markdown/core/component.lua

@@ -8,6 +8,7 @@ local util = require('render-markdown.core.util')
 ---@field private win integer
 ---@field private top integer
 ---@field private bottom integer
+---@field private components table<integer, render.md.CustomComponent>
 ---@field private conceal? table<integer, [integer, integer][]>
 ---@field private links table<integer, [integer, integer, integer][]>
 ---@field last_heading integer
@@ -25,6 +26,7 @@ function Context.new(buf, win, offset)
     local top = util.view(win).topline - 1
     self.top = math.max(top - offset, 0)
     self.bottom = self:compute_bottom(top, offset)
+    self.components = {}
     self.conceal = nil
     self.links = {}
     self.last_heading = -1
@@ -48,6 +50,18 @@ function Context:compute_bottom(top, offset)
     return bottom
 end
 
+---@param info render.md.NodeInfo
+---@return render.md.CustomComponent?
+function Context:get_component(info)
+    return self.components[info.start_row]
+end
+
+---@param info render.md.NodeInfo
+---@param component render.md.CustomComponent
+function Context:add_component(info, component)
+    self.components[info.start_row] = component
+end
+
 ---@param info? render.md.NodeInfo
 ---@return integer
 function Context:width(info)

Diff for: lua/render-markdown/core/context.lua

@@ -1,5 +1,4 @@
 local Context = require('render-markdown.core.context')
-local component = require('render-markdown.core.component')
 local list = require('render-markdown.core.list')
 local logger = require('render-markdown.core.logger')
 local state = require('render-markdown.state')
@@ -106,14 +105,16 @@ function Handler:list_marker(info)
         if not self.config.checkbox.enabled then
             return false
         end
+        if self.context:get_component(info) ~= nil then
+            return true
+        end
         if info:sibling('task_list_marker_unchecked') ~= nil then
             return true
         end
         if info:sibling('task_list_marker_checked') ~= nil then
             return true
         end
-        local paragraph = info:sibling('paragraph')
-        return paragraph ~= nil and component.checkbox(self.config, paragraph.text, 'starts') ~= nil
+        return false
     end
     if sibling_checkbox() then
         -- Hide the list marker for checkboxes rather than replacing with a bullet point

Diff for: lua/render-markdown/handler/markdown.lua

@@ -1,5 +1,4 @@
 local Context = require('render-markdown.core.context')
-local component = require('render-markdown.core.component')
 local list = require('render-markdown.core.list')
 local logger = require('render-markdown.core.logger')
 local state = require('render-markdown.state')
@@ -56,13 +55,13 @@ end
 ---@private
 ---@param info render.md.NodeInfo
 function Handler:shortcut(info)
-    local callout = component.callout(self.config, info.text, 'exact')
+    local callout = self:get_callout(info)
     if callout ~= nil then
         self:callout(info, callout)
         return
     end
 
-    local checkbox = component.checkbox(self.config, info.text, 'exact')
+    local checkbox = self:get_checkbox(info)
     if checkbox ~= nil then
         self:checkbox(info, checkbox)
         return
@@ -75,6 +74,32 @@ function Handler:shortcut(info)
     end
 end
 
+---@private
+---@param info render.md.NodeInfo
+---@return render.md.CustomComponent?
+function Handler:get_callout(info)
+    local text = info.text:lower()
+    for _, callout in pairs(self.config.callout) do
+        if text == callout.raw:lower() then
+            return callout
+        end
+    end
+    return nil
+end
+
+---@private
+---@param info render.md.NodeInfo
+---@return render.md.CustomComponent?
+function Handler:get_checkbox(info)
+    local text = info.text
+    for _, checkbox in pairs(self.config.checkbox.custom) do
+        if text == checkbox.raw then
+            return checkbox
+        end
+    end
+    return nil
+end
+
 ---@private
 ---@param info render.md.NodeInfo
 ---@param callout render.md.CustomComponent
@@ -99,13 +124,16 @@ function Handler:callout(info, callout)
     end
 
     local text, conceal = custom_title()
-    self.marks:add(true, info.start_row, info.start_col, {
+    local added = self.marks:add(true, info.start_row, info.start_col, {
         end_row = info.end_row,
         end_col = info.end_col,
         virt_text = { { text, callout.highlight } },
         virt_text_pos = 'overlay',
         conceal = conceal,
     })
+    if added then
+        self.context:add_component(info, callout)
+    end
 end
 
 ---@private
@@ -117,13 +145,16 @@ function Handler:checkbox(info, checkbox)
     end
     local inline = self.config.checkbox.position == 'inline'
     local icon, highlight = checkbox.rendered, checkbox.highlight
-    self.marks:add(true, info.start_row, info.start_col, {
+    local added = self.marks:add(true, info.start_row, info.start_col, {
         end_row = info.end_row,
         end_col = info.end_col,
         virt_text = { { inline and icon or str.pad_to(info.text, icon), highlight } },
         virt_text_pos = 'inline',
         conceal = '',
     })
+    if added then
+        self.context:add_component(info, checkbox)
+    end
 end
 
 ---@private

Diff for: lua/render-markdown/handler/markdown_inline.lua

@@ -5,7 +5,7 @@ local M = {}
 
 ---@private
 ---@type string
-M.version = '6.3.1'
+M.version = '6.3.2'
 
 function M.check()
     vim.health.start('render-markdown.nvim [version]')

Diff for: lua/render-markdown/health.lua

@@ -1,5 +1,4 @@
 local Base = require('render-markdown.render.base')
-local component = require('render-markdown.core.component')
 local logger = require('render-markdown.core.logger')
 local state = require('render-markdown.state')
 
@@ -25,7 +24,7 @@ function Render:setup()
         return false
     end
 
-    local callout = component.callout(self.config, self.info.text, 'contains')
+    local callout = self.context:get_component(self.info)
     self.highlight = callout ~= nil and callout.highlight or self.quote.highlight
 
     return true