1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
local M = {}
local get_searchdirs = function(dirs)
dirs = dirs or {}
if type(dirs) == 'table' and #dirs > 0 then
return dirs
elseif type(dirs) == 'string' and (dirs == ':workspace') then
dirs = vim.lsp.buf.list_workspace_folders()
elseif type(dirs) == 'string' and not (dirs == '') then
dirs = vim.split(dirs, ',')
end
-- skip asking if we're looking at known filetypes
local ft = vim.api.nvim_buf_get_option(0, 'filetype')
if #dirs == 0 and (ft == 'dirbuf' or ft == 'NvimTree' or ft == 'alpha') then
dirs = { '%:p:h' }
end
-- ask user
if #dirs == 0 then
vim.ui.input({ prompt = 'Search directories: ', default = '%:p:h' }, function(result)
dirs = { result }
end)
end
-- default to something reasonable
if #dirs == 0 then
dirs = { '%:p:h' }
end
-- finally return
return dirs
end
function onchoice(opts, cb)
if not cb then
return
end
opts.attach_mappings = function(prompt_bufnr)
require('telescope.actions').select_default:replace(function()
local selection = require('telescope.actions.state').get_selected_entry()
if selection == nil then
return
end
require('telescope.actions').close(prompt_bufnr)
on_choice(selection.path)
end)
return true
end
end
function search_shim(fn, search_dirs, opts, cb)
opts = opts or {}
opts.search_dirs = get_searchdirs(search_dirs)
opts.prompt_title = vim.fn.expand(table.concat(opts.search_dirs, ','))
onchoice(opts, cb)
fn(opts)
end
function M.live_grep(search_dirs, opts, cb)
search_shim(require('telescope.builtin').live_grep, search_dirs, opts, cb)
end
function M.find_files(search_dirs, opts, cb)
-- convenience aliases
if type(search_dirs) == 'string' and search_dirs == 'data' then
search_dirs = vim.fn.stdpath('data')
end
search_shim(require('telescope.builtin').find_files, search_dirs, opts, cb)
end
function M.setup(opts)
vim.keymap.set('n', ';f', function()
vim.notify('use gsf', vim.log.levels.ERROR)
end, { desc = 'find files' })
vim.keymap.set('n', ';g', function()
vim.notify('use gsg', vim.log.levels.ERROR)
end, { desc = 'live grep' })
vim.keymap.set('n', 'gsg', function()
M.live_grep(':workspace')
end, { desc = 'live grep in workspace' })
vim.keymap.set('n', 'gsG', M.live_grep, { desc = 'live grep ask' })
vim.keymap.set('n', 'gs-', function()
M.live_grep('%:p:h')
end, { desc = 'live grep in parent' })
vim.keymap.set('n', 'gsf', function()
M.find_files(':workspace')
end, { desc = 'find files in workspace' })
vim.keymap.set('n', 'gsF', M.find_files, { desc = 'find files ask' })
vim.api.nvim_create_user_command('Grep', function(o)
M.live_grep(table.concat(o.fargs))
end, {
desc = 'live grep (rg)',
nargs = '*',
complete = 'dir',
})
vim.api.nvim_create_user_command('Find', function(o)
M.find_files(table.concat(o.fargs))
end, {
desc = 'find files (fd)',
nargs = '*',
complete = 'dir',
})
end
return M
|
