98 lines
2.7 KiB
Lua
98 lines
2.7 KiB
Lua
-- lua/chatgpt_nvim/init.lua
|
|
-- Modified to:
|
|
-- 1) Use YAML for config and response parsing.
|
|
-- 2) Assume ChatGPT response is YAML of form:
|
|
-- files:
|
|
-- - path: "somefile.lua"
|
|
-- content: |
|
|
-- multi line
|
|
-- content
|
|
--
|
|
|
|
local M = {}
|
|
|
|
local context = require('chatgpt_nvim.context')
|
|
local handler = require('chatgpt_nvim.handler')
|
|
local config = require('chatgpt_nvim.config')
|
|
|
|
local ok_yaml, lyaml = pcall(require, "lyaml")
|
|
|
|
local function copy_to_clipboard(text)
|
|
vim.fn.setreg('+', text)
|
|
end
|
|
|
|
-- Parse the response from ChatGPT in YAML.
|
|
local function parse_response(raw)
|
|
if not ok_yaml then
|
|
vim.api.nvim_err_writeln("lyaml is not available. Please install lyaml for YAML parsing.")
|
|
return nil
|
|
end
|
|
local ok, data = pcall(lyaml.load, raw)
|
|
if not ok or not data then
|
|
vim.api.nvim_err_writeln("Failed to parse YAML response.")
|
|
return nil
|
|
end
|
|
-- lyaml.load returns a list of documents. We assume only one document is given.
|
|
data = data[1]
|
|
return data
|
|
end
|
|
|
|
function M.run_chatgpt_command()
|
|
local conf = config.load()
|
|
local user_input = vim.fn.input("Message for O1 Model: ")
|
|
if user_input == "" then
|
|
print("No input provided.")
|
|
return
|
|
end
|
|
|
|
local dirs = conf.directories or {"."}
|
|
local project_structure = context.get_project_structure(dirs)
|
|
local all_files = context.get_project_files(dirs)
|
|
local file_sections = context.get_file_contents(all_files)
|
|
local current_file_content, current_file_path = context.get_current_file()
|
|
local readme_content = context.get_readme_content()
|
|
|
|
local sections = {
|
|
conf.initial_prompt .. "\n",
|
|
user_input,
|
|
"\n\nproject structure:\n",
|
|
project_structure,
|
|
"\n\nproject files:\n"
|
|
}
|
|
|
|
-- Add all other files in configured directories
|
|
table.insert(sections, file_sections)
|
|
|
|
local prompt = table.concat(sections, "\n")
|
|
|
|
-- Copy prompt to clipboard
|
|
copy_to_clipboard(prompt)
|
|
print("Prompt copied to clipboard! Please paste it into the ChatGPT O1 model and get the YAML response.")
|
|
end
|
|
|
|
function M.run_chatgpt_paste_command()
|
|
print("Reading ChatGPT YAML response from clipboard...")
|
|
local raw = handler.get_clipboard_content()
|
|
if raw == "" then
|
|
vim.api.nvim_err_writeln("Clipboard is empty. Please copy the YAML response from ChatGPT first.")
|
|
return
|
|
end
|
|
|
|
local data = parse_response(raw)
|
|
if not data or not data.files then
|
|
vim.api.nvim_err_writeln("No 'files' field found in the YAML response.")
|
|
return
|
|
end
|
|
|
|
for _, fileinfo in ipairs(data.files) do
|
|
if fileinfo.path and fileinfo.content then
|
|
handler.write_file(fileinfo.path, fileinfo.content)
|
|
print("Wrote file: " .. fileinfo.path)
|
|
else
|
|
vim.api.nvim_err_writeln("Invalid file entry. Must have 'path' and 'content'.")
|
|
end
|
|
end
|
|
end
|
|
|
|
return M
|