diff options
| -rw-r--r-- | lua/nvim-treesitter/ts_utils.lua | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/lua/nvim-treesitter/ts_utils.lua b/lua/nvim-treesitter/ts_utils.lua index 6f31ef7e6..93ac7027f 100644 --- a/lua/nvim-treesitter/ts_utils.lua +++ b/lua/nvim-treesitter/ts_utils.lua @@ -131,7 +131,8 @@ function M.highlight_range(range, buf, hl_namespace, hl_group) end -- Set visual selection to node --- @param selection_mode One of "charwise" (default), "linewise", "blockwise" +-- @param selection_mode One of "charwise" (default) or "v", "linewise" or "V", +-- "blockwise" or "<C-v>" (as a string with 5 characters or a single character) function M.update_selection(buf, node, selection_mode) selection_mode = selection_mode or "charwise" local start_row, start_col, end_row, end_col = M.get_node_range(node) @@ -152,10 +153,10 @@ function M.update_selection(buf, node, selection_mode) local v_table = {charwise = "v", linewise = "V", blockwise = "<C-v>"} ---- Call to `nvim_replace_termcodes()` is needed for sending appropriate ---- command to enter blockwise mode - local command = vim.api.nvim_replace_termcodes( - "normal! " .. v_table[selection_mode], true, true, true + local mode_string = vim.api.nvim_replace_termcodes( + v_table[selection_mode] or selection_mode, true, true, true ) - vim.cmd(command) + vim.cmd("normal! " .. mode_string) -- Convert exclusive end position to inclusive if end_col == 1 then |