mirror of
https://github.com/harivansh-afk/forge.nvim.git
synced 2026-04-15 04:03:29 +00:00
Harivansh Rathi
f3f66ae560
Wrap the highlighted segment render path so forge only consumes the first return value from fzf-lua's ansi_from_hl helper, and add a regression spec covering the newer two-value return shape. Co-authored-by: Codex <noreply@openai.com>
76 lines
1.7 KiB
Lua
76 lines
1.7 KiB
Lua
local M = {}
|
|
|
|
local fzf_args = (vim.env.FZF_DEFAULT_OPTS or '')
|
|
:gsub('%-%-bind=[^%s]+', '')
|
|
:gsub('%-%-color=[^%s]+', '')
|
|
|
|
---@param key string
|
|
---@return string
|
|
local function to_fzf_key(key)
|
|
if key == '<cr>' then
|
|
return 'default'
|
|
end
|
|
local result = key:gsub('<c%-(%a)>', function(ch)
|
|
return 'ctrl-' .. ch:lower()
|
|
end)
|
|
return result
|
|
end
|
|
|
|
---@param segments forge.Segment[]
|
|
---@return string
|
|
local function render(segments)
|
|
local utils = require('fzf-lua.utils')
|
|
local parts = {}
|
|
for _, seg in ipairs(segments) do
|
|
if seg[2] then
|
|
table.insert(parts, (utils.ansi_from_hl(seg[2], seg[1])))
|
|
else
|
|
table.insert(parts, seg[1])
|
|
end
|
|
end
|
|
return table.concat(parts)
|
|
end
|
|
|
|
---@param opts forge.PickerOpts
|
|
function M.pick(opts)
|
|
local cfg = require('forge').config()
|
|
local keys = cfg.keys
|
|
if keys == false then
|
|
keys = {}
|
|
end
|
|
local bindings = keys[opts.picker_name] or {}
|
|
|
|
local lines = {}
|
|
for i, entry in ipairs(opts.entries) do
|
|
lines[i] = ('%d\t%s'):format(i, render(entry.display))
|
|
end
|
|
|
|
local fzf_actions = {}
|
|
for _, def in ipairs(opts.actions) do
|
|
local key = def.name == 'default' and '<cr>' or bindings[def.name]
|
|
if key then
|
|
fzf_actions[to_fzf_key(key)] = function(selected)
|
|
if not selected[1] then
|
|
def.fn(nil)
|
|
return
|
|
end
|
|
local idx = tonumber(selected[1]:match('^(%d+)'))
|
|
def.fn(idx and opts.entries[idx] or nil)
|
|
end
|
|
end
|
|
end
|
|
|
|
require('fzf-lua').fzf_exec(lines, {
|
|
fzf_args = fzf_args,
|
|
prompt = opts.prompt or '',
|
|
fzf_opts = {
|
|
['--ansi'] = '',
|
|
['--no-multi'] = '',
|
|
['--with-nth'] = '2..',
|
|
['--delimiter'] = '\t',
|
|
},
|
|
actions = fzf_actions,
|
|
})
|
|
end
|
|
|
|
return M
|