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

feat(api): enable customizable calls functions (#457)

Browse Source 
Signed-off-by: Aaron Pham <contact@aarnphm.xyz>
This commit is contained in:
Aaron Pham
2024-09-02 12:22:48 -04:00
committed by GitHub
parent d520f09333
commit 7266661413
9 changed files with 291 additions and 267 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
lua/avante/selection.lua
View File

@@ -367,10 +367,8 @@ function Selection:create_editing_input()
end,
})
local function submit_input()
local lines = api.nvim_buf_get_lines(bufnr, 0, -1, false)
local input = lines[1] or ""
---@param input string
local function submit_input(input)
local full_response = ""
local start_line = self.selection.range.start.line
local finish_line = self.selection.range.finish.line
@@ -426,8 +424,18 @@ function Selection:create_editing_input()
})
end
vim.keymap.set("i", Config.mappings.submit.insert, submit_input, { buffer = bufnr, noremap = true, silent = true })
vim.keymap.set("n", Config.mappings.submit.normal, submit_input, { buffer = bufnr, noremap = true, silent = true })
---@return string
local get_bufnr_input = function()
local lines = api.nvim_buf_get_lines(bufnr, 0, -1, false)
return lines[1] or ""
end
vim.keymap.set("i", Config.mappings.submit.insert, function()
submit_input(get_bufnr_input())
end, { buffer = bufnr, noremap = true, silent = true })
vim.keymap.set("n", Config.mappings.submit.normal, function()
submit_input(get_bufnr_input())
end, { buffer = bufnr, noremap = true, silent = true })
vim.keymap.set("n", "<Esc>", function()
self:close_editing_input()
end, { buffer = bufnr })
@@ -461,6 +469,15 @@ function Selection:create_editing_input()
end
end,
})
api.nvim_create_autocmd("User", {
pattern = "AvanteEditSubmitted",
callback = function(ev)
if ev.data and ev.data.request then
submit_input(ev.data.request)
end
end,
})
end
function Selection:setup_autocmds()