-- lazy.nvim
{
	"robitx/gp.nvim",
	config = function()
		require("gp").setup(conf)
        	-- shortcuts might be setup here (see Usage > Shortcuts in Readme)
	end,
}

-- packer.nvim
use({
    "robitx/gp.nvim",
    config = function()
        require("gp").setup(conf)
        -- shortcuts might be setup here (see Usage > Shortcuts in Readme)
    end,
})

local config = {
	-- required openai api key
	openai_api_key = os.getenv("OPENAI_API_KEY"),
	-- prefix for all commands
	cmd_prefix = "Gp",
	-- example hook functions
	hooks = {
		InspectPlugin = function(plugin)
			print(string.format("Plugin structure:\n%s", vim.inspect(plugin)))
		end,
	},

	-- directory for storing chat files
	chat_dir = os.getenv("HOME") .. "/.local/share/nvim/gp/chats",
	-- chat model
	chat_model = "gpt-3.5-turbo-16k",
	-- chat temperature
	chat_temperature = 0.7,
	-- chat model system prompt
	chat_system_prompt = "You are a general AI assistant.",
	-- chat user prompt prefix
	chat_user_prefix = "🗨:",
	-- chat assistant prompt prefix
	chat_assistant_prefix = "🤖:",
	-- chat topic generation prompt
	chat_topic_gen_prompt = "Summarize the topic of our conversation above"
		.. " in two or three words. Respond only with those words.",
	-- chat topic model
	chat_topic_gen_model = "gpt-3.5-turbo-16k",

	-- command prompt prefix for asking user for input
	command_prompt_prefix = "🤖 ~ ",
	-- command model
	command_model = "gpt-3.5-turbo-16k",
	-- command system prompt
	command_system_prompt = "You are an AI that strictly generates pure formated final code, without providing any comments or explanations.",

	-- templates
	template_selection = "I have the following code from {{filename}}:\n\n```{{filetype}}\n{{selection}}\n```\n\n{{command}}",
	template_rewrite = "I have the following code from {{filename}}:\n\n```{{filetype}}\n{{selection}}\n```\n\n{{command}}"
		.. "\n\nRespond just with the pure formated final code. !!And please: No ``` code ``` blocks.",
	template_command = "{{command}}",
}

...

-- call setup on your config
require("gp").setup(config)

-- shortcuts might be setup here (see Usage > Shortcuts in Readme)

local function keymapOptions(desc)
    return {
        noremap = true,
        silent = true,
        nowait = true,
        desc = "GPT prompt " .. desc,
    }
end

-- Chat commands
vim.keymap.set({"n", "i"}, "<C-g>c", "<cmd>GpChatNew<cr>", keymapOptions("New Chat"))
vim.keymap.set({"n", "i"}, "<C-g>f", "<cmd>GpChatFinder<cr>", keymapOptions("Chat Finder"))
vim.keymap.set({"n", "i"}, "<C-g><C-g>", "<cmd>GpChatRespond<cr>", keymapOptions("Chat Respond"))
vim.keymap.set({"n", "i"}, "<C-g>d", "<cmd>GpChatDelete<cr>", keymapOptions("Chat Delete"))

-- Prompt commands
vim.keymap.set({"n", "i"}, "<C-g>i", "<cmd>GpInline<cr>", keymapOptions("Inline"))
vim.keymap.set({"n", "i"}, "<C-g>a", "<cmd>GpAppend<cr>", keymapOptions("Append"))
vim.keymap.set({"n", "i"}, "<C-g>b", "<cmd>GpPrepend<cr>", keymapOptions("Prepend"))
vim.keymap.set({"n", "i"}, "<C-g>e", "<cmd>GpEnew<cr>", keymapOptions("Enew"))
vim.keymap.set({"n", "i"}, "<C-g>p", "<cmd>GpPopup<cr>", keymapOptions("Popup"))

-- Visual commands
vim.keymap.set("v", "<C-g>c", "<cmd>GpVisualChatNew<cr>", keymapOptions("Visual Chat New"))
vim.keymap.set("v", "<C-g>r", "<cmd>GpVisualRewrite<cr>", keymapOptions("Visual Rewrite"))
vim.keymap.set("v", "<C-g>a", "<cmd>GpVisualAppend<cr>", keymapOptions("Visual Append"))
vim.keymap.set("v", "<C-g>b", "<cmd>GpVisualPrepend<cr>", keymapOptions("Visual Prepend"))
vim.keymap.set("v", "<C-g>e", "<cmd>GpVisualEnew<cr>", keymapOptions("Visual Enew"))
vim.keymap.set("v", "<C-g>p", "<cmd>GpVisualPopup<cr>", keymapOptions("Visual Popup"))

-- VISUAL mode mappings
-- s, x, v modes are handled the same way by which_key
require("which-key").register({
    -- ...
	["<C-g>"] = {
		c = { "<cmd>GpVisualChatNew<cr>", "Visual Chat New" },

		r = { "<cmd>GpVisualRewrite<cr>", "Visual Rewrite" },
		a = { "<cmd>GpVisualAppend<cr>", "Visual Append" },
		b = { "<cmd>GpVisualPrepend<cr>", "Visual Prepend" },
		e = { "<cmd>GpVisualEnew<cr>", "Visual Enew" },
		p = { "<cmd>GpVisualPopup<cr>", "Visual Popup" },
	},
    -- ...
}, {
	mode = "v", -- VISUAL mode
	prefix = "",
	buffer = nil, 
	silent = true,
	noremap = true,
	nowait = true,
})

-- NORMAL mode mappings
require("which-key").register({
    -- ...
	["<C-g>"] = {
		c = { "<cmd>GpChatNew<cr>", "New Chat" },
		f = { "<cmd>GpChatFinder<cr>", "Chat Finder" },
		d = { "<cmd>GpChatDelete<cr>", "Chat Delete" },
		["<C-g>"] = { "<cmd>GpChatRespond<cr>", "Chat Respond" },

		i = { "<cmd>GpInline<cr>", "Inline" },
		a = { "<cmd>GpAppend<cr>", "Append" },
		b = { "<cmd>GpPrepend<cr>", "Prepend" },
		e = { "<cmd>GpEnew<cr>", "Enew" },
		p = { "<cmd>GpPopup<cr>", "Popup" },
	},
    -- ...
}, {
	mode = "n", -- NORMAL mode
	prefix = "",
	buffer = nil, 
	silent = true,
	noremap = true,
	nowait = true,
})

-- INSERT mode mappings
require("which-key").register({
    -- ...
	["<C-g>"] = {
		c = { "<cmd>GpChatNew<cr>", "New Chat" },
		f = { "<cmd>GpChatFinder<cr>", "Chat Finder" },
		d = { "<cmd>GpChatDelete<cr>", "Chat Delete" },
		["<C-g>"] = { "<cmd>GpChatRespond<cr>", "Chat Respond" },

		i = { "<cmd>GpInline<cr>", "Inline" },
		a = { "<cmd>GpAppend<cr>", "Append" },
		b = { "<cmd>GpPrepend<cr>", "Prepend" },
		e = { "<cmd>GpEnew<cr>", "Enew" },
		p = { "<cmd>GpPopup<cr>", "Popup" },
	},
    -- ...
}, {
	mode = "i", -- INSERT mode
	prefix = "",
	buffer = nil, 
	silent = true,
	noremap = true,
	nowait = true,
})