fix: dont pass content around, just bufnr

tris203 · May 3, 2024 · b940437 · b940437
1 parent 7eacf6f
commit b940437
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 8 deletions.
diff --git a/lua/precognition/init.lua b/lua/precognition/init.lua
@@ -97,7 +97,7 @@ end
 ---@return Precognition.GutterHints
 local function build_gutter_hints(buf)
     local gutter_hints = {
-        ["G"] = vm.file_end(vim.api.nvim_buf_get_lines(buf, 0, -1, false)),
+        ["G"] = vm.file_end(vim.api.nvim_get_current_buf()),
         ["gg"] = vm.file_start(),
         ["{"] = vm.prev_paragraph_line(vim.api.nvim_get_current_buf()),
         ["}"] = vm.next_paragraph_line(vim.api.nvim_get_current_buf()),

diff --git a/lua/precognition/vertical_motions.lua b/lua/precognition/vertical_motions.lua
@@ -5,10 +5,10 @@ function M.file_start()
     return 1
 end
 
----@param lines table
+---@param bufnr integer
 ---@return integer
-function M.file_end(lines)
-    return #lines
+function M.file_end(bufnr)
+    return vim.api.nvim_buf_line_count(bufnr)
 end
 
 ---@param buf integer

diff --git a/tests/precognition/vertical_motions_spec.lua b/tests/precognition/vertical_motions_spec.lua
@@ -38,8 +38,7 @@ describe("gutter motion locations", function()
         vim.api.nvim_set_current_buf(testBuf)
         vim.api.nvim_win_set_cursor(0, { 1, 1 })
 
-        local lines = vim.api.nvim_buf_get_lines(testBuf, 0, -1, false)
-        local end_ = vm.file_end(lines)
+        local end_ = vm.file_end(testBuf)
         eq(1, end_)
         eq(vim.api.nvim_buf_line_count(testBuf), end_)
     end)
@@ -59,8 +58,7 @@ describe("gutter motion locations", function()
         vim.api.nvim_set_current_buf(testBuf)
         vim.api.nvim_win_set_cursor(0, { 4, 0 })
 
-        local lines = vim.api.nvim_buf_get_lines(testBuf, 0, -1, false)
-        local end_ = vm.file_end(lines)
+        local end_ = vm.file_end(testBuf)
         eq(8, end_)
         eq(vim.api.nvim_buf_line_count(testBuf), end_)
     end)