set rgb color and set random color #132 and #76

Author: dave-code-ruiz

custom_components/elkbledom/__init__.py

@@ -1,9 +1,13 @@
 from __future__ import annotations
 
+import random
+import voluptuous as vol
+
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.core import HomeAssistant, Event
-from homeassistant.const import CONF_MAC, EVENT_HOMEASSISTANT_STOP
+from homeassistant.core import HomeAssistant, Event, ServiceCall
+from homeassistant.const import CONF_MAC, EVENT_HOMEASSISTANT_STOP, ATTR_ENTITY_ID
 from homeassistant.const import Platform
+from homeassistant.helpers import config_validation as cv
 
 from .const import DOMAIN, CONF_RESET, CONF_DELAY, CONF_MODEL
 from .elkbledom import BLEDOMInstance
@@ -18,6 +22,24 @@
     Platform.SWITCH,
 ]
 
+# Service names
+SERVICE_SET_RANDOM_COLOR = "set_random_color"
+SERVICE_SET_RGB_COLOR = "set_rgb_color"
+
+# Service schemas
+SERVICE_SET_RANDOM_COLOR_SCHEMA = vol.Schema({
+    vol.Required(ATTR_ENTITY_ID): cv.entity_ids,
+    vol.Optional("brightness", default=255): vol.All(vol.Coerce(int), vol.Range(min=0, max=255)),
+})
+
+SERVICE_SET_RGB_COLOR_SCHEMA = vol.Schema({
+    vol.Required(ATTR_ENTITY_ID): cv.entity_ids,
+    vol.Required("r"): vol.All(vol.Coerce(int), vol.Range(min=0, max=255)),
+    vol.Required("g"): vol.All(vol.Coerce(int), vol.Range(min=0, max=255)),
+    vol.Required("b"): vol.All(vol.Coerce(int), vol.Range(min=0, max=255)),
+    vol.Optional("brightness", default=255): vol.All(vol.Coerce(int), vol.Range(min=0, max=255)),
+})
+
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Set up ElkBLEDOM from a config entry."""
     reset = entry.options.get(CONF_RESET, None) or entry.data.get(CONF_RESET, None)
@@ -43,6 +65,73 @@ async def _async_stop(event: Event) -> None:
         hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, _async_stop)
     )
 
+    # Register services (only once for all entries)
+    if not hass.services.has_service(DOMAIN, SERVICE_SET_RANDOM_COLOR):
+        async def handle_set_random_color(call: ServiceCall) -> None:
+            """Handle the set_random_color service call."""
+            entity_ids = call.data[ATTR_ENTITY_ID]
+            brightness = call.data.get("brightness", 255)
+            
+            # Generate random RGB values
+            r = random.randint(0, 255)
+            g = random.randint(0, 255)
+            b = random.randint(0, 255)
+            
+            # Call light.turn_on with random color for each entity
+            await hass.services.async_call(
+                "light",
+                "turn_on",
+                {
+                    ATTR_ENTITY_ID: entity_ids,
+                    "rgb_color": [r, g, b],
+                    "brightness": brightness,
+                },
+                blocking=True,
+            )
+            LOGGER.debug(
+                "Random color set to RGB(%d, %d, %d) with brightness %d for entities: %s",
+                r, g, b, brightness, entity_ids
+            )
+        
+        hass.services.async_register(
+            DOMAIN,
+            SERVICE_SET_RANDOM_COLOR,
+            handle_set_random_color,
+            schema=SERVICE_SET_RANDOM_COLOR_SCHEMA,
+        )
+    
+    if not hass.services.has_service(DOMAIN, SERVICE_SET_RGB_COLOR):
+        async def handle_set_rgb_color(call: ServiceCall) -> None:
+            """Handle the set_rgb_color service call."""
+            entity_ids = call.data[ATTR_ENTITY_ID]
+            r = call.data["r"]
+            g = call.data["g"]
+            b = call.data["b"]
+            brightness = call.data.get("brightness", 255)
+            
+            # Call light.turn_on with specified RGB color
+            await hass.services.async_call(
+                "light",
+                "turn_on",
+                {
+                    ATTR_ENTITY_ID: entity_ids,
+                    "rgb_color": [r, g, b],
+                    "brightness": brightness,
+                },
+                blocking=True,
+            )
+            LOGGER.debug(
+                "RGB color set to (%d, %d, %d) with brightness %d for entities: %s",
+                r, g, b, brightness, entity_ids
+            )
+        
+        hass.services.async_register(
+            DOMAIN,
+            SERVICE_SET_RGB_COLOR,
+            handle_set_rgb_color,
+            schema=SERVICE_SET_RGB_COLOR_SCHEMA,
+        )
+    
     return True
 
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:

custom_components/elkbledom/services.yaml

@@ -0,0 +1,77 @@
+set_random_color:
+  name: Set Random Color
+  description: Set a random RGB color to the LED strip
+  fields:
+    entity_id:
+      name: Entity
+      description: The light entity to control
+      required: true
+      example: "light.led1"
+      selector:
+        entity:
+          domain: light
+    brightness:
+      name: Brightness
+      description: Brightness level (0-255)
+      required: false
+      default: 255
+      example: 200
+      selector:
+        number:
+          min: 0
+          max: 255
+          mode: slider
+
+set_rgb_color:
+  name: Set RGB Color
+  description: Set a specific RGB color to the LED strip
+  fields:
+    entity_id:
+      name: Entity
+      description: The light entity to control
+      required: true
+      example: "light.led1"
+      selector:
+        entity:
+          domain: light
+    r:
+      name: Red
+      description: Red value (0-255)
+      required: true
+      example: 255
+      selector:
+        number:
+          min: 0
+          max: 255
+          mode: slider
+    g:
+      name: Green
+      description: Green value (0-255)
+      required: true
+      example: 128
+      selector:
+        number:
+          min: 0
+          max: 255
+          mode: slider
+    b:
+      name: Blue
+      description: Blue value (0-255)
+      required: true
+      example: 64
+      selector:
+        number:
+          min: 0
+          max: 255
+          mode: slider
+    brightness:
+      name: Brightness
+      description: Brightness level (0-255)
+      required: false
+      default: 255
+      example: 200
+      selector:
+        number:
+          min: 0
+          max: 255
+          mode: slider