feat: changed to and interactive workflow
This commit is contained in:
@@ -1,39 +1,89 @@
|
||||
local M = {}
|
||||
|
||||
local context = require('chatgpt_nvim.context')
|
||||
local handler = require('chatgpt_nvim.handler')
|
||||
|
||||
function M.run_chatgpt_command()
|
||||
vim.schedule(function()
|
||||
-- Prompt the user for input
|
||||
local user_input = vim.fn.input("Message for O1 Model: ")
|
||||
if user_input == "" then
|
||||
print("No input provided.")
|
||||
return
|
||||
-- Prompt the user for input
|
||||
local user_input = vim.fn.input("Message for O1 Model: ")
|
||||
if user_input == "" then
|
||||
print("No input provided.")
|
||||
return
|
||||
end
|
||||
|
||||
local project_structure = context.get_project_structure()
|
||||
local current_file_content, current_file_path = context.get_current_file()
|
||||
local readme_content = context.get_readme_content()
|
||||
|
||||
local sections = {
|
||||
"Following you will get instructions for a development project. First you get a prompt of the user which starts with \"<<<CGP User Message\" in a line and ends with \"<<<CGP User Message End\" in a line.",
|
||||
"\nThen you get the project structure which starts with \"<<<CGPT Project Structure\" in a line and ends with \"<<<CGPT Project Structure End\" in a line",
|
||||
"\nThen you get the current file for context. It contains at the first line the file path, followed by its content.It starts with \"<<<CGPT Current File\" in a line and ends with \"<<<CGPT Current File End\" in a line",
|
||||
"\nLastly you get the content of the README.md for context if it is present. It starts with \"<<<CGPT README\" in a line and ends with \"<<<CGPT README End\" in a line",
|
||||
"\nReturn code blocks for the changes.",
|
||||
"\nIf it makes sense, also update the README.md to reflect the changes made.",
|
||||
"\n\n<<<CGPT User Message\n",
|
||||
user_input,
|
||||
"\n<<<CGPT User Message END",
|
||||
"\n\n<<<CGPT Project Structure\n",
|
||||
project_structure,
|
||||
"\n<<<CGPT Project Structure END",
|
||||
"\n\n<<<CGPT Current File\n",
|
||||
current_file_path .. "\n" .. current_file_content,
|
||||
"\n\n<<<CGPT Current File END\n",
|
||||
}
|
||||
|
||||
if readme_content then
|
||||
table.insert(sections, "\n\n<<<CGPT README\n" .. readme_content .. "\n<<<CGPT README END")
|
||||
end
|
||||
|
||||
local prompt = table.concat(sections, "\n")
|
||||
|
||||
-- Copy prompt to clipboard so user can paste it into the website form.
|
||||
vim.fn.setreg('+', prompt)
|
||||
print("Prompt copied to clipboard! Please paste it into the website form and get the response from the ChatGPT O1 model.")
|
||||
print("Press <Enter> once you've pasted the prompt into the website and have the first file ready to copy.")
|
||||
vim.fn.input("") -- wait for user to press Enter
|
||||
|
||||
while true do
|
||||
-- Ask user for filepath
|
||||
local filepath = ""
|
||||
while true do
|
||||
print("Please copy the FILE PATH (the first line of the code block) to your clipboard.")
|
||||
print("Press <Enter> when done.")
|
||||
vim.fn.input("") -- Wait for user confirmation
|
||||
filepath = handler.get_clipboard_content()
|
||||
if filepath == "" then
|
||||
vim.api.nvim_err_writeln("Clipboard is empty. Please copy the file path and try again.")
|
||||
else
|
||||
print("Got file path: " .. filepath)
|
||||
break
|
||||
end
|
||||
end
|
||||
|
||||
local project_structure = context.get_project_structure()
|
||||
local current_file_content, current_file_path = context.get_current_file()
|
||||
local readme_content = context.get_readme_content()
|
||||
|
||||
local sections = {
|
||||
"User Message:\n" .. user_input,
|
||||
"\nProject Structure:\n" .. project_structure,
|
||||
"\nCurrent File:\nPath: " .. current_file_path .. "\n" .. current_file_content,
|
||||
"\nInstructions:\nPlease return code blocks for the changes. Each code block must start with the file path on the first line, followed by the code.",
|
||||
"Do not comment out the file path. Just plain text.",
|
||||
"If multiple files are needed, return multiple code blocks each starting with a file path."
|
||||
}
|
||||
|
||||
if readme_content then
|
||||
table.insert(sections, "\nREADME Content:\n" .. readme_content)
|
||||
-- Ask user for file content
|
||||
local filecontent = ""
|
||||
while true do
|
||||
print("Now copy the FILE CONTENT (everything after the first line) to your clipboard.")
|
||||
print("Press <Enter> when done.")
|
||||
vim.fn.input("") -- Wait for user confirmation
|
||||
filecontent = handler.get_clipboard_content()
|
||||
if filecontent == "" then
|
||||
vim.api.nvim_err_writeln("Clipboard is empty. Please copy the file content and try again.")
|
||||
else
|
||||
break
|
||||
end
|
||||
end
|
||||
|
||||
local prompt = table.concat(sections, "\n")
|
||||
handler.write_file(filepath, filecontent)
|
||||
print("Wrote file: " .. filepath)
|
||||
|
||||
-- Copy prompt to clipboard so user can paste it into the website form.
|
||||
vim.fn.setreg('+', prompt)
|
||||
print("Prompt (including README if found) copied to clipboard! Paste it into the website form.")
|
||||
end)
|
||||
local another = vim.fn.input("Do you want to paste another file? (y/n): ")
|
||||
if another:lower() ~= "y" then
|
||||
handler.finish()
|
||||
break
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
|
||||
Reference in New Issue
Block a user