feat: add full_view_ask api (#2658)

lua/avante/api.lua

@@ -113,6 +113,18 @@ end
 ---@field sidebar_pre_render? fun(sidebar: avante.Sidebar)
 ---@field sidebar_post_render? fun(sidebar: avante.Sidebar)
 ---@field project_root? string optional project root
+---@field show_logo? boolean whether to show the logo
+
+function M.full_view_ask()
+  M.ask({
+    show_logo = true,
+    sidebar_post_render = function(sidebar)
+      sidebar:toggle_code_window()
+      -- vim.wo[sidebar.containers.result.winid].number = true
+      -- vim.wo[sidebar.containers.result.winid].relativenumber = true
+    end,
+  })
+end
 
 ---@param opts? AskOptions
 function M.ask(opts)

lua/avante/highlights.lua

@@ -51,6 +51,21 @@ local Highlights = {
   AVANTE_TASK_COMPLETED = { name = "AvanteTaskCompleted", fg = "#98c379", bg_link = "Normal" },
   AVANTE_TASK_FAILED = { name = "AvanteTaskFailed", fg = "#e06c75", bg_link = "Normal" },
   AVANTE_THINKING = { name = "AvanteThinking", fg = "#c678dd", bg_link = "Normal" },
+  -- Gradient logo highlights
+  AVANTE_LOGO_LINE_1 = { name = "AvanteLogoLine1", fg = "#f5f5f5" },
+  AVANTE_LOGO_LINE_2 = { name = "AvanteLogoLine2", fg = "#e8e8e8" },
+  AVANTE_LOGO_LINE_3 = { name = "AvanteLogoLine3", fg = "#dbdbdb" },
+  AVANTE_LOGO_LINE_4 = { name = "AvanteLogoLine4", fg = "#cfcfcf" },
+  AVANTE_LOGO_LINE_5 = { name = "AvanteLogoLine5", fg = "#c2c2c2" },
+  AVANTE_LOGO_LINE_6 = { name = "AvanteLogoLine6", fg = "#b5b5b5" },
+  AVANTE_LOGO_LINE_7 = { name = "AvanteLogoLine7", fg = "#a8a8a8" },
+  AVANTE_LOGO_LINE_8 = { name = "AvanteLogoLine8", fg = "#9b9b9b" },
+  AVANTE_LOGO_LINE_9 = { name = "AvanteLogoLine9", fg = "#8e8e8e" },
+  AVANTE_LOGO_LINE_10 = { name = "AvanteLogoLine10", fg = "#818181" },
+  AVANTE_LOGO_LINE_11 = { name = "AvanteLogoLine11", fg = "#747474" },
+  AVANTE_LOGO_LINE_12 = { name = "AvanteLogoLine12", fg = "#676767" },
+  AVANTE_LOGO_LINE_13 = { name = "AvanteLogoLine13", fg = "#5a5a5a" },
+  AVANTE_LOGO_LINE_14 = { name = "AvanteLogoLine14", fg = "#4d4d4d" },
 }
 
 Highlights.conflict = {

lua/avante/init.lua

@@ -94,15 +94,12 @@ function H.keymaps()
       function() require("avante.api").ask() end,
       { desc = "avante: ask" }
     )
-    Utils.safe_keymap_set({ "n", "v" }, Config.mappings.full_view_ask, function()
+    Utils.safe_keymap_set(
-      require("avante.api").ask({
+      { "n", "v" },
-        sidebar_post_render = function(sidebar)
+      Config.mappings.full_view_ask,
-          sidebar:toggle_code_window()
+      function() require("avante.api").full_view_ask() end,
-          vim.wo[sidebar.containers.result.winid].number = true
+      { desc = "avante: ask with full result view" }
-          vim.wo[sidebar.containers.result.winid].relativenumber = true
+    )
-        end,
-      })
-    end, { desc = "avante: ask with full result view" })
     Utils.safe_keymap_set(
       { "n", "v" },
       Config.mappings.new_ask,

lua/avante/sidebar.lua

@@ -20,6 +20,7 @@ local History = require("avante.history")
 local Render = require("avante.history.render")
 local Line = require("avante.ui.line")
 local LRUCache = require("avante.utils.lru_cache")
+local logo = require("avante.utils.logo")
 
 local RESULT_BUF_NAME = "AVANTE_RESULT"
 local VIEW_BUFFER_UPDATED_PATTERN = "AvanteViewBufferUpdated"
@@ -75,6 +76,7 @@ Sidebar.__index = Sidebar
 ---@field token_count integer | nil
 ---@field acp_client ACPClient | nil
 ---@field acp_session_id string | nil
+---@field post_render? fun(sidebar: avante.Sidebar)
 
 ---@param id integer the tabpage id retrieved from api.nvim_get_current_tabpage()
 function Sidebar:new(id)
@@ -105,6 +107,7 @@ function Sidebar:new(id)
     -- Cache-related fields
     _cached_history_lines = nil,
     _history_cache_invalidated = true,
+    post_render = nil,
   }, Sidebar)
 end
 
@@ -148,6 +151,7 @@ end
 ---@param opts SidebarOpenOptions
 function Sidebar:open(opts)
   opts = opts or {}
+  self.show_logo = opts.show_logo
   local in_visual_mode = Utils.in_visual_mode() and self:in_code_win()
   if not self:is_open() then
     self:reset()
@@ -1478,6 +1482,30 @@ function Sidebar:resize()
   vim.defer_fn(function() vim.cmd("AvanteRefresh") end, 200)
 end
 
+function Sidebar:render_logo()
+  local logo_lines = vim.split(logo, "\n")
+  local max_width = 30
+  --- get editor width
+  local editor_width = vim.api.nvim_win_get_width(self.containers.result.winid)
+  local padding = math.floor((editor_width - max_width) / 2)
+  Utils.unlock_buf(self.containers.result.bufnr)
+  for i, line in ipairs(logo_lines) do
+    --- center logo
+    line = vim.trim(line)
+    vim.api.nvim_buf_set_lines(self.containers.result.bufnr, i - 1, i, false, { string.rep(" ", padding) .. line })
+    --- apply gradient color
+    if line ~= "" then
+      local hl_group = "AvanteLogoLine" .. i
+      vim.api.nvim_buf_set_extmark(self.containers.result.bufnr, RESULT_BUF_HL_NAMESPACE, i - 1, padding, {
+        end_col = padding + #line,
+        hl_group = hl_group,
+      })
+    end
+  end
+  Utils.lock_buf(self.containers.result.bufnr)
+  return #logo_lines
+end
+
 function Sidebar:toggle_code_window()
   local win_width = api.nvim_win_get_width(self.code.winid)
   if win_width == 0 then
@@ -1559,9 +1587,6 @@ end, 50)
 ---@param content string concatenated content of the buffer
 ---@param opts? {focus?: boolean, scroll?: boolean, backspace?: integer, callback?: fun(): nil} whether to focus the result view
 function Sidebar:update_content(content, opts)
-  if not self.containers.result or not self.containers.result.bufnr then return end
-
-  -- 提前验证容器有效性，避免后续无效操作
   if not Utils.is_valid_container(self.containers.result) then return end
 
   local should_auto_scroll = self:should_auto_scroll()
@@ -1572,10 +1597,9 @@ function Sidebar:update_content(content, opts)
     opts or {}
   )
 
-  -- 缓存历史行，避免重复计算
   local history_lines
   if not self._cached_history_lines or self._history_cache_invalidated then
-    history_lines = self.get_history_lines(self.chat_history)
+    history_lines = self.get_history_lines(self.chat_history, self.show_logo)
     self._cached_history_lines = history_lines
     self._history_cache_invalidated = false
   else
@@ -1589,26 +1613,23 @@ function Sidebar:update_content(content, opts)
     end
   end
 
-  -- 使用 vim.schedule 而不是 vim.defer_fn(0)，性能更好
-  -- 再次检查容器有效性
   if not Utils.is_valid_container(self.containers.result) then return end
 
   self:clear_state()
 
-  -- 批量更新操作
+  local skip_line_count = 0
+  if self.show_logo then skip_line_count = self:render_logo() end
+
   local bufnr = self.containers.result.bufnr
   Utils.unlock_buf(bufnr)
 
-  Utils.update_buffer_lines(RESULT_BUF_HL_NAMESPACE, bufnr, self.old_result_lines, history_lines)
+  Utils.update_buffer_lines(RESULT_BUF_HL_NAMESPACE, bufnr, self.old_result_lines, history_lines, skip_line_count)
 
-  -- 缓存结果行
   self.old_result_lines = history_lines
 
-  -- 批量设置选项
   api.nvim_set_option_value("filetype", "Avante", { buf = bufnr })
   Utils.lock_buf(bufnr)
 
-  -- 处理焦点和滚动
   if opts.focus and not self:is_focused_on_result() then
     xpcall(function() api.nvim_set_current_win(self.containers.result.winid) end, function(err)
       Utils.debug("Failed to set current win:", err)
@@ -1694,11 +1715,12 @@ end
 ---@param message avante.HistoryMessage
 ---@param messages avante.HistoryMessage[]
 ---@param ctx table
+---@param ignore_record_prefix boolean | nil
 ---@return avante.ui.Line[]
-local function _get_message_lines(message, messages, ctx)
+local function _get_message_lines(message, messages, ctx, ignore_record_prefix)
   if message.visible == false then return {} end
   local lines = Render.message_to_lines(message, messages)
-  if message.is_user_submission then
+  if message.is_user_submission and not ignore_record_prefix then
     ctx.selected_filepaths = message.selected_filepaths
     local text = table.concat(vim.tbl_map(function(line) return tostring(line) end, lines), "\n")
     local prefix = render_chat_record_prefix(
@@ -1744,9 +1766,12 @@ local _message_to_lines_lru_cache = LRUCache:new(100)
 ---@param message avante.HistoryMessage
 ---@param messages avante.HistoryMessage[]
 ---@param ctx table
+---@param ignore_record_prefix boolean | nil
 ---@return avante.ui.Line[]
-local function get_message_lines(message, messages, ctx)
+local function get_message_lines(message, messages, ctx, ignore_record_prefix)
-  if message.state == "generating" or message.is_calling then return _get_message_lines(message, messages, ctx) end
+  if message.state == "generating" or message.is_calling then
+    return _get_message_lines(message, messages, ctx, ignore_record_prefix)
+  end
   local text_len = 0
   local content = message.message.content
   if type(content) == "table" then
@@ -1765,20 +1790,21 @@ local function get_message_lines(message, messages, ctx)
   local cache_key = message.uuid .. ":" .. tostring(text_len)
   local cached_lines = _message_to_lines_lru_cache:get(cache_key)
   if cached_lines then return cached_lines end
-  local lines = _get_message_lines(message, messages, ctx)
+  local lines = _get_message_lines(message, messages, ctx, ignore_record_prefix)
   _message_to_lines_lru_cache:set(cache_key, lines)
   return lines
 end
 
 ---@param history avante.ChatHistory
+---@param ignore_record_prefix boolean | nil
 ---@return avante.ui.Line[]
-function Sidebar.get_history_lines(history)
+function Sidebar.get_history_lines(history, ignore_record_prefix)
   local history_messages = History.get_history_messages(history)
   local ctx = {}
   ---@type avante.ui.Line[][]
   local group = {}
   for _, message in ipairs(history_messages) do
-    local lines = get_message_lines(message, history_messages, ctx)
+    local lines = get_message_lines(message, history_messages, ctx, ignore_record_prefix)
     if #lines == 0 then goto continue end
     if message.is_user_submission then table.insert(group, {}) end
     local last_item = group[#group]
@@ -2030,8 +2056,13 @@ function Sidebar:new_chat(args, cb)
   self.current_state = nil
   self.acp_session_id = nil
   self:update_content("New chat", { focus = false, scroll = false, callback = function() self:focus_input() end })
+  --- goto first line then go to last line
+  vim.schedule(function()
+    vim.api.nvim_win_call(self.containers.result.winid, function() vim.cmd("normal! ggG") end)
+  end)
   if cb then cb(args) end
   vim.schedule(function() self:create_todos_container() end)
+  if self.post_render then vim.defer_fn(function() self.post_render(self) end, 100) end
 end
 
 local debounced_save_history = Utils.debounce(
@@ -2971,8 +3002,6 @@ function Sidebar:render(opts)
 
   self:create_selected_files_container()
 
-  self:update_content_with_history()
-
   if self.code.bufnr and api.nvim_buf_is_valid(self.code.bufnr) then
     -- reset states when buffer is closed
     api.nvim_buf_attach(self.code.bufnr, false, {
@@ -2995,7 +3024,16 @@ function Sidebar:render(opts)
 
   self:setup_colors()
 
-  if opts.sidebar_post_render then vim.defer_fn(function() opts.sidebar_post_render(self) end, 100) end
+  if opts.sidebar_post_render then
+    self.post_render = opts.sidebar_post_render
+    vim.defer_fn(function()
+      opts.sidebar_post_render(self)
+      self:update_content_with_history()
+    end, 100)
+  else
+    self:update_content_with_history()
+  end
+
   return self
 end
 

lua/avante/utils/init.lua

@@ -1297,7 +1297,9 @@ end
 ---@param bufnr integer
 ---@param old_lines avante.ui.Line[]
 ---@param new_lines avante.ui.Line[]
-function M.update_buffer_lines(ns_id, bufnr, old_lines, new_lines)
+---@param skip_line_count? integer
+function M.update_buffer_lines(ns_id, bufnr, old_lines, new_lines, skip_line_count)
+  skip_line_count = skip_line_count or 0
   local diffs = get_lines_diff(old_lines, new_lines)
   if #diffs == 0 then return end
   for _, diff in ipairs(diffs) do
@@ -1309,9 +1311,15 @@ function M.update_buffer_lines(ns_id, bufnr, old_lines, new_lines)
       local lines_ = vim.split(line, "\n")
       cleaned_lines = vim.list_extend(cleaned_lines, lines_)
     end
-    vim.api.nvim_buf_set_lines(bufnr, diff.start_line - 1, diff.end_line - 1, false, cleaned_lines)
+    vim.api.nvim_buf_set_lines(
+      bufnr,
+      skip_line_count + diff.start_line - 1,
+      skip_line_count + diff.end_line - 1,
+      false,
+      cleaned_lines
+    )
     for i, line in ipairs(lines) do
-      line:set_highlights(ns_id, bufnr, diff.start_line + i - 2)
+      line:set_highlights(ns_id, bufnr, skip_line_count + diff.start_line + i - 2)
     end
     vim.cmd("redraw")
   end

lua/avante/utils/logo.lua

@@ -0,0 +1,31 @@
+local logo = [[
+
+⠀⢠⣤⣤⣤⡄⣷⣄⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀
+⠀⣿⣿⣿⣿⢁⣿⣿⣷⡶⠶⠶⠶⠶⠶⠶⠶⠶⠶⠶⠶⠶⠶⠶⠶⠶⣤⡀⠀⠀
+⣀⣛⣛⣛⣛⣸⣿⣿⣿⢁⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⢸⣿⣿⣦⡀
+⠈⠻⣿⣿⣿⣿⣿⣿⣿⢸⣿⣿⣿⡿⠿⠿⠿⠿⠿⠿⢿⣿⣿⣿⡟⢸⣿⣿⣿⠇
+⠀⠀⢸⢛⣛⣛⣛⣛⣃⣼⣿⣿⣿⠁⣿⣿⣿⣿⣿⣿⢠⣿⣿⣿⡇⣿⣿⣿⣿⠀
+⠀⠀⡇⣾⣿⣿⣿⣿⣿⣿⣿⣿⣏⠸⠿⠿⠿⠿⠿⠏⣼⣿⣿⣿⢸⣿⣿⣿⡇⠀
+⠀⢸⢸⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⡇⣾⣿⣿⣿⠀⠀
+⠀⡼⢸⣿⣿⣿⣿⣿⣿⣿⣿⣿⢰⣶⣶⣶⣶⣶⣶⢸⣿⣿⣿⠃⣿⣿⣿⡿⠀⠀
+⢸⣀⣉⣉⣉⣉⣉⣉⣉⣉⣉⣡⣿⣿⣿⡿⠀⢸⣅⣉⣉⣉⣁⣿⣿⣿⣿⣿⣦⡀
+⠈⠻⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⡇⠀⠀⠙⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿
+⠀⠀⠈⠛⠛⠛⠛⠛⠛⠛⠛⠛⠛⠛⠛⠃⠀⠀⠀⠈⠻⣿⣿⣿⣿⣿⣿⣿⣿⡇
+]]
+
+local logo_ = [[
+
+⡉⢭⣭⣭⣭⠅⣿⣄⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀
+⠌⣾⢲⡳⣞⢄⡿⣪⢷⠖⢖⠖⢖⠖⢖⠖⢖⠖⢖⠖⢖⠖⢖⠖⠖⠶⣦⡀⠀⠀
+⣇⣉⣃⣙⣈⣸⡳⣝⢷⢑⣟⢽⡫⣟⢽⡫⣟⢽⡫⣟⢽⡫⣟⢽⡻⢸⣏⢿⣦⠀
+⠈⠻⣿⡿⣿⣟⣿⣜⠯⢰⣏⢷⡝⡮⠳⠝⠮⠳⠝⠮⢳⣝⢮⡳⡏⣸⢯⢷⣽⠁
+⠀⠀⣸⢛⣑⣛⣊⣛⣃⡼⣧⢻⣺⠠⣿⣟⣿⣻⣟⣿⢠⡯⣳⢽⠅⣿⡹⣗⣽⠀
+⠀⠀⡇⣻⢺⢮⡺⣕⢯⡺⣕⢯⡇⠼⠫⠻⠕⠷⠵⠏⣼⢫⣞⡽⢰⡯⣻⡺⡇⠀
+⠀⢨⢃⡿⣕⢯⡞⣵⡫⣯⢺⡵⣫⡻⣛⢟⣝⢟⣝⢯⡺⣵⢣⡏⢸⣯⣫⢟⠅⠀
+⠀⡞⢸⡗⣽⢺⡵⣫⢞⡵⣫⠾⢰⣶⣶⣶⣶⣶⡶⢸⡳⣝⡽⢂⣿⡺⣳⡏⠀⠀
+⣸⣈⣃⣋⣊⣑⣉⣊⣣⣙⣘⣤⣟⢷⣫⡗⠀⢸⣄⣋⣜⣙⣢⡽⣳⢯⢾⡽⣦⡀
+⠈⠛⣿⢿⡿⣿⢿⣿⣻⣟⣿⣳⣟⣷⢝⡇⠀⠀⠹⢿⣻⢿⣽⣻⣗⢿⡵⣻⡺⣏
+⠀⠀⠈⠛⠙⠛⠓⠛⠚⠙⠚⠓⠛⠚⠛⠁⠀⠀⠀⠈⠹⣟⣾⣳⣟⢿⣞⣽⣝⡇
+]]
+
+return logo