feat:stop rendering when edit big file shellRaining#85

add new options choke_at_linecount
check_choke

lua/hlchunk/base_mod.lua

@@ -8,6 +8,7 @@ local fn = vim.fn
 ---@field exclude_filetypes table<string, boolean>
 ---@field support_filetypes table<string>
 ---@field notify boolean
+---@field choke_at_linecount number
 
 ---@class RuntimeVar
 ---@field old_win_info table<number, number>
@@ -36,6 +37,7 @@ local BaseMod = {
         exclude_filetypes = {},
         support_filetypes = {},
         notify = false,
+        choke_at_linecount = -1,
     },
     ns_id = -1,
     old_win_info = fn.winsaveview(),
@@ -181,6 +183,13 @@ function BaseMod:set_hl()
     end
 end
 
+function BaseMod:check_choke(line_count)
+    if self.options.choke_at_linecount ~= -1 and line_count >= self.options.choke_at_linecount then
+        return true
+    else
+        return false
+    end
+end
 -- set options for mod, if the mod dont have default config, it will notify you
 ---@param options BaseModOpts
 function BaseMod:set_options(options)

lua/hlchunk/mods/blank.lua

@@ -25,6 +25,7 @@ local blank_mod = BaseMod:new({
             api.nvim_get_hl(0, { name = "Whitespace" }),
         },
         exclude_filetypes = ft.exclude_filetypes,
+        choke_at_linecount = -1,
     },
 })
 
@@ -60,7 +61,12 @@ function blank_mod:render_line(index, indent)
 end
 
 function blank_mod:render()
-    if (not self.options.enable) or self.options.exclude_filetypes[vim.bo.filetype] or fn.shiftwidth() == 0 then
+    if
+        not self.options.enable
+        or self.options.exclude_filetypes[vim.bo.filetype]
+        or fn.shiftwidth() == 0
+        or BaseMod.check_choke(self, vim.api.nvim_buf_line_count(0))
+    then
         return
     end
 

lua/hlchunk/mods/chunk.lua

@@ -38,6 +38,7 @@ local chunk_mod = BaseMod:new({
         textobject = "",
         max_file_size = 1024 * 1024,
         error_sign = true,
+        choke_at_linecount = -1,
     },
 })
 
@@ -49,7 +50,11 @@ end
 
 -- set new virtual text to the right place
 function chunk_mod:render(opts)
-    if not self.options.enable or self.options.exclude_filetypes[vim.bo.ft] then
+    if
+        not self.options.enable
+        or self.options.exclude_filetypes[vim.bo.ft]
+        or BaseMod.check_choke(self, vim.api.nvim_buf_line_count(0))
+    then
         return
     end
 

lua/hlchunk/mods/context.lua

@@ -23,11 +23,16 @@ local context_mod = BaseMod:new({
             "#806d9c",
         },
         exclude_filetypes = ft.exclude_filetypes,
+        choke_at_linecount = -1,
     },
 })
 
 function context_mod:render()
-    if (not self.options.enable) or self.options.exclude_filetypes[vim.bo.filetype] then
+    if
+        not self.options.enable
+        or self.options.exclude_filetypes[vim.bo.filetype]
+        or BaseMod.check_choke(self, vim.api.nvim_buf_line_count(0))
+    then
         return
     end
 

lua/hlchunk/mods/indent.lua

@@ -26,6 +26,7 @@ local indent_mod = BaseMod:new({
             fn.synIDattr(fn.synIDtrans(fn.hlID("Whitespace")), "fg", "gui"),
         },
         exclude_filetypes = ft.exclude_filetypes,
+        choke_at_linecount = -1,
     },
 })
 
@@ -67,7 +68,12 @@ function indent_mod:render_line(index, indent)
 end
 
 function indent_mod:render()
-    if (not self.options.enable) or self.options.exclude_filetypes[vim.bo.filetype] or fn.shiftwidth() == 0 then
+    if
+        not self.options.enable
+        or self.options.exclude_filetypes[vim.bo.filetype]
+        or fn.shiftwidth() == 0
+        or BaseMod.check_choke(self, vim.api.nvim_buf_line_count(0))
+    then
         return
     end
 

lua/hlchunk/mods/line_num.lua

@@ -19,11 +19,16 @@ local line_num_mod = BaseMod:new({
         style = "#806d9c",
         support_filetypes = ft.support_filetypes,
         exclude_filetypes = ft.exclude_filetypes,
+        choke_at_linecount = -1,
     },
 })
 
 function line_num_mod:render()
-    if not self.options.enable or self.options.exclude_filetypes[vim.bo.ft] then
+    if
+        not self.options.enable
+        or self.options.exclude_filetypes[vim.bo.ft]
+        or BaseMod.check_choke(self, vim.api.nvim_buf_line_count(0))
+    then
         return
     end