mirror of
https://github.com/LazyVim/LazyVim.git
synced 2026-07-22 04:21:08 +00:00
Merge 8c179b1efe into 459a4c3b10
This commit is contained in:
G
GitHub
commit
4fa091afc5
1 changed files with 296 additions and 0 deletions
296
lua/lazyvim/plugins/extras/ai/opencode.lua
Normal file
296
lua/lazyvim/plugins/extras/ai/opencode.lua
Normal file
|
|
@ -0,0 +1,296 @@
|
|||
if not vim.fn.executable("opencode") then
|
||||
vim.notify("opencode: binary not installed. See opencode.ai", vim.log.levels.ERROR)
|
||||
return {}
|
||||
end
|
||||
|
||||
local function ask(prompt, opts)
|
||||
if not vim.fn.executable("opencode") then
|
||||
vim.notify("opencode: binary not installed. See opencode.ai", vim.log.levels.ERROR)
|
||||
return
|
||||
end
|
||||
require("opencode").ask(prompt, opts)
|
||||
end
|
||||
|
||||
return {
|
||||
-- which-key
|
||||
{
|
||||
"folke/which-key.nvim",
|
||||
optional = true,
|
||||
opts = function(_, opts)
|
||||
opts.spec = opts.spec or {}
|
||||
table.insert(opts.spec, { "<leader>a", group = "OpenCode" })
|
||||
table.insert(opts.spec, { "<leader>ag", group = "git" })
|
||||
table.insert(opts.spec, { "<leader>ap", group = "prompt" })
|
||||
end,
|
||||
},
|
||||
|
||||
{
|
||||
"Nickvandyke/opencode.nvim",
|
||||
version = "*",
|
||||
dependencies = {
|
||||
"nvim-lua/plenary.nvim",
|
||||
},
|
||||
cmd = { "OpenCode", "OpenCodeToggle" },
|
||||
keys = {
|
||||
{ "gO", false },
|
||||
|
||||
{
|
||||
"<leader>aa",
|
||||
function()
|
||||
require("opencode").toggle()
|
||||
end,
|
||||
mode = { "n" },
|
||||
desc = "Toggle OpenCode",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>aq",
|
||||
function()
|
||||
require("opencode").stop()
|
||||
end,
|
||||
mode = { "n" },
|
||||
desc = "Stop/Close OpenCode",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>as",
|
||||
function()
|
||||
require("opencode").select()
|
||||
end,
|
||||
mode = { "n", "x" },
|
||||
desc = "Select action",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>ai",
|
||||
function()
|
||||
ask("", { submit = true, focus = false })
|
||||
end,
|
||||
mode = { "n", "x" },
|
||||
desc = "Ask (empty)",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>aI",
|
||||
function()
|
||||
ask("@this: ", { submit = true, focus = false })
|
||||
end,
|
||||
mode = { "n", "x" },
|
||||
desc = "Ask with context",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>ab",
|
||||
function()
|
||||
ask("@buffer ", { submit = true, focus = false })
|
||||
end,
|
||||
mode = { "n", "x" },
|
||||
desc = "Ask about buffer",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>agd",
|
||||
function()
|
||||
ask("@diff ", { submit = true, focus = false })
|
||||
end,
|
||||
mode = { "n", "x" },
|
||||
desc = "Git diff",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>ax",
|
||||
function()
|
||||
ask("@quickfix ", { submit = true, focus = false })
|
||||
end,
|
||||
mode = { "n", "x" },
|
||||
desc = "Ask with quickfix",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>av",
|
||||
function()
|
||||
ask("@visible ", { submit = true, focus = false })
|
||||
end,
|
||||
mode = { "n", "x" },
|
||||
desc = "Ask with visible text",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>aB",
|
||||
function()
|
||||
ask("@buffers ", { submit = true, focus = false })
|
||||
end,
|
||||
mode = { "n", "x" },
|
||||
desc = "Ask with all buffers",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>am",
|
||||
function()
|
||||
ask("@marks ", { submit = true, focus = false })
|
||||
end,
|
||||
mode = { "n", "x" },
|
||||
desc = "Ask with marks",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>ape",
|
||||
function()
|
||||
ask("@this explain: ", { submit = true, focus = false })
|
||||
end,
|
||||
mode = { "n", "x" },
|
||||
desc = "Explain",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>apf",
|
||||
function()
|
||||
ask("@this fix: ", { submit = true, focus = false })
|
||||
end,
|
||||
mode = { "n", "x" },
|
||||
desc = "Fix",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>apd",
|
||||
function()
|
||||
ask("@this diagnose: ", { submit = true, focus = false })
|
||||
end,
|
||||
mode = { "n", "x" },
|
||||
desc = "Diagnose",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>apr",
|
||||
function()
|
||||
ask("@this review: ", { submit = true, focus = false })
|
||||
end,
|
||||
mode = { "n", "x" },
|
||||
desc = "Review",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>apt",
|
||||
function()
|
||||
ask("@this test: ", { submit = true, focus = false })
|
||||
end,
|
||||
mode = { "n", "x" },
|
||||
desc = "Test",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>apo",
|
||||
function()
|
||||
ask("@this optimize: ", { submit = true, focus = false })
|
||||
end,
|
||||
mode = { "n", "x" },
|
||||
desc = "Optimize",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>an",
|
||||
function()
|
||||
require("opencode").command("session.new")
|
||||
end,
|
||||
mode = { "n" },
|
||||
desc = "New session",
|
||||
},
|
||||
|
||||
{
|
||||
"<leader>ac",
|
||||
function()
|
||||
require("opencode").command("session.close")
|
||||
end,
|
||||
mode = { "n" },
|
||||
desc = "Close session",
|
||||
},
|
||||
|
||||
{
|
||||
"go",
|
||||
function()
|
||||
return require("opencode").operator("@this ")
|
||||
end,
|
||||
expr = true,
|
||||
mode = { "n", "x" },
|
||||
desc = "Add range to OpenCode",
|
||||
},
|
||||
|
||||
{
|
||||
"goo",
|
||||
function()
|
||||
return require("opencode").operator("@this ") .. "_"
|
||||
end,
|
||||
expr = true,
|
||||
mode = { "n" },
|
||||
desc = "Add line to OpenCode",
|
||||
},
|
||||
},
|
||||
config = function(_, opts)
|
||||
local config = require("opencode.config")
|
||||
local opencode_cmd = "opencode --port"
|
||||
local snacks_terminal_opts = {
|
||||
win = {
|
||||
position = "right",
|
||||
on_win = function(win)
|
||||
-- Set up keymaps and cleanup for an arbitrary terminal
|
||||
require("opencode.terminal").setup(win.win)
|
||||
end,
|
||||
enter = false,
|
||||
},
|
||||
-- Otherwise, we get error `Terminal exited with code` when closing Opencode because
|
||||
-- the `stop` command already closes the terminal window, but the Snacks.terminal `TermClose`
|
||||
-- autocmd kicks in and tries to close it again, causing the error.
|
||||
auto_close = false,
|
||||
}
|
||||
|
||||
config.opts = vim.tbl_deep_extend(
|
||||
"force",
|
||||
config.opts,
|
||||
{
|
||||
server = {
|
||||
start = function()
|
||||
require("snacks.terminal").open(opencode_cmd, snacks_terminal_opts)
|
||||
end,
|
||||
stop = function()
|
||||
local win, created = require("snacks.terminal").get(opencode_cmd, snacks_terminal_opts)
|
||||
if win and created then
|
||||
win:close()
|
||||
vim.cmd.stopinsert()
|
||||
elseif win and not created then
|
||||
win:close()
|
||||
end
|
||||
end,
|
||||
toggle = function()
|
||||
require("snacks.terminal").toggle(opencode_cmd, snacks_terminal_opts)
|
||||
end,
|
||||
},
|
||||
},
|
||||
-- We merge here also `opts` so we can bypass the following limitation with `vim.g` https://github.com/nickjvandyke/opencode.nvim/issues/36
|
||||
opts or {},
|
||||
vim.g.opencode_opts or {}
|
||||
)
|
||||
|
||||
vim.api.nvim_create_autocmd({ "TermOpen" }, {
|
||||
group = vim.api.nvim_create_augroup("opencode_integrated", { clear = true }),
|
||||
pattern = "*:opencode --port*",
|
||||
callback = function(event)
|
||||
vim.bo[event.buf].buflisted = false
|
||||
vim.bo[event.buf].filetype = "opencode"
|
||||
|
||||
vim.keymap.set("t", "<C-U>", function()
|
||||
require("opencode").command("session.half.page.up")
|
||||
end, { buffer = event.buf, desc = "Half scroll back" })
|
||||
vim.keymap.set("t", "<C-D>", function()
|
||||
require("opencode").command("session.half.page.down")
|
||||
end, { buffer = event.buf, desc = "Half scroll forward" })
|
||||
vim.keymap.set("t", "<C-B>", function()
|
||||
require("opencode").command("session.page.up")
|
||||
end, { buffer = event.buf, desc = "Scroll backward" })
|
||||
vim.keymap.set("t", "<C-F>", function()
|
||||
require("opencode").command("session.page.down")
|
||||
end, { buffer = event.buf, desc = "Scroll forward" })
|
||||
end,
|
||||
})
|
||||
end,
|
||||
},
|
||||
}
|
||||
Loading…
Add table
Reference in a new issue