fix(undo): fix undo after opening an old session

Author: sudo-tee

lua/opencode/ui/renderer/events.lua

@@ -224,10 +224,11 @@ function M.on_message_updated(message, revert_index)
     replay_orphan_parts(msg.info.id)
     flush.mark_message_dirty(msg.info.id)
     state.renderer.set_current_message(msg)
-    if message.info.role == 'user' then
-      state.renderer.set_last_user_message(msg)
-      scroll(true)
-    end
+  end
+
+  if msg.info.role == 'user' then
+    state.renderer.set_last_user_message(msg)
+    scroll(true)
   end
 
   update_stats(msg)