1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
local copilot_fn = function()
local copilot_config = {
schema = {
model = {
default = "claude-3.5-sonnet",
},
},
}
return require("codecompanion.adapters").extend("copilot", copilot_config)
end
local supported_adapters = {
copilot = copilot_fn,
}
local function save_path()
local Path = require("plenary.path")
local p = Path:new(vim.fn.stdpath("data") .. "/cc_saves")
p:mkdir({ parents = true })
return p
end
--- Load a saved codecompanion.nvim chat file into a new CodeCompanion chat buffer.
--- Usage: CodeCompanionLoad
vim.api.nvim_create_user_command("CodeCompanionLoad", function()
local fzf = require("fzf-lua")
local function select_adapter(filepath)
local adapters = vim.tbl_keys(supported_adapters)
fzf.fzf_exec(adapters, {
prompt = "Select CodeCompanion Adapter> ",
actions = {
["default"] = function(selected)
local adapter = selected[1]
-- Open new CodeCompanion chat with selected adapter
vim.cmd("CodeCompanionChat " .. adapter)
-- Read contents of saved chat file
local lines = vim.fn.readfile(filepath)
-- Get the current buffer (which should be the new CodeCompanion chat)
local current_buf = vim.api.nvim_get_current_buf()
-- Paste contents into the new chat buffer
vim.api.nvim_buf_set_lines(current_buf, 0, -1, false, lines)
end,
},
})
end
local function start_picker()
local files = vim.fn.glob(save_path() .. "/*", false, true)
fzf.fzf_exec(files, {
prompt = "Saved CodeCompanion Chats | <c-r>: remove >",
previewer = "builtin",
actions = {
["default"] = function(selected)
if #selected > 0 then
local filepath = selected[1]
select_adapter(filepath)
end
end,
["ctrl-r"] = function(selected)
if #selected > 0 then
local filepath = selected[1]
os.remove(filepath)
-- Refresh the picker
start_picker()
end
end,
},
})
end
start_picker()
end, {})
--- Save the current codecompanion.nvim chat buffer to a file in the save_folder.
--- Usage: CodeCompanionSave <filename>.md
---@param opts table
vim.api.nvim_create_user_command("CodeCompanionSave", function(opts)
local codecompanion = require("codecompanion")
local success, chat = pcall(function()
return codecompanion.buf_get_chat(0)
end)
if not success or chat == nil then
vim.notify("CodeCompanionSave should only be called from CodeCompanion chat buffers", vim.log.levels.ERROR)
return
end
if #opts.fargs == 0 then
vim.notify("CodeCompanionSave requires at least 1 arg to make a file name", vim.log.levels.ERROR)
end
local save_name = table.concat(opts.fargs, "-") .. ".md"
local save_file = save_path():joinpath(save_name)
local lines = vim.api.nvim_buf_get_lines(0, 0, -1, false)
save_file:write(table.concat(lines, "\n"), "w")
end, { nargs = "*" })
|
