dotfiles/nvim/lua/tel.lua

local telescope = require('telescope')

telescope.setup({
	defaults = {
		file_ignore_patterns = {'.git'},
		mappings = {
			i = {
				['<C-o>'] = { '<Esc>', type = 'command' },
				['<Esc>'] = 'close',

				['<C-s>'] = 'select_horizontal',
			},
			n = {
				q = 'close',

				['<C-o>'] = { '<Nop>', type = 'command' },

				['<C-s>'] = 'select_horizontal',
			},
		},
		sorting_strategy = 'ascending',
		layout_config = {
			prompt_position = 'top',
		},
		prompt_prefix = '❯ ',
		selection_caret = '❯ ',
		multi_icon = '●',
	},
	pickers = {
		buffers = {
			mappings = {
				n = {
					dd = 'delete_buffer',
				},
			},
		},
		find_files = {
			hidden = true,
		},
		lsp_code_actions = {
			initial_mode = 'normal',
		},
		lsp_range_code_actions = {
			initial_mode = 'normal',
		},
	},
})

telescope.load_extension('fzf')

-- taken from https://github.com/nvim-telescope/telescope-ui-select.nvim for code actions
telescope.register_extension {
	setup = function(topts)
		local specific_opts = vim.F.if_nil(topts.specific_opts, {})
		topts.specific_opts = nil

		if #topts == 1 and topts[1] ~= nil then
			topts = topts[1]
		end

		local pickers = require "telescope.pickers"
		local finders = require "telescope.finders"
		local conf = require("telescope.config").values
		local actions = require "telescope.actions"
		local action_state = require "telescope.actions.state"
		local strings = require "plenary.strings"
		local entry_display = require "telescope.pickers.entry_display"
		local utils = require "telescope.utils"

		__TelescopeUISelectSpecificOpts = vim.F.if_nil(
			__TelescopeUISelectSpecificOpts,
			vim.tbl_extend("keep", specific_opts, {
				["codeaction"] = {
					make_indexed = function(items)
						local indexed_items = {}
						local widths = {
							idx = 0,
							command_title = 0,
							client_name = 0,
						}
						for idx, item in ipairs(items) do
							local client = vim.lsp.get_client_by_id(item[1])
							local entry = {
								idx = idx,
								["add"] = {
									command_title = item[2].title:gsub("\r\n", "\\r\\n"):gsub("\n", "\\n"),
									client_name = client and client.name or "",
								},
								text = item,
							}
							table.insert(indexed_items, entry)
							widths.idx = math.max(widths.idx, strings.strdisplaywidth(entry.idx))
							widths.command_title = math.max(widths.command_title, strings.strdisplaywidth(entry.add.command_title))
							widths.client_name = math.max(widths.client_name, strings.strdisplaywidth(entry.add.client_name))
						end
						return indexed_items, widths
					end,
					make_displayer = function(widths)
						return entry_display.create {
							separator = " ",
							items = {
								{ width = widths.idx + 1 }, -- +1 for ":" suffix
								{ width = widths.command_title },
								{ width = widths.client_name },
							},
						}
					end,
					make_display = function(displayer)
						return function(e)
							return displayer {
								{ e.value.idx .. ":", "TelescopePromptPrefix" },
								{ e.value.add.command_title },
								{ e.value.add.client_name, "TelescopeResultsComment" },
							}
						end
					end,
					make_ordinal = function(e)
						return e.idx .. e.add["command_title"]
					end,
				},
			})
		)

		vim.ui.select = function(items, opts, on_choice)
			opts = opts or {}
			local prompt = vim.F.if_nil(opts.prompt, "Select one of")
			if prompt:sub(-1, -1) == ":" then
				prompt = prompt:sub(1, -2)
			end
			opts.format_item = vim.F.if_nil(opts.format_item, function(e)
				return tostring(e)
			end)

			-- We want or here because __TelescopeUISelectSpecificOpts[x] can be either nil or even false -> {}
			local sopts = __TelescopeUISelectSpecificOpts[vim.F.if_nil(opts.kind, "")] or {}
			local indexed_items, widths = vim.F.if_nil(sopts.make_indexed, function(items_)
				local indexed_items = {}
				for idx, item in ipairs(items_) do
					table.insert(indexed_items, { idx = idx, text = item })
				end
				return indexed_items
			end)(items)
			local displayer = vim.F.if_nil(sopts.make_displayer, function() end)(widths)
			local make_display = vim.F.if_nil(sopts.make_display, function(_)
				return function(e)
					local x, _ = opts.format_item(e.value.text)
					return x
				end
			end)(displayer)
			local make_ordinal = vim.F.if_nil(sopts.make_ordinal, function(e)
				return opts.format_item(e.text)
			end)
			pickers.new(topts, {
				prompt_title = prompt,
				finder = finders.new_table {
					results = indexed_items,
					entry_maker = function(e)
						return {
							value = e,
							display = make_display,
							ordinal = make_ordinal(e),
						}
					end,
				},
				attach_mappings = function(prompt_bufnr)
					actions.select_default:replace(function()
						local selection = action_state.get_selected_entry()
						if selection == nil then
							utils.__warn_no_selection "ui-select"
							return
						end
						actions.close(prompt_bufnr)
						on_choice(selection.value.text, selection.value.idx)
					end)
					return true
				end,
				sorter = conf.generic_sorter(topts),
			}):find()
		end
	end,
}

local map = vim.api.nvim_set_keymap

map('n', '<Leader>pf', '<Cmd>Telescope find_files<CR>', {noremap = true})
map('n', '<Leader>pb', '<Cmd>Telescope buffers<CR>', {noremap = true})
map('n', '<Leader>po', '<Cmd>Telescope lsp_document_symbols<CR>', {noremap = true})
map('n', '<Leader>ps', '<Cmd>Telescope lsp_workspace_symbols<CR>', {noremap = true})
map('n', '<Leader>pd', '<Cmd>Telescope lsp_definitions<CR>', {noremap = true})
map('n', '<Leader>pr', '<Cmd>Telescope lsp_references<CR>', {noremap = true})
map('n', '<Leader>p', ':Telescope ', {noremap = true})
map('n', '<Leader>p ', ':Telescope ', {noremap = true})

map('n', '<Leader>.', ':lua vim.lsp.buf.code_action()<CR>', {noremap = true})

-- vim.cmd [[
-- 	function! s:telescope_hlsetup() abort
-- 		highlight link TelescopeBorder NonText 
-- 		highlight link TelescopeTitle NonText
-- 	endfunction
-- 	call s:telescope_hlsetup()
-- 	augroup telescope_hlsetup
-- 		au!
-- 		au ColorScheme * call s:telescope_hlsetup()
-- 	augroup END
-- ]]