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)
This commit is contained in:
Santos Gallegos 2021-04-16 00:51:40 -05:00 committed by GitHub
parent 9fb2b0c0b6
commit 2abad14069
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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