68 lines
1.7 KiB
Lua
68 lines
1.7 KiB
Lua
local Utils = require("avante.utils")
|
|
local Path = require("avante.path")
|
|
local Config = require("avante.config")
|
|
local Selector = require("avante.ui.selector")
|
|
|
|
---@class avante.HistorySelector
|
|
local M = {}
|
|
|
|
---@param history avante.ChatHistory
|
|
---@return table?
|
|
local function to_selector_item(history)
|
|
local messages = Utils.get_history_messages(history)
|
|
local timestamp = #messages > 0 and messages[#messages].timestamp or history.timestamp
|
|
local name = history.title .. " - " .. timestamp .. " (" .. #messages .. ")"
|
|
name = name:gsub("\n", "\\n")
|
|
return {
|
|
name = name,
|
|
filename = history.filename,
|
|
}
|
|
end
|
|
|
|
---@param bufnr integer
|
|
---@param cb fun(filename: string)
|
|
function M.open(bufnr, cb)
|
|
local selector_items = {}
|
|
|
|
local histories = Path.history.list(bufnr)
|
|
|
|
for _, history in ipairs(histories) do
|
|
table.insert(selector_items, to_selector_item(history))
|
|
end
|
|
|
|
if #selector_items == 0 then
|
|
Utils.warn("No models available in config")
|
|
return
|
|
end
|
|
|
|
local selector = Selector:new({
|
|
provider = Config.selector.provider,
|
|
title = "Select Avante History",
|
|
items = vim
|
|
.iter(selector_items)
|
|
:map(
|
|
function(item)
|
|
return {
|
|
id = item.filename,
|
|
title = item.name,
|
|
}
|
|
end
|
|
)
|
|
:totable(),
|
|
on_select = function(item_ids)
|
|
if not item_ids then return end
|
|
if #item_ids == 0 then return end
|
|
cb(item_ids[1])
|
|
end,
|
|
get_preview_content = function(item_id)
|
|
local history = Path.history.load(vim.api.nvim_get_current_buf(), item_id)
|
|
local Sidebar = require("avante.sidebar")
|
|
local content = Sidebar.render_history_content(history)
|
|
return content, "markdown"
|
|
end,
|
|
})
|
|
selector:open()
|
|
end
|
|
|
|
return M
|