1
0
mirror of https://github.com/SpaceVim/SpaceVim.git synced 2025-02-19 08:53:44 +08:00
SpaceVim/bundle/quickfix.nvim/lua/quickfix.lua

290 lines
9.1 KiB
Lua
Vendored

--=============================================================================
-- qf.lua
-- Copyright 2025 Eric Wong
-- Author: Eric Wong < wsdjeg@outlook.com >
-- URL: https://spacevim.org
-- License: GPLv3
--=============================================================================
local M = {}
local qf_historys = {}
local location_history = {}
local is_loclist = function(win)
if not win then
win = vim.api.nvim_get_current_win()
end
return vim.fn.getwininfo(win)[1].loclist == 1
end
-- This function only can be called on location list windows
local function save_location_list_to_history(loclist)
local locwinid = vim.fn.getloclist(0, { filewinid = 0 })
if not location_history[locwinid] then
location_history[locwinid] = {}
end
table.insert(location_history[locwinid], vim.deepcopy(loclist))
end
M.setup = function()
local group = vim.api.nvim_create_augroup('quickfix_mapping', {
clear = true,
})
vim.api.nvim_create_autocmd({ 'FileType' }, {
pattern = 'qf',
group = group,
callback = function(ev)
vim.keymap.set('n', vim.g.quickfix_mapping_delete or 'dd', function()
if is_loclist() then
local loclist = vim.fn.getloclist(0)
local line = vim.fn.line('.')
if #loclist >= 1 then
local locwinid = vim.fn.getloclist(0, { filewinid = 0 })
if not location_history[locwinid] then
location_history[locwinid] = {}
end
table.insert(location_history[locwinid], vim.deepcopy(loclist))
table.remove(loclist, line)
end
vim.fn.setloclist(0, loclist)
vim.cmd(tostring(line))
else
local qflist = vim.fn.getqflist()
local line = vim.fn.line('.')
if #qflist >= 1 then
table.insert(qf_historys, vim.deepcopy(qflist))
table.remove(qflist, line)
end
vim.fn.setqflist(qflist)
vim.cmd(tostring(line))
end
end, { buffer = ev.buf })
end,
})
local function table_remove(t, from, to)
local rst = {}
for i = 1, #t, 1 do
if i < from or i > to then
rst[#rst + 1] = t[i]
end
end
return rst
end
vim.api.nvim_create_autocmd({ 'FileType' }, {
pattern = 'qf',
group = group,
callback = function(ev)
vim.keymap.set('v', vim.g.quickfix_mapping_visual_delete or 'd', function()
local esc = vim.api.nvim_replace_termcodes('<esc>', true, false, true)
vim.api.nvim_feedkeys(esc, 'x', false)
if is_loclist() then
local loclist = vim.fn.getloclist(0)
local from = vim.fn.getpos("'<")[2]
local to = vim.fn.getpos("'>")[2]
if from > to then
from, to = to, from
end
local lc = table_remove(loclist, from, to)
if #lc ~= #loclist and #lc ~= 0 then
local locwinid = vim.fn.getloclist(0, { filewinid = 0 })
if not location_history[locwinid] then
location_history[locwinid] = {}
end
table.insert(location_history[locwinid], loclist)
end
vim.fn.setloclist(0, lc)
vim.cmd(tostring(from))
else
local qflist = vim.fn.getqflist()
local from = vim.fn.getpos("'<")[2]
local to = vim.fn.getpos("'>")[2]
if from > to then
from, to = to, from
end
local qf = table_remove(qflist, from, to)
if #qf ~= #qflist and #qflist ~= 0 then
table.insert(qf_historys, qflist)
end
vim.fn.setqflist(qf)
vim.cmd(tostring(from))
end
end, { buffer = ev.buf })
end,
})
vim.api.nvim_create_autocmd({ 'FileType' }, {
pattern = 'qf',
group = group,
callback = function(ev)
vim.keymap.set('n', vim.g.quickfix_mapping_filter_filename or 'c', function()
local input_pattern = vim.fn.input('filter pattern:')
-- vim.cmd('noautocmd normal! :')
if is_loclist() then
local re = vim.regex(input_pattern)
local lc = {}
local loclist = vim.fn.getloclist(0)
for _, item in ipairs(loclist) do
if not re:match_str(vim.fn.bufname(item.bufnr)) then
table.insert(lc, item)
end
end
if #lc ~= #loclist and #loclist ~= 0 then
save_location_list_to_history(loclist)
end
vim.fn.setloclist(0, lc)
else
local re = vim.regex(input_pattern)
local qflist = vim.fn.getqflist()
local qf = {}
for _, item in ipairs(qflist) do
if not re:match_str(vim.fn.bufname(item.bufnr)) then
table.insert(qf, item)
end
end
if #qf ~= #qflist and #qflist ~= 0 then
table.insert(qf_historys, qflist)
end
vim.fn.setqflist(qf)
end
end, { buffer = ev.buf })
end,
})
vim.api.nvim_create_autocmd({ 'FileType' }, {
pattern = 'qf',
group = group,
callback = function(ev)
vim.keymap.set('n', vim.g.quickfix_mapping_rfilter_filename or 'C', function()
local input_pattern = vim.fn.input('filter pattern:')
if is_loclist() then
local re = vim.regex(input_pattern)
local lc = {}
local loclist = vim.fn.getloclist(0)
for _, item in ipairs(loclist) do
if re:match_str(vim.fn.bufname(item.bufnr)) then
table.insert(lc, item)
end
end
if #lc ~= #loclist and #loclist ~= 0 then
save_location_list_to_history(loclist)
end
vim.fn.setloclist(0, lc)
else
local re = vim.regex(input_pattern)
local qf = {}
local qflist = vim.fn.getqflist()
for _, item in ipairs(qflist) do
if re:match_str(vim.fn.bufname(item.bufnr)) then
table.insert(qf, item)
end
end
if #qf ~= #qflist and #qflist ~= 0 then
table.insert(qf_historys, qflist)
end
vim.fn.setqflist(qf)
end
end, { buffer = ev.buf })
end,
})
vim.api.nvim_create_autocmd({ 'FileType' }, {
pattern = 'qf',
group = group,
callback = function(ev)
vim.keymap.set('n', vim.g.quickfix_mapping_filter_text or 'o', function()
local input_pattern = vim.fn.input('filter pattern:')
-- vim.cmd('noautocmd normal! :')
if is_loclist() then
local re = vim.regex(input_pattern)
local lc = {}
local loclist = vim.fn.getloclist(0)
for _, item in ipairs(loclist) do
if not re:match_str(item.text) then
table.insert(lc, item)
end
end
if #lc ~= #loclist and #loclist ~= 0 then
save_location_list_to_history(loclist)
end
vim.fn.setloclist(0, lc)
else
local re = vim.regex(input_pattern)
local qf = {}
local qflist = vim.fn.getqflist()
for _, item in ipairs(qflist) do
if not re:match_str(item.text) then
table.insert(qf, item)
end
end
if #qf ~= #qflist and #qflist ~= 0 then
table.insert(qf_historys, qflist)
end
vim.fn.setqflist(qf)
end
end, { buffer = ev.buf })
end,
})
vim.api.nvim_create_autocmd({ 'FileType' }, {
pattern = 'qf',
group = group,
callback = function(ev)
vim.keymap.set('n', vim.g.quickfix_mapping_rfilter_text or 'O', function()
local input_pattern = vim.fn.input('filter pattern:')
-- vim.cmd('noautocmd normal! :')
if is_loclist() then
local re = vim.regex(input_pattern)
local lc = {}
local loclist = vim.fn.getloclist(0)
for _, item in ipairs(loclist) do
if re:match_str(item.text) then
table.insert(lc, item)
end
end
if #lc ~= #loclist and #loclist ~= 0 then
save_location_list_to_history(loclist)
end
vim.fn.setloclist(0, lc)
else
local re = vim.regex(input_pattern)
local qf = {}
local qflist = vim.fn.getqflist()
for _, item in ipairs(qflist) do
if re:match_str(item.text) then
table.insert(qf, item)
end
end
if #qf ~= #qflist and #qflist ~= 0 then
table.insert(qf_historys, qflist)
end
vim.fn.setqflist(qf)
end
end, { buffer = ev.buf })
end,
})
vim.api.nvim_create_autocmd({ 'FileType' }, {
pattern = 'qf',
group = group,
callback = function(ev)
vim.keymap.set('n', vim.g.quickfix_mapping_undo or 'u', function()
if is_loclist() then
local locwinid = vim.fn.getloclist(0, { filewinid = 0 })
local lochis = location_history[locwinid] or {}
local lc = table.remove(lochis)
if lc then
vim.fn.setloclist(0, lc)
end
else
local qf = table.remove(qf_historys)
if qf then
vim.fn.setqflist(qf)
end
end
end, { buffer = ev.buf })
end,
})
end
return M