button.py: show feedback on display

ThomasFarstrike · ThomasFarstrike · commit 3addeb846af0 · 2025-05-06T16:17:25.000+02:00
diff --git a/internal_filesystem/system/button.py b/internal_filesystem/system/button.py
@@ -2,6 +2,7 @@
 
 from machine import Pin, Timer
 import time
+import _thread
 
 # Configure IO0 as input with pull-up resistor
 button = Pin(0, Pin.IN, Pin.PULL_UP)
@@ -14,32 +15,41 @@
 # Timer for checking long press
 timer = Timer(-1)
 
+message = "Bootloader mode activated.\nYou can now install firmware over USB.\n\n"
+
+def handle_long_press():
+    print(message)
+    import lvgl as lv
+    screen = lv.obj()
+    label = lv.label(screen)
+    label.set_text(message)
+    label.center()
+    lv.screen_load(screen)
+    time.sleep(1)
+    import machine
+    machine.bootloader()
 
 def on_long_press(t): # Callback for when long press duration is reached.
+    global timer
     timer.deinit()  # Stop the timer
     global is_pressed
     if is_pressed and button.value() == 0:  # Ensure button is still pressed
-        print("Button IO0 long pressed, going into bootloader mode...")
-        import machine
-        machine.bootloader()
+        _thread.start_new_thread(handle_long_press, ())
     else:
         is_pressed = False
 
 
 def button_handler(pin):
     """Interrupt handler for button press and release."""
-    global press_start_time, is_pressed
-    # Debounce: Ignore interrupts within 50ms of the last event
-    if time.ticks_diff(time.ticks_ms(), press_start_time) < 50:
-        return
+    global press_start_time, is_pressed, timer
     if button.value() == 0:  # Button pressed (LOW due to pull-up)
-        print("Button IO0 pressed.")
+        print("Button IO0 pressed")
         press_start_time = time.ticks_ms()
         is_pressed = True
         # Start timer to check for long press after long_press_duration
         timer.init(mode=Timer.ONE_SHOT, period=long_press_duration, callback=on_long_press)
     else:  # Button released (HIGH)
-        print("Button IO0 released.")
+        print("Button IO0 released")
         timer.deinit()  # Cancel timer if button is released early
         is_pressed = False
 
