From 4c4964a740aa5ea1f00d01ab17425cc8221e60b0 Mon Sep 17 00:00:00 2001
From: Nayan <nayan8teen@gmail.com>
Date: Thu, 2 Oct 2025 14:42:15 -0400
Subject: [PATCH] Device: Wake for all visual alerts (#1309)

this should do it
---
 selfdrive/ui/sunnypilot/qt/onroad/hud.cc | 2 ++
 selfdrive/ui/sunnypilot/ui.cc            | 3 +--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/selfdrive/ui/sunnypilot/qt/onroad/hud.cc b/selfdrive/ui/sunnypilot/qt/onroad/hud.cc
index 607e0cbbb..21d66ffa9 100644
--- a/selfdrive/ui/sunnypilot/qt/onroad/hud.cc
+++ b/selfdrive/ui/sunnypilot/qt/onroad/hud.cc
@@ -210,6 +210,8 @@ void HudRendererSP::draw(QPainter &p, const QRect &surface_rect) {
     // Green Light & Lead Depart Alerts
     if (greenLightAlert or leadDepartAlert) {
       e2eAlertDisplayTimer = 3 * UI_FREQ;
+      // reset onroad sleep timer for e2e alerts
+      uiStateSP()->reset_onroad_sleep_timer();
     }
 
     if (e2eAlertDisplayTimer > 0) {
diff --git a/selfdrive/ui/sunnypilot/ui.cc b/selfdrive/ui/sunnypilot/ui.cc
index 6cbdeadf5..7f4fc8823 100644
--- a/selfdrive/ui/sunnypilot/ui.cc
+++ b/selfdrive/ui/sunnypilot/ui.cc
@@ -14,8 +14,7 @@ void UIStateSP::updateStatus() {
 
   if (scene.started && scene.onroadScreenOffControl) {
     auto selfdriveState = (*sm)["selfdriveState"].getSelfdriveState();
-    if (selfdriveState.getAlertSize() != cereal::SelfdriveState::AlertSize::NONE &&
-        selfdriveState.getAlertStatus() != cereal::SelfdriveState::AlertStatus::NORMAL) {
+    if (selfdriveState.getAlertSize() != cereal::SelfdriveState::AlertSize::NONE) {
       reset_onroad_sleep_timer();
     } else if (scene.onroadScreenOffTimer > 0) {
       scene.onroadScreenOffTimer--;