fix: failed to write correct entry into prompt log file at very begin… (#2581)
This commit is contained in:
@@ -38,17 +38,31 @@ function M.log_prompt(request)
|
||||
}
|
||||
|
||||
-- Remove any existing entries with the same input
|
||||
if #entries > 1 then
|
||||
for i = #entries - 1, 1, -1 do
|
||||
if entries[i].input == entry.input then table.remove(entries, i) end
|
||||
end
|
||||
-- Add the new entry
|
||||
for i = #entries - 1, 1, -1 do
|
||||
if entries[i].input == entry.input then table.remove(entries, i) end
|
||||
end
|
||||
|
||||
-- Add the new entry
|
||||
if #entries > 0 then
|
||||
table.insert(entries, #entries, entry)
|
||||
idx = #entries - 1
|
||||
else
|
||||
table.insert(entries, entry)
|
||||
end
|
||||
|
||||
local max = Config.prompt_logger.max_entries
|
||||
|
||||
-- Left trim entries if the count exceeds max_entries
|
||||
-- We need to keep the last entry (current input) and trim from the beginning
|
||||
if max > 0 and #entries > max + 1 then
|
||||
-- Calculate how many entries to remove
|
||||
local to_remove = #entries - max - 1
|
||||
-- Remove oldest entries from the beginning
|
||||
for _ = 1, to_remove do
|
||||
table.remove(entries, 1)
|
||||
end
|
||||
end
|
||||
|
||||
local file = io.open(log_file, "w")
|
||||
if file then
|
||||
-- Write all entries to the log file, except the last one
|
||||
@@ -68,8 +82,16 @@ local function _read_log(delta)
|
||||
return entries[idx + 1]
|
||||
end
|
||||
|
||||
local function update_current_input()
|
||||
if idx == #entries - 1 then
|
||||
local lines = vim.api.nvim_buf_get_lines(0, 0, -1, false)
|
||||
entries[#entries].input = table.concat(lines, "\n")
|
||||
end
|
||||
end
|
||||
|
||||
function M.on_log_retrieve(delta)
|
||||
return function()
|
||||
update_current_input()
|
||||
local res = _read_log(delta)
|
||||
if not res or not res.input then
|
||||
vim.notify("No log entry found.", vim.log.levels.WARN)
|
||||
@@ -83,11 +105,4 @@ function M.on_log_retrieve(delta)
|
||||
end
|
||||
end
|
||||
|
||||
function M.update_current_input()
|
||||
if idx == #entries - 1 then
|
||||
local lines = vim.api.nvim_buf_get_lines(0, 0, -1, false)
|
||||
entries[#entries].input = table.concat(lines, "\n")
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
|
||||
Reference in New Issue
Block a user