mirror of
https://github.com/SpaceVim/SpaceVim.git
synced 2025-01-24 02:50:03 +08:00
f8b280e03e
neovim 0.10.0-dev always show unknown option: t_ve when use `vim.o.t_ve`. but `exists('&t_ve')` return 1
843 lines
24 KiB
Lua
843 lines
24 KiB
Lua
--=============================================================================
|
|
-- flygrep.lua --- grep on the fly in SpaceVim
|
|
-- Copyright (c) 2016-2023 Wang Shidong & Contributors
|
|
-- Author: Wang Shidong < wsdjeg@outlook.com >
|
|
-- URL: https://spacevim.org
|
|
-- License: GPLv3
|
|
--=============================================================================
|
|
|
|
local M = {}
|
|
|
|
local logger = require('spacevim.logger').derive('flygrep')
|
|
local mpt = require('spacevim.api').import('prompt')
|
|
local hi = require('spacevim.api').import('vim.highlight')
|
|
local regex = require('spacevim.api').import('vim.regex')
|
|
local Key = require('spacevim.api').import('vim.keys')
|
|
local buffer = require('spacevim.api').import('vim.buffer')
|
|
local window = require('spacevim.api').import('vim.window')
|
|
local sl = require('spacevim.api').import('vim.statusline')
|
|
local nt = require('spacevim.api').import('notify')
|
|
local job = require('spacevim.api.job')
|
|
|
|
-- set commandline mpt
|
|
|
|
mpt._prompt.mpt = vim.g.spacevim_commandline_prompt .. ' '
|
|
|
|
-- compatibility functions
|
|
local function empty(expr)
|
|
return vim.fn.empty(expr) == 1
|
|
end
|
|
local function isdirectory(dir)
|
|
return vim.fn.isdirectory(dir) == 1
|
|
end
|
|
|
|
local function noautocmd(f) -- {{{
|
|
local ei = vim.o.eventignore
|
|
vim.o.eventignore = 'all'
|
|
pcall(f)
|
|
vim.o.eventignore = ei
|
|
end
|
|
-- }}}
|
|
local timer_start = vim.fn.timer_start
|
|
local timer_stop = vim.fn.timer_stop
|
|
|
|
-- the script local values, same as s: in vim script
|
|
local previous_winid = -1
|
|
local grep_expr = ''
|
|
local grep_default_exe, grep_default_opt, grep_default_ropt, grep_default_expr_opt, grep_default_fix_string_opt, grep_default_ignore_case, grep_default_smart_case =
|
|
require('spacevim.plugin.search').default_tool()
|
|
|
|
local grep_timer_id = -1
|
|
local preview_timer_id = -1
|
|
local preview_bufnr = -1
|
|
local grepid = 0
|
|
local mode = ''
|
|
local buffer_id = -1
|
|
local flygrep_win_id = -1
|
|
--- @type string|table
|
|
local grep_files = {}
|
|
local grep_dir = ''
|
|
local grep_exe = ''
|
|
local grep_opt = {}
|
|
local grep_ropt = {}
|
|
local grep_ignore_case = {}
|
|
local grep_smart_case = {}
|
|
local grep_expr_opt = {}
|
|
local search_hi_id = -1
|
|
local filter_hi_id = -1
|
|
local grep_mode = 'expr'
|
|
local filename_pattern = [[[^:]*:\d\+:\d\+:]]
|
|
local preview_able = false
|
|
local grep_history = {}
|
|
local preview_win_id = -1
|
|
local filter_file = ''
|
|
|
|
--- @return table # a list of searching pattern history
|
|
local function read_histroy()
|
|
if
|
|
vim.fn.filereadable(vim.fn.expand(vim.g.spacevim_data_dir .. 'SpaceVim/flygrep_history'))
|
|
== 1
|
|
then
|
|
local _his = vim.fn.json_decode(
|
|
vim.fn.join(
|
|
vim.fn.readfile(vim.fn.expand(vim.g.spacevim_data_dir .. 'SpaceVim/flygrep_history'), ''),
|
|
''
|
|
)
|
|
)
|
|
if type(_his) == table then
|
|
return _his or {}
|
|
else
|
|
return {}
|
|
end
|
|
else
|
|
return {}
|
|
end
|
|
end
|
|
grep_history = read_histroy()
|
|
|
|
local function update_history()
|
|
if vim.fn.index(grep_history, grep_expr) >= 0 then
|
|
vim.fn.remove(grep_history, vim.fn.index(grep_history, grep_expr))
|
|
end
|
|
table.insert(grep_history, grep_expr)
|
|
if vim.fn.isdirectory(vim.fn.expand(vim.g.spacevim_data_dir .. 'SpaceVim')) == 0 then
|
|
vim.fn.mkdir(vim.fn.expand(vim.g.spacevim_data_dir .. 'SpaceVim'))
|
|
end
|
|
if
|
|
vim.fn.filereadable(vim.fn.expand(vim.g.spacevim_data_dir .. 'SpaceVim/flygrep_history'))
|
|
== 1
|
|
then
|
|
vim.fn.writefile(
|
|
{ vim.fn.json_encode(grep_history) },
|
|
vim.fn.expand(vim.g.spacevim_data_dir .. 'SpaceVim/flygrep_history')
|
|
)
|
|
end
|
|
end
|
|
|
|
local function append(t1, t2)
|
|
for _, v in pairs(t2) do
|
|
table.insert(t1, v)
|
|
end
|
|
end
|
|
|
|
--- @param expr string # searching pattern
|
|
--- @return table<string> # searching command
|
|
local function get_search_cmd(expr)
|
|
local cmd = { grep_exe }
|
|
append(cmd, grep_opt)
|
|
if vim.o.ignorecase then
|
|
append(cmd, grep_ignore_case)
|
|
end
|
|
if vim.o.smartcase then
|
|
append(cmd, grep_smart_case)
|
|
end
|
|
if grep_mode == 'string' then
|
|
append(cmd, grep_default_fix_string_opt)
|
|
end
|
|
append(cmd, grep_expr_opt)
|
|
if not empty(grep_files) and vim.fn.type(grep_files) == 3 then
|
|
append(cmd, { expr })
|
|
append(cmd, grep_files)
|
|
elseif not empty(grep_files) and vim.fn.type(grep_files) == 1 then
|
|
append(cmd, { expr })
|
|
append(cmd, { grep_files })
|
|
elseif not empty(grep_dir) then
|
|
if grep_exe == 'findstr' then
|
|
append(cmd, { grep_dir, expr, [[%CD%\*]] })
|
|
else
|
|
append(cmd, { expr, grep_dir })
|
|
end
|
|
else
|
|
append(cmd, { expr })
|
|
if grep_exe == 'rg' or grep_exe == 'ag' or grep_exe == 'pt' then
|
|
append(cmd, { '.' })
|
|
end
|
|
append(cmd, grep_ropt)
|
|
end
|
|
return cmd
|
|
end
|
|
|
|
local function update_statusline()
|
|
if sl.support_float() and vim.fn.win_id2tabwin(flygrep_win_id)[1] == vim.fn.tabpagenr() then
|
|
sl.open_float({
|
|
{ 'FlyGrep ', 'SpaceVim_statusline_a_bold' },
|
|
{ ' ', 'SpaceVim_statusline_a_SpaceVim_statusline_b' },
|
|
{ M.mode() .. ' ', 'SpaceVim_statusline_b' },
|
|
{ ' ', 'SpaceVim_statusline_b_SpaceVim_statusline_c' },
|
|
{ vim.fn.getcwd() .. ' ', 'SpaceVim_statusline_c' },
|
|
{ ' ', 'SpaceVim_statusline_c_SpaceVim_statusline_b' },
|
|
{ M.lineNr() .. ' ', 'SpaceVim_statusline_b' },
|
|
{ ' ', 'SpaceVim_statusline_b_SpaceVim_statusline_z' },
|
|
{ vim.fn['repeat'](' ', vim.o.columns - 11), 'SpaceVim_statusline_z' },
|
|
})
|
|
end
|
|
end
|
|
|
|
local complete_input_history_num = { 0, 0 }
|
|
|
|
local function grep_stdout(id, data, _)
|
|
-- ignore previous result
|
|
if id ~= grepid then
|
|
return
|
|
end
|
|
noautocmd(function()
|
|
local datas = vim.fn.filter(data, '!empty(v:val)')
|
|
if vim.fn.getbufline(buffer_id, 1)[1] == '' then
|
|
vim.api.nvim_buf_set_lines(buffer_id, 0, -1, false, datas)
|
|
else
|
|
vim.api.nvim_buf_set_lines(buffer_id, -1, -1, false, datas)
|
|
end
|
|
vim.cmd('redraw')
|
|
update_statusline()
|
|
end)
|
|
end
|
|
|
|
local function grep_stderr(_, data, _)
|
|
for _, d in pairs(data) do
|
|
logger.info('grep stderr:' .. d)
|
|
end
|
|
end
|
|
|
|
local function close_statusline()
|
|
sl.close_float()
|
|
end
|
|
|
|
local function grep_exit(id, data, _)
|
|
if id ~= grepid then
|
|
return
|
|
end
|
|
logger.info('grep exit:' .. data)
|
|
update_statusline()
|
|
vim.cmd('redraw')
|
|
mpt._build_prompt()
|
|
grepid = 0
|
|
end
|
|
|
|
-- The available options are:
|
|
-- - input: string, the default input pattern
|
|
-- - files: a list of string or `@buffers`
|
|
-- - cmd: list
|
|
-- - opt: list
|
|
-- - ropt: list
|
|
-- - ignore_case: boolean
|
|
-- - smart_case: boolean
|
|
-- - expr_opt:
|
|
|
|
local current_grep_pattern = ''
|
|
local function grep_timer(_)
|
|
if grep_mode == 'expr' then
|
|
current_grep_pattern = vim.fn.join(vim.fn.split(grep_expr), '.*')
|
|
else
|
|
current_grep_pattern = grep_expr
|
|
end
|
|
local cmd = get_search_cmd(current_grep_pattern)
|
|
logger.info('grep cmd:' .. vim.inspect(cmd))
|
|
grepid = job.start(cmd, {
|
|
on_stdout = grep_stdout,
|
|
on_stderr = grep_stderr,
|
|
on_exit = grep_exit,
|
|
})
|
|
logger.info('flygrep job id is:' .. grepid)
|
|
end
|
|
|
|
local function matchadd(group, pattern, p)
|
|
local _, id = pcall(vim.fn.matchadd, group, pattern, p)
|
|
return id
|
|
end
|
|
|
|
local function expr_to_pattern(expr)
|
|
if grep_mode == 'expr' then
|
|
local items = vim.fn.split(expr)
|
|
local pattern = vim.fn.join(items, '.*')
|
|
local ignorecase = ''
|
|
if vim.o.ignorecase then
|
|
ignorecase = [[\c]]
|
|
else
|
|
ignorecase = [[\C]]
|
|
end
|
|
pattern = filename_pattern .. [[.*\zs]] .. ignorecase .. regex.parser(pattern, false)
|
|
logger.info('matchadd pattern: ' .. pattern)
|
|
return pattern
|
|
else
|
|
return expr
|
|
end
|
|
end
|
|
|
|
local function flygrep(t)
|
|
-- if the insert text is empty, clear grepid
|
|
grepid = 0
|
|
update_statusline()
|
|
mpt._build_prompt()
|
|
if t == '' then
|
|
return
|
|
end
|
|
pcall(vim.fn.matchdelete, search_hi_id)
|
|
search_hi_id = matchadd('FlygrepSearchPattern', expr_to_pattern(t), 2)
|
|
grep_expr = t
|
|
timer_stop(grep_timer_id)
|
|
grep_timer_id = timer_start(200, grep_timer, { ['repeat'] = 1 })
|
|
end
|
|
|
|
local function close_flygrep_win()
|
|
pcall(vim.api.nvim_win_close, flygrep_win_id, true)
|
|
vim.fn.win_gotoid(previous_winid)
|
|
end
|
|
|
|
local function get_file_pos(line)
|
|
local filename = vim.fn.fnameescape(vim.fn.split(line, [[:\d\+:]])[1])
|
|
local linenr = vim.fn.str2nr(string.sub(vim.fn.matchstr(line, [[:\d\+:]]), 2, -2))
|
|
local colum = vim.fn.str2nr(string.sub(vim.fn.matchstr(line, [[\(:\d\+\)\@<=:\d\+:]]), 2, -2))
|
|
return filename, linenr, colum
|
|
end
|
|
|
|
local function preview_timer(_)
|
|
local cursor = vim.api.nvim_win_get_cursor(flygrep_win_id)
|
|
local line = vim.api.nvim_buf_get_lines(buffer_id, cursor[1] - 1, cursor[1], false)[1]
|
|
if line == '' then
|
|
return
|
|
end
|
|
local filename, liner, colum = get_file_pos(line)
|
|
if vim.fn.bufexists(preview_bufnr) ~= 1 then
|
|
preview_bufnr = vim.api.nvim_create_buf(false, true)
|
|
end
|
|
local flygrep_win_height = 16
|
|
if not window.is_float(preview_win_id) then
|
|
noautocmd(function()
|
|
preview_win_id = vim.api.nvim_open_win(preview_bufnr, false, {
|
|
relative = 'editor',
|
|
width = vim.o.columns,
|
|
height = 8,
|
|
row = vim.o.lines - flygrep_win_height - 2 - 8,
|
|
col = 0,
|
|
})
|
|
end)
|
|
end
|
|
vim.api.nvim_buf_set_lines(preview_bufnr, 0, -1, false, vim.fn.readfile(filename, ''))
|
|
local ft = vim.filetype.match({ filename = filename })
|
|
if ft then
|
|
vim.api.nvim_buf_set_option(preview_bufnr, 'syntax', ft)
|
|
else
|
|
local ftdetect_autocmd = vim.api.nvim_get_autocmds({
|
|
group = 'filetypedetect',
|
|
event = 'BufRead',
|
|
pattern = '*.' .. vim.fn.fnamemodify(filename, ':e'),
|
|
})
|
|
-- logger.info(vim.inspect(ftdetect_autocmd))
|
|
if ftdetect_autocmd[1] then
|
|
if
|
|
ftdetect_autocmd[1].command and vim.startswith(ftdetect_autocmd[1].command, 'set filetype=')
|
|
then
|
|
ft = ftdetect_autocmd[1].command:gsub('set filetype=', '')
|
|
vim.api.nvim_buf_set_option(preview_bufnr, 'syntax', ft)
|
|
end
|
|
end
|
|
end
|
|
vim.api.nvim_win_set_cursor(preview_win_id, { liner, colum })
|
|
mpt._build_prompt()
|
|
end
|
|
|
|
local function preview()
|
|
timer_stop(preview_timer_id)
|
|
preview_timer_id = timer_start(200, preview_timer, { ['repeat'] = 1 })
|
|
end
|
|
|
|
local function close_preview_win()
|
|
pcall(vim.api.nvim_win_close, preview_win_id, true)
|
|
end
|
|
|
|
local function close_buffer()
|
|
if grepid > 0 then
|
|
grepid = 0
|
|
job.stop(grepid)
|
|
end
|
|
timer_stop(grep_timer_id)
|
|
timer_stop(preview_timer_id)
|
|
if preview_able then
|
|
close_preview_win()
|
|
preview_able = false
|
|
end
|
|
close_flygrep_win()
|
|
vim.cmd('noautocmd normal :')
|
|
end
|
|
|
|
mpt._onclose = close_buffer
|
|
|
|
local function close_grep_job()
|
|
if grepid > 0 then
|
|
pcall(job.stop, grepid)
|
|
end
|
|
timer_stop(grep_timer_id)
|
|
timer_stop(preview_timer_id)
|
|
vim.api.nvim_buf_set_lines(buffer_id, 0, -1, false, {})
|
|
update_statusline()
|
|
complete_input_history_num = { 0, 0 }
|
|
end
|
|
|
|
mpt._oninputpro = close_grep_job
|
|
|
|
local function next_item()
|
|
local cursor = vim.api.nvim_win_get_cursor(flygrep_win_id)
|
|
if cursor[1] >= vim.api.nvim_buf_line_count(buffer_id) then
|
|
cursor[1] = 1
|
|
else
|
|
cursor[1] = cursor[1] + 1
|
|
end
|
|
vim.api.nvim_win_set_cursor(flygrep_win_id, cursor)
|
|
if preview_able then
|
|
preview()
|
|
end
|
|
update_statusline()
|
|
vim.cmd('redraw')
|
|
mpt._build_prompt()
|
|
end
|
|
|
|
local function previous_item()
|
|
local cursor = vim.api.nvim_win_get_cursor(flygrep_win_id)
|
|
if cursor[1] == 1 then
|
|
cursor[1] = vim.api.nvim_buf_line_count(buffer_id)
|
|
else
|
|
cursor[1] = cursor[1] - 1
|
|
end
|
|
vim.api.nvim_win_set_cursor(flygrep_win_id, cursor)
|
|
if preview_able then
|
|
preview()
|
|
end
|
|
update_statusline()
|
|
vim.cmd('redraw')
|
|
mpt._build_prompt()
|
|
end
|
|
|
|
local function open_item(...)
|
|
local argv = { ... }
|
|
local edit_command = argv[1] or 'edit'
|
|
mpt._handle_fly = flygrep
|
|
local cursor = vim.api.nvim_win_get_cursor(flygrep_win_id)
|
|
local line = vim.api.nvim_buf_get_lines(buffer_id, cursor[1] - 1, cursor[1], false)[1]
|
|
-- print(vim.inspect(line))
|
|
if line ~= '' then
|
|
if grepid ~= 0 then
|
|
-- change grepid to 0, and callback function will be skipped
|
|
grepid = 0
|
|
job.stop(grepid)
|
|
end
|
|
mpt._clear_prompt()
|
|
mpt._quit = true
|
|
local filename, liner, colum = get_file_pos(line)
|
|
if preview_able then
|
|
close_preview_win()
|
|
end
|
|
preview_able = false
|
|
close_flygrep_win()
|
|
update_history()
|
|
buffer.open_pos(edit_command, filename, liner, colum)
|
|
vim.cmd('noautocmd normal! :')
|
|
end
|
|
end
|
|
|
|
local function open_item_in_tab()
|
|
open_item('tabedit')
|
|
end
|
|
|
|
local function open_item_vertically()
|
|
open_item('vsplit')
|
|
end
|
|
|
|
local function open_item_horizontally()
|
|
open_item('split')
|
|
end
|
|
|
|
local function move_cursor()
|
|
if vim.v.mouse_winid == flygrep_win_id then
|
|
vim.api.nvim_win_set_cursor(flygrep_win_id, { vim.v.mouse_lnum, 0 })
|
|
end
|
|
mpt._build_prompt()
|
|
end
|
|
|
|
local function double_click()
|
|
if vim.v.mouse_winid == flygrep_win_id then
|
|
vim.api.nvim_win_set_cursor(flygrep_win_id, { vim.v.mouse_lnum, 0 })
|
|
end
|
|
open_item()
|
|
end
|
|
|
|
local function toggle_expr_mode()
|
|
if grep_mode == 'expr' then
|
|
grep_mode = 'string'
|
|
else
|
|
grep_mode = 'expr'
|
|
end
|
|
mpt._oninputpro()
|
|
mpt._handle_fly(mpt._prompt.cursor_begin .. mpt._prompt.cursor_char .. mpt._prompt.cursor_end)
|
|
end
|
|
|
|
local function apply_to_quickfix()
|
|
mpt._handle_fly = flygrep
|
|
if vim.fn.getbufline(buffer_id, 1)[1] ~= '' then
|
|
if grepid ~= 0 then
|
|
-- stop job, and skip callback function
|
|
grepid = 0
|
|
job.stop(grepid)
|
|
end
|
|
mpt._quit = true
|
|
if preview_able then
|
|
close_preview_win()
|
|
end
|
|
preview_able = false
|
|
local searching_result = vim.api.nvim_buf_get_lines(buffer_id, 0, -1, false)
|
|
close_flygrep_win()
|
|
update_history()
|
|
if vim.fn.empty(searching_result) == 0 then
|
|
-- vim.cmd('cgetexpr ' .. vim.fn.join(searching_result, "\n"))
|
|
-- vim.cmd([[
|
|
-- cgetexpr join(luaeval(searching_result), "\n")
|
|
-- ]])
|
|
-- vim.fn.setqflist({})
|
|
vim.fn.setqflist({}, 'r', {
|
|
title = 'FlyGrep partten:'
|
|
.. mpt._prompt.cursor_begin
|
|
.. mpt._prompt.cursor_char
|
|
.. mpt._prompt.cursor_end,
|
|
lines = searching_result,
|
|
})
|
|
mpt._clear_prompt()
|
|
-- use botright to make sure quicfix windows width same as screen
|
|
vim.cmd('botright copen')
|
|
end
|
|
vim.cmd('noautocmd normal! :')
|
|
end
|
|
end
|
|
|
|
local function toggle_preview()
|
|
if not preview_able then
|
|
preview_able = true
|
|
preview()
|
|
else
|
|
close_preview_win()
|
|
preview_able = false
|
|
end
|
|
vim.cmd('redraw')
|
|
mpt._build_prompt()
|
|
end
|
|
|
|
local function get_filter_cmd(expr)
|
|
local cmd = { grep_exe }
|
|
append(cmd, require('spacevim.plugin.search').getFopt(grep_exe))
|
|
append(cmd, { expr, filter_file })
|
|
return cmd
|
|
end
|
|
|
|
local function filter_timer(_)
|
|
local cmd = get_filter_cmd(vim.fn.join(vim.fn.split(grep_expr), '.*'))
|
|
grepid = job.start(cmd, {
|
|
on_stdout = grep_stdout,
|
|
on_exit = grep_exit,
|
|
})
|
|
end
|
|
|
|
local function filter(expr)
|
|
mpt._build_prompt()
|
|
pcall(vim.fn.matchdelete, filter_hi_id)
|
|
if expr == '' then
|
|
-- if the mpt is empty, put context in filter_file into flygrep buffer
|
|
if vim.fn.filereadable(filter_file) then
|
|
vim.api.nvim_buf_set_lines(buffer_id, 0, -1, false, vim.fn.readfile(filter_file, ''))
|
|
vim.cmd('redraw')
|
|
end
|
|
return
|
|
end
|
|
filter_hi_id = matchadd('FlygrepFilterPattern', expr_to_pattern(expr), 2)
|
|
grep_expr = expr
|
|
grep_timer_id = timer_start(200, filter_timer, { ['repeat'] = 1 })
|
|
end
|
|
|
|
local function start_filter()
|
|
mode = 'f'
|
|
update_statusline()
|
|
mpt._handle_fly = filter
|
|
mpt._clear_prompt()
|
|
filter_file = vim.fn.tempname()
|
|
local context = vim.api.nvim_buf_get_lines(buffer_id, 0, -1, false)
|
|
local ok, _ = pcall(vim.fn.writefile, context, filter_file, 'b')
|
|
if not ok then
|
|
logger.info('Failed to write filter content to temp file')
|
|
end
|
|
mpt._build_prompt()
|
|
end
|
|
|
|
local function tbl_filter(func, t) -- {{{
|
|
local rettab = {}
|
|
for _, entry in pairs(t) do
|
|
if func(entry) then
|
|
table.insert(rettab, entry)
|
|
end
|
|
end
|
|
return rettab
|
|
end
|
|
-- }}}
|
|
|
|
local function complete_input_history(str, num) -- {{{
|
|
-- logger.info(vim.inspect(grep_history))
|
|
-- local results = vim.fn.filter(, "v:val =~# '^' . a:str")
|
|
local results = tbl_filter(function(node)
|
|
-- here the note sometimes do not have title, then it is nil
|
|
if type(node) ~= 'string' then
|
|
return false
|
|
end
|
|
return vim.startswith(node, str)
|
|
end, vim.deepcopy(grep_history))
|
|
logger.info(vim.inspect(results))
|
|
local complete_items
|
|
if not empty(results) and results[-1] ~= str then
|
|
complete_items = results
|
|
table.insert(complete_items, str)
|
|
elseif empty(results) then
|
|
complete_items = { str }
|
|
else
|
|
complete_items = results
|
|
end
|
|
-- 5 0 6
|
|
local patch = (num[1] - num[2]) % vim.fn.len(complete_items)
|
|
local index
|
|
if patch >= 0 then
|
|
index = vim.fn.len(complete_items) - patch
|
|
else
|
|
index = vim.fn.abs(patch)
|
|
end
|
|
return complete_items[index]
|
|
end
|
|
-- }}}
|
|
|
|
local complete_input_history_base = ''
|
|
local function previous_match_history()
|
|
if complete_input_history_num[1] == 0 and complete_input_history_num[2] == 0 then
|
|
complete_input_history_base = mpt._prompt.cursor_begin
|
|
mpt._prompt.cursor_char = ''
|
|
mpt._prompt.cursor_end = ''
|
|
end
|
|
complete_input_history_num[1] = complete_input_history_num[1] + 1
|
|
mpt._prompt.cursor_begin =
|
|
complete_input_history(complete_input_history_base, complete_input_history_num)
|
|
vim.api.nvim_buf_set_lines(buffer_id, 0, -1, false, {})
|
|
mpt._handle_fly(mpt._prompt.cursor_begin .. mpt._prompt.cursor_char .. mpt._prompt.cursor_end)
|
|
end
|
|
|
|
local function next_match_history()
|
|
if complete_input_history_num[1] == 0 and complete_input_history_num[2] == 0 then
|
|
complete_input_history_base = mpt._prompt.cursor_begin
|
|
mpt._prompt.cursor_char = ''
|
|
mpt._prompt.cursor_end = ''
|
|
end
|
|
complete_input_history_num[2] = complete_input_history_num[2] + 1
|
|
mpt._prompt.cursor_begin =
|
|
complete_input_history(complete_input_history_base, complete_input_history_num)
|
|
vim.api.nvim_buf_set_lines(buffer_id, 0, -1, false, {})
|
|
mpt._handle_fly(mpt._prompt.cursor_begin .. mpt._prompt.cursor_char .. mpt._prompt.cursor_end)
|
|
end
|
|
local function page_up()
|
|
-- exe "noautocmd normal! \<PageUp>"
|
|
vim.api.nvim_win_call(flygrep_win_id, function()
|
|
vim.api.nvim_feedkeys(Key.t('<PageUp>'), 'x', false)
|
|
end)
|
|
if preview_able then
|
|
preview()
|
|
end
|
|
update_statusline()
|
|
vim.cmd('redraw')
|
|
mpt._build_prompt()
|
|
end
|
|
|
|
local function page_down()
|
|
-- exe "noautocmd normal! \<PageUp>"
|
|
vim.api.nvim_win_call(flygrep_win_id, function()
|
|
vim.api.nvim_feedkeys(Key.t('<PageDown>'), 'x', false)
|
|
end)
|
|
if preview_able then
|
|
preview()
|
|
end
|
|
update_statusline()
|
|
vim.cmd('redraw')
|
|
mpt._build_prompt()
|
|
end
|
|
|
|
local function update_files(f) -- {{{
|
|
end
|
|
-- }}}
|
|
|
|
local function flygrep_result_to_files() -- {{{
|
|
end
|
|
-- }}}
|
|
|
|
local function start_replace()
|
|
mode = 'r'
|
|
pcall(vim.fn.matchdelete, search_hi_id)
|
|
if grepid ~= 0 then
|
|
job.stop(grepid)
|
|
end
|
|
local replace_text = current_grep_pattern
|
|
local rst
|
|
if not empty(replace_text) then
|
|
rst = require('spacevim.plugin.iedit').start({ expr = replace_text }, 1, vim.fn.line('$'))
|
|
end
|
|
search_hi_id = vim.fn.matchadd('FlyGrepPattern', expr_to_pattern(rst), 2)
|
|
update_statusline()
|
|
if rst ~= replace_text then
|
|
update_files(flygrep_result_to_files())
|
|
vim.cmd('checktime')
|
|
end
|
|
end
|
|
|
|
mpt._function_key = {
|
|
[Key.t('<Tab>')] = next_item,
|
|
[Key.t('<C-j>')] = next_item,
|
|
[Key.t('<ScrollWheelDown>')] = next_item,
|
|
[Key.t('<S-tab>')] = previous_item,
|
|
[Key.t('<C-k>')] = previous_item,
|
|
[Key.t('<ScrollWheelUp>')] = previous_item,
|
|
[Key.t('<Return>')] = open_item,
|
|
[Key.t('<C-t>')] = open_item_in_tab,
|
|
[Key.t('<LeftMouse>')] = move_cursor,
|
|
[Key.t('<2-LeftMouse>')] = double_click,
|
|
[Key.t('<C-f>')] = start_filter,
|
|
[Key.t('<C-v>')] = open_item_vertically,
|
|
[Key.t('<C-s>')] = open_item_horizontally,
|
|
[Key.t('<C-q>')] = apply_to_quickfix,
|
|
[Key.t('<M-r>')] = start_replace,
|
|
[Key.t('<C-p>')] = toggle_preview,
|
|
[Key.t('<C-e>')] = toggle_expr_mode,
|
|
[Key.t('<Up>')] = previous_match_history,
|
|
[Key.t('<Down>')] = next_match_history,
|
|
[Key.t('<PageDown>')] = page_down,
|
|
[Key.t('<PageUp>')] = page_up,
|
|
-- [Key.t('<C-End>')] = page_end,
|
|
-- [Key.t('<C-Home>')] = page_home,
|
|
[Key.t('x80\xfdK')] = previous_item,
|
|
[Key.t('x80\xfc \x80\xfdK')] = previous_item,
|
|
[Key.t('x80\xfc@\x80\xfdK')] = previous_item,
|
|
[Key.t('x80\xfc`\x80\xfdK')] = previous_item,
|
|
[Key.t('x80\xfdL')] = next_item,
|
|
[Key.t('x80\xfc \x80\xfdL')] = next_item,
|
|
[Key.t('x80\xfc@\x80\xfdL')] = next_item,
|
|
[Key.t('x80\xfc`\x80\xfdL')] = next_item,
|
|
}
|
|
|
|
function M.mode()
|
|
local _, iedit_mode = pcall(vim.api.nvim_win_get_var, flygrep_win_id, 'spacevim_iedit_mode')
|
|
if iedit_mode == 'n' then
|
|
return 'iedit-normal'
|
|
elseif iedit_mode == 'i' then
|
|
return 'iedit-insert'
|
|
else
|
|
if mode == '' then
|
|
return grep_mode
|
|
else
|
|
return grep_mode .. '(' .. mode .. ')'
|
|
end
|
|
end
|
|
end
|
|
|
|
function M.lineNr()
|
|
if vim.fn.getbufline(buffer_id, 1)[1] == '' then
|
|
return 'no result'
|
|
else
|
|
local current = vim.api.nvim_win_get_cursor(flygrep_win_id)[1]
|
|
local total = vim.api.nvim_buf_line_count(buffer_id)
|
|
return current .. '/' .. total
|
|
end
|
|
end
|
|
|
|
function M.open(argv)
|
|
previous_winid = vim.fn.win_getid()
|
|
if empty(grep_default_exe) then
|
|
logger.warn('make sure you have one search tool in your PATH')
|
|
nt.notify('make sure you have one search tool in your PATH')
|
|
return
|
|
elseif type(argv.cmd) == "string" and vim.fn.empty(argv.cmd) == 0 and vim.fn.executable(argv.cmd) == 0 then
|
|
logger.warn(argv.cmd .. ' is not executable, make sure ' .. argv.cmd .. ' is in your PATH')
|
|
nt.notify(argv.cmd .. ' is not executable,\nmake sure ' .. argv.cmd .. ' is in your PATH', 'WarningMsg')
|
|
return
|
|
end
|
|
mode = ''
|
|
mpt._handle_fly = flygrep
|
|
buffer_id = vim.api.nvim_create_buf(false, true)
|
|
local flygrep_win_height = 16
|
|
noautocmd(function()
|
|
flygrep_win_id = vim.api.nvim_open_win(buffer_id, true, {
|
|
relative = 'editor',
|
|
width = vim.o.columns,
|
|
height = flygrep_win_height,
|
|
row = vim.o.lines - flygrep_win_height - 2,
|
|
col = 0,
|
|
})
|
|
end)
|
|
|
|
if vim.fn.exists('&winhighlight') == 1 then
|
|
vim.cmd('set winhighlight=Normal:Pmenu,EndOfBuffer:Pmenu,CursorLine:PmenuSel')
|
|
end
|
|
vim.cmd(
|
|
'setlocal buftype=nofile bufhidden=wipe nobuflisted nolist noswapfile nowrap cursorline nospell nonu norelativenumber'
|
|
)
|
|
vim.opt_local.fillchars = { eob = ' ' }
|
|
local cursor_hi = {}
|
|
cursor_hi = hi.group2dict('Cursor')
|
|
local lcursor_hi = {}
|
|
lcursor_hi = hi.group2dict('lCursor')
|
|
local guicursor = vim.o.guicursor
|
|
hi.hide_in_normal('Cursor')
|
|
hi.hide_in_normal('lCursor')
|
|
if vim.fn.has('nvim') == 1 then
|
|
vim.cmd('set guicursor+=a:Cursor/lCursor')
|
|
end
|
|
vim.cmd('setf SpaceVimFlyGrep')
|
|
update_statusline()
|
|
matchadd('FileName', filename_pattern, 3)
|
|
vim.cmd('hi def link FlygrepSearchPattern MoreMsg')
|
|
vim.cmd('hi def link FlygrepFilterPattern Question')
|
|
mpt._prompt.cursor_begin = argv.input or ''
|
|
local fs = argv.files or ''
|
|
if fs == '@buffers' then
|
|
grep_files = vim.fn.map(buffer.listed_buffers(), 'bufname(v:val)')
|
|
elseif not empty(fs) then
|
|
grep_files = fs
|
|
else
|
|
grep_files = ''
|
|
end
|
|
|
|
local dir = vim.fn.expand(argv.dir or '')
|
|
|
|
if not empty(dir) and isdirectory(dir) then
|
|
grep_dir = dir
|
|
else
|
|
grep_dir = ''
|
|
end
|
|
grep_exe = argv.cmd or grep_default_exe
|
|
if empty(grep_dir) and empty(grep_files) and grep_exe == 'findstr' then
|
|
grep_files = '*.*'
|
|
elseif grep_exe == 'findstr' and not empty(grep_dir) then
|
|
grep_dir = '/D:' .. grep_dir
|
|
end
|
|
grep_opt = argv.opt or grep_default_opt
|
|
grep_ropt = argv.ropt or grep_default_ropt
|
|
grep_ignore_case = argv.ignore_case or grep_default_ignore_case
|
|
grep_smart_case = argv.smart_case or grep_default_smart_case
|
|
grep_expr_opt = argv.expr_opt or grep_default_expr_opt
|
|
logger.info('FlyGrep startting ===========================')
|
|
logger.info(' executable : ' .. grep_exe)
|
|
logger.info(' option : ' .. vim.fn.string(grep_opt))
|
|
logger.info(' r_option : ' .. vim.fn.string(grep_ropt))
|
|
logger.info(' files : ' .. vim.fn.string(grep_files))
|
|
logger.info(' dir : ' .. vim.fn.string(grep_dir))
|
|
logger.info(' ignore_case : ' .. vim.fn.string(grep_ignore_case))
|
|
logger.info(' smart_case : ' .. vim.fn.string(grep_smart_case))
|
|
logger.info(' expr opt : ' .. vim.fn.string(grep_expr_opt))
|
|
mpt.open()
|
|
if sl.support_float() then
|
|
close_statusline()
|
|
end
|
|
logger.info('FlyGrep ending =====================')
|
|
hi.hi(cursor_hi)
|
|
hi.hi(lcursor_hi)
|
|
vim.o.guicursor = guicursor
|
|
end
|
|
|
|
return M
|