From 2abad14069534f5589cab43f744cdf7be93a5464 Mon Sep 17 00:00:00 2001 From: Santos Gallegos Date: Fri, 16 Apr 2021 00:51:40 -0500 Subject: Fix TSEditQuery with several files (#1191) - Choice already starts with 1, there isn't need to increment 1 - The first item is the prompt as recommended in `:h inputlist()` (this way the choice matches when using the mouse) --- lua/nvim-treesitter/configs.lua | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) (limited to 'lua') diff --git a/lua/nvim-treesitter/configs.lua b/lua/nvim-treesitter/configs.lua index 1cc9756e..333cf003 100644 --- a/lua/nvim-treesitter/configs.lua +++ b/lua/nvim-treesitter/configs.lua @@ -207,9 +207,18 @@ function M.edit_query_file(query_group, lang) vim.cmd(':edit '..files[1]) else local counter = 0 - local choice = vim.fn.inputlist(vim.tbl_map(function(f) counter = counter + 1;return counter..'. '..f end, files)) - if choice > 0 then - vim.cmd(':edit '..files[choice + 1]) + local choices = { + 'Select a file:', + table.unpack(vim.tbl_map(function(f) + counter = counter + 1 + return counter..'. '..f + end, + files + )) + } + local choice = vim.fn.inputlist(choices) + if choice > 0 and choice <= #files then + vim.cmd(':edit '..files[choice]) end end end -- cgit v1.2.3