fix(context-actions): fix off by one context actions

There was 2 off by one errors when adding contextual actions

When adding them in snapshots

When refreshing them after a part update

A couple of test needed to be regenerated for it to work

I added the possibility to regenerate a test snapshot by it's name or
file-name

Author: sudo-tee

lua/opencode/ui/formatter.lua

@@ -154,66 +154,45 @@ function M._format_revert_message(session_data, start_idx)
   return output
 end
 
+local function add_action(output, text, action_type, args, key, line)
+  line = (line or output:get_line_count()) - 1
+  output:add_action({
+    text = text,
+    type = action_type,
+    args = args,
+    key = key,
+    display_line = line,
+    range = { from = line, to = line },
+  })
+end
+
 ---@param output Output Output object to write to
 ---@param part OpencodeMessagePart
 function M._format_patch(output, part)
+  if not part.hash then
+    return
+  end
+
   local restore_points = snapshot.get_restore_points_by_parent(part.hash) or {}
   M._format_action(output, icons.get('snapshot') .. ' Created Snapshot', vim.trim(part.hash:sub(1, 8)))
-  local snapshot_header_line = output:get_line_count()
 
   -- Anchor all snapshot-level actions to the snapshot header line
-  output:add_action({
-    text = '[R]evert file',
-    type = 'diff_revert_selected_file',
-    args = { part.hash },
-    key = 'R',
-    display_line = snapshot_header_line,
-    range = { from = snapshot_header_line, to = snapshot_header_line },
-  })
-  output:add_action({
-    text = 'Revert [A]ll',
-    type = 'diff_revert_all',
-    args = { part.hash },
-    key = 'A',
-    display_line = snapshot_header_line,
-    range = { from = snapshot_header_line, to = snapshot_header_line },
-  })
-  output:add_action({
-    text = '[D]iff',
-    type = 'diff_open',
-    args = { part.hash },
-    key = 'D',
-    display_line = snapshot_header_line,
-    range = { from = snapshot_header_line, to = snapshot_header_line },
-  })
+  add_action(output, '[R]evert file', 'diff_revert_selected_file', { part.hash }, 'R')
+  add_action(output, 'Revert [A]ll', 'diff_revert_all', { part.hash }, 'A')
+  add_action(output, '[D]iff', 'diff_open', { part.hash }, 'D')
 
   if #restore_points > 0 then
     for _, restore_point in ipairs(restore_points) do
       output:add_line(
         string.format(
-          '  %s Restore point `%s` - %s',
+          '  %s Restore point `%s` - %s ',
           icons.get('restore_point'),
-          restore_point.id:sub(1, 8),
+          vim.trim(restore_point.id:sub(1, 8)),
           util.format_time(restore_point.created_at)
         )
       )
-      local restore_line = output:get_line_count()
-      output:add_action({
-        text = 'Restore [A]ll',
-        type = 'diff_restore_snapshot_all',
-        args = { restore_point.id },
-        key = 'A',
-        display_line = restore_line,
-        range = { from = restore_line, to = restore_line },
-      })
-      output:add_action({
-        text = '[R]estore file',
-        type = 'diff_restore_snapshot_file',
-        args = { restore_point.id },
-        key = 'R',
-        display_line = restore_line,
-        range = { from = restore_line, to = restore_line },
-      })
+      add_action(output, 'Restore [A]ll', 'diff_restore_snapshot_all', { restore_point.id }, 'A')
+      add_action(output, '[R]estore file', 'diff_restore_snapshot_file', { restore_point.id }, 'R')
     end
   end
 end
@@ -282,10 +261,10 @@ function M.format_message_header(message)
     and (not message.parts or #message.parts == 0)
   then
     local error = message.info.error
-    local error_messgage = error.data and error.data.message or vim.inspect(error)
+    local error_message = error.data and error.data.message or vim.inspect(error)
 
     output:add_line('')
-    M._format_callout(output, 'ERROR', error_messgage)
+    M._format_callout(output, 'ERROR', error_message)
   end
 
   output:add_line('')
@@ -797,8 +776,8 @@ function M.format_part(part, message, is_last_part)
 
   if is_last_part and role == 'assistant' and message.info.error and message.info.error ~= '' then
     local error = message.info.error
-    local error_messgage = error.data and error.data.message or vim.inspect(error)
-    M._format_callout(output, 'ERROR', error_messgage)
+    local error_message = error.data and error.data.message or vim.inspect(error)
+    M._format_callout(output, 'ERROR', error_message)
     output:add_empty_line()
   end
 

lua/opencode/ui/render_state.lua

@@ -77,8 +77,8 @@ end
 ---@param snapshot_id string Call ID
 ---@return OpencodeMessagePart? part Part if found
 function RenderState:get_part_by_snapshot_id(snapshot_id)
-  for _, rendered_message in pairs(self._messages) do
-    for _, part in ipairs(rendered_message.message.parts) do
+  for _, rendered_message in pairs(self._messages or {}) do
+    for _, part in ipairs(rendered_message.message.parts or {}) do
       if part.type == 'patch' and part.hash == snapshot_id then
         return part
       end

lua/opencode/ui/renderer.lua

@@ -413,7 +413,7 @@ function M._replace_part_in_buffer(part_id, formatted_data)
   output_window.set_extmarks(formatted_data.extmarks, cached.line_start)
 
   if formatted_data.actions then
-    M._render_state:add_actions(part_id, formatted_data.actions, cached.line_start)
+    M._render_state:add_actions(part_id, formatted_data.actions, cached.line_start + 1)
   end
 
   M._render_state:update_part_lines(part_id, cached.line_start, new_line_end)

tests/data/diagnostics.expected.json

@@ -1 +1 @@
-{"extmarks":[[1,1,0,{"virt_text_pos":"win_col","virt_text_hide":false,"virt_text_win_col":-3,"ns_id":3,"virt_text_repeat_linebreak":false,"priority":10,"right_gravity":true,"virt_text":[["▌󰭻 ","OpencodeMessageRoleUser"],[" "],["USER","OpencodeMessageRoleUser"],["","OpencodeHint"],[" (2025-10-12 06:42:56)","OpencodeHint"],[" [msg_9d7287269001C5gRusYfX7A1w1]","OpencodeHint"]]}],[2,2,0,{"virt_text_pos":"win_col","virt_text_hide":false,"virt_text_win_col":-3,"ns_id":3,"virt_text_repeat_linebreak":true,"priority":4096,"right_gravity":true,"virt_text":[["▌","OpencodeMessageRoleUser"]]}],[3,3,0,{"virt_text_pos":"win_col","virt_text_hide":false,"virt_text_win_col":-3,"ns_id":3,"virt_text_repeat_linebreak":true,"priority":4096,"right_gravity":true,"virt_text":[["▌","OpencodeMessageRoleUser"]]}],[4,3,39,{"right_gravity":true,"end_row":3,"end_col":53,"hl_group":"OpencodeMention","ns_id":3,"priority":1000,"hl_eol":false,"end_right_gravity":false}],[5,4,0,{"virt_text_pos":"win_col","virt_text_hide":false,"virt_text_win_col":-3,"ns_id":3,"virt_text_repeat_linebreak":true,"priority":4096,"right_gravity":true,"virt_text":[["▌","OpencodeMessageRoleUser"]]}],[6,5,0,{"virt_text_pos":"win_col","virt_text_hide":false,"virt_text_win_col":-3,"ns_id":3,"virt_text_repeat_linebreak":true,"priority":4096,"right_gravity":true,"virt_text":[["▌","OpencodeMessageRoleUser"]]}],[7,8,0,{"virt_text_pos":"win_col","virt_text_hide":false,"virt_text_win_col":-3,"ns_id":3,"virt_text_repeat_linebreak":false,"priority":10,"right_gravity":true,"virt_text":[[" ","OpencodeMessageRoleAssistant"],[" "],["BUILD","OpencodeMessageRoleAssistant"],[" claude-sonnet-4.5","OpencodeHint"],[" (2025-10-12 06:42:56)","OpencodeHint"],[" [msg_9d7287287001HVwpPaH7WkRVdN]","OpencodeHint"]]}],[8,10,0,{"virt_text_pos":"win_col","virt_text_hide":false,"virt_text_win_col":-1,"ns_id":3,"virt_text_repeat_linebreak":true,"priority":4096,"right_gravity":true,"virt_text":[["▌","OpencodeToolBorder"]]}],[9,11,0,{"virt_text_pos":"win_col","virt_text_hide":false,"virt_text_win_col":-1,"ns_id":3,"virt_text_repeat_linebreak":true,"priority":4096,"right_gravity":true,"virt_text":[["▌","OpencodeToolBorder"]]}],[10,12,0,{"virt_text_pos":"win_col","virt_text_hide":false,"virt_text_win_col":-1,"ns_id":3,"virt_text_repeat_linebreak":true,"priority":4096,"right_gravity":true,"virt_text":[["▌","OpencodeToolBorder"]]}],[11,13,0,{"virt_text_pos":"overlay","virt_text_hide":false,"end_col":0,"ns_id":3,"end_right_gravity":false,"virt_text_repeat_linebreak":false,"end_row":14,"hl_group":"OpencodeDiffDelete","right_gravity":true,"priority":5000,"hl_eol":true,"virt_text":[["-","OpencodeDiffDelete"]]}],[12,13,0,{"virt_text_pos":"win_col","virt_text_hide":false,"virt_text_win_col":-1,"ns_id":3,"virt_text_repeat_linebreak":true,"priority":4096,"right_gravity":true,"virt_text":[["▌","OpencodeToolBorder"]]}],[13,14,0,{"virt_text_pos":"overlay","virt_text_hide":false,"end_col":0,"ns_id":3,"end_right_gravity":false,"virt_text_repeat_linebreak":false,"end_row":15,"hl_group":"OpencodeDiffAdd","right_gravity":true,"priority":5000,"hl_eol":true,"virt_text":[["+","OpencodeDiffAdd"]]}],[14,14,0,{"virt_text_pos":"win_col","virt_text_hide":false,"virt_text_win_col":-1,"ns_id":3,"virt_text_repeat_linebreak":true,"priority":4096,"right_gravity":true,"virt_text":[["▌","OpencodeToolBorder"]]}],[15,15,0,{"virt_text_pos":"win_col","virt_text_hide":false,"virt_text_win_col":-1,"ns_id":3,"virt_text_repeat_linebreak":true,"priority":4096,"right_gravity":true,"virt_text":[["▌","OpencodeToolBorder"]]}],[16,16,0,{"virt_text_pos":"win_col","virt_text_hide":false,"virt_text_win_col":-1,"ns_id":3,"virt_text_repeat_linebreak":true,"priority":4096,"right_gravity":true,"virt_text":[["▌","OpencodeToolBorder"]]}],[17,21,0,{"virt_text_pos":"win_col","virt_text_hide":false,"virt_text_win_col":-3,"ns_id":3,"virt_text_repeat_linebreak":false,"priority":10,"right_gravity":true,"virt_text":[[" ","OpencodeMessageRoleAssistant"],[" "],["BUILD","OpencodeMessageRoleAssistant"],[" claude-sonnet-4.5","OpencodeHint"],[" (2025-10-12 06:43:03)","OpencodeHint"],[" [msg_9d7288f2f001hW6NqqhtBc72UU]","OpencodeHint"]]}]],"timestamp":1762908306,"lines":["----","","","can you add \"great\" before \"string\" in @diff-test.txt?","","[diff-test.txt](diff-test.txt)","","----","","","**  edit** `diff-test.txt`","","`````txt"," this is a string"," this is a great string","","`````","","**󰻛  Created Snapshot** `1f593f7e`","","----","","",""],"actions":[{"key":"R","text":"[R]evert file","display_line":20,"args":["1f593f7ed419c95d3995f8ef4b98d4e571c3a492"],"range":{"from":20,"to":20},"type":"diff_revert_selected_file"},{"key":"A","text":"Revert [A]ll","display_line":20,"args":["1f593f7ed419c95d3995f8ef4b98d4e571c3a492"],"range":{"from":20,"to":20},"type":"diff_revert_all"},{"key":"D","text":"[D]iff","display_line":20,"args":["1f593f7ed419c95d3995f8ef4b98d4e571c3a492"],"range":{"from":20,"to":20},"type":"diff_open"}]}
+{"actions":[{"range":{"from":19,"to":19},"args":["1f593f7ed419c95d3995f8ef4b98d4e571c3a492"],"type":"diff_revert_selected_file","text":"[R]evert file","display_line":19,"key":"R"},{"range":{"from":19,"to":19},"args":["1f593f7ed419c95d3995f8ef4b98d4e571c3a492"],"type":"diff_revert_all","text":"Revert [A]ll","display_line":19,"key":"A"},{"range":{"from":19,"to":19},"args":["1f593f7ed419c95d3995f8ef4b98d4e571c3a492"],"type":"diff_open","text":"[D]iff","display_line":19,"key":"D"}],"timestamp":1763482495,"lines":["----","","","can you add \"great\" before \"string\" in @diff-test.txt?","","[diff-test.txt](diff-test.txt)","","----","","","**  edit** `diff-test.txt`","","`````txt"," this is a string"," this is a great string","","`````","","**󰻛  Created Snapshot** `1f593f7e`","","----","","",""],"extmarks":[[1,1,0,{"ns_id":3,"right_gravity":true,"virt_text_pos":"win_col","virt_text_hide":false,"priority":10,"virt_text_repeat_linebreak":false,"virt_text":[["▌󰭻 ","OpencodeMessageRoleUser"],[" "],["USER","OpencodeMessageRoleUser"],["","OpencodeHint"],[" (2025-10-12 06:42:56)","OpencodeHint"],[" [msg_9d7287269001C5gRusYfX7A1w1]","OpencodeHint"]],"virt_text_win_col":-3}],[2,2,0,{"ns_id":3,"right_gravity":true,"virt_text_pos":"win_col","virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_win_col":-3}],[3,3,0,{"ns_id":3,"right_gravity":true,"virt_text_pos":"win_col","virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_win_col":-3}],[4,3,39,{"ns_id":3,"right_gravity":true,"end_col":53,"end_row":3,"priority":1000,"hl_eol":false,"end_right_gravity":false,"hl_group":"OpencodeMention"}],[5,4,0,{"ns_id":3,"right_gravity":true,"virt_text_pos":"win_col","virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_win_col":-3}],[6,5,0,{"ns_id":3,"right_gravity":true,"virt_text_pos":"win_col","virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeMessageRoleUser"]],"virt_text_win_col":-3}],[7,8,0,{"ns_id":3,"right_gravity":true,"virt_text_pos":"win_col","virt_text_hide":false,"priority":10,"virt_text_repeat_linebreak":false,"virt_text":[[" ","OpencodeMessageRoleAssistant"],[" "],["BUILD","OpencodeMessageRoleAssistant"],[" claude-sonnet-4.5","OpencodeHint"],[" (2025-10-12 06:42:56)","OpencodeHint"],[" [msg_9d7287287001HVwpPaH7WkRVdN]","OpencodeHint"]],"virt_text_win_col":-3}],[8,10,0,{"ns_id":3,"right_gravity":true,"virt_text_pos":"win_col","virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeToolBorder"]],"virt_text_win_col":-1}],[9,11,0,{"ns_id":3,"right_gravity":true,"virt_text_pos":"win_col","virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeToolBorder"]],"virt_text_win_col":-1}],[10,12,0,{"ns_id":3,"right_gravity":true,"virt_text_pos":"win_col","virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeToolBorder"]],"virt_text_win_col":-1}],[11,13,0,{"hl_eol":true,"right_gravity":true,"end_col":0,"end_row":14,"virt_text_repeat_linebreak":false,"hl_group":"OpencodeDiffDelete","ns_id":3,"virt_text":[["-","OpencodeDiffDelete"]],"virt_text_pos":"overlay","priority":5000,"end_right_gravity":false,"virt_text_hide":false}],[12,13,0,{"ns_id":3,"right_gravity":true,"virt_text_pos":"win_col","virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeToolBorder"]],"virt_text_win_col":-1}],[13,14,0,{"hl_eol":true,"right_gravity":true,"end_col":0,"end_row":15,"virt_text_repeat_linebreak":false,"hl_group":"OpencodeDiffAdd","ns_id":3,"virt_text":[["+","OpencodeDiffAdd"]],"virt_text_pos":"overlay","priority":5000,"end_right_gravity":false,"virt_text_hide":false}],[14,14,0,{"ns_id":3,"right_gravity":true,"virt_text_pos":"win_col","virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeToolBorder"]],"virt_text_win_col":-1}],[15,15,0,{"ns_id":3,"right_gravity":true,"virt_text_pos":"win_col","virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeToolBorder"]],"virt_text_win_col":-1}],[16,16,0,{"ns_id":3,"right_gravity":true,"virt_text_pos":"win_col","virt_text_hide":false,"priority":4096,"virt_text_repeat_linebreak":true,"virt_text":[["▌","OpencodeToolBorder"]],"virt_text_win_col":-1}],[17,21,0,{"ns_id":3,"right_gravity":true,"virt_text_pos":"win_col","virt_text_hide":false,"priority":10,"virt_text_repeat_linebreak":false,"virt_text":[[" ","OpencodeMessageRoleAssistant"],[" "],["BUILD","OpencodeMessageRoleAssistant"],[" claude-sonnet-4.5","OpencodeHint"],[" (2025-10-12 06:43:03)","OpencodeHint"],[" [msg_9d7288f2f001hW6NqqhtBc72UU]","OpencodeHint"]],"virt_text_win_col":-3}]]}