132 lines
3.8 KiB
Lua
132 lines
3.8 KiB
Lua
-- lua/gemini/init.lua
|
|
|
|
local api = require("gemini.api")
|
|
local chat = require("gemini.chat")
|
|
local config = require("gemini.config")
|
|
local completion = require("gemini.completion")
|
|
|
|
local M = {}
|
|
|
|
local function get_current_buffer_content()
|
|
local lines = vim.api.nvim_buf_get_lines(0, 0, -1, false)
|
|
return table.concat(lines, "\n")
|
|
end
|
|
|
|
function M.prompt_query(context)
|
|
local prompt = context and "Gemini (with buffer context): " or "Gemini: "
|
|
vim.ui.input({ prompt = prompt }, function(input)
|
|
if input then
|
|
M.query(input, context)
|
|
end
|
|
end)
|
|
end
|
|
|
|
function M.query(prompt, context)
|
|
chat.set_context(context)
|
|
chat.create_window()
|
|
|
|
local initial_content = "User: " .. prompt .. "\n\nAssistant: Thinking..."
|
|
chat.update_content(initial_content, true, true)
|
|
|
|
api.get_response(prompt, context, function(response, error)
|
|
if error then
|
|
vim.notify("Failed to get response: " .. error, vim.log.levels.ERROR)
|
|
return
|
|
end
|
|
|
|
local content = "User: " .. prompt .. "\n\nAssistant: " .. response
|
|
chat.update_content(content, true)
|
|
end)
|
|
end
|
|
|
|
function M.setup(opts)
|
|
config.setup(opts)
|
|
|
|
-- Set up completion
|
|
completion.setup()
|
|
|
|
-- Enable completion by default for all buffers
|
|
vim.api.nvim_create_autocmd("BufEnter", {
|
|
pattern = "*",
|
|
callback = function()
|
|
vim.b.gemini_completion_enabled = true
|
|
end
|
|
})
|
|
|
|
-- Completion keymaps
|
|
vim.keymap.set('i', '<leader>gg', function()
|
|
debug_print("Keymap triggered")
|
|
-- Enable completion for current buffer
|
|
vim.b.gemini_completion_enabled = true
|
|
completion.trigger_completion()
|
|
end, { desc = 'Trigger Gemini completion' })
|
|
|
|
pcall(vim.treesitter.language.require_language, "markdown")
|
|
|
|
-- Set up commands
|
|
vim.api.nvim_create_user_command("Gemini", function(opts)
|
|
if opts.args == "" then
|
|
vim.notify("Please provide a prompt for Gemini.", vim.log.levels.WARN)
|
|
return
|
|
end
|
|
M.query(opts.args)
|
|
end, {
|
|
desc = "Query Google AI",
|
|
nargs = "+",
|
|
complete = "shellcmd",
|
|
})
|
|
|
|
vim.api.nvim_create_user_command("GeminiClearChat", function()
|
|
api.clear_conversation()
|
|
chat.clear()
|
|
vim.notify("Chat history cleared", vim.log.levels.INFO)
|
|
end, {
|
|
desc = "Clear Gemini chat history"
|
|
})
|
|
|
|
-- Chat keymaps
|
|
vim.keymap.set("n", "<leader>gc", function()
|
|
M.prompt_query()
|
|
end, { desc = "Chat with Gemini AI" })
|
|
|
|
vim.keymap.set("n", "<leader>gs", function()
|
|
M.prompt_query(get_current_buffer_content())
|
|
end, { desc = "Chat with Gemini AI (with buffer context)" })
|
|
|
|
vim.keymap.set("n", "<leader>gq", function()
|
|
api.clear_conversation()
|
|
chat.clear()
|
|
vim.notify("Chat history cleared", vim.log.levels.INFO)
|
|
end, { desc = "Clear Gemini chat history" })
|
|
|
|
-- Add a command to toggle completion
|
|
vim.api.nvim_create_user_command("GeminiToggleCompletion", function()
|
|
vim.b.gemini_completion_enabled = not vim.b.gemini_completion_enabled
|
|
vim.notify(
|
|
string.format("Gemini completion %s",
|
|
vim.b.gemini_completion_enabled and "enabled" or "disabled"),
|
|
vim.log.levels.INFO
|
|
)
|
|
end, {
|
|
desc = "Toggle Gemini completion"
|
|
})
|
|
end
|
|
|
|
function M.complete(findstart, base)
|
|
if findstart == 1 then
|
|
local line = vim.api.nvim_get_current_line()
|
|
local col = vim.api.nvim_win_get_cursor(0)[2]
|
|
local start = col
|
|
while start > 0 and string.match(line:sub(start, start), "[%w_]") do
|
|
start = start - 1
|
|
end
|
|
return start
|
|
else
|
|
-- Don't return "Loading..." immediately
|
|
-- Instead, return empty list and let completion handle it
|
|
return {}
|
|
end
|
|
end
|
|
|
|
return M
|