5fa4f701dd
* perf(ai): token streaming with quick refactoring Signed-off-by: Aaron Pham <contact@aarnphm.xyz> * fix: window resize and AvanteSwitchProvider Signed-off-by: Aaron Pham <contact@aarnphm.xyz> * revert: config change Signed-off-by: Aaron Pham <contact@aarnphm.xyz> * chore: return early Signed-off-by: Aaron Pham <contact@aarnphm.xyz> --------- Signed-off-by: Aaron Pham <contact@aarnphm.xyz>
73 lines
2.3 KiB
Lua
73 lines
2.3 KiB
Lua
local Config = require("avante.config")
|
|
local api = vim.api
|
|
|
|
---@class avante.View
|
|
---@field buf integer
|
|
---@field win integer
|
|
---@field RESULT_BUF_NAME string
|
|
local View = {}
|
|
|
|
local RESULT_BUF_NAME = "AVANTE_RESULT"
|
|
|
|
function View:new()
|
|
return setmetatable({ buf = nil, win = nil }, { __index = View })
|
|
end
|
|
|
|
---setup view buffer
|
|
---@param split_command string A split command to position the side bar to
|
|
---@param size integer a given % to resize the chat window
|
|
---@return avante.View
|
|
function View:setup(split_command, size)
|
|
-- create a scratch unlisted buffer
|
|
self.buf = api.nvim_create_buf(false, true)
|
|
|
|
-- set filetype
|
|
api.nvim_set_option_value("filetype", "Avante", { buf = self.buf })
|
|
api.nvim_set_option_value("bufhidden", "wipe", { buf = self.buf })
|
|
api.nvim_set_option_value("modifiable", false, { buf = self.buf })
|
|
api.nvim_set_option_value("swapfile", false, { buf = self.buf })
|
|
|
|
-- create a split
|
|
vim.cmd(split_command)
|
|
|
|
--get current window and attach the buffer to it
|
|
self.win = api.nvim_get_current_win()
|
|
api.nvim_win_set_buf(self.win, self.buf)
|
|
|
|
vim.cmd("vertical resize " .. size)
|
|
|
|
-- win stuff
|
|
api.nvim_set_option_value("spell", false, { win = self.win })
|
|
api.nvim_set_option_value("signcolumn", "no", { win = self.win })
|
|
api.nvim_set_option_value("foldcolumn", "0", { win = self.win })
|
|
api.nvim_set_option_value("number", false, { win = self.win })
|
|
api.nvim_set_option_value("relativenumber", false, { win = self.win })
|
|
api.nvim_set_option_value("winfixwidth", true, { win = self.win })
|
|
api.nvim_set_option_value("list", false, { win = self.win })
|
|
api.nvim_set_option_value("wrap", Config.windows.wrap_line, { win = self.win })
|
|
api.nvim_set_option_value("winhl", "", { win = self.win })
|
|
api.nvim_set_option_value("linebreak", true, { win = self.win }) -- only has effect when wrap=true
|
|
api.nvim_set_option_value("breakindent", true, { win = self.win }) -- only has effect when wrap=true
|
|
|
|
-- buffer stuff
|
|
xpcall(function()
|
|
api.nvim_buf_set_name(self.buf, RESULT_BUF_NAME)
|
|
end, function(_) end)
|
|
|
|
return self
|
|
end
|
|
|
|
function View:close()
|
|
if self.win then
|
|
api.nvim_win_close(self.win, true)
|
|
self.win = nil
|
|
self.buf = nil
|
|
end
|
|
end
|
|
|
|
function View:is_open()
|
|
return self.win and self.buf and api.nvim_buf_is_valid(self.buf) and api.nvim_win_is_valid(self.win)
|
|
end
|
|
|
|
return View
|