From 1bfd33fbd12d997db50fe5fc6d72af990e89fa33 Mon Sep 17 00:00:00 2001
From: cyberneel <neelesh.chevuri25@gmail.com>
Date: Tue, 17 Dec 2024 09:23:20 -0600
Subject: [PATCH] Added a variable to ignore button press during certion
 actions.

I think pressing the side button as the auto snooze function is running causes a crash.
---
 src/displayapp/screens/Sleep.cpp | 6 ++++--
 src/displayapp/screens/Sleep.h   | 2 ++
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/displayapp/screens/Sleep.cpp b/src/displayapp/screens/Sleep.cpp
index 9f2bcf9b7e..21b5cb1aba 100644
--- a/src/displayapp/screens/Sleep.cpp
+++ b/src/displayapp/screens/Sleep.cpp
@@ -24,13 +24,15 @@ namespace {
   }
 
   void SnoozeAlarmTaskCallback(lv_task_t* task) {
-    auto* screen = static_cast<Sleep*>(task->user_data);
     lv_task_set_prio(task, LV_TASK_PRIO_OFF);
+    auto* screen = static_cast<Sleep*>(task->user_data);
+    screen->ignoreButtonPush = true;
     screen->StopAlerting(false);
     screen->UpdateDisplay();
     screen->SnoozeWakeAlarm();
     screen->displayState = Sleep::SleepDisplayState::Info;
     screen->UpdateDisplay();
+    screen->ignoreButtonPush = false;
   }
 
   void PressesToStopAlarmTimeoutCallback(lv_task_t* task) {
@@ -555,7 +557,7 @@ void Sleep::OnButtonEvent(lv_obj_t* obj, lv_event_t event) {
 }
 
 bool Sleep::OnButtonPushed() {
-  if (infiniSleepController.IsAlerting()) {
+  if (infiniSleepController.IsAlerting() || ignoreButtonPush) {
     return true;
   }
   if (displayState != SleepDisplayState::Info) {
diff --git a/src/displayapp/screens/Sleep.h b/src/displayapp/screens/Sleep.h
index ed3d49e1a5..a38194775a 100644
--- a/src/displayapp/screens/Sleep.h
+++ b/src/displayapp/screens/Sleep.h
@@ -37,6 +37,8 @@ namespace Pinetime {
 
         Controllers::InfiniSleepController& infiniSleepController;
 
+        bool ignoreButtonPush = false;
+
       private:
         System::WakeLock wakeLock;
         Controllers::MotorController& motorController;