You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

284 lines
8.3 KiB
Lua

3 years ago
if not pcall(require, "fzf") then
return
end
local action = require("fzf.actions").action
3 years ago
local core = require "fzf-lua.core"
local path = require "fzf-lua.path"
local utils = require "fzf-lua.utils"
local config = require "fzf-lua.config"
local actions = require "fzf-lua.actions"
local fn, api = vim.fn, vim.api
local M = {}
local function getbufnumber(line)
return tonumber(string.match(line, "%[(%d+)"))
end
local function getfilename(line)
-- return string.match(line, "%[.*%] (.+)")
-- greedy match anything after last nbsp
return line:match("[^" .. utils.nbsp .. "]*$")
3 years ago
end
M.buffers = function(opts)
opts = config.normalize_opts(opts, config.globals.buffers)
if not opts then return end
3 years ago
local act = action(function (items, fzf_lines, _)
-- only preview first item
local item = items[1]
local buf = getbufnumber(item)
if api.nvim_buf_is_loaded(buf) then
return api.nvim_buf_get_lines(buf, 0, fzf_lines, false)
else
local name = getfilename(item)
if fn.filereadable(name) ~= 0 then
return fn.readfile(name, "", fzf_lines)
end
return "UNLOADED: " .. name
end
end)
coroutine.wrap(function ()
local items = {}
local bufnrs = vim.tbl_filter(function(b)
if 1 ~= vim.fn.buflisted(b) then
return false
end
-- only hide unloaded buffers if opts.show_all_buffers is false, keep them listed if true or nil
if opts.show_all_buffers == false and not vim.api.nvim_buf_is_loaded(b) then
return false
end
if opts.ignore_current_buffer and b == vim.api.nvim_get_current_buf() then
return false
end
if opts.cwd_only and not path.is_relative(vim.api.nvim_buf_get_name(b), vim.loop.cwd()) then
3 years ago
return false
end
return true
3 years ago
end, opts._list_bufs and opts._list_bufs() or vim.api.nvim_list_bufs())
3 years ago
if not next(bufnrs) then return end
local header_line = false
local buffers = {}
for _, bufnr in ipairs(bufnrs) do
local flag = bufnr == vim.fn.bufnr('') and '%' or (bufnr == vim.fn.bufnr('#') and '#' or ' ')
local element = {
bufnr = bufnr,
flag = flag,
info = vim.fn.getbufinfo(bufnr)[1],
}
if opts.sort_lastused and (flag == "#" or flag == "%") then
if flag == "%" then header_line = true end
local idx = ((buffers[1] ~= nil and buffers[1].flag == "%") and 2 or 1)
table.insert(buffers, idx, element)
else
table.insert(buffers, element)
end
end
for _, buf in pairs(buffers) do
-- local hidden = buf.info.hidden == 1 and 'h' or 'a'
local hidden = ''
local readonly = vim.api.nvim_buf_get_option(buf.bufnr, 'readonly') and '=' or ' '
local changed = buf.info.changed == 1 and '+' or ' '
local flags = hidden .. readonly .. changed
local leftbr = utils.ansi_codes.clear('[')
local rightbr = utils.ansi_codes.clear(']')
local bufname = string.format("%s:%s",
utils._if(#buf.info.name>0, path.relative(buf.info.name, vim.loop.cwd()), "[No Name]"),
utils._if(buf.info.lnum>0, buf.info.lnum, ""))
if buf.flag == '%' then
flags = utils.ansi_codes.red(buf.flag) .. flags
leftbr, rightbr = '[', ']'
if not header_line then
leftbr = utils.ansi_codes.green(leftbr)
rightbr = utils.ansi_codes.green(rightbr)
bufname = utils.ansi_codes.green(bufname)
end
3 years ago
elseif buf.flag == '#' then
flags = utils.ansi_codes.cyan(buf.flag) .. flags
else
flags = utils.nbsp .. flags
3 years ago
end
local bufnrstr = string.format("%s%s%s", leftbr,
utils.ansi_codes.yellow(string.format(buf.bufnr)), rightbr)
local buficon = ''
local hl = ''
3 years ago
if opts.file_icons then
if utils.is_term_bufname(buf.info.name) then
-- get shell-like icon for terminal buffers
buficon, hl = core.get_devicon(buf.info.name, "sh")
else
local filename = path.tail(buf.info.name)
local extension = path.extension(filename)
buficon, hl = core.get_devicon(filename, extension)
end
3 years ago
if opts.color_icons then
buficon = utils.ansi_codes[hl](buficon)
3 years ago
end
end
local item_str = string.format("%s%s%s%s%s%s%s",
utils._if(header_line and vim.tbl_isempty(items),
string.format("%-16s", bufnrstr),
string.format("%-32s", bufnrstr)),
utils.nbsp,
flags,
utils.nbsp,
buficon,
utils.nbsp,
bufname)
3 years ago
table.insert(items, item_str)
end
opts.preview = act
opts._fzf_cli_args = utils._if(
header_line and not opts.ignore_current_buffer,
'--header-lines=1', ''
)
local selected = core.fzf(opts, items)
3 years ago
if not selected then return end
if #selected > 1 then
for i = 2, #selected do
selected[i] = tostring(getbufnumber(selected[i]))
end
end
actions.act(opts.actions, selected)
3 years ago
end)()
end
M.lines = function(opts)
opts = config.normalize_opts(opts, config.globals.lines)
M.buffer_lines(opts)
end
M.blines = function(opts)
opts = config.normalize_opts(opts, config.globals.blines)
opts.current_buffer_only = true
M.buffer_lines(opts)
end
M.buffer_lines = function(opts)
if not opts then return end
local buffers = {}
if opts.current_buffer_only then
table.insert(buffers, vim.api.nvim_get_current_buf())
else
buffers = vim.tbl_filter(function(b)
if 1 ~= vim.fn.buflisted(b) or utils.is_term_buffer(b) then
return false
end
-- only hide unloaded buffers if opts.show_all_buffers is false, keep them listed if true or nil
if opts.show_all_buffers == false and not vim.api.nvim_buf_is_loaded(b) then
return false
end
if opts.ignore_current_buffer and b == vim.api.nvim_get_current_buf() then
return false
end
if opts.cwd_only and not path.is_relative(vim.api.nvim_buf_get_name(b), vim.loop.cwd()) then
return false
end
return true
end, vim.api.nvim_list_bufs())
end
coroutine.wrap(function()
local items = {}
for _, bufnr in ipairs(buffers) do
if api.nvim_buf_is_loaded(bufnr) then
local bufname = path.basename(api.nvim_buf_get_name(bufnr))
local data = api.nvim_buf_get_lines(bufnr, 0, -1, false)
local buficon, hl
if opts.file_icons then
local filename = path.tail(bufname)
local extension = path.extension(filename)
buficon, hl = core.get_devicon(filename, extension)
if opts.color_icons then
buficon = utils.ansi_codes[hl](buficon)
end
end
for l, text in ipairs(data) do
table.insert(items, ("[%s]%s%s%s%s:%s: %s"):format(
utils.ansi_codes.yellow(tostring(bufnr)),
utils.nbsp,
buficon or '',
buficon and utils.nbsp or '',
utils.ansi_codes.magenta(bufname),
utils.ansi_codes.green(tostring(l)),
text))
end
end
end
-- ignore bufnr when searching
-- disable multi-select
opts.nomulti = true
opts.preview_window = 'hidden:right:0'
opts._fzf_cli_args = "--delimiter=']' --nth 2,-1"
local selected = core.fzf(opts, items)
if not selected then return end
-- get the line number
local line = tonumber(selected[2]:match(":(%d+):"))
if #selected > 1 then
for i = 2, #selected do
selected[i] = tostring(getbufnumber(selected[i]))
end
end
actions.act(opts.actions, selected)
if line then vim.api.nvim_win_set_cursor(0, {line, 0}) end
end)()
end
3 years ago
M.tabs = function(opts)
opts = config.normalize_opts(opts, config.globals.tabs)
if not opts then return end
local buf_to_tab = {}
opts._list_bufs = function()
local res = {}
for _, t in ipairs(vim.api.nvim_list_tabpages()) do
local w = vim.api.nvim_tabpage_get_win(t)
local b = vim.api.nvim_win_get_buf(w)
buf_to_tab[b] = t
table.insert(res, b)
end
return res
end
opts.nomulti = true
opts.preview_window = 'hidden:right:0'
opts.fzf_cli_args = "--delimiter=']' --nth 2,-1" --with-nth 2"
opts.actions["default"] = function(selected)
local bufnr = tonumber(selected[2])
local tabnr = buf_to_tab[bufnr]
vim.cmd("tabn " .. tabnr)
end
M.buffers(opts)
end
3 years ago
return M