Add long press and release event support

custom_components/button_plus/button.py

@@ -3,19 +3,29 @@
 from __future__ import annotations
 
 import logging
+from datetime import timedelta
+from functools import cached_property
+from typing import Any
 
 from homeassistant.components.button import ButtonEntity, ButtonDeviceClass
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.device_registry import DeviceInfo
+from homeassistant.helpers.entity_platform import (
+    AddEntitiesCallback,
+    async_get_current_platform,
+)
 
 from .button_plus_api.model import Connector, ConnectorEnum
 from .const import DOMAIN
 from . import ButtonPlusHub
 
 _LOGGER = logging.getLogger(__name__)
 
+SCAN_INTERVAL = timedelta(seconds=30)
+SERVICE_LONG_PRESS = "long_press"
+SERVICE_RELEASE = "release"
+
 
 async def async_setup_entry(
     hass: HomeAssistant,
@@ -47,8 +57,23 @@ async def async_setup_entry(
 
     async_add_entities(button_entities)
 
+    platform = async_get_current_platform()
+    platform.async_register_entity_service(
+        SERVICE_LONG_PRESS,
+        {},
+        "_async_long_press_action",
+    )
+
+    platform.async_register_entity_service(
+        SERVICE_RELEASE,
+        {},
+        "_async_release_action",
+    )
+
 
 class ButtonPlusButton(ButtonEntity):
+    _attr_click_type: str | None = None
+
     def __init__(self, btn_id: int, hub: ButtonPlusHub):
         self._is_on = False
         self._hub_id = hub.hub_id
@@ -107,3 +132,24 @@ def device_info(self) -> DeviceInfo:
     async def async_press(self) -> None:
         """Handle the button press."""
         _LOGGER.debug(f"async press from mqtt button: {self._btn_id}")
+
+    async def _async_long_press_action(self) -> None:
+        self._attr_click_type = "long"
+        await super()._async_press_action()
+
+    async def _async_press_action(self) -> None:
+        self._attr_click_type = "single"
+        await super()._async_press_action()
+
+    async def _async_release_action(self) -> None:
+        pass
+
+    @property
+    def state_attributes(self) -> dict[str, Any] | None:
+        return {
+            "click_type": self._attr_click_type,
+        }
+
+    @cached_property
+    def click_type(self) -> str | None:
+        return self._attr_click_type

custom_components/button_plus/config_flow.py

@@ -342,6 +342,14 @@ def add_topics_to_buttons(
                 }
             )
 
+            # Create topics for button click
+            button.topics.append({
+                "brokerid": "ha-button-plus",
+                "topic": f"buttonplus/{device_id}/button/{button.button_id}/long_press",
+                "payload": "press",
+                "eventtype": EventType.LONG_PRESS
+            })
+
         return device_config
 
     def get_mqtt_endpoint(self, endpoint: str) -> str:

custom_components/button_plus/coordinator.py

@@ -29,33 +29,25 @@ def __init__(self, hass: HomeAssistant, hub: ButtonPlusHub):
         self.hub = hub
         self._hass = hass
         self._mqtt_subscribed_buttons = False
-        self._mqtt_topic_buttons = f"buttonplus/{hub.hub_id}/button/+/click"
-        self._mqtt_topic_brightness = f"buttonplus/{hub.hub_id}/brightness/+"
-        self._mqtt_topic_page = f"buttonplus/{hub.hub_id}/page/+"
+        self._mqtt_topics = [
+            (f"buttonplus/{hub.hub_id}/button/+/click", self.mqtt_button_callback),
+            (f"buttonplus/{hub.hub_id}/button/+/long_press", self.mqtt_button_long_press_callback),
+            (f"buttonplus/{hub.hub_id}/brightness/+", self.mqtt_brightness_callback),
+            (f"buttonplus/{hub.hub_id}/page/+", self.mqtt_page_callback),
+        ]
 
     async def _async_update_data(self):
         """Create MQTT subscriptions for buttonplus"""
         _LOGGER.debug("Initial data fetch from coordinator")
         if not self._mqtt_subscribed_buttons:
-            self.unsubscribe_mqtt = await mqtt.async_subscribe(
-                self._hass, self._mqtt_topic_buttons, self.mqtt_button_callback, 0
-            )
-            _LOGGER.debug(f"MQTT subscribed to {self._mqtt_topic_buttons}")
-
-        if not self._mqtt_subscribed_buttons:
-            self.unsubscribe_mqtt = await mqtt.async_subscribe(
-                self._hass,
-                self._mqtt_topic_brightness,
-                self.mqtt_brightness_callback,
-                0,
-            )
-            _LOGGER.debug(f"MQTT subscribed to {self._mqtt_topic_brightness}")
-
-        if not self._mqtt_subscribed_buttons:
-            self.unsubscribe_mqtt = await mqtt.async_subscribe(
-                self._hass, self._mqtt_topic_page, self.mqtt_page_callback, 0
-            )
-            _LOGGER.debug(f"MQTT subscribed to {self._mqtt_topic_page}")
+            for topic, cb in self._mqtt_topics:
+                self.unsubscribe_mqtt = await mqtt.async_subscribe(
+                    self._hass,
+                    topic,
+                    cb,
+                    0
+                )
+                _LOGGER.debug(f"MQTT subscribed to {topic}")
 
     @callback
     async def mqtt_page_callback(self, message: ReceiveMessage):
@@ -92,3 +84,18 @@ async def mqtt_button_callback(self, message: ReceiveMessage):
         await self.hass.services.async_call(
             "button", "press", target={"entity_id": entity.entity_id}
         )
+
+    @callback
+    async def mqtt_button_long_press_callback(self, message: ReceiveMessage):
+        # Handle the message here
+        _LOGGER.debug(f"Received message on topic {message.topic}: {message.payload}")
+        match = re.search(r'/(\d+)/long_press', message.topic)
+        btn_id = int(match.group(1)) if match else None
+
+        entity: ButtonEntity = self.hub.button_entities[str(btn_id)]
+
+        await self.hass.services.async_call(
+            DOMAIN,
+            'long_press',
+            target={"entity_id": entity.entity_id}
+        )

custom_components/button_plus/services.yaml

@@ -0,0 +1,10 @@
+long_press:
+  description: Long press of a button on a Button+
+  target:
+    entity:
+      domain: button_plus
+release:
+    description: Release of a button on a Button+
+    target:
+        entity:
+          domain: button_plus

custom_components/button_plus/strings.json

@@ -0,0 +1,14 @@
+{
+  "config": {
+    "services": {
+      "long_press": {
+        "name": "Long press",
+        "description": "Long press on a button on button+"
+      },
+      "release": {
+        "name": "Release",
+        "description": "Release of a button on button+"
+      }
+    }
+  }
+}