From 06f1d710bd0e48de1bcd6a883873247b955c7c7b Mon Sep 17 00:00:00 2001
From: yetone <yetoneful@gmail.com>
Date: Wed, 28 Aug 2024 16:13:04 +0800
Subject: [PATCH] fix: end_col out of range (#310)

---
 lua/avante/selection.lua | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/lua/avante/selection.lua b/lua/avante/selection.lua
index 5b71e32..30a23f0 100644
--- a/lua/avante/selection.lua
+++ b/lua/avante/selection.lua
@@ -280,6 +280,9 @@ function Selection:create_editing_input()
 
   self.selection = Utils.get_visual_selection_and_range()
 
+  local end_row = self.selection.range.finish.line - 1
+  local end_col = math.min(self.selection.range.finish.col, #code_lines[self.selection.range.finish.line])
+
   self.selected_code_extmark_id = api.nvim_buf_set_extmark(
     code_bufnr,
     SELECTED_CODE_NAMESPACE,
@@ -288,8 +291,8 @@ function Selection:create_editing_input()
     {
       hl_group = "Visual",
       hl_mode = "combine",
-      end_row = self.selection.range.finish.line - 1,
-      end_col = self.selection.range.finish.col,
+      end_row = end_row,
+      end_col = end_col,
       priority = PRIORITY,
     }
   )