import logging

from homeassistant.components.button import ButtonEntity

_LOGGER = logging.getLogger(__name__)

DOMAIN = "plejd"

async def async_setup_entry(hass, config_entry, async_add_entities):
    scenes = hass.data[DOMAIN]["scenes"].get(config_entry.entry_id, [])

    entities = []
    for s in scenes:
        button = PlejdSceneButton(s, config_entry.entry_id)
        entities.append(button)
    async_add_entities(entities, False)

class PlejdSceneButton(ButtonEntity):

    def __init__(self, device, entry_id):
        super().__init__()
        self.device = device
        self.entry_id = entry_id

    @property
    def device_info(self):
        return {
            "identifiers": {(DOMAIN, f"{self.entry_id}:{self.device.index}")},
            "name": self.device.name,
            "manufacturer": "Plejd",
            #"connections": ???,
        }

    @property
    def has_entity_name(self):
        return True
    
    @property
    def name(self):
        return None

    @property
    def unique_id(self):
        return f"{self.entry_id}:{self.device.index}"

    async def async_press(self):
        await self.device.activate()