feat: highlight assistant file references

Author: jensenojs

lua/opencode/ui/formatter.lua

@@ -690,6 +690,25 @@ local function add_symbol_reference_highlights(output, rendered, rendered_refere
   end
 end
 
+local function add_file_reference_highlights(output, rendered, rendered_reference_ranges, first_line_idx)
+  local line_start = 1
+
+  for line_idx, line in ipairs(vim.split(rendered, '\n')) do
+    local line_end = line_start + #line - 1
+    for _, range in ipairs(rendered_reference_ranges) do
+      if ranges_overlap(line_start, line_end, range[1], range[2]) then
+        output:add_extmark(first_line_idx + line_idx - 1, {
+          start_col = math.max(range[1], line_start) - line_start,
+          end_col = math.min(range[2], line_end) - line_start + 1,
+          hl_group = 'OpencodeReference',
+          priority = 1000,
+        })
+      end
+    end
+    line_start = line_start + #line + 1
+  end
+end
+
 ---@param output Output Output object to write to
 ---@param text string
 ---@param message_id string|nil Optional message ID for reference parsing
@@ -701,6 +720,7 @@ function M._format_assistant_message(output, text, message_id)
   local first_line_idx = output:get_line_count()
 
   output:add_lines(vim.split(rendered, '\n'))
+  add_file_reference_highlights(output, rendered, rendered_reference_ranges, first_line_idx)
 
   -- Render-time symbol highlights are only visual hints. This intentionally
   -- rebuilds from the current conversation refs instead of storing targets on

lua/opencode/ui/highlight.lua

@@ -41,7 +41,7 @@ function M.setup()
     vim.api.nvim_set_hl(0, 'OpencodeContextSwitchOn', { link = '@label', default = true })
     vim.api.nvim_set_hl(0, 'OpencodePickerTime', { link = 'Comment', default = true })
     vim.api.nvim_set_hl(0, 'OpencodeDebugText', { link = 'Comment', default = true })
-    vim.api.nvim_set_hl(0, 'OpencodeReference', { fg = '#1976D2', default = true })
+    vim.api.nvim_set_hl(0, 'OpencodeReference', { fg = '#5F7896', default = true })
     vim.api.nvim_set_hl(0, 'OpencodeSymbolReference', { link = 'Identifier', default = true })
     vim.api.nvim_set_hl(0, 'OpencodeReasoningText', { link = 'Comment', default = true })
     vim.api.nvim_set_hl(0, 'OpencodePermissionTitle', { fg = '#FF9E3B', default = true })
@@ -90,7 +90,7 @@ function M.setup()
     vim.api.nvim_set_hl(0, 'OpencodeContextSwitchOn', { link = '@label', default = true })
     vim.api.nvim_set_hl(0, 'OpencodePickerTime', { link = 'Comment', default = true })
     vim.api.nvim_set_hl(0, 'OpencodeDebugText', { link = 'Comment', default = true })
-    vim.api.nvim_set_hl(0, 'OpencodeReference', { fg = '#7AA2F7', default = true })
+    vim.api.nvim_set_hl(0, 'OpencodeReference', { fg = '#8AA6C8', default = true })
     vim.api.nvim_set_hl(0, 'OpencodeSymbolReference', { link = 'Identifier', default = true })
     vim.api.nvim_set_hl(0, 'OpencodeReasoningText', { link = 'Comment', default = true })
     vim.api.nvim_set_hl(0, 'OpencodePermissionTitle', { fg = '#FF9E3B', default = true })

tests/data/api-abort.expected.json

@@ -194,6 +194,21 @@
         "virt_text_pos": "right_align",
         "virt_text_repeat_linebreak": false
       }
+    ],
+    [
+      9,
+      10,
+      86,
+      {
+        "end_col": 99,
+        "end_right_gravity": false,
+        "end_row": 10,
+        "hl_eol": false,
+        "hl_group": "OpencodeReference",
+        "ns_id": 3,
+        "priority": 1000,
+        "right_gravity": true
+      }
     ]
   ],
   "lines": [

tests/data/cursor_data.expected.json

@@ -314,6 +314,21 @@
         "virt_text_pos": "right_align",
         "virt_text_repeat_linebreak": false
       }
+    ],
+    [
+      15,
+      17,
+      95,
+      {
+        "end_col": 124,
+        "end_right_gravity": false,
+        "end_row": 17,
+        "hl_eol": false,
+        "hl_group": "OpencodeReference",
+        "ns_id": 3,
+        "priority": 1000,
+        "right_gravity": true
+      }
     ]
   ],
   "lines": [