feat(references): parse references on message completion and display icons

- Parse file references when session becomes idle (AI done responding)
- Cache parsed references per-message for faster picker opening
- Handle session load from previous sessions
- Display reference icon before recognized file paths in output
- Subscribe to 'messages' state changes to parse loaded sessions
- Update test expected data files with new icon rendering

Author: aweis89

lua/opencode/init.lua

@@ -16,6 +16,7 @@ function M.setup(opts)
   require('opencode.event_manager').setup()
   require('opencode.context').setup()
   require('opencode.ui.context_bar').setup()
+  require('opencode.ui.reference_picker').setup()
 end
 
 return M

lua/opencode/types.lua

@@ -285,6 +285,7 @@
 ---@class OpencodeMessage
 ---@field info MessageInfo Metadata about the message
 ---@field parts OpencodeMessagePart[] Parts that make up the message
+---@field references CodeReference[]|nil Parsed file references from text parts (cached)
 
 ---@class MessageInfo
 ---@field id string Unique message identifier
@@ -488,7 +489,8 @@
 ---@field file_path string Relative or absolute file path
 ---@field line number|nil Line number (1-indexed)
 ---@field column number|nil Column number (optional)
----@field context string Surrounding text for display in picker
 ---@field message_id string ID of the message containing this reference
 ---@field match_start number Start position of match in original text
 ---@field match_end number End position of match in original text
+---@field file string Absolute file path (for Snacks picker preview)
+---@field pos number[]|nil Position as {line, col} for Snacks picker preview

lua/opencode/ui/formatter.lua

@@ -426,9 +426,39 @@ end
 ---@param output Output Output object to write to
 ---@param text string
 function M._format_assistant_message(output, text)
-  -- Simply add the text as-is
-  -- Reference detection is handled by the reference_picker module for navigation
-  output:add_lines(vim.split(text, '\n'))
+  local reference_picker = require('opencode.ui.reference_picker')
+  local references = reference_picker.parse_references(text, '')
+
+  -- If no references, just add the text as-is
+  if #references == 0 then
+    output:add_lines(vim.split(text, '\n'))
+    return
+  end
+
+  -- Sort references by match_start position (ascending)
+  table.sort(references, function(a, b)
+    return a.match_start < b.match_start
+  end)
+
+  -- Build a new text with icons inserted before each reference
+  local result = ''
+  local last_pos = 1
+  local ref_icon = icons.get('reference')
+
+  for _, ref in ipairs(references) do
+    -- Add text before this reference
+    result = result .. text:sub(last_pos, ref.match_start - 1)
+    -- Add the icon and the reference
+    result = result .. ref_icon .. text:sub(ref.match_start, ref.match_end)
+    last_pos = ref.match_end + 1
+  end
+
+  -- Add any remaining text after the last reference
+  if last_pos <= #text then
+    result = result .. text:sub(last_pos)
+  end
+
+  output:add_lines(vim.split(result, '\n'))
 end
 
 ---@param output Output Output object to write to

lua/opencode/ui/reference_picker.lua

@@ -243,15 +243,11 @@ function M.collect_references()
     local msg = state.messages[i]
 
     -- Only process assistant messages
-    if msg.info and msg.info.role == 'assistant' and msg.parts then
-      for _, part in ipairs(msg.parts) do
-        -- Only process text parts (not tool calls)
-        if part.type == 'text' and part.text then
-          local refs = M.parse_references(part.text, msg.info.id)
-          for _, ref in ipairs(refs) do
-            table.insert(all_references, ref)
-          end
-        end
+    if msg.info and msg.info.role == 'assistant' then
+      -- Use cached references if available, otherwise parse on-demand
+      local refs = msg.references or M._parse_message_references(msg)
+      for _, ref in ipairs(refs) do
+        table.insert(all_references, ref)
       end
     end
   end
@@ -270,11 +266,57 @@ function M.collect_references()
   return deduplicated
 end
 
----Get references for a specific text (used by formatter for visual indicators)
----@param text string
+---Parse references from a single message's text parts
+---@param msg OpencodeMessage
 ---@return CodeReference[]
-function M.get_references_for_text(text)
-  return M.parse_references(text, '')
+function M._parse_message_references(msg)
+  local refs = {}
+  if not msg.parts then
+    return refs
+  end
+
+  local message_id = msg.info and msg.info.id or ''
+  for _, part in ipairs(msg.parts) do
+    if part.type == 'text' and part.text then
+      local part_refs = M.parse_references(part.text, message_id)
+      for _, ref in ipairs(part_refs) do
+        table.insert(refs, ref)
+      end
+    end
+  end
+  return refs
+end
+
+---Parse and cache references for all assistant messages in the current session
+function M._parse_session_messages()
+  if not state.messages then
+    return
+  end
+
+  for _, msg in ipairs(state.messages) do
+    -- Only parse assistant messages that don't already have references cached
+    if msg.info and msg.info.role == 'assistant' and not msg.references then
+      msg.references = M._parse_message_references(msg)
+    end
+  end
+end
+
+---Setup reference picker event subscriptions
+---Should be called once during plugin initialization
+function M.setup()
+  -- Subscribe to session.idle to parse references when AI is done responding
+  if state.event_manager then
+    state.event_manager:subscribe('session.idle', function()
+      M._parse_session_messages()
+    end)
+  end
+
+  -- Subscribe to messages changes to handle session loads
+  state.subscribe('messages', function()
+    -- Parse any messages that don't have cached references
+    -- This handles loading previous sessions
+    M._parse_session_messages()
+  end)
 end
 
 ---Format a reference for display in the picker

tests/data/cursor_data.expected.json

@@ -1 +1 @@
-{"actions":[],"timestamp":1762903196,"lines":["----","","","what is this doing","","[lua/opencode/context.lua](lua/opencode/context.lua)","","Line 59:","`````lua","local is_enabled = vim.tbl_get(config, 'context', context_key, 'enabled')","`````","","----","","","**Explanation**","","- The line `local is_enabled = vim.tbl_get(config, 'context', context_key, 'enabled')` (in `lua/opencode/context.lua:58`) uses `vim.tbl_get` to safely read a nested field from a table.","- Concretely it attempts to read `config.context[context_key].enabled` but without throwing an error if `config.context` or `config.context[context_key]` is nil. If any intermediate key is missing it returns `nil`.","- In the surrounding function `M.is_context_enabled`, that value is the default config value for the given context key. The function then checks the state override:","  - If `state.current_context_config[context_key].enabled` is not `nil`, that state value (true/false) is returned.","  - Otherwise the `is_enabled` value (from `config`) is returned.","- Example: if `config.context.selection.enabled == true` but `state.current_context_config.selection.enabled == false`, the function returns `false` (state overrides config). If the state value is `nil`, the config value is used.","- Why this matters: `vim.tbl_get` provides safe nested access; using it avoids runtime errors when some parts of the nested config are absent.","",""],"extmarks":[[1,1,0,{"virt_text_hide":false,"priority":10,"virt_text_repeat_linebreak":false,"virt_text_win_col":-3,"ns_id":3,"right_gravity":true,"virt_text":[["▌󰭻 ","OpencodeMessageRoleUser"],[" "],["USER","OpencodeMessageRoleUser"],["","OpencodeHint"],[" (2025-10-30 17:42:54)","OpencodeHint"],[" [msg_a3637244a001FDRDfoBYVPEGpd]","OpencodeHint"]],"virt_text_pos":"win_col"}],[2,2,0,{"virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text_win_col":-3,"ns_id":3,"right_gravity":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col"}],[3,3,0,{"virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text_win_col":-3,"ns_id":3,"right_gravity":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col"}],[4,4,0,{"virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text_win_col":-3,"ns_id":3,"right_gravity":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col"}],[5,5,0,{"virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text_win_col":-3,"ns_id":3,"right_gravity":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col"}],[6,6,0,{"virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text_win_col":-3,"ns_id":3,"right_gravity":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col"}],[7,7,0,{"virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text_win_col":-3,"ns_id":3,"right_gravity":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col"}],[8,8,0,{"virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text_win_col":-3,"ns_id":3,"right_gravity":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col"}],[9,9,0,{"virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text_win_col":-3,"ns_id":3,"right_gravity":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col"}],[10,10,0,{"virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text_win_col":-3,"ns_id":3,"right_gravity":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col"}],[11,11,0,{"virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text_win_col":-3,"ns_id":3,"right_gravity":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col"}],[12,13,0,{"virt_text_hide":false,"priority":10,"virt_text_repeat_linebreak":false,"virt_text_win_col":-3,"ns_id":3,"right_gravity":true,"virt_text":[[" ","OpencodeMessageRoleAssistant"],[" "],["BUILD","OpencodeMessageRoleAssistant"],[" gpt-5-mini","OpencodeHint"],[" (2025-10-30 17:42:56)","OpencodeHint"],[" [msg_a36372b9a001M1lQEix4SK5QE5]","OpencodeHint"]],"virt_text_pos":"win_col"}]]}
+{"extmarks":[[1,1,0,{"ns_id":3,"virt_text_hide":false,"right_gravity":true,"virt_text_repeat_linebreak":false,"virt_text":[["▌󰭻 ","OpencodeMessageRoleUser"],[" "],["USER","OpencodeMessageRoleUser"],["","OpencodeHint"],[" (2025-10-30 17:42:54)","OpencodeHint"],[" [msg_a3637244a001FDRDfoBYVPEGpd]","OpencodeHint"]],"virt_text_pos":"win_col","virt_text_win_col":-3,"priority":10}],[2,2,0,{"ns_id":3,"virt_text_hide":false,"right_gravity":true,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col","virt_text_win_col":-3,"priority":4096}],[3,3,0,{"ns_id":3,"virt_text_hide":false,"right_gravity":true,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col","virt_text_win_col":-3,"priority":4096}],[4,4,0,{"ns_id":3,"virt_text_hide":false,"right_gravity":true,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col","virt_text_win_col":-3,"priority":4096}],[5,5,0,{"ns_id":3,"virt_text_hide":false,"right_gravity":true,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col","virt_text_win_col":-3,"priority":4096}],[6,6,0,{"ns_id":3,"virt_text_hide":false,"right_gravity":true,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col","virt_text_win_col":-3,"priority":4096}],[7,7,0,{"ns_id":3,"virt_text_hide":false,"right_gravity":true,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col","virt_text_win_col":-3,"priority":4096}],[8,8,0,{"ns_id":3,"virt_text_hide":false,"right_gravity":true,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col","virt_text_win_col":-3,"priority":4096}],[9,9,0,{"ns_id":3,"virt_text_hide":false,"right_gravity":true,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col","virt_text_win_col":-3,"priority":4096}],[10,10,0,{"ns_id":3,"virt_text_hide":false,"right_gravity":true,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col","virt_text_win_col":-3,"priority":4096}],[11,11,0,{"ns_id":3,"virt_text_hide":false,"right_gravity":true,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_pos":"win_col","virt_text_win_col":-3,"priority":4096}],[12,13,0,{"ns_id":3,"virt_text_hide":false,"right_gravity":true,"virt_text_repeat_linebreak":false,"virt_text":[[" ","OpencodeMessageRoleAssistant"],[" "],["BUILD","OpencodeMessageRoleAssistant"],[" gpt-5-mini","OpencodeHint"],[" (2025-10-30 17:42:56)","OpencodeHint"],[" [msg_a36372b9a001M1lQEix4SK5QE5]","OpencodeHint"]],"virt_text_pos":"win_col","virt_text_win_col":-3,"priority":10}]],"timestamp":1766369775,"actions":[],"lines":["----","","","what is this doing","","[lua/opencode/context.lua](lua/opencode/context.lua)","","Line 59:","`````lua","local is_enabled = vim.tbl_get(config, 'context', context_key, 'enabled')","`````","","----","","","**Explanation**","","- The line `local is_enabled = vim.tbl_get(config, 'context', context_key, 'enabled')` (in ` lua/opencode/context.lua:58`) uses `vim.tbl_get` to safely read a nested field from a table.","- Concretely it attempts to read `config.context[context_key].enabled` but without throwing an error if `config.context` or `config.context[context_key]` is nil. If any intermediate key is missing it returns `nil`.","- In the surrounding function `M.is_context_enabled`, that value is the default config value for the given context key. The function then checks the state override:","  - If `state.current_context_config[context_key].enabled` is not `nil`, that state value (true/false) is returned.","  - Otherwise the `is_enabled` value (from `config`) is returned.","- Example: if `config.context.selection.enabled == true` but `state.current_context_config.selection.enabled == false`, the function returns `false` (state overrides config). If the state value is `nil`, the config value is used.","- Why this matters: `vim.tbl_get` provides safe nested access; using it avoids runtime errors when some parts of the nested config are absent.","",""]}