phyer/avante.nvim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add rounded opt (#139)

Browse Source
This commit is contained in:
yetone 2024-08-21 23:31:11 +08:00 committed by GitHub
parent 8a62e3e225
commit 58bc74d4f8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 25 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lua/avante/config.lua
View File

@ -100,8 +100,9 @@ M.defaults = {
windows = { windows = {
wrap_line = true, -- similar to vim.o.wrap wrap_line = true, -- similar to vim.o.wrap
width = 30, -- default % based on available width width = 30, -- default % based on available width
sidebar = { sidebar_header = {
align = "center", -- left, center, right for title align = "center", -- left, center, right for title
rounded = true,
}, },
}, },
--- @class AvanteConflictUserConfig --- @class AvanteConflictUserConfig

31
lua/avante/sidebar.lua
View File

@ -331,19 +331,25 @@ function Sidebar:do_render_header(winid, bufnr, header_text, hl, reverse_hl)
return return
end end
local reversed_hl_off = 0
if Config.windows.sidebar_header.rounded then
reversed_hl_off = 1
header_text = "" .. header_text .. ""
else
header_text = " " .. header_text .. " " header_text = " " .. header_text .. " "
end
local width = api.nvim_win_get_width(winid) local width = api.nvim_win_get_width(winid)
local header_text_length = vim.fn.strdisplaywidth(header_text) local header_text_length = vim.fn.strdisplaywidth(header_text)
local prefix_padding, suffix_padding = 0, 0 local prefix_padding, suffix_padding = 0, 0
if Config.windows.sidebar.align == "center" then if Config.windows.sidebar_header.align == "center" then
prefix_padding = math.floor((width - header_text_length) / 2) prefix_padding = math.floor((width - header_text_length) / 2)
suffix_padding = width - header_text_length - prefix_padding suffix_padding = width - header_text_length - prefix_padding
elseif Config.windows.sidebar.align == "right" then elseif Config.windows.sidebar_header.align == "right" then
prefix_padding = width - header_text_length prefix_padding = width - header_text_length
suffix_padding = 0 suffix_padding = 0
elseif Config.windows.sidebar.align == "left" then elseif Config.windows.sidebar_header.align == "left" then
suffix_padding = width - header_text_length suffix_padding = width - header_text_length
prefix_padding = 0 prefix_padding = 0
end end
@ -352,18 +358,25 @@ function Sidebar:do_render_header(winid, bufnr, header_text, hl, reverse_hl)
local suffix_padding_text = string.rep(" ", suffix_padding) local suffix_padding_text = string.rep(" ", suffix_padding)
Utils.unlock_buf(bufnr) Utils.unlock_buf(bufnr)
api.nvim_buf_set_lines(bufnr, 0, -1, false, { prefix_padding_text .. header_text .. suffix_padding_text }) api.nvim_buf_set_lines(bufnr, 0, -1, false, { prefix_padding_text .. header_text .. suffix_padding_text })
api.nvim_buf_add_highlight(bufnr, -1, "WinSeparator", 0, 0, #prefix_padding_text) api.nvim_buf_add_highlight(bufnr, -1, "WinSeparator", 0, 0, #prefix_padding_text - reversed_hl_off)
api.nvim_buf_add_highlight(bufnr, -1, reverse_hl, 0, #prefix_padding_text, #prefix_padding_text) api.nvim_buf_add_highlight(bufnr, -1, reverse_hl, 0, #prefix_padding_text, #prefix_padding_text + reversed_hl_off)
api.nvim_buf_add_highlight(bufnr, -1, hl, 0, #prefix_padding_text, #prefix_padding_text + #header_text) api.nvim_buf_add_highlight(
bufnr,
-1,
hl,
0,
#prefix_padding_text + reversed_hl_off,
#prefix_padding_text + #header_text - reversed_hl_off * 3
)
api.nvim_buf_add_highlight( api.nvim_buf_add_highlight(
bufnr, bufnr,
-1, -1,
reverse_hl, reverse_hl,
0, 0,
#prefix_padding_text + #header_text, #prefix_padding_text + #header_text - reversed_hl_off * 3,
#prefix_padding_text + #header_text #prefix_padding_text + #header_text - reversed_hl_off * 2
) )
api.nvim_buf_add_highlight(bufnr, -1, "WinSeparator", 0, #prefix_padding_text + #header_text, -1) api.nvim_buf_add_highlight(bufnr, -1, "WinSeparator", 0, #prefix_padding_text + #header_text - reversed_hl_off, -1)
Utils.lock_buf(bufnr) Utils.lock_buf(bufnr)
end end