diff options
Diffstat (limited to 'lua/nvim-treesitter/ts_utils.lua')
| -rw-r--r-- | lua/nvim-treesitter/ts_utils.lua | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/lua/nvim-treesitter/ts_utils.lua b/lua/nvim-treesitter/ts_utils.lua index f9c3de10..ebeb1235 100644 --- a/lua/nvim-treesitter/ts_utils.lua +++ b/lua/nvim-treesitter/ts_utils.lua @@ -325,7 +325,12 @@ end -- Set visual selection to node function M.update_selection(buf, node) - local start_row, start_col, end_row, end_col = node:range() + local start_row, start_col, end_row, end_col + if type(node) == 'table' then + start_row, start_col, end_row, end_col = unpack(node) + else + start_row, start_col, end_row, end_col = node:range() + end if end_row == vim.fn.line('$') then end_col = #vim.fn.getline('$') |