From 3387a4ab40d7b9adbf9f0d45bbd65b4570299140 Mon Sep 17 00:00:00 2001
From: tychovrahe <brunam@seznam.cz>
Date: Thu, 20 Mar 2025 14:47:25 +0100
Subject: [PATCH] feat(core/prodtest): add hibernation on power button
 long-press

[no changelog]
---
 .../projects/prodtest/cmd/prodtest_rgbled.c   | 24 ++++++++++++++
 .../projects/prodtest/cmd/prodtest_rgbled.h   | 24 ++++++++++++++
 core/embed/projects/prodtest/main.c           | 33 +++++++++++++++++++
 3 files changed, 81 insertions(+)
 create mode 100644 core/embed/projects/prodtest/cmd/prodtest_rgbled.h

diff --git a/core/embed/projects/prodtest/cmd/prodtest_rgbled.c b/core/embed/projects/prodtest/cmd/prodtest_rgbled.c
index 56ac01ef66..e14ee341e0 100644
--- a/core/embed/projects/prodtest/cmd/prodtest_rgbled.c
+++ b/core/embed/projects/prodtest/cmd/prodtest_rgbled.c
@@ -23,6 +23,29 @@
 
 #include <io/rgb_led.h>
 #include <rtl/cli.h>
+#include <sys/systimer.h>
+
+static bool g_prodtest_rgbled_start = false;
+
+static void prodtest_rgbled_timer_cb(void* context) {
+  if (g_prodtest_rgbled_start) {
+    rgb_led_set_color(0);
+    g_prodtest_rgbled_start = false;
+  }
+}
+
+void prodtest_rgbled_init(void) {
+  static systimer_t* timer = NULL;
+
+  timer = systimer_create(prodtest_rgbled_timer_cb, NULL);
+
+  if (timer == NULL) {
+    return;
+  }
+  systimer_set(timer, 1000);
+  rgb_led_set_color(0xFF00);
+  g_prodtest_rgbled_start = true;
+}
 
 static void prodtest_rgbled_set(cli_t* cli) {
   uint32_t r = 0;
@@ -53,6 +76,7 @@ static void prodtest_rgbled_set(cli_t* cli) {
 
   uint32_t rgb = (r << 16) | (g << 8) | b;
 
+  g_prodtest_rgbled_start = false;
   rgb_led_set_color(rgb);
 
   cli_ok(cli, "");
diff --git a/core/embed/projects/prodtest/cmd/prodtest_rgbled.h b/core/embed/projects/prodtest/cmd/prodtest_rgbled.h
new file mode 100644
index 0000000000..5992f9cfb3
--- /dev/null
+++ b/core/embed/projects/prodtest/cmd/prodtest_rgbled.h
@@ -0,0 +1,24 @@
+/*
+ * This file is part of the Trezor project, https://trezor.io/
+ *
+ * Copyright (c) SatoshiLabs
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#pragma once
+
+#include <trezor_types.h>
+
+void prodtest_rgbled_init(void);
diff --git a/core/embed/projects/prodtest/main.c b/core/embed/projects/prodtest/main.c
index b787447bea..2693b197bb 100644
--- a/core/embed/projects/prodtest/main.c
+++ b/core/embed/projects/prodtest/main.c
@@ -24,6 +24,7 @@
 #include <io/usb.h>
 #include <rtl/cli.h>
 #include <sys/system.h>
+#include <sys/systick.h>
 #include <util/flash_otp.h>
 #include <util/rsod.h>
 #include <util/unit_properties.h>
@@ -66,6 +67,7 @@
 
 #ifdef USE_RGB_LED
 #include <io/rgb_led.h>
+#include "cmd/prodtest_rgbled.h"
 #endif
 
 #ifdef USE_HASH_PROCESSOR
@@ -213,6 +215,7 @@ static void drivers_init(void) {
 #endif
 #ifdef USE_RGB_LED
   rgb_led_init();
+  prodtest_rgbled_init();
 #endif
 #ifdef USE_BLE
   unit_properties_init();
@@ -252,6 +255,10 @@ int main(void) {
   pair_optiga(&g_cli);
 #endif
 
+#if defined USE_BUTTON && defined USE_POWERCTL
+  uint32_t btn_deadline = 0;
+#endif
+
   while (true) {
     cli_event_t cli_event = {0};
 
@@ -262,6 +269,32 @@ int main(void) {
         cli_process_command(&g_cli, cli_event.data.command_received.command);
       }
     }
+
+#if defined USE_BUTTON && defined USE_POWERCTL
+    button_event_t btn_event = {0};
+    bool btn = button_get_event(&btn_event);
+
+    if (btn) {
+      if (btn_event.button == BTN_POWER &&
+          btn_event.event_type == BTN_EVENT_DOWN) {
+        btn_deadline = ticks_timeout(1000);
+      }
+      if (btn_event.button == BTN_POWER &&
+          btn_event.event_type == BTN_EVENT_UP) {
+        if (ticks_expired(systick_ms())) {
+          powerctl_hibernate();
+          rgb_led_set_color(0xffff00);
+          systick_delay_ms(1000);
+          rgb_led_set_color(0);
+        }
+      }
+    }
+
+    if (button_is_down(BTN_POWER) && ticks_expired(btn_deadline)) {
+      rgb_led_set_color(0xff0000);
+    }
+
+#endif
   }
 
   return 0;