feat(prompt_logger): Save prompts to disk and add buffer navigation via keymaps (#2075)

lua/avante/utils/init.lua

@@ -9,6 +9,7 @@ local lsp = vim.lsp
 ---@field path avante.utils.path
 ---@field environment avante.utils.environment
 ---@field lsp avante.utils.lsp
+---@field logger avante.utils.promptLogger
 local M = {}
 
 setmetatable(M, {

lua/avante/utils/promptLogger.lua

@@ -0,0 +1,99 @@
+local Config = require("avante.config")
+local Utils = require("avante.utils")
+
+---@class avante.utils.promptLogger
+local M = {}
+
+function M.log_prompt(request)
+  local log_dir = Config.prompt_logger.log_dir
+  local log_file = Utils.join_paths(log_dir, "avante_prompt_" .. os.date("%Y%m%d_%H%M%S") .. ".log")
+
+  if vim.fn.isdirectory(log_dir) == 0 then vim.fn.mkdir(log_dir, "p") end
+
+  local file = io.open(log_file, "w")
+  if file then
+    file:write(request)
+    file:close()
+    if Config.prompt_logger and Config.prompt_logger.fortune_cookie_on_success then
+      local handle = io.popen("fortune -s -n 100")
+      if handle then
+        local fortune_msg = handle:read("*a")
+        handle:close()
+        if fortune_msg and #fortune_msg > 0 then vim.notify(fortune_msg, vim.log.levels.INFO, { title = "" }) end
+      end
+    end
+  else
+    vim.notify("Failed to log prompt", vim.log.levels.ERROR)
+  end
+end
+
+-- Cache + helper
+local logs, idx = {}, 0
+
+local function refresh_logs()
+  local dir = Config.prompt_logger.log_dir
+  logs = vim.fn.glob(Utils.join_paths(dir, "avante_prompt_*.log"), false, true)
+  table.sort(logs, function(a, b) -- newest first
+    return a > b
+  end)
+end
+
+---@param step integer 0 = keep | 1 = newer | -1 = older
+local function load_log(step)
+  if #logs == 0 then refresh_logs() end
+  if #logs == 0 then
+    vim.notify("No prompt logs found 🤷", vim.log.levels.WARN)
+    return
+  end
+  idx = (idx + step) -- turn wheel
+  if idx < 1 then idx = #logs end -- wrap around
+  if idx > #logs then idx = 1 end
+
+  local fp = io.open(logs[idx], "r")
+  if not fp then
+    vim.notify("Could not open " .. logs[idx], vim.log.levels.ERROR)
+    return
+  end
+  local content = fp:read("*a")
+  fp:close()
+
+  local buf = vim.api.nvim_get_current_buf()
+  vim.api.nvim_buf_set_lines(buf, 0, -1, false, vim.split(content, "\n", { plain = true }))
+  vim.bo[buf].modifiable = true
+  vim.b[buf].avante_logpath = logs[idx]
+end
+
+function M.next_log() load_log(1) end
+
+function M.prev_log() load_log(-1) end
+
+local function _read_log(delta)
+  if #logs == 0 then refresh_logs() end
+  if #logs == 0 then return nil end
+
+  local target = idx + delta
+  if target < 1 then target = 1 end
+  if target > #logs then target = #logs end
+
+  idx = target
+
+  local fp = io.open(logs[idx], "r")
+  if not fp then return nil end
+  local txt = fp:read("*a")
+  fp:close()
+  return { txt = txt, path = logs[idx] }
+end
+
+function M.on_log_retrieve(delta)
+  return function()
+    local res = _read_log(delta)
+    if not res then
+      vim.notify("No logs available", vim.log.levels.WARN)
+      return
+    end
+    vim.api.nvim_buf_set_lines(0, 0, -1, false, vim.split(res.txt, "\n", { plain = true }))
+    vim.api.nvim_win_set_cursor(0, { 1, 0 })
+  end
+end
+
+return M