Add buffer_characters

lua/fittencode/actions/identify_programming_language.lua

@@ -34,11 +34,10 @@ local function _identify_current_buffer()
     return
   end
 
-  local count = 0
-  local lines = api.nvim_buf_get_lines(buffer, 0, -1, false)
-  vim.tbl_map(function(line)
-    count = count + #line
-  end, lines)
+  local count, lines = Base.buffer_characters(buffer)
+  if not count or not lines then
+    return
+  end
   if count > Config.options.prompt.max_characters then
     return
   end

lua/fittencode/base.lua

@@ -186,6 +186,20 @@ function M.rfind(s, sub)
   end)()
 end
 
+---@param buffer? number
+function M.buffer_characters(buffer)
+  buffer = buffer or api.nvim_get_current_buf()
+  if not api.nvim_buf_is_valid(buffer) then
+    return
+  end
+  local count = 0
+  local lines = api.nvim_buf_get_lines(buffer, 0, -1, false)
+  vim.tbl_map(function(line)
+    count = count + #line
+  end, lines)
+  return count, lines
+end
+
 ---@class NeovimVersion
 ---@field nvim string
 ---@field buildtype string

lua/fittencode/prompt_providers/default.lua

@@ -1,5 +1,6 @@
 local api = vim.api
 
+local Base = require('fittencode.base')
 local Config = require('fittencode.config')
 local Log = require('fittencode.log')
 
@@ -37,11 +38,10 @@ function M:execute(ctx)
     return
   end
 
-  local count = 0
-  local lines = api.nvim_buf_get_lines(ctx.buffer, 0, -1, false)
-  vim.tbl_map(function(line)
-    count = count + #line
-  end, lines)
+  local count = Base.buffer_characters(ctx.buffer)
+  if not count then
+    return
+  end
   if count > Config.options.prompt.max_characters then
     return
   end