CarGDev/avante.nvim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(utils): simplify and annotate count_lines() (#2446)

Browse Source
This commit is contained in:
Dmitry Torokhov
2025-07-15 01:42:30 -07:00
committed by GitHub
parent 5e50db3e05
commit d3dde0c5ff
1 changed files with 6 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
lua/avante/utils/init.lua
View File

@@ -1774,20 +1774,15 @@ function M.message_to_text(message, messages)
return "" return ""
end end
---Counts number of strings in text, accounting for possibility of a trailing newline
---@param str string | nil
---@return integer
function M.count_lines(str) function M.count_lines(str)
if not str or str == "" then return 0 end if not str or str == "" then return 0 end
local count = 1 local _, count = str:gsub("\n", "\n")
local len = #str -- Number of lines is one more than number of newlines unless we have a trailing newline
local newline_byte = string.byte("\n") return str:sub(-1) ~= "\n" and count + 1 or count
for i = 1, len do
if str:byte(i) == newline_byte then count = count + 1 end
end
if str:byte(len) == newline_byte then count = count - 1 end
return count
end end
function M.tbl_override(value, override) function M.tbl_override(value, override)