Initial commit

custom_components/smartthings_find/button.py

@@ -0,0 +1,62 @@
+import logging
+from homeassistant.components.button import ButtonEntity
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity import DeviceInfo
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+
+from .const import DOMAIN
+from .utils import fetch_csrf
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None:
+    """Set up SmartThings Find button entities."""
+    devices = hass.data[DOMAIN]["devices"]
+    entities = []
+    for device in devices:
+        entities += [RingButton(hass, device)]
+    async_add_entities(entities)
+
+
+class RingButton(ButtonEntity):
+    """Representation a button entity to make a SmartThings Find device ring."""
+
+    def __init__(self, hass: HomeAssistant, device):
+        """Initialize the button."""
+        self._attr_unique_id = f"stf_ring_button_{device['data']['dvceID']}"
+        self._attr_name = f"{device['data']['modelName']} Ring"
+
+        if 'icons' in device['data'] and 'coloredIcon' in device['data']['icons']:
+            self._attr_entity_picture = device['data']['icons']['coloredIcon']
+        self._attr_icon = 'mdi:nfc-search-variant'
+        # self.hass = hass
+        self.device = device['data']
+        self._attr_device_info = device['ha_dev_info']
+
+    async def async_press(self):
+        """Handle the button press."""
+        session = self.hass.data[DOMAIN]["session"]
+        csrf_token = self.hass.data[DOMAIN]["_csrf"]
+        ring_payload = {
+            "dvceId": self.device['dvceID'],
+            "operation": "RING",
+            "usrId": self.device['usrId'],
+            "status": "start",
+            "lockMessage": "Home Assistant is ringing your device!"
+        }
+        url = f"https://smartthingsfind.samsung.com/dm/addOperation.do?_csrf={
+            csrf_token}"
+
+        try:
+            async with session.post(url, json=ring_payload) as response:
+                _LOGGER.debug("HTTP response status: %s", response.status)
+                if response.status == 200:
+                    _LOGGER.info(f"Successfully rang device {self.device['modelName']}")
+                    _LOGGER.debug(f"Response: {await response.text()}")
+                else:
+                    # Fetch a new CSRF token to make sure we're still logged in
+                    await fetch_csrf(self.hass, session)
+        except Exception as e:
+            _LOGGER.error(f"Exception occurred while ringing '{self.device['modelName']}': %s", e)