feat: restore mentions when navigating history

Author: sudo-tee

lua/opencode/api.lua

@@ -216,13 +216,15 @@ function M.prev_history()
   local prev_prompt = history.prev()
   if prev_prompt then
     input_window.set_content(prev_prompt)
+    require('opencode.ui.mention').restore_mentions(state.windows.input_buf)
   end
 end
 
 function M.next_history()
   local next_prompt = history.next()
   if next_prompt then
     input_window.set_content(next_prompt)
+    require('opencode.ui.mention').restore_mentions(state.windows.input_buf)
   end
 end
 

lua/opencode/context.lua

@@ -108,6 +108,11 @@ function M.add_file(file)
   end
 end
 
+M.clear_files = function()
+  M.context.mentioned_files = nil
+  M.context.mentioned_files_content = nil
+end
+
 function M.add_subagent(subagent)
   if not M.context.mentioned_subagents then
     M.context.mentioned_subagents = {}
@@ -118,6 +123,10 @@ function M.add_subagent(subagent)
   end
 end
 
+M.clear_subagents = function()
+  M.context.mentioned_subagents = nil
+end
+
 ---@param opts OpencodeContextConfig
 function M.delta_context(opts)
   opts = opts or config.context

lua/opencode/ui/mention.lua

@@ -2,7 +2,7 @@ local M = {}
 
 local mentions_namespace = vim.api.nvim_create_namespace('OpencodeMentions')
 
-function M.highlight_all_mentions(buf)
+function M.highlight_all_mentions(buf, callback)
   -- Pattern for mentions
   local mention_pattern = '@[%w_%-%./][%w_%-%./]*'
 
@@ -24,6 +24,10 @@ function M.highlight_all_mentions(buf)
         break
       end
 
+      if callback then
+        callback(line:sub(mention_start + 1, mention_end), row, mention_start, mention_end)
+      end
+
       -- Add extmark for this mention
       vim.api.nvim_buf_set_extmark(buf, mentions_namespace, row - 1, mention_start - 1, {
         end_col = mention_end,
@@ -66,4 +70,25 @@ function M.mention(get_name)
   end)
 end
 
+function M.restore_mentions(buf)
+  local context = require('opencode.context')
+
+  context.clear_subagents()
+  context.clear_files()
+
+  M.highlight_all_mentions(buf, function(name)
+    local agents = require('opencode.config_file').get_subagents()
+
+    if vim.tbl_contains(agents, name) then
+      require('opencode.context').add_subagent(name)
+      return
+    end
+
+    if vim.fn.filereadable(name) == 1 then
+      require('opencode.context').add_file(name)
+      return
+    end
+  end)
+end
+
 return M