From b480d25155d69d53fb4aaaacdfdfef410ea8ddc9 Mon Sep 17 00:00:00 2001 From: Stephan Seitz Date: Wed, 15 Jul 2020 19:32:41 +0200 Subject: feat(textobjects): Add `start` to include preceding things like documentation --- lua/nvim-treesitter/ts_utils.lua | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'lua/nvim-treesitter/ts_utils.lua') 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('$') -- cgit v1.2.3