diff options
| -rw-r--r-- | lua/99/prompt-settings.lua | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/lua/99/prompt-settings.lua b/lua/99/prompt-settings.lua index 258de34..a688a0e 100644 --- a/lua/99/prompt-settings.lua +++ b/lua/99/prompt-settings.lua @@ -4,10 +4,10 @@ local function get_surrounding_context(range, n) local start_row, _ = range.start:to_vim() local end_row, _ = range.end_:to_vim() - -- nvim_buf_get_lines safely clamps out-of-bounds indexes - local from = start_row - n - local to = end_row + 1 + n -- +1 because end_row is end-inclusive - local lines = vim.api.nvim_buf_get_lines(range.buffer, from, to, false) -- nvim_buf_get_lines is end-exclusive + local line_count = vim.api.nvim_buf_line_count(range.buffer) + local from = math.max(start_row - n, 0) + local to = math.min(end_row + 1 + n, line_count) + local lines = vim.api.nvim_buf_get_lines(range.buffer, from, to, false) return table.concat(lines, "\n") end |