Add leds utils

Author: mirkokurt

src/arduino/app_utils/__init__.py

@@ -12,7 +12,7 @@
 from .jsonparser import *
 from .logger import *
 from .slidingwindowbuffer import *
-from .userinput import *
+from .leds import *
 
 __all__ = [
     "App",
@@ -31,4 +31,5 @@
     "SineGenerator",
     "SlidingWindowBuffer",
     "UserTextInput",
+    "Leds",
 ]

src/arduino/app_utils/leds.py

@@ -0,0 +1,55 @@
+from arduino.app_utils import Logger, Bridge
+
+logger = Logger(__name__)
+
+LED_IDS = [1, 2, 3, 4]  # Supported LED IDs
+
+LED_BRIGHTNESS_FILES = [
+    "/sys/class/leds/red:user/brightness",
+    "/sys/class/leds/green:user/brightness",
+    "/sys/class/leds/blue:user/brightness",
+    "/sys/class/leds/red:panic/brightness",
+    "/sys/class/leds/green:wlan/brightness",
+    "/sys/class/leds/blue:bt/brightness",
+]
+
+def set_led1_color(r, g, b):
+    write_led_file(LED_BRIGHTNESS_FILES[0], r)
+    write_led_file(LED_BRIGHTNESS_FILES[1], g)
+    write_led_file(LED_BRIGHTNESS_FILES[2], b)
+
+def set_led2_color(r, g, b):
+    write_led_file(LED_BRIGHTNESS_FILES[3], r)
+    write_led_file(LED_BRIGHTNESS_FILES[4], g)
+    write_led_file(LED_BRIGHTNESS_FILES[5], b)
+
+def write_led_file(led_file, color):
+    try:
+        with open(led_file, "w") as f:
+            f.write(f"{color}\n")
+    except Exception as e:
+        print(f"Error writing to {led_file}: {e}")
+
+class Leds:
+    @staticmethod
+    def set_led_color(ledid: int, rgb_color: dict):
+        try:
+            if ledid not in LED_IDS:
+                raise ValueError(f"Unknown led '{ledid}'")
+
+            if not rgb_color or not all(k in rgb_color for k in ("r", "g", "b")):
+                raise ValueError("Color must be an object with 'r', 'g', 'b' keys")
+
+            # Led 1 and 2 are controlled by Python code directly (MPU), while Led 3 and 4 are controlled via Bridge (MCU)
+            match ledid:
+                case 1:
+                    set_led1_color(rgb_color['r'], rgb_color['g'], rgb_color['b'])
+                case 2:
+                    set_led2_color(rgb_color['r'], rgb_color['g'], rgb_color['b'])
+                case 3 | 4:
+                    Bridge.call("set_led_color", ledid, rgb_color['r'], rgb_color['g'], rgb_color['b'])
+
+        except Exception as e:
+            Logger(__name__).error(f"LED color set error: {e}")
+
+Leds = Leds()