mirror of
https://github.com/folke/snacks.nvim
synced 2025-07-07 13:15:08 +00:00
1244 lines
35 KiB
Lua
1244 lines
35 KiB
Lua
---@class snacks.win
|
|
---@field id number
|
|
---@field buf? number
|
|
---@field scratch_buf? number
|
|
---@field win? number
|
|
---@field opts snacks.win.Config
|
|
---@field augroup? number
|
|
---@field backdrop? snacks.win
|
|
---@field keys snacks.win.Keys[]
|
|
---@field events (snacks.win.Event|{event:string|string[]})[]
|
|
---@field meta table<string, any>
|
|
---@field closed? boolean
|
|
---@overload fun(opts? :snacks.win.Config|{}): snacks.win
|
|
local M = setmetatable({}, {
|
|
__call = function(t, ...)
|
|
return t.new(...)
|
|
end,
|
|
})
|
|
M.__index = M
|
|
|
|
M.meta = {
|
|
desc = "Create and manage floating windows or splits",
|
|
}
|
|
|
|
---@class snacks.win.Keys: vim.api.keyset.keymap
|
|
---@field [1]? string
|
|
---@field [2]? string|string[]|fun(self: snacks.win): string?
|
|
---@field mode? string|string[]
|
|
|
|
---@class snacks.win.Event: vim.api.keyset.create_autocmd
|
|
---@field buf? true
|
|
---@field win? true
|
|
---@field callback? fun(self: snacks.win, ev:vim.api.keyset.create_autocmd.callback_args):boolean?
|
|
|
|
---@class snacks.win.Backdrop
|
|
---@field bg? string
|
|
---@field blend? number
|
|
---@field transparent? boolean defaults to true
|
|
---@field win? snacks.win.Config overrides the backdrop window config
|
|
|
|
---@class snacks.win.Dim
|
|
---@field width number width of the window, without borders
|
|
---@field height number height of the window, without borders
|
|
---@field row number row of the window (0-indexed)
|
|
---@field col number column of the window (0-indexed)
|
|
---@field border? boolean whether the window has a border
|
|
|
|
---@alias snacks.win.Action.fn fun(self: snacks.win):(boolean|string?)
|
|
---@alias snacks.win.Action.spec snacks.win.Action|snacks.win.Action.fn
|
|
---@class snacks.win.Action
|
|
---@field action snacks.win.Action.fn
|
|
---@field desc? string
|
|
|
|
---@class snacks.win.Config: vim.api.keyset.win_config
|
|
---@field style? string merges with config from `Snacks.config.styles[style]`
|
|
---@field show? boolean Show the window immediately (default: true)
|
|
---@field height? number|fun(self:snacks.win):number Height of the window. Use <1 for relative height. 0 means full height. (default: 0.9)
|
|
---@field width? number|fun(self:snacks.win):number Width of the window. Use <1 for relative width. 0 means full width. (default: 0.9)
|
|
---@field min_height? number Minimum height of the window
|
|
---@field max_height? number Maximum height of the window
|
|
---@field min_width? number Minimum width of the window
|
|
---@field max_width? number Maximum width of the window
|
|
---@field col? number|fun(self:snacks.win):number Column of the window. Use <1 for relative column. (default: center)
|
|
---@field row? number|fun(self:snacks.win):number Row of the window. Use <1 for relative row. (default: center)
|
|
---@field minimal? boolean Disable a bunch of options to make the window minimal (default: true)
|
|
---@field position? "float"|"bottom"|"top"|"left"|"right"
|
|
---@field border? "none"|"top"|"right"|"bottom"|"left"|"hpad"|"vpad"|"rounded"|"single"|"double"|"solid"|"shadow"|string[]|false
|
|
---@field buf? number If set, use this buffer instead of creating a new one
|
|
---@field file? string If set, use this file instead of creating a new buffer
|
|
---@field enter? boolean Enter the window after opening (default: false)
|
|
---@field backdrop? number|false|snacks.win.Backdrop Opacity of the backdrop (default: 60)
|
|
---@field wo? vim.wo|{} window options
|
|
---@field bo? vim.bo|{} buffer options
|
|
---@field b? table<string, any> buffer local variables
|
|
---@field w? table<string, any> window local variables
|
|
---@field ft? string filetype to use for treesitter/syntax highlighting. Won't override existing filetype
|
|
---@field scratch_ft? string filetype to use for scratch buffers
|
|
---@field keys? table<string, false|string|fun(self: snacks.win)|snacks.win.Keys> Key mappings
|
|
---@field on_buf? fun(self: snacks.win) Callback after opening the buffer
|
|
---@field on_win? fun(self: snacks.win) Callback after opening the window
|
|
---@field on_close? fun(self: snacks.win) Callback after closing the window
|
|
---@field fixbuf? boolean don't allow other buffers to be opened in this window
|
|
---@field text? string|string[]|fun():(string[]|string) Initial lines to set in the buffer
|
|
---@field actions? table<string, snacks.win.Action.spec> Actions that can be used in key mappings
|
|
---@field resize? boolean Automatically resize the window when the editor is resized
|
|
local defaults = {
|
|
show = true,
|
|
fixbuf = true,
|
|
relative = "editor",
|
|
position = "float",
|
|
minimal = true,
|
|
wo = {
|
|
winhighlight = "Normal:SnacksNormal,NormalNC:SnacksNormalNC,WinBar:SnacksWinBar,WinBarNC:SnacksWinBarNC",
|
|
},
|
|
bo = {},
|
|
keys = {
|
|
q = "close",
|
|
},
|
|
}
|
|
|
|
Snacks.config.style("float", {
|
|
position = "float",
|
|
backdrop = 60,
|
|
height = 0.9,
|
|
width = 0.9,
|
|
zindex = 50,
|
|
})
|
|
|
|
Snacks.config.style("help", {
|
|
position = "float",
|
|
backdrop = false,
|
|
border = "top",
|
|
row = -1,
|
|
width = 0,
|
|
height = 0.3,
|
|
})
|
|
|
|
Snacks.config.style("split", {
|
|
position = "bottom",
|
|
height = 0.4,
|
|
width = 0.4,
|
|
})
|
|
|
|
Snacks.config.style("minimal", {
|
|
wo = {
|
|
cursorcolumn = false,
|
|
cursorline = false,
|
|
cursorlineopt = "both",
|
|
colorcolumn = "",
|
|
fillchars = "eob: ,lastline:…",
|
|
list = false,
|
|
listchars = "extends:…,tab: ",
|
|
number = false,
|
|
relativenumber = false,
|
|
signcolumn = "no",
|
|
spell = false,
|
|
winbar = "",
|
|
statuscolumn = "",
|
|
wrap = false,
|
|
sidescrolloff = 0,
|
|
},
|
|
})
|
|
|
|
local SCROLL_UP, SCROLL_DOWN = Snacks.util.keycode("<c-u>"), Snacks.util.keycode("<c-d>")
|
|
|
|
local split_commands = {
|
|
editor = {
|
|
top = "topleft",
|
|
right = "vertical botright",
|
|
bottom = "botright",
|
|
left = "vertical topleft",
|
|
},
|
|
win = {
|
|
top = "aboveleft",
|
|
right = "vertical rightbelow",
|
|
bottom = "belowright",
|
|
left = "vertical leftabove",
|
|
},
|
|
}
|
|
|
|
local win_opts = {
|
|
"anchor",
|
|
"border",
|
|
"bufpos",
|
|
"col",
|
|
"external",
|
|
"fixed",
|
|
"focusable",
|
|
"footer",
|
|
"footer_pos",
|
|
"height",
|
|
"hide",
|
|
"noautocmd",
|
|
"relative",
|
|
"row",
|
|
"style",
|
|
"title",
|
|
"title_pos",
|
|
"width",
|
|
"win",
|
|
"zindex",
|
|
}
|
|
|
|
---@type table<string, string[]>
|
|
local borders = {
|
|
left = { "", "", "", "", "", "", "", "│" },
|
|
right = { "", "", "", "│", "", "", "", "" },
|
|
top = { "", "─", "", "", "", "", "", "" },
|
|
bottom = { "", "", "", "", "", "─", "", "" },
|
|
hpad = { "", "", "", " ", "", "", "", " " },
|
|
vpad = { "", " ", "", "", "", " ", "", "" },
|
|
}
|
|
|
|
Snacks.util.set_hl({
|
|
Backdrop = { bg = "#000000" },
|
|
Normal = "NormalFloat",
|
|
NormalNC = "NormalFloat",
|
|
WinBar = "Title",
|
|
WinBarNC = "SnacksWinBar",
|
|
WinKey = "Keyword",
|
|
WinKeySep = "NonText",
|
|
WinKeyDesc = "Function",
|
|
}, { prefix = "Snacks", default = true })
|
|
|
|
local id = 0
|
|
local event_stack = {} ---@type string[]
|
|
|
|
--@private
|
|
---@param ...? snacks.win.Config|string|{}
|
|
---@return snacks.win.Config
|
|
function M.resolve(...)
|
|
local done = {} ---@type table<string, boolean>
|
|
local merge = {} ---@type snacks.win.Config[]
|
|
local stack = {}
|
|
for i = 1, select("#", ...) do
|
|
local next = select(i, ...) ---@type snacks.win.Config|string?
|
|
if next then
|
|
table.insert(stack, next)
|
|
end
|
|
end
|
|
while #stack > 0 do
|
|
local next = table.remove(stack)
|
|
next = type(next) == "string" and Snacks.config.styles[next] or next
|
|
---@cast next snacks.win.Config?
|
|
if next and type(next) == "table" then
|
|
table.insert(merge, 1, next)
|
|
if next.style and not done[next.style] then
|
|
done[next.style] = true
|
|
table.insert(stack, next.style)
|
|
end
|
|
end
|
|
end
|
|
local ret = #merge == 0 and {} or #merge == 1 and merge[1] or vim.tbl_deep_extend("force", {}, unpack(merge))
|
|
ret.style = nil
|
|
return ret
|
|
end
|
|
|
|
---@param opts? snacks.win.Config|{}
|
|
---@return snacks.win
|
|
function M.new(opts)
|
|
local self = setmetatable({}, M)
|
|
id = id + 1
|
|
self.id = id
|
|
self.meta = {}
|
|
opts = M.resolve(Snacks.config.get("win", defaults), opts)
|
|
if opts.minimal then
|
|
opts = M.resolve("minimal", opts)
|
|
end
|
|
if opts.position == "float" then
|
|
opts = M.resolve("float", opts)
|
|
else
|
|
opts = M.resolve("split", opts)
|
|
local vertical = opts.position == "left" or opts.position == "right"
|
|
opts.wo.winfixheight = not vertical
|
|
opts.wo.winfixwidth = vertical
|
|
end
|
|
if opts.relative == "win" then
|
|
opts.win = opts.win or vim.api.nvim_get_current_win()
|
|
end
|
|
|
|
self.keys = {}
|
|
self.events = {}
|
|
local done = {} ---@type table<string, snacks.win.Keys>
|
|
for key, spec in pairs(opts.keys) do
|
|
if spec then
|
|
if type(spec) == "string" then
|
|
spec = { key, spec, desc = spec }
|
|
elseif type(spec) == "function" then
|
|
spec = { key, spec }
|
|
elseif type(spec) == "table" and spec[1] and not spec[2] then
|
|
spec = vim.deepcopy(spec) -- deepcopy just in case
|
|
spec[1], spec[2] = key, spec[1]
|
|
end
|
|
---@cast spec snacks.win.Keys
|
|
local lhs = Snacks.util.normkey(spec[1] or "")
|
|
local mode = type(spec.mode) == "table" and spec.mode or { spec.mode or "n" }
|
|
---@cast mode string[]
|
|
mode = #mode == 0 and { "n" } or mode
|
|
for _, m in ipairs(mode) do
|
|
local k = m .. ":" .. lhs
|
|
if done[k] then
|
|
Snacks.notify.warn(
|
|
("# Duplicate key mapping for `%s` mode=%s (check case):\n```lua\n%s\n```\n```lua\n%s\n```"):format(
|
|
lhs,
|
|
m,
|
|
vim.inspect(done[k]),
|
|
vim.inspect(spec)
|
|
)
|
|
)
|
|
end
|
|
done[k] = spec
|
|
end
|
|
table.insert(self.keys, spec)
|
|
end
|
|
end
|
|
|
|
self:on("WinClosed", self.on_close, { win = true })
|
|
|
|
-- update window size when resizing
|
|
self:on("VimResized", self.on_resize)
|
|
|
|
---@cast opts snacks.win.Config
|
|
self.opts = opts
|
|
if opts.show ~= false then
|
|
self:show()
|
|
end
|
|
return self
|
|
end
|
|
|
|
function M:on_resize()
|
|
if self.opts.resize ~= false then
|
|
self:update()
|
|
end
|
|
end
|
|
|
|
---@param actions string|string[]
|
|
function M:execute(actions)
|
|
return self:action(actions)()
|
|
end
|
|
|
|
---@param actions string|string[]
|
|
---@return (fun(): boolean|string?) action, string? desc
|
|
function M:action(actions)
|
|
actions = type(actions) == "string" and { actions } or actions
|
|
---@cast actions string[]
|
|
local desc = {} ---@type string[]
|
|
for a, name in ipairs(actions) do
|
|
desc[a] = name:gsub("_", " ")
|
|
if self.opts.actions and self.opts.actions[name] then
|
|
local action = self.opts.actions[name]
|
|
desc[a] = type(action) == "table" and action.desc and action.desc or desc[a]
|
|
end
|
|
end
|
|
return function()
|
|
for _, name in ipairs(actions) do
|
|
if self.opts.actions and self.opts.actions[name] then
|
|
local a = self.opts.actions[name]
|
|
local fn = type(a) == "function" and a or a.action
|
|
local ret = fn(self)
|
|
if ret then
|
|
return type(ret) == "string" and ret or nil
|
|
end
|
|
elseif self[name] then
|
|
self[name](self)
|
|
return
|
|
else
|
|
return name
|
|
end
|
|
end
|
|
end,
|
|
table.concat(desc, ", ")
|
|
end
|
|
|
|
---@param opts? {col_width?: number, key_width?: number, win?: snacks.win.Config}
|
|
function M:toggle_help(opts)
|
|
opts = opts or {}
|
|
local col_width, key_width = opts.col_width or 30, opts.key_width or 10
|
|
for _, win in ipairs(vim.api.nvim_tabpage_list_wins(0)) do
|
|
local buf = vim.api.nvim_win_get_buf(win)
|
|
if vim.bo[buf].filetype == "snacks_win_help" then
|
|
vim.api.nvim_win_close(win, true)
|
|
return
|
|
end
|
|
end
|
|
local ns = vim.api.nvim_create_namespace("snacks.win.help")
|
|
local win = M.new(M.resolve({ style = "help" }, opts.win or {}, {
|
|
show = false,
|
|
focusable = false,
|
|
zindex = self.opts.zindex + 1,
|
|
bo = { filetype = "snacks_win_help" },
|
|
}))
|
|
self:on("WinClosed", function()
|
|
win:close()
|
|
end, { win = true })
|
|
self:on("BufLeave", function()
|
|
win:close()
|
|
end, { buf = true })
|
|
local dim = win:dim()
|
|
|
|
-- NOTE: we use the actual buffer keymaps instead of self.keys,
|
|
-- since we want to show all keymaps, not just the ones we've defined on the window
|
|
local keys = {} ---@type vim.api.keyset.get_keymap[]
|
|
vim.list_extend(keys, vim.api.nvim_buf_get_keymap(self.buf, "n"))
|
|
vim.list_extend(keys, vim.api.nvim_buf_get_keymap(self.buf, "i"))
|
|
table.sort(keys, function(a, b)
|
|
return (a.desc or a.lhs or "") < (b.desc or b.lhs or "")
|
|
end)
|
|
|
|
local done = {} ---@type table<string, boolean>
|
|
keys = vim.tbl_filter(function(keymap)
|
|
local key = Snacks.util.normkey(keymap.lhs or "")
|
|
if done[key] or (keymap.desc and keymap.desc:find("which%-key")) then
|
|
return false
|
|
end
|
|
done[key] = true
|
|
return true
|
|
end, keys)
|
|
|
|
local cols = math.floor((dim.width - 1) / col_width)
|
|
local rows = math.ceil(#keys / cols)
|
|
win.opts.height = rows
|
|
local help = {} ---@type {[1]:string, [2]:string}[][]
|
|
local row, col = 0, 1
|
|
|
|
---@param str string
|
|
---@param len number
|
|
---@param align? "left"|"right"
|
|
local function trunc(str, len, align)
|
|
local w = vim.api.nvim_strwidth(str)
|
|
if w > len then
|
|
return vim.fn.strcharpart(str, 0, len - 1) .. "…"
|
|
end
|
|
return align == "right" and (string.rep(" ", len - w) .. str) or (str .. string.rep(" ", len - w))
|
|
end
|
|
|
|
for _, keymap in ipairs(keys) do
|
|
local key = Snacks.util.normkey(keymap.lhs or "")
|
|
row = row + 1
|
|
if row > rows then
|
|
row, col = 1, col + 1
|
|
end
|
|
help[row] = help[row] or {}
|
|
vim.list_extend(help[row], {
|
|
{ trunc(key, key_width, "right"), "SnacksWinKey" },
|
|
{ " " },
|
|
{ "➜", "SnacksWinKeySep" },
|
|
{ " " },
|
|
{ trunc(keymap.desc or "", col_width - key_width - 3), "SnacksWinKeyDesc" },
|
|
})
|
|
end
|
|
win:show()
|
|
for l, line in ipairs(help) do
|
|
vim.api.nvim_buf_set_lines(win.buf, l - 1, l, false, { "" })
|
|
vim.api.nvim_buf_set_extmark(win.buf, ns, l - 1, 0, {
|
|
virt_text = line,
|
|
virt_text_pos = "overlay",
|
|
})
|
|
end
|
|
end
|
|
|
|
---@param event string|string[]
|
|
---@param cb fun(self: snacks.win, ev:vim.api.keyset.create_autocmd.callback_args):boolean?
|
|
---@param opts? snacks.win.Event
|
|
function M:on(event, cb, opts)
|
|
opts = opts or {}
|
|
opts.callback = cb
|
|
table.insert(self.events, vim.tbl_extend("keep", { event = event }, opts))
|
|
if self:valid() then
|
|
self:_on(event, opts)
|
|
end
|
|
end
|
|
|
|
---@param event string|string[]
|
|
---@param opts snacks.win.Event
|
|
function M:_on(event, opts)
|
|
local event_opts = {} ---@type vim.api.keyset.create_autocmd
|
|
local skip = { "buf", "win", "event" }
|
|
for k, v in pairs(opts) do
|
|
if not vim.tbl_contains(skip, k) then
|
|
event_opts[k] = v
|
|
end
|
|
end
|
|
event_opts.group = event_opts.group or self.augroup
|
|
event_opts.callback = function(ev)
|
|
table.insert(event_stack, ev.event)
|
|
local ok, err = pcall(opts.callback, self, ev)
|
|
table.remove(event_stack)
|
|
return not ok and error(err) or err
|
|
end
|
|
if event_opts.pattern or event_opts.buffer then
|
|
-- don't alter the pattern or buffer
|
|
elseif opts.win then
|
|
event_opts.pattern = self.win .. ""
|
|
elseif opts.buf then
|
|
event_opts.buffer = self.buf
|
|
end
|
|
vim.api.nvim_create_autocmd(event, event_opts)
|
|
end
|
|
|
|
function M:focus()
|
|
if self:valid() then
|
|
vim.api.nvim_set_current_win(self.win)
|
|
end
|
|
end
|
|
|
|
function M:redraw()
|
|
if vim.api.nvim__redraw then
|
|
vim.api.nvim__redraw({ win = self.win, valid = false, flush = true, cursor = false })
|
|
else
|
|
vim.cmd("redraw")
|
|
end
|
|
end
|
|
|
|
---@param left? boolean
|
|
function M:hscroll(left)
|
|
vim.api.nvim_win_call(self.win, function()
|
|
vim.cmd(("normal! %s"):format(left and "zh" or "zl"))
|
|
end)
|
|
end
|
|
|
|
---@param up? boolean
|
|
function M:scroll(up)
|
|
vim.api.nvim_win_call(self.win, function()
|
|
vim.cmd(("normal! %s"):format(up and SCROLL_UP or SCROLL_DOWN))
|
|
end)
|
|
end
|
|
|
|
function M:destroy()
|
|
self:close()
|
|
self.events = {}
|
|
self.keys = {}
|
|
self.meta = {}
|
|
-- self.opts = {}
|
|
end
|
|
|
|
---@param opts? { buf: boolean }
|
|
function M:close(opts)
|
|
opts = opts or {}
|
|
local wipe = opts.buf ~= false and self.buf == self.scratch_buf
|
|
|
|
local win = self.win
|
|
local buf = wipe and self.buf
|
|
local scratch_buf = self.scratch_buf ~= self.buf and self.scratch_buf or nil
|
|
self:on_close()
|
|
|
|
self.win = nil
|
|
self.scratch_buf = nil
|
|
if buf then
|
|
self.buf = nil
|
|
end
|
|
|
|
local close = function()
|
|
if win and vim.api.nvim_win_is_valid(win) then
|
|
local ok, err = pcall(vim.api.nvim_win_close, win, true)
|
|
if not ok and (err and err:find("E444")) then
|
|
-- last window, so creat a split and close it again
|
|
vim.cmd("silent! vsplit")
|
|
pcall(vim.api.nvim_win_close, win, true)
|
|
elseif not ok then
|
|
error(err)
|
|
end
|
|
end
|
|
if buf and vim.api.nvim_buf_is_valid(buf) then
|
|
vim.api.nvim_buf_delete(buf, { force = true })
|
|
end
|
|
if scratch_buf and vim.api.nvim_buf_is_valid(scratch_buf) then
|
|
vim.api.nvim_buf_delete(scratch_buf, { force = true })
|
|
end
|
|
if self.augroup then
|
|
pcall(vim.api.nvim_del_augroup_by_id, self.augroup)
|
|
self.augroup = nil
|
|
end
|
|
end
|
|
local retries = 0
|
|
local try_close ---@type fun()
|
|
try_close = function()
|
|
local ok, err = pcall(close)
|
|
if ok or not err then
|
|
return
|
|
end
|
|
|
|
-- command window is open
|
|
if err:find("E11") then
|
|
vim.defer_fn(try_close, 200)
|
|
return
|
|
end
|
|
|
|
-- text lock
|
|
if err:find("E565") and retries < 20 then
|
|
retries = retries + 1
|
|
vim.defer_fn(try_close, 50)
|
|
return
|
|
end
|
|
|
|
if not ok then
|
|
Snacks.notify.error("Failed to close window: " .. err)
|
|
end
|
|
end
|
|
-- HACK: WinClosed is not recursive, so we need to schedule it
|
|
-- if we're in a WinClosed event
|
|
if vim.tbl_contains(event_stack, "WinClosed") or not pcall(close) then
|
|
vim.schedule(try_close)
|
|
end
|
|
end
|
|
|
|
function M:hide()
|
|
self:close({ buf = false })
|
|
return self
|
|
end
|
|
|
|
function M:toggle()
|
|
if self:valid() then
|
|
self:hide()
|
|
else
|
|
self:show()
|
|
end
|
|
return self
|
|
end
|
|
|
|
---@param title string|{[1]:string, [2]:string}[]
|
|
---@param pos? "center"|"left"|"right"
|
|
function M:set_title(title, pos)
|
|
if not self:has_border() then
|
|
return
|
|
end
|
|
if type(title) == "string" then
|
|
title = vim.trim(title)
|
|
if title ~= "" then
|
|
-- HACK: add extra space when last char is non word
|
|
-- like for icons etc
|
|
if not title:sub(-1):match("%w") then
|
|
title = title .. " "
|
|
end
|
|
title = " " .. title .. " "
|
|
end
|
|
elseif #title == 0 then
|
|
title = ""
|
|
end
|
|
pos = pos or self.opts.title_pos or "center"
|
|
if vim.deep_equal(self.opts.title, title) and self.opts.title_pos == pos then
|
|
return
|
|
end
|
|
self.opts.title = title
|
|
self.opts.title_pos = pos
|
|
if not self:valid() then
|
|
return
|
|
end
|
|
-- Don't try to update if the relative window is invalid.
|
|
-- It will be fixed once a full update is done.
|
|
local relative_win = vim.api.nvim_win_get_config(self.win).win
|
|
if relative_win and not vim.api.nvim_win_is_valid(relative_win) then
|
|
return
|
|
end
|
|
vim.api.nvim_win_set_config(self.win, {
|
|
title = self.opts.title,
|
|
title_pos = self.opts.title_pos,
|
|
})
|
|
end
|
|
|
|
---@private
|
|
function M:open_buf()
|
|
if self.buf and vim.api.nvim_buf_is_valid(self.buf) then
|
|
-- keep existing buffer
|
|
self.buf = self.buf
|
|
elseif self.scratch_buf and vim.api.nvim_buf_is_valid(self.scratch_buf) then
|
|
-- keep existing scratch buffer
|
|
self.buf = self.scratch_buf
|
|
elseif self.opts.file then
|
|
self.buf = vim.fn.bufadd(self.opts.file)
|
|
if not vim.api.nvim_buf_is_loaded(self.buf) then
|
|
vim.bo[self.buf].readonly = true
|
|
vim.bo[self.buf].swapfile = false
|
|
vim.fn.bufload(self.buf)
|
|
vim.bo[self.buf].modifiable = false
|
|
end
|
|
elseif self.opts.buf then
|
|
self.buf = self.opts.buf
|
|
else
|
|
self:scratch()
|
|
end
|
|
return self.buf
|
|
end
|
|
|
|
function M:scratch()
|
|
if self.buf == self.scratch_buf and self:buf_valid() then
|
|
return
|
|
end
|
|
self.buf = vim.api.nvim_create_buf(false, true)
|
|
vim.bo[self.buf].swapfile = false
|
|
self.scratch_buf = self.buf
|
|
local text = type(self.opts.text) == "function" and self.opts.text() or self.opts.text
|
|
text = type(text) == "string" and vim.split(text, "\n") or text
|
|
if text then
|
|
---@cast text string[]
|
|
vim.api.nvim_buf_set_lines(self.buf, 0, -1, false, text)
|
|
end
|
|
if not self.opts.bo.filetype then
|
|
if self.opts.scratch_ft then
|
|
vim.bo[self.buf].filetype = self.opts.scratch_ft
|
|
else
|
|
vim.bo[self.buf].filetype = self.opts.bo.filetype or "snacks_win"
|
|
end
|
|
vim.bo[self.buf].syntax = ""
|
|
end
|
|
if self:win_valid() then
|
|
vim.api.nvim_win_set_buf(self.win, self.buf)
|
|
end
|
|
end
|
|
|
|
---@private
|
|
function M:open_win()
|
|
local relative = self.opts.relative or "editor"
|
|
local position = self.opts.position or "float"
|
|
local enter = self.opts.enter == nil or self.opts.enter or false
|
|
if self.opts.focusable == false then
|
|
enter = false
|
|
end
|
|
local opts = self:win_opts()
|
|
if position == "float" then
|
|
self.win = vim.api.nvim_open_win(self.buf, enter, opts)
|
|
elseif position == "current" then
|
|
self.win = vim.api.nvim_get_current_win()
|
|
else
|
|
local parent = self.opts.win or 0
|
|
local vertical = position == "left" or position == "right"
|
|
if parent == 0 then
|
|
for _, win in ipairs(vim.api.nvim_tabpage_list_wins(0)) do
|
|
if
|
|
vim.w[win].snacks_win
|
|
and vim.w[win].snacks_win.relative == relative
|
|
and vim.w[win].snacks_win.position == position
|
|
then
|
|
parent = win
|
|
relative = "win"
|
|
position = vertical and "bottom" or "right"
|
|
vertical = not vertical
|
|
break
|
|
end
|
|
end
|
|
end
|
|
local cmd = split_commands[relative][position]
|
|
local size = vertical and opts.width or opts.height
|
|
vim.api.nvim_win_call(parent, function()
|
|
vim.cmd("silent noswapfile " .. cmd .. " " .. size .. "split")
|
|
vim.api.nvim_win_set_buf(0, self.buf)
|
|
self.win = vim.api.nvim_get_current_win()
|
|
end)
|
|
if enter then
|
|
vim.api.nvim_set_current_win(self.win)
|
|
end
|
|
vim.schedule(function()
|
|
self:equalize()
|
|
end)
|
|
end
|
|
vim.w[self.win].snacks_win = {
|
|
id = self.id,
|
|
position = self.opts.position,
|
|
relative = self.opts.relative,
|
|
}
|
|
end
|
|
|
|
---@private
|
|
function M:equalize()
|
|
if self:is_floating() then
|
|
return
|
|
end
|
|
local all = vim.tbl_filter(function(win)
|
|
return vim.w[win].snacks_win
|
|
and vim.w[win].snacks_win.relative == self.opts.relative
|
|
and vim.w[win].snacks_win.position == self.opts.position
|
|
end, vim.api.nvim_tabpage_list_wins(0))
|
|
if #all <= 1 then
|
|
return
|
|
end
|
|
local vertical = self.opts.position == "left" or self.opts.position == "right"
|
|
local parent_size = self:parent_size()[vertical and "height" or "width"]
|
|
local size = math.floor(parent_size / #all)
|
|
for _, win in ipairs(all) do
|
|
vim.api.nvim_win_call(win, function()
|
|
vim.cmd(("%s resize %s"):format(vertical and "horizontal" or "vertical", size))
|
|
end)
|
|
end
|
|
end
|
|
|
|
function M:update()
|
|
if self:valid() then
|
|
Snacks.util.bo(self.buf, self.opts.bo)
|
|
Snacks.util.wo(self.win, self.opts.wo)
|
|
if self:is_floating() then
|
|
local opts = self:win_opts()
|
|
opts.noautocmd = nil
|
|
vim.api.nvim_win_set_config(self.win, opts)
|
|
end
|
|
end
|
|
end
|
|
|
|
function M:on_current_tab()
|
|
return self:win_valid() and vim.api.nvim_get_current_tabpage() == vim.api.nvim_win_get_tabpage(self.win)
|
|
end
|
|
|
|
function M:show()
|
|
if self:valid() then
|
|
self:update()
|
|
return self
|
|
end
|
|
self.augroup = vim.api.nvim_create_augroup("snacks_win_" .. self.id, { clear = true })
|
|
|
|
self:open_buf()
|
|
|
|
-- buffer local variables
|
|
for k, v in pairs(self.opts.b or {}) do
|
|
vim.b[self.buf][k] = v
|
|
end
|
|
|
|
-- OPTIM: prevent treesitter or syntax highlighting to attach on FileType if it's not already enabled
|
|
local optim_hl = not vim.b[self.buf].ts_highlight and vim.bo[self.buf].syntax == ""
|
|
vim.b[self.buf].ts_highlight = optim_hl or vim.b[self.buf].ts_highlight
|
|
Snacks.util.bo(self.buf, self.opts.bo)
|
|
vim.b[self.buf].ts_highlight = not optim_hl and vim.b[self.buf].ts_highlight or nil
|
|
|
|
if self.opts.on_buf then
|
|
self.opts.on_buf(self)
|
|
end
|
|
|
|
self:open_win()
|
|
self.closed = false
|
|
-- window local variables
|
|
for k, v in pairs(self.opts.w or {}) do
|
|
vim.w[self.win][k] = v
|
|
end
|
|
if Snacks.util.is_transparent() then
|
|
self.opts.wo.winblend = 0
|
|
end
|
|
Snacks.util.wo(self.win, self.opts.wo)
|
|
if self.opts.on_win then
|
|
self.opts.on_win(self)
|
|
end
|
|
|
|
-- syntax highlighting
|
|
local ft = self.opts.ft or vim.bo[self.buf].filetype
|
|
if ft and not ft:find("^snacks_") and not vim.b[self.buf].ts_highlight and vim.bo[self.buf].syntax == "" then
|
|
local lang = vim.treesitter.language.get_lang(ft)
|
|
if not (lang and pcall(vim.treesitter.start, self.buf, lang)) then
|
|
vim.bo[self.buf].syntax = ft
|
|
end
|
|
end
|
|
|
|
for _, event in ipairs(self.events) do
|
|
self:_on(event.event, event)
|
|
end
|
|
|
|
-- swap buffers when opening a new buffer in the same window
|
|
vim.api.nvim_create_autocmd("BufWinEnter", {
|
|
group = self.augroup,
|
|
nested = true,
|
|
callback = function()
|
|
return self:fixbuf()
|
|
end,
|
|
})
|
|
|
|
self:map()
|
|
self:drop()
|
|
|
|
return self
|
|
end
|
|
|
|
function M:fixbuf()
|
|
-- window closes, so delete the autocmd
|
|
if not self:win_valid() then
|
|
return true
|
|
end
|
|
|
|
if not self:on_current_tab() then
|
|
return
|
|
end
|
|
|
|
local buf = vim.api.nvim_win_get_buf(self.win)
|
|
|
|
-- same buffer
|
|
if buf == self.buf then
|
|
return
|
|
end
|
|
|
|
-- don't swap if fixbuf is disabled
|
|
if self.opts.fixbuf == false then
|
|
self.buf = buf
|
|
-- update window options
|
|
Snacks.util.wo(self.win, self.opts.wo)
|
|
return
|
|
end
|
|
|
|
-- another buffer was opened in this window
|
|
-- find another window to swap with
|
|
local main ---@type number?
|
|
for _, win in ipairs(vim.api.nvim_tabpage_list_wins(0)) do
|
|
local win_buf = vim.api.nvim_win_get_buf(win)
|
|
local is_float = vim.api.nvim_win_get_config(win).zindex ~= nil
|
|
if win ~= self.win and not is_float then
|
|
if vim.bo[win_buf].buftype == "" or vim.b[win_buf].snacks_main or vim.w[win].snacks_main then
|
|
main = win
|
|
break
|
|
end
|
|
end
|
|
end
|
|
|
|
if main then
|
|
vim.api.nvim_win_set_buf(self.win, self.buf)
|
|
vim.api.nvim_win_set_buf(main, buf)
|
|
vim.api.nvim_set_current_win(main)
|
|
vim.cmd.stopinsert()
|
|
else
|
|
-- no main window found, so close this window
|
|
vim.api.nvim_win_set_buf(self.win, self.buf)
|
|
vim.schedule(function()
|
|
vim.cmd.stopinsert()
|
|
vim.cmd("sbuffer " .. buf)
|
|
if self.win and vim.api.nvim_win_is_valid(self.win) then
|
|
vim.api.nvim_win_close(self.win, true)
|
|
end
|
|
end)
|
|
end
|
|
end
|
|
|
|
---@param buf number
|
|
function M:set_buf(buf)
|
|
assert(self:valid(), "Window is not valid")
|
|
self.buf = buf
|
|
vim.api.nvim_win_set_buf(self.win, buf)
|
|
Snacks.util.wo(self.win, self.opts.wo)
|
|
end
|
|
|
|
function M:map()
|
|
if not self:buf_valid() then
|
|
return
|
|
end
|
|
for _, spec in pairs(self.keys) do
|
|
local opts = vim.deepcopy(spec)
|
|
opts[1] = nil
|
|
opts[2] = nil
|
|
opts.mode = nil
|
|
---@diagnostic disable-next-line: cast-type-mismatch
|
|
---@cast opts vim.keymap.set.Opts
|
|
opts.buffer = self.buf
|
|
opts.nowait = true
|
|
local rhs = spec[2]
|
|
local is_action = type(rhs) == "string" or type(rhs) == "table"
|
|
if is_action then
|
|
local desc = spec.desc
|
|
---@cast rhs string|string[]
|
|
rhs, desc = self:action(rhs)
|
|
opts.desc = opts.desc or desc
|
|
else
|
|
rhs = function()
|
|
return spec[2](self)
|
|
end
|
|
end
|
|
spec.desc = spec.desc or opts.desc
|
|
---@cast spec snacks.win.Keys
|
|
vim.keymap.set(spec.mode or "n", spec[1], rhs, opts)
|
|
end
|
|
end
|
|
|
|
---@private
|
|
function M:on_close()
|
|
-- close the backdrop
|
|
if self.backdrop then
|
|
self.backdrop:close()
|
|
self.backdrop = nil
|
|
end
|
|
if self.closed then
|
|
return
|
|
end
|
|
self.closed = true
|
|
if self.opts.on_close then
|
|
self.opts.on_close(self)
|
|
end
|
|
-- Go back to the previous window when closing,
|
|
-- and it's the current window
|
|
if vim.api.nvim_get_current_win() == self.win then
|
|
pcall(vim.cmd.wincmd, "p")
|
|
end
|
|
end
|
|
|
|
function M:add_padding()
|
|
local listchars = vim.split(self.opts.wo.listchars or "", ",")
|
|
listchars = vim.tbl_filter(function(s)
|
|
return not s:find("eol:") and s ~= ""
|
|
end, listchars)
|
|
table.insert(listchars, "eol: ")
|
|
self.opts.wo.listchars = table.concat(listchars, ",")
|
|
self.opts.wo.list = true
|
|
self.opts.wo.statuscolumn = " "
|
|
end
|
|
|
|
function M:is_floating()
|
|
return self:valid() and vim.api.nvim_win_get_config(self.win).zindex ~= nil
|
|
end
|
|
|
|
---@private
|
|
function M:drop()
|
|
if self.backdrop then
|
|
self.backdrop:close()
|
|
self.backdrop = nil
|
|
end
|
|
local backdrop = self.opts.backdrop
|
|
if not backdrop then
|
|
return
|
|
end
|
|
backdrop = type(backdrop) == "number" and { blend = backdrop } or backdrop
|
|
backdrop = backdrop == true and {} or backdrop
|
|
backdrop = vim.tbl_extend("force", { bg = "#000000", blend = 60, transparent = true }, backdrop)
|
|
---@cast backdrop snacks.win.Backdrop
|
|
|
|
if
|
|
(Snacks.util.is_transparent() and backdrop.transparent)
|
|
or not vim.o.termguicolors
|
|
or backdrop.blend == 100
|
|
or not self:is_floating()
|
|
then
|
|
return
|
|
end
|
|
|
|
local bg, winblend = backdrop.bg or "#000000", backdrop.blend
|
|
if not backdrop.transparent then
|
|
if Snacks.util.is_transparent() then
|
|
bg = nil
|
|
else
|
|
bg = Snacks.util.blend(Snacks.util.color("Normal", "bg"), bg, winblend / 100)
|
|
end
|
|
winblend = 0
|
|
end
|
|
|
|
local group = ("SnacksBackdrop_%s"):format(bg and bg:sub(2) or "T")
|
|
vim.api.nvim_set_hl(0, group, { bg = bg })
|
|
|
|
self.backdrop = M.new(M.resolve({
|
|
enter = false,
|
|
backdrop = false,
|
|
relative = "editor",
|
|
height = 0,
|
|
width = 0,
|
|
style = "minimal",
|
|
border = "none",
|
|
focusable = false,
|
|
zindex = self.opts.zindex - 1,
|
|
wo = {
|
|
winhighlight = "Normal:" .. group,
|
|
winblend = winblend,
|
|
colorcolumn = "",
|
|
},
|
|
bo = {
|
|
buftype = "nofile",
|
|
filetype = "snacks_win_backdrop",
|
|
},
|
|
}, backdrop.win))
|
|
end
|
|
|
|
function M:line(line)
|
|
return self:lines(line, line)[1] or ""
|
|
end
|
|
|
|
---@param from? number 1-indexed, inclusive
|
|
---@param to? number 1-indexed, inclusive
|
|
function M:lines(from, to)
|
|
return self:buf_valid() and vim.api.nvim_buf_get_lines(self.buf, from and from - 1 or 0, to or -1, false) or {}
|
|
end
|
|
|
|
---@param from? number 1-indexed, inclusive
|
|
---@param to? number 1-indexed, inclusive
|
|
function M:text(from, to)
|
|
return table.concat(self:lines(from, to), "\n")
|
|
end
|
|
|
|
---@return { height: number, width: number }
|
|
function M:parent_size()
|
|
return {
|
|
height = self.opts.relative == "win" and vim.api.nvim_win_get_height(self.opts.win) or vim.o.lines,
|
|
width = self.opts.relative == "win" and vim.api.nvim_win_get_width(self.opts.win) or vim.o.columns,
|
|
}
|
|
end
|
|
|
|
---@private
|
|
function M:win_opts()
|
|
local opts = {} ---@type vim.api.keyset.win_config
|
|
for _, k in ipairs(win_opts) do
|
|
opts[k] = self.opts[k]
|
|
end
|
|
|
|
opts.border = opts.border and (borders[opts.border] or opts.border) or "none"
|
|
|
|
if opts.relative == "cursor" then
|
|
self.opts.row = self.opts.row or 0
|
|
self.opts.col = self.opts.col or 0
|
|
end
|
|
|
|
local dim = self:dim()
|
|
opts.height, opts.width = dim.height, dim.width
|
|
opts.row, opts.col = dim.row, dim.col
|
|
|
|
if opts.title_pos and not opts.title then
|
|
opts.title_pos = nil
|
|
end
|
|
if opts.footer_pos and not opts.footer then
|
|
opts.footer_pos = nil
|
|
end
|
|
|
|
if vim.fn.has("nvim-0.10") == 0 then
|
|
opts.footer, opts.footer_pos = nil, nil
|
|
end
|
|
|
|
if not self:has_border() then
|
|
opts.title, opts.footer = nil, nil
|
|
opts.title_pos, opts.footer_pos = nil, nil
|
|
end
|
|
return opts
|
|
end
|
|
|
|
---@return { height: number, width: number }
|
|
function M:size()
|
|
local opts = self:win_opts()
|
|
local height = opts.height
|
|
local width = opts.width
|
|
if self:has_border() then
|
|
height = height + 2
|
|
width = width + 2
|
|
end
|
|
return { height = height, width = width }
|
|
end
|
|
|
|
function M:has_border()
|
|
return self.opts.border and self.opts.border ~= "" and self.opts.border ~= "none"
|
|
end
|
|
|
|
--- Calculate the size of the border
|
|
function M:border_size()
|
|
-- The array specifies the eight
|
|
-- chars building up the border in a clockwise fashion
|
|
-- starting with the top-left corner.
|
|
-- { "╔", "═" ,"╗", "║", "╝", "═", "╚", "║" }
|
|
local border = self:has_border() and self.opts.border or { "" }
|
|
border = type(border) == "string" and borders[border] or border
|
|
border = type(border) == "string" and { "x" } or border
|
|
assert(type(border) == "table", "Invalid border type")
|
|
---@cast border string[]
|
|
while #border < 8 do
|
|
vim.list_extend(border, border)
|
|
end
|
|
-- remove border hl groups
|
|
border = vim.tbl_map(function(b)
|
|
return type(b) == "table" and b[1] or b
|
|
end, border)
|
|
local function size(from, to)
|
|
for i = from, to do
|
|
if border[i] ~= "" then
|
|
return 1
|
|
end
|
|
end
|
|
return 0
|
|
end
|
|
---@type { top: number, right: number, bottom: number, left: number }
|
|
return {
|
|
top = size(1, 3),
|
|
right = size(3, 5),
|
|
bottom = size(5, 7),
|
|
left = math.max(size(7, 8), size(1, 1)),
|
|
}
|
|
end
|
|
|
|
function M:border_text_width()
|
|
if not self:has_border() then
|
|
return 0
|
|
end
|
|
local ret = 0
|
|
for _, t in ipairs({ "title", "footer" }) do
|
|
local str = self.opts[t] or {}
|
|
str = type(str) == "string" and { str } or str
|
|
---@cast str (string|string[])[]
|
|
ret = math.max(ret, #table.concat(
|
|
vim.tbl_map(function(s)
|
|
return type(s) == "string" and s or s[1]
|
|
end, str),
|
|
""
|
|
))
|
|
end
|
|
return ret
|
|
end
|
|
|
|
function M:buf_valid()
|
|
return self.buf and vim.api.nvim_buf_is_valid(self.buf)
|
|
end
|
|
|
|
function M:win_valid()
|
|
return self.win and vim.api.nvim_win_is_valid(self.win)
|
|
end
|
|
|
|
function M:valid()
|
|
return self:win_valid() and self:buf_valid() and vim.api.nvim_win_get_buf(self.win) == self.buf
|
|
end
|
|
|
|
---@param parent? snacks.win.Dim
|
|
function M:dim(parent)
|
|
parent = parent or self:parent_size()
|
|
---@type snacks.win.Dim
|
|
local ret = {
|
|
height = 0,
|
|
width = 0,
|
|
col = 0,
|
|
row = 0,
|
|
border = self:has_border(),
|
|
}
|
|
|
|
---@param s? number|fun(win:snacks.win):number? size
|
|
---@param ps number parent size
|
|
local function size(s, ps, border_offset)
|
|
s = type(s) == "function" and s(self) or s or 0
|
|
---@cast s number
|
|
if s == 0 then -- full size
|
|
return ps - border_offset
|
|
elseif s < 1 then -- relative size
|
|
return math.floor(ps * s) - border_offset
|
|
end
|
|
return s
|
|
end
|
|
|
|
---@param p? number|fun(win:snacks.win):number? pos
|
|
---@param s number size
|
|
---@param ps number parent size
|
|
local function pos(p, s, ps, border_from, border_to)
|
|
p = type(p) == "function" and p(self) or p
|
|
---@cast p number?
|
|
if self.opts.relative == "cursor" then
|
|
return p or 0
|
|
end
|
|
if not p then -- center
|
|
return math.floor((ps - s) / 2) - border_from
|
|
end
|
|
---@cast p number
|
|
if p < 0 then -- negative position
|
|
return ps - s + p - border_from - border_to
|
|
elseif p < 1 and p > 0 then -- relative position
|
|
return math.floor(ps * p) + border_from
|
|
end
|
|
return p
|
|
end
|
|
|
|
local border = self:border_size()
|
|
|
|
ret.height = size(self.opts.height, parent.height, border.top + border.bottom)
|
|
ret.height = math.max(ret.height, self.opts.min_height or 0, 1)
|
|
ret.height = math.min(ret.height, self.opts.max_height or ret.height, parent.height)
|
|
ret.height = math.max(ret.height, 1)
|
|
|
|
ret.width = size(self.opts.width, parent.width, border.left + border.right)
|
|
ret.width = math.max(ret.width, self.opts.min_width or 0, 1)
|
|
ret.width = math.min(ret.width, self.opts.max_width or ret.width, parent.width)
|
|
ret.width = math.max(ret.width, 1)
|
|
|
|
ret.row = pos(self.opts.row, ret.height, parent.height, border.top, border.bottom)
|
|
ret.col = pos(self.opts.col, ret.width, parent.width, border.left, border.right)
|
|
|
|
return ret
|
|
end
|
|
|
|
return M
|