free/gemini
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

debug completion

Browse Source
This commit is contained in:
Jonas Widen 2025-03-17 07:40:02 +01:00
parent 600d0f8b43
commit bd75f7dd32
1 changed files with 13 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
lua/gemini/completion.lua
View File

@ -58,37 +58,32 @@ local function show_suggestion(suggestion, start_col)
local line = vim.api.nvim_win_get_cursor(0)[1] - 1 local line = vim.api.nvim_win_get_cursor(0)[1] - 1
local line_text = vim.api.nvim_buf_get_lines(0, line, line + 1, true)[1] local line_text = vim.api.nvim_buf_get_lines(0, line, line + 1, true)[1]
-- Get text before cursor on current line -- Get text after cursor on current line
local prefix = string.sub(line_text, 1, start_col) local input_after_cursor = string.sub(line_text, start_col + 1)
-- Process first line of suggestion
local first_line = suggestion_lines[1] local first_line = suggestion_lines[1]
-- Find the longest common prefix between the input and suggestion -- Find the longest common prefix between the existing text and suggestion
local common_length = 0 local common_length = 0
local input_after_cursor = string.sub(line_text, start_col + 1) while common_length < #input_after_cursor and common_length < #first_line do
local suggestion_text = first_line
while common_length < #input_after_cursor and common_length < #suggestion_text do
if string.sub(input_after_cursor, common_length + 1, common_length + 1) == if string.sub(input_after_cursor, common_length + 1, common_length + 1) ==
string.sub(suggestion_text, common_length + 1, common_length + 1) then string.sub(first_line, common_length + 1, common_length + 1) then
common_length = common_length + 1 common_length = common_length + 1
else else
break break
end end
end end
-- Only show the part of suggestion that doesn't overlap with existing text -- Only show the part of suggestion that doesn't overlap
first_line = string.sub(suggestion_text, common_length + 1) first_line = string.sub(first_line, common_length + 1)
if first_line == "" and #suggestion_lines == 1 then return end if first_line == "" and #suggestion_lines == 1 then return end
current_suggestion.text = suggestion current_suggestion.text = suggestion
current_suggestion.start_col = start_col current_suggestion.start_col = start_col
-- Show first line as virtual text only if there's non-overlapping content -- Show first line as virtual text
if first_line ~= "" then if first_line ~= "" then
vim.api.nvim_buf_set_extmark(0, current_suggestion.namespace_id, line, start_col + common_length, { vim.api.nvim_buf_set_extmark(0, current_suggestion.namespace_id, line, start_col, {
virt_text = {{first_line, 'GeminiSuggestion'}}, virt_text = {{first_line, 'GeminiSuggestion'}},
virt_text_pos = 'inline', virt_text_pos = 'inline',
virt_text_hide = true, virt_text_hide = true,
@ -102,13 +97,13 @@ local function show_suggestion(suggestion, start_col)
table.insert(virt_lines, {{suggestion_lines[i], 'GeminiSuggestion'}}) table.insert(virt_lines, {{suggestion_lines[i], 'GeminiSuggestion'}})
end end
vim.api.nvim_buf_set_extmark(0, current_suggestion.namespace_id, line, start_col + common_length, { vim.api.nvim_buf_set_extmark(0, current_suggestion.namespace_id, line, start_col, {
virt_lines = virt_lines, virt_lines = virt_lines,
virt_lines_above = false, virt_lines_above = false,
}) })
end end
debug_print("Showing suggestion (%d lines, common prefix length: %d)", #suggestion_lines, common_length) debug_print("Showing suggestion: '%s' (common length: %d)", first_line, common_length)
end end
function M.accept_suggestion() function M.accept_suggestion()
@ -119,7 +114,7 @@ function M.accept_suggestion()
local line = vim.api.nvim_win_get_cursor(0)[1] - 1 local line = vim.api.nvim_win_get_cursor(0)[1] - 1
local col = vim.api.nvim_win_get_cursor(0)[2] local col = vim.api.nvim_win_get_cursor(0)[2]
-- Only accept if we're still at or before the suggestion start -- Only accept if we're still at or after the suggestion start
if col < current_suggestion.start_col then if col < current_suggestion.start_col then
return false return false
end end
@ -132,7 +127,7 @@ function M.accept_suggestion()
local suggestion_lines = vim.split(current_suggestion.text, "\n") local suggestion_lines = vim.split(current_suggestion.text, "\n")
if #suggestion_lines == 0 then return false end if #suggestion_lines == 0 then return false end
-- For the first line, only insert the part that doesn't overlap with existing text -- For the first line, only insert the part that doesn't overlap
local first_line = suggestion_lines[1] local first_line = suggestion_lines[1]
local common_length = 0 local common_length = 0
while common_length < #text_after_cursor and common_length < #first_line do while common_length < #text_after_cursor and common_length < #first_line do