Add config.max_file_length

lewis6991 · Feb 27, 2021 · 093da28 · 093da28
1 parent b866c83
commit 093da28
Show file tree

Hide file tree

Showing 6 changed files with 34 additions and 0 deletions.
diff --git a/doc/gitsigns.txt b/doc/gitsigns.txt
@@ -254,6 +254,11 @@ status_formatter                            *gitsigns-config-status_formatter*
 <
         Function used to format `b:gitsigns_status`.
 
+max_file_length                              *gitsigns-config-max_file_length*
+        Type: `number`, Default: `40000`
+
+      Max file length to attach to.
+
 debug_mode                                        *gitsigns-config-debug_mode*
         Type: `boolean`, Default: `false`
 

diff --git a/lua/gitsigns.lua b/lua/gitsigns.lua
@@ -430,6 +430,12 @@ local attach = throttle_leading(100, sync(function()
    end
    dprint('Attaching', cbuf, 'attach')
 
+   local lc = api.nvim_buf_line_count(cbuf)
+   if lc > config.max_file_length then
+      dprint('Exceeds max_file_length', cbuf, 'attach')
+      return
+   end
+
    if api.nvim_buf_get_option(cbuf, 'buftype') ~= '' then
       dprint('Non-normal buffer', cbuf, 'attach')
       return

diff --git a/lua/gitsigns/config.lua b/lua/gitsigns/config.lua
@@ -149,6 +149,14 @@ local schema = {
     ]],
    },
 
+   max_file_length = {
+      type = 'number',
+      default = 40000,
+      description = [[
+      Max file length to attach to.
+    ]],
+   },
+
    debug_mode = {
       type = 'boolean',
       default = false,

diff --git a/teal/gitsigns.tl b/teal/gitsigns.tl
@@ -430,6 +430,12 @@ local attach = throttle_leading(100, sync(function()
   end
   dprint('Attaching', cbuf, 'attach')
 
+  local lc = api.nvim_buf_line_count(cbuf)
+  if lc > config.max_file_length then
+    dprint('Exceeds max_file_length', cbuf, 'attach')
+    return
+  end
+
   if api.nvim_buf_get_option(cbuf, 'buftype') ~= '' then
     dprint('Non-normal buffer', cbuf, 'attach')
     return

diff --git a/teal/gitsigns/config.tl b/teal/gitsigns/config.tl
@@ -149,6 +149,14 @@ local schema: {string:SchemaElem} = {
     ]]
   },
 
+  max_file_length = {
+    type = 'number',
+    default = 40000,
+    description = [[
+      Max file length to attach to.
+    ]]
+  },
+
   debug_mode = {
     type = 'boolean',
     default = false,

diff --git a/teal/gitsigns/types.tl b/teal/gitsigns/types.tl
@@ -50,6 +50,7 @@ global record Config
     record watch_index
         interval: number
     end
+    max_file_length: number
     status_formatter: function(StatusObj): string
 end