chore: use stylua for formatting (#1040)
* chore: stylua job and config * reformat with stylua
This commit is contained in:
@@ -1,38 +1,38 @@
|
||||
local a = vim.api
|
||||
local popup = require('popup')
|
||||
local popup = require "popup"
|
||||
|
||||
local async_lib = require('plenary.async_lib')
|
||||
local async_lib = require "plenary.async_lib"
|
||||
local async_util = async_lib.util
|
||||
|
||||
local async = async_lib.async
|
||||
local await = async_lib.await
|
||||
local channel = async_util.channel
|
||||
|
||||
require('telescope')
|
||||
require "telescope"
|
||||
|
||||
local actions = require('telescope.actions')
|
||||
local action_set = require('telescope.actions.set')
|
||||
local config = require('telescope.config')
|
||||
local debounce = require('telescope.debounce')
|
||||
local deprecated = require('telescope.deprecated')
|
||||
local log = require('telescope.log')
|
||||
local mappings = require('telescope.mappings')
|
||||
local state = require('telescope.state')
|
||||
local utils = require('telescope.utils')
|
||||
local actions = require "telescope.actions"
|
||||
local action_set = require "telescope.actions.set"
|
||||
local config = require "telescope.config"
|
||||
local debounce = require "telescope.debounce"
|
||||
local deprecated = require "telescope.deprecated"
|
||||
local log = require "telescope.log"
|
||||
local mappings = require "telescope.mappings"
|
||||
local state = require "telescope.state"
|
||||
local utils = require "telescope.utils"
|
||||
|
||||
local entry_display = require('telescope.pickers.entry_display')
|
||||
local p_highlighter = require('telescope.pickers.highlights')
|
||||
local p_scroller = require('telescope.pickers.scroller')
|
||||
local p_window = require('telescope.pickers.window')
|
||||
local entry_display = require "telescope.pickers.entry_display"
|
||||
local p_highlighter = require "telescope.pickers.highlights"
|
||||
local p_scroller = require "telescope.pickers.scroller"
|
||||
local p_window = require "telescope.pickers.window"
|
||||
|
||||
local EntryManager = require('telescope.entry_manager')
|
||||
local MultiSelect = require('telescope.pickers.multi')
|
||||
local EntryManager = require "telescope.entry_manager"
|
||||
local MultiSelect = require "telescope.pickers.multi"
|
||||
|
||||
local get_default = utils.get_default
|
||||
|
||||
local ns_telescope_matching = a.nvim_create_namespace('telescope_matching')
|
||||
local ns_telescope_prompt = a.nvim_create_namespace('telescope_prompt')
|
||||
local ns_telescope_prompt_prefix = a.nvim_create_namespace('telescope_prompt_prefix')
|
||||
local ns_telescope_matching = a.nvim_create_namespace "telescope_matching"
|
||||
local ns_telescope_prompt = a.nvim_create_namespace "telescope_prompt"
|
||||
local ns_telescope_prompt_prefix = a.nvim_create_namespace "telescope_prompt_prefix"
|
||||
|
||||
local pickers = {}
|
||||
|
||||
@@ -48,7 +48,7 @@ function Picker:new(opts)
|
||||
opts = opts or {}
|
||||
|
||||
if opts.layout_strategy and opts.get_window_options then
|
||||
error("layout_strategy and get_window_options are not compatible keys")
|
||||
error "layout_strategy and get_window_options are not compatible keys"
|
||||
end
|
||||
|
||||
-- Reset actions for any replaced / enhanced actions.
|
||||
@@ -76,7 +76,7 @@ function Picker:new(opts)
|
||||
_on_input_filter_cb = opts.on_input_filter_cb or function() end,
|
||||
|
||||
finder = opts.finder,
|
||||
sorter = opts.sorter or require('telescope.sorters').empty(),
|
||||
sorter = opts.sorter or require("telescope.sorters").empty(),
|
||||
|
||||
all_previewers = opts.previewer,
|
||||
current_previewer_index = 1,
|
||||
@@ -141,7 +141,7 @@ end
|
||||
---@param index number: The row being displayed
|
||||
---@return number The row for the picker to display in
|
||||
function Picker:get_row(index)
|
||||
if self.sorting_strategy == 'ascending' then
|
||||
if self.sorting_strategy == "ascending" then
|
||||
return index - 1
|
||||
else
|
||||
return self.max_results - index
|
||||
@@ -153,7 +153,7 @@ end
|
||||
---@param row number: The row being displayed
|
||||
---@return number The index in line_manager
|
||||
function Picker:get_index(row)
|
||||
if self.sorting_strategy == 'ascending' then
|
||||
if self.sorting_strategy == "ascending" then
|
||||
return row + 1
|
||||
else
|
||||
return self.max_results - row
|
||||
@@ -161,7 +161,7 @@ function Picker:get_index(row)
|
||||
end
|
||||
|
||||
function Picker:get_reset_row()
|
||||
if self.sorting_strategy == 'ascending' then
|
||||
if self.sorting_strategy == "ascending" then
|
||||
return 0
|
||||
else
|
||||
return self.max_results - 1
|
||||
@@ -169,12 +169,14 @@ function Picker:get_reset_row()
|
||||
end
|
||||
|
||||
function Picker:is_done()
|
||||
if not self.manager then return true end
|
||||
if not self.manager then
|
||||
return true
|
||||
end
|
||||
end
|
||||
|
||||
function Picker:clear_extra_rows(results_bufnr)
|
||||
if self:is_done() then
|
||||
log.trace("Not clearing due to being already complete")
|
||||
log.trace "Not clearing due to being already complete"
|
||||
return
|
||||
end
|
||||
|
||||
@@ -184,7 +186,7 @@ function Picker:clear_extra_rows(results_bufnr)
|
||||
end
|
||||
|
||||
local worst_line, ok, msg
|
||||
if self.sorting_strategy == 'ascending' then
|
||||
if self.sorting_strategy == "ascending" then
|
||||
local num_results = self.manager:num_results()
|
||||
worst_line = self.max_results - num_results
|
||||
|
||||
@@ -231,28 +233,21 @@ function Picker:highlight_one_row(results_bufnr, prompt, display, row)
|
||||
if highlights then
|
||||
for _, hl in ipairs(highlights) do
|
||||
local highlight, start, finish
|
||||
if type(hl) == 'table' then
|
||||
highlight = hl.highlight or 'TelescopeMatching'
|
||||
if type(hl) == "table" then
|
||||
highlight = hl.highlight or "TelescopeMatching"
|
||||
start = hl.start
|
||||
finish = hl.finish or hl.start
|
||||
elseif type(hl) == 'number' then
|
||||
highlight = 'TelescopeMatching'
|
||||
elseif type(hl) == "number" then
|
||||
highlight = "TelescopeMatching"
|
||||
start = hl
|
||||
finish = hl
|
||||
else
|
||||
error('Invalid higlighter fn')
|
||||
error "Invalid higlighter fn"
|
||||
end
|
||||
|
||||
self:_increment('highlights')
|
||||
self:_increment "highlights"
|
||||
|
||||
vim.api.nvim_buf_add_highlight(
|
||||
results_bufnr,
|
||||
ns_telescope_matching,
|
||||
highlight,
|
||||
row,
|
||||
start - 1,
|
||||
finish
|
||||
)
|
||||
vim.api.nvim_buf_add_highlight(results_bufnr, ns_telescope_matching, highlight, row, start - 1, finish)
|
||||
end
|
||||
end
|
||||
|
||||
@@ -261,7 +256,7 @@ function Picker:highlight_one_row(results_bufnr, prompt, display, row)
|
||||
end
|
||||
|
||||
function Picker:can_select_row(row)
|
||||
if self.sorting_strategy == 'ascending' then
|
||||
if self.sorting_strategy == "ascending" then
|
||||
return row <= self.manager:num_results()
|
||||
else
|
||||
return row >= 0 and row <= self.max_results and row >= self.max_results - self.manager:num_results()
|
||||
@@ -304,47 +299,48 @@ function Picker:find()
|
||||
popup_opts.preview.minheight = popup_opts.preview.height
|
||||
end
|
||||
|
||||
local results_win, results_opts = popup.create('', popup_opts.results)
|
||||
local results_win, results_opts = popup.create("", popup_opts.results)
|
||||
local results_bufnr = a.nvim_win_get_buf(results_win)
|
||||
|
||||
self.results_bufnr = results_bufnr
|
||||
self.results_win = results_win
|
||||
|
||||
-- TODO: Should probably always show all the line for results win, so should implement a resize for the windows
|
||||
a.nvim_win_set_option(results_win, 'wrap', false)
|
||||
a.nvim_win_set_option(results_win, 'winhl', 'Normal:TelescopeNormal')
|
||||
a.nvim_win_set_option(results_win, 'winblend', self.window.winblend)
|
||||
a.nvim_win_set_option(results_win, "wrap", false)
|
||||
a.nvim_win_set_option(results_win, "winhl", "Normal:TelescopeNormal")
|
||||
a.nvim_win_set_option(results_win, "winblend", self.window.winblend)
|
||||
local results_border_win = results_opts.border and results_opts.border.win_id
|
||||
if results_border_win then
|
||||
vim.api.nvim_win_set_option(results_border_win, 'winhl', 'Normal:TelescopeResultsBorder')
|
||||
vim.api.nvim_win_set_option(results_border_win, "winhl", "Normal:TelescopeResultsBorder")
|
||||
end
|
||||
|
||||
|
||||
local preview_win, preview_opts, preview_bufnr
|
||||
if popup_opts.preview then
|
||||
preview_win, preview_opts = popup.create('', popup_opts.preview)
|
||||
preview_win, preview_opts = popup.create("", popup_opts.preview)
|
||||
preview_bufnr = a.nvim_win_get_buf(preview_win)
|
||||
|
||||
a.nvim_win_set_option(preview_win, 'winhl', 'Normal:TelescopePreviewNormal')
|
||||
a.nvim_win_set_option(preview_win, 'winblend', self.window.winblend)
|
||||
a.nvim_win_set_option(preview_win, "winhl", "Normal:TelescopePreviewNormal")
|
||||
a.nvim_win_set_option(preview_win, "winblend", self.window.winblend)
|
||||
local preview_border_win = preview_opts and preview_opts.border and preview_opts.border.win_id
|
||||
if preview_border_win then
|
||||
vim.api.nvim_win_set_option(preview_border_win, 'winhl', 'Normal:TelescopePreviewBorder')
|
||||
vim.api.nvim_win_set_option(preview_border_win, "winhl", "Normal:TelescopePreviewBorder")
|
||||
end
|
||||
end
|
||||
|
||||
-- TODO: We need to center this and make it prettier...
|
||||
local prompt_win, prompt_opts = popup.create('', popup_opts.prompt)
|
||||
local prompt_win, prompt_opts = popup.create("", popup_opts.prompt)
|
||||
local prompt_bufnr = a.nvim_win_get_buf(prompt_win)
|
||||
a.nvim_win_set_option(prompt_win, 'winhl', 'Normal:TelescopeNormal')
|
||||
a.nvim_win_set_option(prompt_win, 'winblend', self.window.winblend)
|
||||
a.nvim_win_set_option(prompt_win, "winhl", "Normal:TelescopeNormal")
|
||||
a.nvim_win_set_option(prompt_win, "winblend", self.window.winblend)
|
||||
local prompt_border_win = prompt_opts.border and prompt_opts.border.win_id
|
||||
if prompt_border_win then vim.api.nvim_win_set_option(prompt_border_win, 'winhl', 'Normal:TelescopePromptBorder') end
|
||||
if prompt_border_win then
|
||||
vim.api.nvim_win_set_option(prompt_border_win, "winhl", "Normal:TelescopePromptBorder")
|
||||
end
|
||||
|
||||
-- Prompt prefix
|
||||
local prompt_prefix = self.prompt_prefix
|
||||
if prompt_prefix ~= '' then
|
||||
a.nvim_buf_set_option(prompt_bufnr, 'buftype', 'prompt')
|
||||
if prompt_prefix ~= "" then
|
||||
a.nvim_buf_set_option(prompt_bufnr, "buftype", "prompt")
|
||||
vim.fn.prompt_setprompt(prompt_bufnr, prompt_prefix)
|
||||
end
|
||||
self.prompt_prefix = prompt_prefix
|
||||
@@ -377,8 +373,12 @@ function Picker:find()
|
||||
return
|
||||
end
|
||||
|
||||
if not first_line then first_line = 0 end
|
||||
if not last_line then last_line = 1 end
|
||||
if not first_line then
|
||||
first_line = 0
|
||||
end
|
||||
if not last_line then
|
||||
last_line = 1
|
||||
end
|
||||
|
||||
if first_line > 0 or last_line > 1 then
|
||||
log.debug("ON_LINES: Bad range", first_line, last_line, self:_get_prompt())
|
||||
@@ -396,7 +396,9 @@ function Picker:find()
|
||||
self.finder = finder
|
||||
end
|
||||
|
||||
if self.sorter then self.sorter:_start(prompt) end
|
||||
if self.sorter then
|
||||
self.sorter:_start(prompt)
|
||||
end
|
||||
|
||||
-- TODO: Entry manager should have a "bulk" setter. This can prevent a lot of redraws from display
|
||||
self.manager = EntryManager:new(self.max_results, self.entry_adder, self.stats)
|
||||
@@ -428,23 +430,27 @@ function Picker:find()
|
||||
self.closed = true
|
||||
|
||||
-- TODO: Should we actually do this?
|
||||
collectgarbage(); collectgarbage()
|
||||
collectgarbage()
|
||||
collectgarbage()
|
||||
end,
|
||||
})
|
||||
|
||||
if self.sorter then self.sorter:_init() end
|
||||
if self.sorter then
|
||||
self.sorter:_init()
|
||||
end
|
||||
async_lib.run(main_loop())
|
||||
status_updater()
|
||||
|
||||
-- TODO: Use WinLeave as well?
|
||||
local on_buf_leave = string.format(
|
||||
[[ autocmd BufLeave <buffer> ++nested ++once :silent lua require('telescope.pickers').on_close_prompt(%s)]],
|
||||
prompt_bufnr)
|
||||
prompt_bufnr
|
||||
)
|
||||
|
||||
vim.cmd([[augroup PickerInsert]])
|
||||
vim.cmd([[ au!]])
|
||||
vim.cmd( on_buf_leave)
|
||||
vim.cmd([[augroup END]])
|
||||
vim.cmd [[augroup PickerInsert]]
|
||||
vim.cmd [[ au!]]
|
||||
vim.cmd(on_buf_leave)
|
||||
vim.cmd [[augroup END]]
|
||||
|
||||
self.prompt_bufnr = prompt_bufnr
|
||||
|
||||
@@ -452,26 +458,31 @@ function Picker:find()
|
||||
self.preview_border = preview_border
|
||||
local preview_border_win = (preview_border and preview_border.win_id) and preview_border.win_id
|
||||
|
||||
state.set_status(prompt_bufnr, setmetatable({
|
||||
prompt_bufnr = prompt_bufnr,
|
||||
prompt_win = prompt_win,
|
||||
prompt_border_win = prompt_border_win,
|
||||
state.set_status(
|
||||
prompt_bufnr,
|
||||
setmetatable({
|
||||
prompt_bufnr = prompt_bufnr,
|
||||
prompt_win = prompt_win,
|
||||
prompt_border_win = prompt_border_win,
|
||||
|
||||
results_bufnr = results_bufnr,
|
||||
results_win = results_win,
|
||||
results_border_win = results_border_win,
|
||||
results_bufnr = results_bufnr,
|
||||
results_win = results_win,
|
||||
results_border_win = results_border_win,
|
||||
|
||||
preview_bufnr = preview_bufnr,
|
||||
preview_win = preview_win,
|
||||
preview_border_win = preview_border_win,
|
||||
picker = self,
|
||||
}, { __mode = 'kv' }))
|
||||
preview_bufnr = preview_bufnr,
|
||||
preview_win = preview_win,
|
||||
preview_border_win = preview_border_win,
|
||||
picker = self,
|
||||
}, {
|
||||
__mode = "kv",
|
||||
})
|
||||
)
|
||||
|
||||
mappings.apply_keymap(prompt_bufnr, self.attach_mappings, config.values.mappings)
|
||||
|
||||
-- Do filetype last, so that users can register at the last second.
|
||||
pcall(a.nvim_buf_set_option, prompt_bufnr, 'filetype', 'TelescopePrompt')
|
||||
pcall(a.nvim_buf_set_option, results_bufnr, 'filetype', 'TelescopeResults')
|
||||
pcall(a.nvim_buf_set_option, prompt_bufnr, "filetype", "TelescopePrompt")
|
||||
pcall(a.nvim_buf_set_option, results_bufnr, "filetype", "TelescopeResults")
|
||||
|
||||
if self.default_text then
|
||||
self:set_prompt(self.default_text)
|
||||
@@ -529,7 +540,9 @@ function Picker:delete_selection(delete_cb)
|
||||
|
||||
-- Sort in reverse order as removing an entry from the table shifts down the
|
||||
-- other elements to close the hole.
|
||||
table.sort(selection_index, function(x, y) return x > y end)
|
||||
table.sort(selection_index, function(x, y)
|
||||
return x > y
|
||||
end)
|
||||
for _, index in ipairs(selection_index) do
|
||||
local selection = table.remove(self.finder.results, index)
|
||||
delete_cb(selection)
|
||||
@@ -548,7 +561,7 @@ end
|
||||
function Picker:set_prompt(str)
|
||||
-- TODO(conni2461): As soon as prompt_buffers are fix use this:
|
||||
-- vim.api.nvim_buf_set_lines(self.prompt_bufnr, 0, 1, false, { str })
|
||||
vim.api.nvim_feedkeys(str, 'n', false)
|
||||
vim.api.nvim_feedkeys(str, "n", false)
|
||||
end
|
||||
|
||||
function Picker.close_windows(status)
|
||||
@@ -566,9 +579,7 @@ function Picker.close_windows(status)
|
||||
end
|
||||
|
||||
local bufnr = vim.api.nvim_win_get_buf(win_id)
|
||||
if bdelete
|
||||
and vim.api.nvim_buf_is_valid(bufnr)
|
||||
and not vim.api.nvim_buf_get_option(bufnr, 'buflisted') then
|
||||
if bdelete and vim.api.nvim_buf_is_valid(bufnr) and not vim.api.nvim_buf_get_option(bufnr, "buflisted") then
|
||||
vim.cmd(string.format("silent! bdelete! %s", bufnr))
|
||||
end
|
||||
|
||||
@@ -650,10 +661,11 @@ end
|
||||
function Picker:_reset_prefix_color(hl_group)
|
||||
self._current_prefix_hl_group = hl_group or nil
|
||||
|
||||
if self.prompt_prefix ~= '' then
|
||||
vim.api.nvim_buf_add_highlight(self.prompt_bufnr,
|
||||
if self.prompt_prefix ~= "" then
|
||||
vim.api.nvim_buf_add_highlight(
|
||||
self.prompt_bufnr,
|
||||
ns_telescope_prompt_prefix,
|
||||
self._current_prefix_hl_group or 'TelescopePromptPrefix',
|
||||
self._current_prefix_hl_group or "TelescopePromptPrefix",
|
||||
0,
|
||||
0,
|
||||
#self.prompt_prefix
|
||||
@@ -665,26 +677,28 @@ end
|
||||
-- TODO(conni2461): Next two functions only work together otherwise color doesn't work
|
||||
-- Probably a issue with prompt buffers
|
||||
function Picker:change_prompt_prefix(new_prefix, hl_group)
|
||||
if not new_prefix then return end
|
||||
if not new_prefix then
|
||||
return
|
||||
end
|
||||
|
||||
if new_prefix ~= '' then
|
||||
if new_prefix ~= "" then
|
||||
vim.fn.prompt_setprompt(self.prompt_bufnr, new_prefix)
|
||||
else
|
||||
vim.api.nvim_buf_set_text(self.prompt_bufnr, 0, 0, 0, #self.prompt_prefix, {})
|
||||
vim.api.nvim_buf_set_option(self.prompt_bufnr, 'buftype', '')
|
||||
vim.api.nvim_buf_set_option(self.prompt_bufnr, "buftype", "")
|
||||
end
|
||||
self.prompt_prefix = new_prefix
|
||||
self:_reset_prefix_color(hl_group)
|
||||
end
|
||||
|
||||
function Picker:reset_prompt(text)
|
||||
local prompt_text = self.prompt_prefix .. (text or '')
|
||||
local prompt_text = self.prompt_prefix .. (text or "")
|
||||
vim.api.nvim_buf_set_lines(self.prompt_bufnr, 0, -1, false, { prompt_text })
|
||||
|
||||
self:_reset_prefix_color(self._current_prefix_hl_group)
|
||||
|
||||
if text then
|
||||
vim.api.nvim_win_set_cursor(self.prompt_win, {1, #prompt_text})
|
||||
vim.api.nvim_win_set_cursor(self.prompt_win, { 1, #prompt_text })
|
||||
end
|
||||
end
|
||||
|
||||
@@ -693,10 +707,14 @@ end
|
||||
function Picker:refresh(finder, opts)
|
||||
opts = opts or {}
|
||||
if opts.new_prefix then
|
||||
local handle = type(opts.new_prefix) == 'table' and unpack or function(x) return x end
|
||||
local handle = type(opts.new_prefix) == "table" and unpack or function(x)
|
||||
return x
|
||||
end
|
||||
self:change_prompt_prefix(handle(opts.new_prefix))
|
||||
end
|
||||
if opts.reset_prompt then self:reset_prompt() end
|
||||
if opts.reset_prompt then
|
||||
self:reset_prompt()
|
||||
end
|
||||
|
||||
if finder then
|
||||
self.finder:close()
|
||||
@@ -708,7 +726,9 @@ function Picker:refresh(finder, opts)
|
||||
end
|
||||
|
||||
function Picker:set_selection(row)
|
||||
if not self.manager then return end
|
||||
if not self.manager then
|
||||
return
|
||||
end
|
||||
|
||||
row = self.scroller(self.max_results, self.manager:num_results(), row)
|
||||
|
||||
@@ -729,11 +749,9 @@ function Picker:set_selection(row)
|
||||
end
|
||||
|
||||
if row > a.nvim_buf_line_count(results_bufnr) then
|
||||
log.debug(string.format(
|
||||
"Should not be possible to get row this large %s %s",
|
||||
row,
|
||||
a.nvim_buf_line_count(results_bufnr)
|
||||
))
|
||||
log.debug(
|
||||
string.format("Should not be possible to get row this large %s %s", row, a.nvim_buf_line_count(results_bufnr))
|
||||
)
|
||||
|
||||
return
|
||||
end
|
||||
@@ -741,7 +759,9 @@ function Picker:set_selection(row)
|
||||
local entry = self.manager:get_entry(self:get_index(row))
|
||||
state.set_global_key("selected_entry", entry)
|
||||
|
||||
if not entry then return end
|
||||
if not entry then
|
||||
return
|
||||
end
|
||||
|
||||
-- TODO: Probably should figure out what the rows are that made this happen...
|
||||
-- Probably something with setting a row that's too high for this?
|
||||
@@ -754,14 +774,13 @@ function Picker:set_selection(row)
|
||||
-- Only change the first couple characters, nvim_buf_set_text leaves the existing highlights
|
||||
a.nvim_buf_set_text(
|
||||
results_bufnr,
|
||||
self._selection_row, 0,
|
||||
self._selection_row, #self.selection_caret,
|
||||
self._selection_row,
|
||||
0,
|
||||
self._selection_row,
|
||||
#self.selection_caret,
|
||||
{ self.entry_prefix }
|
||||
)
|
||||
self.highlighter:hi_multiselect(
|
||||
self._selection_row,
|
||||
self:is_multi_selected(self._selection_entry)
|
||||
)
|
||||
self.highlighter:hi_multiselect(self._selection_row, self:is_multi_selected(self._selection_entry))
|
||||
|
||||
-- local display = a.nvim_buf_get_lines(results_bufnr, old_row, old_row + 1, false)[1]
|
||||
-- display = ' ' .. display
|
||||
@@ -781,10 +800,10 @@ function Picker:set_selection(row)
|
||||
-- TODO: You should go back and redraw the highlights for this line from the sorter.
|
||||
-- That's the only smart thing to do.
|
||||
if not a.nvim_buf_is_valid(results_bufnr) then
|
||||
log.debug("Invalid buf somehow...")
|
||||
log.debug "Invalid buf somehow..."
|
||||
return
|
||||
end
|
||||
a.nvim_buf_set_lines(results_bufnr, row, row + 1, false, {display})
|
||||
a.nvim_buf_set_lines(results_bufnr, row, row + 1, false, { display })
|
||||
|
||||
-- don't highlight the ' ' at the end of caret
|
||||
self.highlighter:hi_selection(row, caret:sub(1, -2))
|
||||
@@ -813,12 +832,9 @@ end
|
||||
function Picker:refresh_previewer()
|
||||
local status = state.get_status(self.prompt_bufnr)
|
||||
if status.preview_win and self.previewer then
|
||||
self:_increment("previewed")
|
||||
self:_increment "previewed"
|
||||
|
||||
self.previewer:preview(
|
||||
self._selection_entry,
|
||||
status
|
||||
)
|
||||
self.previewer:preview(self._selection_entry, status)
|
||||
if self.preview_border then
|
||||
if config.values.dynamic_preview_title == true then
|
||||
self.preview_border:change_title(self.previewer:dynamic_title(self._selection_entry))
|
||||
@@ -831,7 +847,9 @@ end
|
||||
|
||||
function Picker:cycle_previewers(next)
|
||||
local size = #self.all_previewers
|
||||
if size == 1 then return end
|
||||
if size == 1 then
|
||||
return
|
||||
end
|
||||
|
||||
self.current_previewer_index = self.current_previewer_index + next
|
||||
if self.current_previewer_index > size then
|
||||
@@ -845,7 +863,9 @@ function Picker:cycle_previewers(next)
|
||||
end
|
||||
|
||||
function Picker:entry_adder(index, entry, _, insert)
|
||||
if not entry then return end
|
||||
if not entry then
|
||||
return
|
||||
end
|
||||
|
||||
local row = self:get_row(index)
|
||||
|
||||
@@ -867,13 +887,13 @@ function Picker:entry_adder(index, entry, _, insert)
|
||||
local prefix = self.entry_prefix
|
||||
display = prefix .. display
|
||||
|
||||
self:_increment("displayed")
|
||||
self:_increment "displayed"
|
||||
|
||||
-- TODO: Don't need to schedule this if we schedule the adder.
|
||||
local offset = insert and 0 or 1
|
||||
vim.schedule(function()
|
||||
if not vim.api.nvim_buf_is_valid(self.results_bufnr) then
|
||||
log.debug("ON_ENTRY: Invalid buffer")
|
||||
log.debug "ON_ENTRY: Invalid buffer"
|
||||
return
|
||||
end
|
||||
|
||||
@@ -885,12 +905,12 @@ function Picker:entry_adder(index, entry, _, insert)
|
||||
end
|
||||
|
||||
if insert then
|
||||
if self.sorting_strategy == 'descending' then
|
||||
if self.sorting_strategy == "descending" then
|
||||
vim.api.nvim_buf_set_lines(self.results_bufnr, 0, 1, false, {})
|
||||
end
|
||||
end
|
||||
|
||||
local set_ok, msg = pcall(vim.api.nvim_buf_set_lines, self.results_bufnr, row, row + offset, false, {display})
|
||||
local set_ok, msg = pcall(vim.api.nvim_buf_set_lines, self.results_bufnr, row, row + offset, false, { display })
|
||||
if set_ok and display_highlights then
|
||||
self.highlighter:hi_display(row, prefix, display_highlights)
|
||||
end
|
||||
@@ -901,14 +921,13 @@ function Picker:entry_adder(index, entry, _, insert)
|
||||
|
||||
-- This pretty much only fails when people leave newlines in their results.
|
||||
-- So we'll clean it up for them if it fails.
|
||||
if not set_ok and display:find("\n") then
|
||||
if not set_ok and display:find "\n" then
|
||||
display = display:gsub("\n", " | ")
|
||||
vim.api.nvim_buf_set_lines(self.results_bufnr, row, row + 1, false, {display})
|
||||
vim.api.nvim_buf_set_lines(self.results_bufnr, row, row + 1, false, { display })
|
||||
end
|
||||
end)
|
||||
end
|
||||
|
||||
|
||||
function Picker:_reset_track()
|
||||
self.stats.processed = 0
|
||||
self.stats.displayed = 0
|
||||
@@ -945,7 +964,6 @@ function Picker:_decrement(key)
|
||||
self.stats[key] = (self.stats[key] or 0) - 1
|
||||
end
|
||||
|
||||
|
||||
-- TODO: Decide how much we want to use this.
|
||||
-- Would allow for better debugging of items.
|
||||
function Picker:register_completion_callback(cb)
|
||||
@@ -967,7 +985,9 @@ end
|
||||
function Picker:get_status_updater(prompt_win, prompt_bufnr)
|
||||
return function()
|
||||
local text = self:get_status_text()
|
||||
if self.closed or not vim.api.nvim_buf_is_valid(prompt_bufnr) then return end
|
||||
if self.closed or not vim.api.nvim_buf_is_valid(prompt_bufnr) then
|
||||
return
|
||||
end
|
||||
local current_prompt = vim.api.nvim_buf_get_lines(prompt_bufnr, 0, 1, false)[1]
|
||||
if not current_prompt then
|
||||
return
|
||||
@@ -981,13 +1001,7 @@ function Picker:get_status_updater(prompt_win, prompt_bufnr)
|
||||
|
||||
local padding = string.rep(" ", vim.api.nvim_win_get_width(prompt_win) - prompt_len - #text - 3)
|
||||
vim.api.nvim_buf_clear_namespace(prompt_bufnr, ns_telescope_prompt, 0, 1)
|
||||
vim.api.nvim_buf_set_virtual_text(
|
||||
prompt_bufnr,
|
||||
ns_telescope_prompt,
|
||||
0,
|
||||
{ {padding .. text, "NonText"} },
|
||||
{}
|
||||
)
|
||||
vim.api.nvim_buf_set_virtual_text(prompt_bufnr, ns_telescope_prompt, 0, { { padding .. text, "NonText" } }, {})
|
||||
|
||||
-- TODO: Wait for bfredl
|
||||
-- vim.api.nvim_buf_set_extmark(prompt_bufnr, ns_telescope_prompt, 0, 0, {
|
||||
@@ -997,11 +1011,10 @@ function Picker:get_status_updater(prompt_win, prompt_bufnr)
|
||||
-- virt_text_pos = "eol",
|
||||
-- })
|
||||
|
||||
self:_increment("status")
|
||||
self:_increment "status"
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
function Picker:get_result_processor(find_id, prompt, status_updater)
|
||||
local cb_add = function(score, entry)
|
||||
self.manager:add_entry(self, score, entry)
|
||||
@@ -1009,17 +1022,15 @@ function Picker:get_result_processor(find_id, prompt, status_updater)
|
||||
end
|
||||
|
||||
local cb_filter = function(_)
|
||||
self:_increment("filtered")
|
||||
self:_increment "filtered"
|
||||
end
|
||||
|
||||
return function(entry)
|
||||
if find_id ~= self._find_id
|
||||
or self.closed
|
||||
or self:is_done() then
|
||||
if find_id ~= self._find_id or self.closed or self:is_done() then
|
||||
return true
|
||||
end
|
||||
|
||||
self:_increment("processed")
|
||||
self:_increment "processed"
|
||||
|
||||
if not entry or entry.valid == false then
|
||||
return
|
||||
@@ -1029,11 +1040,11 @@ function Picker:get_result_processor(find_id, prompt, status_updater)
|
||||
-- a ton of time on large results.
|
||||
log.trace("Processing result... ", entry)
|
||||
for _, v in ipairs(self.file_ignore_patterns or {}) do
|
||||
local file = type(entry.value) == 'string' and entry.value or entry.filename
|
||||
local file = type(entry.value) == "string" and entry.value or entry.filename
|
||||
if file then
|
||||
if string.find(file, v) then
|
||||
log.trace("SKIPPING", entry.value, "because", v)
|
||||
self:_decrement("processed")
|
||||
self:_decrement "processed"
|
||||
return
|
||||
end
|
||||
end
|
||||
@@ -1045,18 +1056,20 @@ end
|
||||
|
||||
function Picker:get_result_completor(results_bufnr, find_id, prompt, status_updater)
|
||||
return function()
|
||||
if self.closed == true or self:is_done() then return end
|
||||
if self.closed == true or self:is_done() then
|
||||
return
|
||||
end
|
||||
|
||||
local selection_strategy = self.selection_strategy or 'reset'
|
||||
local selection_strategy = self.selection_strategy or "reset"
|
||||
|
||||
-- TODO: Either: always leave one result or make sure we actually clean up the results when nothing matches
|
||||
if selection_strategy == 'row' then
|
||||
if selection_strategy == "row" then
|
||||
if self._selection_row == nil and self.default_selection_index ~= nil then
|
||||
self:set_selection(self:get_row(self.default_selection_index))
|
||||
else
|
||||
self:set_selection(self:get_selection_row())
|
||||
end
|
||||
elseif selection_strategy == 'follow' then
|
||||
elseif selection_strategy == "follow" then
|
||||
if self._selection_row == nil and self.default_selection_index ~= nil then
|
||||
self:set_selection(self:get_row(self.default_selection_index))
|
||||
else
|
||||
@@ -1069,53 +1082,53 @@ function Picker:get_result_completor(results_bufnr, find_id, prompt, status_upda
|
||||
self:set_selection(self:get_reset_row())
|
||||
end
|
||||
end
|
||||
elseif selection_strategy == 'reset' then
|
||||
elseif selection_strategy == "reset" then
|
||||
if self.default_selection_index ~= nil then
|
||||
self:set_selection(self:get_row(self.default_selection_index))
|
||||
else
|
||||
self:set_selection(self:get_reset_row())
|
||||
end
|
||||
elseif selection_strategy == 'closest' then
|
||||
elseif selection_strategy == "closest" then
|
||||
if prompt == "" and self.default_selection_index ~= nil then
|
||||
self:set_selection(self:get_row(self.default_selection_index))
|
||||
else
|
||||
self:set_selection(self:get_reset_row())
|
||||
end
|
||||
else
|
||||
error('Unknown selection strategy: ' .. selection_strategy)
|
||||
error("Unknown selection strategy: " .. selection_strategy)
|
||||
end
|
||||
|
||||
local current_line = vim.api.nvim_get_current_line():sub(self.prompt_prefix:len() + 1)
|
||||
state.set_global_key('current_line', current_line)
|
||||
state.set_global_key("current_line", current_line)
|
||||
|
||||
status_updater()
|
||||
|
||||
self:clear_extra_rows(results_bufnr)
|
||||
self:highlight_displayed_rows(results_bufnr, prompt)
|
||||
if self.sorter then self.sorter:_finish(prompt) end
|
||||
if self.sorter then
|
||||
self.sorter:_finish(prompt)
|
||||
end
|
||||
|
||||
self:_on_complete()
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
|
||||
pickers.new = function(opts, defaults)
|
||||
local result = {}
|
||||
|
||||
for k, v in pairs(opts or {}) do
|
||||
assert(type(k) == 'string', "Should be string, opts")
|
||||
assert(type(k) == "string", "Should be string, opts")
|
||||
result[k] = v
|
||||
end
|
||||
|
||||
for k, v in pairs(defaults or {}) do
|
||||
if result[k] == nil then
|
||||
assert(type(k) == 'string', "Should be string, defaults")
|
||||
assert(type(k) == "string", "Should be string, defaults")
|
||||
result[k] = v
|
||||
else
|
||||
-- For attach mappings, we want people to be able to pass in another function
|
||||
-- and apply their mappings after we've applied our defaults.
|
||||
if k == 'attach_mappings' then
|
||||
if k == "attach_mappings" then
|
||||
local opt_value = result[k]
|
||||
result[k] = function(...)
|
||||
v(...)
|
||||
@@ -1154,5 +1167,4 @@ end
|
||||
|
||||
pickers._Picker = Picker
|
||||
|
||||
|
||||
return pickers
|
||||
|
||||
Reference in New Issue
Block a user