fix: add opts.bufnr and opts.winnr to builtin picker (#1706)

This commit is contained in:
James Trew
2022-02-14 11:39:58 -05:00
committed by GitHub
parent c948263d89
commit 206e71d133
8 changed files with 39 additions and 56 deletions

View File

@@ -1311,9 +1311,6 @@ builtin.lsp_references({opts}) *builtin.lsp_references()*
Parameters: ~ Parameters: ~
{opts} (table) options to pass to the picker {opts} (table) options to pass to the picker
Options: ~
{timeout} (number) timeout for the sync call (default: 10000)
builtin.lsp_definitions({opts}) *builtin.lsp_definitions()* builtin.lsp_definitions({opts}) *builtin.lsp_definitions()*
Goto the definition of the word under the cursor, if there's only one, Goto the definition of the word under the cursor, if there's only one,
@@ -1324,8 +1321,6 @@ builtin.lsp_definitions({opts}) *builtin.lsp_definitions()*
{opts} (table) options to pass to the picker {opts} (table) options to pass to the picker
Options: ~ Options: ~
{timeout} (number) timeout for the sync call (default:
10000)
{jump_type} (string) how to goto definition if there is only {jump_type} (string) how to goto definition if there is only
one, values: "tab", "split", "vsplit", one, values: "tab", "split", "vsplit",
"never" "never"
@@ -1341,8 +1336,6 @@ builtin.lsp_type_definitions({opts}) *builtin.lsp_type_definitions()*
{opts} (table) options to pass to the picker {opts} (table) options to pass to the picker
Options: ~ Options: ~
{timeout} (number) timeout for the sync call (default:
10000)
{jump_type} (string) how to goto definition if there is only {jump_type} (string) how to goto definition if there is only
one, values: "tab", "split", "vsplit", one, values: "tab", "split", "vsplit",
"never" "never"
@@ -1358,8 +1351,6 @@ builtin.lsp_implementations({opts}) *builtin.lsp_implementations()*
{opts} (table) options to pass to the picker {opts} (table) options to pass to the picker
Options: ~ Options: ~
{timeout} (number) timeout for the sync call (default:
10000)
{jump_type} (string) how to goto implementation if there is {jump_type} (string) how to goto implementation if there is
only one, values: "tab", "split", only one, values: "tab", "split",
"vsplit", "never" "vsplit", "never"
@@ -1404,8 +1395,6 @@ builtin.lsp_document_symbols({opts}) *builtin.lsp_document_symbols()*
{opts} (table) options to pass to the picker {opts} (table) options to pass to the picker
Options: ~ Options: ~
{timeout} (number) timeout for the sync call
(default: 10000)
{ignore_filename} (boolean) dont show filenames (default: {ignore_filename} (boolean) dont show filenames (default:
true) true)
{show_line} (boolean) if true, shows the content of the {show_line} (boolean) if true, shows the content of the
@@ -1429,8 +1418,6 @@ builtin.lsp_workspace_symbols({opts}) *builtin.lsp_workspace_symbols()*
Options: ~ Options: ~
{query} (string) for what to query the workspace {query} (string) for what to query the workspace
(default: "") (default: "")
{timeout} (number) timeout for the sync call
(default: 10000)
{ignore_filename} (boolean) dont show filenames (default: {ignore_filename} (boolean) dont show filenames (default:
false) false)
{show_line} (boolean) if true, shows the content of the {show_line} (boolean) if true, shows the content of the

View File

@@ -86,6 +86,9 @@ local diagnostics_to_tbl = function(opts)
end end
diagnostics.get = function(opts) diagnostics.get = function(opts)
if opts.bufnr ~= 0 then
opts.bufnr = nil
end
if opts.bufnr == nil then if opts.bufnr == nil then
opts.path_display = vim.F.if_nil(opts.path_display, {}) opts.path_display = vim.F.if_nil(opts.path_display, {})
end end

View File

@@ -271,16 +271,14 @@ files.treesitter = function(opts)
end end
local parsers = require "nvim-treesitter.parsers" local parsers = require "nvim-treesitter.parsers"
if not parsers.has_parser() then if not parsers.has_parser(parsers.get_buf_lang(opts.bufnr)) then
print "No parser for the current buffer" print "No parser for the current buffer"
return return
end end
local ts_locals = require "nvim-treesitter.locals" local ts_locals = require "nvim-treesitter.locals"
local bufnr = opts.bufnr or vim.api.nvim_get_current_buf()
local results = {} local results = {}
for _, definitions in ipairs(ts_locals.get_definitions(bufnr)) do for _, definitions in ipairs(ts_locals.get_definitions(opts.bufnr)) do
local entries = prepare_match(definitions) local entries = prepare_match(definitions)
for _, entry in ipairs(entries) do for _, entry in ipairs(entries) do
table.insert(results, entry) table.insert(results, entry)
@@ -307,17 +305,16 @@ end
files.current_buffer_fuzzy_find = function(opts) files.current_buffer_fuzzy_find = function(opts)
-- All actions are on the current buffer -- All actions are on the current buffer
local bufnr = vim.api.nvim_get_current_buf() local filename = vim.fn.expand(vim.api.nvim_buf_get_name(opts.bufnr))
local filename = vim.fn.expand(vim.api.nvim_buf_get_name(bufnr)) local filetype = vim.api.nvim_buf_get_option(opts.bufnr, "filetype")
local filetype = vim.api.nvim_buf_get_option(bufnr, "filetype")
local lines = vim.api.nvim_buf_get_lines(0, 0, -1, false) local lines = vim.api.nvim_buf_get_lines(opts.bufnr, 0, -1, false)
local lines_with_numbers = {} local lines_with_numbers = {}
for lnum, line in ipairs(lines) do for lnum, line in ipairs(lines) do
table.insert(lines_with_numbers, { table.insert(lines_with_numbers, {
lnum = lnum, lnum = lnum,
bufnr = bufnr, bufnr = opts.bufnr,
filename = filename, filename = filename,
text = line, text = line,
}) })
@@ -329,9 +326,9 @@ files.current_buffer_fuzzy_find = function(opts)
end end
local _, ts_configs = pcall(require, "nvim-treesitter.configs") local _, ts_configs = pcall(require, "nvim-treesitter.configs")
local parser_ok, parser = pcall(vim.treesitter.get_parser, bufnr, filetype) local parser_ok, parser = pcall(vim.treesitter.get_parser, opts.bufnr, filetype)
local query_ok, query = pcall(vim.treesitter.get_query, filetype, "highlights") local query_ok, query = pcall(vim.treesitter.get_query, filetype, "highlights")
if parser_ok and query_ok and ts_ok and ts_configs.is_enabled("highlight", filetype, bufnr) then if parser_ok and query_ok and ts_ok and ts_configs.is_enabled("highlight", filetype, opts.bufnr) then
local root = parser:parse()[1]:root() local root = parser:parse()[1]:root()
local highlighter = vim.treesitter.highlighter.new(parser) local highlighter = vim.treesitter.highlighter.new(parser)
@@ -344,7 +341,7 @@ files.current_buffer_fuzzy_find = function(opts)
return obj return obj
end, end,
}) })
for id, node in query:iter_captures(root, bufnr, 0, -1) do for id, node in query:iter_captures(root, opts.bufnr, 0, -1) do
local hl = highlighter_query:_get_hl_from_capture(id) local hl = highlighter_query:_get_hl_from_capture(id)
if hl and type(hl) ~= "number" then if hl and type(hl) ~= "number" then
local row1, col1, row2, col2 = node:range() local row1, col1, row2, col2 = node:range()

View File

@@ -101,8 +101,8 @@ local get_current_buf_line = function(winnr)
end end
git.bcommits = function(opts) git.bcommits = function(opts)
opts.current_line = (opts.current_file == nil) and get_current_buf_line(0) or nil opts.current_line = (opts.current_file == nil) and get_current_buf_line(opts.winnr) or nil
opts.current_file = vim.F.if_nil(opts.current_file, vim.fn.expand "%:p") opts.current_file = vim.F.if_nil(opts.current_file, vim.api.nvim_buf_get_name(opts.bufnr))
opts.entry_maker = vim.F.if_nil(opts.entry_maker, make_entry.gen_from_git_commits(opts)) opts.entry_maker = vim.F.if_nil(opts.entry_maker, make_entry.gen_from_git_commits(opts))
local git_command = vim.F.if_nil(opts.git_command, { "git", "log", "--pretty=oneline", "--abbrev-commit" }) local git_command = vim.F.if_nil(opts.git_command, { "git", "log", "--pretty=oneline", "--abbrev-commit" })

View File

@@ -352,12 +352,10 @@ builtin.jumplist = require_on_exported_call("telescope.builtin.internal").jumpli
--- Lists LSP references for word under the cursor, jumps to reference on `<cr>` --- Lists LSP references for word under the cursor, jumps to reference on `<cr>`
---@param opts table: options to pass to the picker ---@param opts table: options to pass to the picker
---@field timeout number: timeout for the sync call (default: 10000)
builtin.lsp_references = require_on_exported_call("telescope.builtin.lsp").references builtin.lsp_references = require_on_exported_call("telescope.builtin.lsp").references
--- Goto the definition of the word under the cursor, if there's only one, otherwise show all options in Telescope --- Goto the definition of the word under the cursor, if there's only one, otherwise show all options in Telescope
---@param opts table: options to pass to the picker ---@param opts table: options to pass to the picker
---@field timeout number: timeout for the sync call (default: 10000)
---@field jump_type string: how to goto definition if there is only one, values: "tab", "split", "vsplit", "never" ---@field jump_type string: how to goto definition if there is only one, values: "tab", "split", "vsplit", "never"
---@field ignore_filename boolean: dont show filenames (default: true) ---@field ignore_filename boolean: dont show filenames (default: true)
builtin.lsp_definitions = require_on_exported_call("telescope.builtin.lsp").definitions builtin.lsp_definitions = require_on_exported_call("telescope.builtin.lsp").definitions
@@ -365,14 +363,12 @@ builtin.lsp_definitions = require_on_exported_call("telescope.builtin.lsp").defi
--- Goto the definition of the type of the word under the cursor, if there's only one, --- Goto the definition of the type of the word under the cursor, if there's only one,
--- otherwise show all options in Telescope --- otherwise show all options in Telescope
---@param opts table: options to pass to the picker ---@param opts table: options to pass to the picker
---@field timeout number: timeout for the sync call (default: 10000)
---@field jump_type string: how to goto definition if there is only one, values: "tab", "split", "vsplit", "never" ---@field jump_type string: how to goto definition if there is only one, values: "tab", "split", "vsplit", "never"
---@field ignore_filename boolean: dont show filenames (default: true) ---@field ignore_filename boolean: dont show filenames (default: true)
builtin.lsp_type_definitions = require("telescope.builtin.lsp").type_definitions builtin.lsp_type_definitions = require("telescope.builtin.lsp").type_definitions
--- Goto the implementation of the word under the cursor if there's only one, otherwise show all options in Telescope --- Goto the implementation of the word under the cursor if there's only one, otherwise show all options in Telescope
---@param opts table: options to pass to the picker ---@param opts table: options to pass to the picker
---@field timeout number: timeout for the sync call (default: 10000)
---@field jump_type string: how to goto implementation if there is only one, values: "tab", "split", "vsplit", "never" ---@field jump_type string: how to goto implementation if there is only one, values: "tab", "split", "vsplit", "never"
---@field ignore_filename boolean: dont show filenames (default: true) ---@field ignore_filename boolean: dont show filenames (default: true)
builtin.lsp_implementations = require_on_exported_call("telescope.builtin.lsp").implementations builtin.lsp_implementations = require_on_exported_call("telescope.builtin.lsp").implementations
@@ -393,7 +389,6 @@ builtin.lsp_range_code_actions = require_on_exported_call("telescope.builtin.lsp
--- - Default keymaps: --- - Default keymaps:
--- - `<C-l>`: show autocompletion menu to prefilter your query by type of symbol you want to see (i.e. `:variable:`) --- - `<C-l>`: show autocompletion menu to prefilter your query by type of symbol you want to see (i.e. `:variable:`)
---@param opts table: options to pass to the picker ---@param opts table: options to pass to the picker
---@field timeout number: timeout for the sync call (default: 10000)
---@field ignore_filename boolean: dont show filenames (default: true) ---@field ignore_filename boolean: dont show filenames (default: true)
---@field show_line boolean: if true, shows the content of the line the tag is found on (default: false) ---@field show_line boolean: if true, shows the content of the line the tag is found on (default: false)
---@field symbols string|table: filter results by symbol kind(s) ---@field symbols string|table: filter results by symbol kind(s)
@@ -405,7 +400,6 @@ builtin.lsp_document_symbols = require_on_exported_call("telescope.builtin.lsp")
--- - `<C-l>`: show autocompletion menu to prefilter your query by type of symbol you want to see (i.e. `:variable:`) --- - `<C-l>`: show autocompletion menu to prefilter your query by type of symbol you want to see (i.e. `:variable:`)
---@param opts table: options to pass to the picker ---@param opts table: options to pass to the picker
---@field query string: for what to query the workspace (default: "") ---@field query string: for what to query the workspace (default: "")
---@field timeout number: timeout for the sync call (default: 10000)
---@field ignore_filename boolean: dont show filenames (default: false) ---@field ignore_filename boolean: dont show filenames (default: false)
---@field show_line boolean: if true, shows the content of the line the tag is found on (default: false) ---@field show_line boolean: if true, shows the content of the line the tag is found on (default: false)
---@field symbols string|table: filter results by symbol kind(s) ---@field symbols string|table: filter results by symbol kind(s)
@@ -448,6 +442,8 @@ local apply_config = function(mod)
for k, v in pairs(mod) do for k, v in pairs(mod) do
mod[k] = function(opts) mod[k] = function(opts)
opts = opts or {} opts = opts or {}
opts.bufnr = opts.bufnr or vim.api.nvim_get_current_buf()
opts.winnr = opts.winnr or vim.api.nvim_get_current_win()
local pconf = pickers_conf[k] or {} local pconf = pickers_conf[k] or {}
local defaults = (function() local defaults = (function()
if pconf.theme then if pconf.theme then

View File

@@ -62,6 +62,8 @@ internal.builtin = function(opts)
end end
end end
opts.bufnr = vim.api.nvim_get_current_buf()
opts.winnr = vim.api.nvim_get_current_win()
pickers.new(opts, { pickers.new(opts, {
prompt_title = title, prompt_title = title,
finder = finders.new_table { finder = finders.new_table {

View File

@@ -13,10 +13,10 @@ local utils = require "telescope.utils"
local lsp = {} local lsp = {}
lsp.references = function(opts) lsp.references = function(opts)
local params = vim.lsp.util.make_position_params() local params = vim.lsp.util.make_position_params(opts.winnr)
params.context = { includeDeclaration = true } params.context = { includeDeclaration = true }
vim.lsp.buf_request(0, "textDocument/references", params, function(err, result, ctx, _config) vim.lsp.buf_request(opts.bufnr, "textDocument/references", params, function(err, result, ctx, _config)
if err then if err then
vim.api.nvim_err_writeln("Error when finding references: " .. err.message) vim.api.nvim_err_writeln("Error when finding references: " .. err.message)
return return
@@ -49,8 +49,8 @@ end
local function list_or_jump(action, title, opts) local function list_or_jump(action, title, opts)
opts = opts or {} opts = opts or {}
local params = vim.lsp.util.make_position_params() local params = vim.lsp.util.make_position_params(opts.winnr)
vim.lsp.buf_request(0, action, params, function(err, result, ctx, _config) vim.lsp.buf_request(opts.bufnr, action, params, function(err, result, ctx, _config)
if err then if err then
vim.api.nvim_err_writeln("Error when executing " .. action .. " : " .. err.message) vim.api.nvim_err_writeln("Error when executing " .. action .. " : " .. err.message)
return return
@@ -106,9 +106,8 @@ lsp.implementations = function(opts)
end end
lsp.document_symbols = function(opts) lsp.document_symbols = function(opts)
local bufnr = vim.api.nvim_get_current_buf() local params = vim.lsp.util.make_position_params(opts.winnr)
local params = vim.lsp.util.make_position_params() vim.lsp.buf_request(opts.bufnr, "textDocument/documentSymbol", params, function(err, result, _, _)
vim.lsp.buf_request(0, "textDocument/documentSymbol", params, function(err, result, _ctx, _config)
if err then if err then
vim.api.nvim_err_writeln("Error when finding document symbols: " .. err.message) vim.api.nvim_err_writeln("Error when finding document symbols: " .. err.message)
return return
@@ -119,7 +118,7 @@ lsp.document_symbols = function(opts)
return return
end end
local locations = vim.lsp.util.symbols_to_items(result or {}, bufnr) or {} local locations = vim.lsp.util.symbols_to_items(result or {}, opts.bufnr) or {}
locations = utils.filter_symbols(locations, opts) locations = utils.filter_symbols(locations, opts)
if locations == nil then if locations == nil then
-- error message already printed in `utils.filter_symbols` -- error message already printed in `utils.filter_symbols`
@@ -127,6 +126,7 @@ lsp.document_symbols = function(opts)
end end
if vim.tbl_isempty(locations) then if vim.tbl_isempty(locations) then
print "locations table empty"
return return
end end
@@ -147,14 +147,15 @@ lsp.document_symbols = function(opts)
end end
lsp.code_actions = function(opts) lsp.code_actions = function(opts)
local params = vim.F.if_nil(opts.params, vim.lsp.util.make_range_params()) local params = vim.F.if_nil(opts.params, vim.lsp.util.make_range_params(opts.winnr))
local lnum = vim.api.nvim_win_get_cursor(opts.winnr)[1]
params.context = { params.context = {
diagnostics = vim.lsp.diagnostic.get_line_diagnostics(), diagnostics = vim.lsp.diagnostic.get_line_diagnostics(opts.bufnr, lnum - 1),
} }
local results_lsp, err = vim.lsp.buf_request_sync( local results_lsp, err = vim.lsp.buf_request_sync(
0, opts.bufnr,
"textDocument/codeAction", "textDocument/codeAction",
params, params,
vim.F.if_nil(opts.timeout, 10000) vim.F.if_nil(opts.timeout, 10000)
@@ -334,20 +335,19 @@ lsp.code_actions = function(opts)
end end
lsp.range_code_actions = function(opts) lsp.range_code_actions = function(opts)
opts.params = vim.lsp.util.make_given_range_params({ opts.start_line, 1 }, { opts.end_line, 1 }) opts.params = vim.lsp.util.make_given_range_params({ opts.start_line, 1 }, { opts.end_line, 1 }, opts.bufnr)
lsp.code_actions(opts) lsp.code_actions(opts)
end end
lsp.workspace_symbols = function(opts) lsp.workspace_symbols = function(opts)
local bufnr = vim.api.nvim_get_current_buf()
local params = { query = opts.query or "" } local params = { query = opts.query or "" }
vim.lsp.buf_request(0, "workspace/symbol", params, function(err, server_result, _ctx, _config) vim.lsp.buf_request(opts.bufnr, "workspace/symbol", params, function(err, server_result, _, _)
if err then if err then
vim.api.nvim_err_writeln("Error when finding workspace symbols: " .. err.message) vim.api.nvim_err_writeln("Error when finding workspace symbols: " .. err.message)
return return
end end
local locations = vim.lsp.util.symbols_to_items(server_result or {}, bufnr) or {} local locations = vim.lsp.util.symbols_to_items(server_result or {}, opts.bufnr) or {}
locations = utils.filter_symbols(locations, opts) locations = utils.filter_symbols(locations, opts)
if locations == nil then if locations == nil then
-- error message already printed in `utils.filter_symbols` -- error message already printed in `utils.filter_symbols`
@@ -400,21 +400,19 @@ local function get_workspace_symbols_requester(bufnr, opts)
end end
lsp.dynamic_workspace_symbols = function(opts) lsp.dynamic_workspace_symbols = function(opts)
local curr_bufnr = vim.api.nvim_get_current_buf()
pickers.new(opts, { pickers.new(opts, {
prompt_title = "LSP Dynamic Workspace Symbols", prompt_title = "LSP Dynamic Workspace Symbols",
finder = finders.new_dynamic { finder = finders.new_dynamic {
entry_maker = opts.entry_maker or make_entry.gen_from_lsp_symbols(opts), entry_maker = opts.entry_maker or make_entry.gen_from_lsp_symbols(opts),
fn = get_workspace_symbols_requester(curr_bufnr, opts), fn = get_workspace_symbols_requester(opts.bufnr, opts),
}, },
previewer = conf.qflist_previewer(opts), previewer = conf.qflist_previewer(opts),
sorter = conf.generic_sorter(opts), sorter = conf.generic_sorter(opts),
}):find() }):find()
end end
local function check_capabilities(feature) local function check_capabilities(feature, bufnr)
local clients = vim.lsp.buf_get_clients(0) local clients = vim.lsp.buf_get_clients(bufnr)
local supported_client = false local supported_client = false
for _, client in pairs(clients) do for _, client in pairs(clients) do
@@ -452,7 +450,7 @@ local function apply_checks(mod)
opts = opts or {} opts = opts or {}
local feature_name = feature_map[k] local feature_name = feature_map[k]
if feature_name and not check_capabilities(feature_name) then if feature_name and not check_capabilities(feature_name, opts.bufnr) then
return return
end end
v(opts) v(opts)

View File

@@ -536,7 +536,7 @@ function make_entry.gen_from_treesitter(opts)
return function(entry) return function(entry)
local ts_utils = require "nvim-treesitter.ts_utils" local ts_utils = require "nvim-treesitter.ts_utils"
local start_row, start_col, end_row, _ = ts_utils.get_node_range(entry.node) local start_row, start_col, end_row, _ = ts_utils.get_node_range(entry.node)
local node_text = ts_utils.get_node_text(entry.node)[1] local node_text = ts_utils.get_node_text(entry.node, bufnr)[1]
return { return {
valid = true, valid = true,
@@ -906,7 +906,7 @@ function make_entry.gen_from_ctags(opts)
opts = opts or {} opts = opts or {}
local cwd = vim.fn.expand(opts.cwd or vim.loop.cwd()) local cwd = vim.fn.expand(opts.cwd or vim.loop.cwd())
local current_file = Path:new(vim.fn.expand "%"):normalize(cwd) local current_file = Path:new(vim.api.nvim_buf_get_name(opts.bufnr)):normalize(cwd)
local display_items = { local display_items = {
{ remaining = true }, { remaining = true },