core/homeassistant/components/telegram_bot/event.py

"""Event platform for Telegram bot integration."""

from typing import Any

from homeassistant.components.event import EventEntity, EventEntityDescription
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback

from .bot import TelegramBotConfigEntry
from .const import (
    EVENT_TELEGRAM_ATTACHMENT,
    EVENT_TELEGRAM_CALLBACK,
    EVENT_TELEGRAM_COMMAND,
    EVENT_TELEGRAM_SENT,
    EVENT_TELEGRAM_TEXT,
)
from .entity import TelegramBotEntity
from .helpers import signal


async def async_setup_entry(
    hass: HomeAssistant,
    config_entry: TelegramBotConfigEntry,
    async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
    """Set up the event platform."""
    async_add_entities([TelegramBotEventEntity(config_entry)])


class TelegramBotEventEntity(TelegramBotEntity, EventEntity):
    """An event entity."""

    _attr_event_types = [
        EVENT_TELEGRAM_ATTACHMENT,
        EVENT_TELEGRAM_CALLBACK,
        EVENT_TELEGRAM_COMMAND,
        EVENT_TELEGRAM_TEXT,
        EVENT_TELEGRAM_SENT,
    ]

    def __init__(
        self,
        config_entry: TelegramBotConfigEntry,
    ) -> None:
        """Initialize the entity."""

        super().__init__(
            config_entry,
            EventEntityDescription(key="update_event", translation_key="update_event"),
        )

    async def async_added_to_hass(self) -> None:
        """Register callbacks."""
        self.async_on_remove(
            async_dispatcher_connect(
                self.hass,
                signal(self.config_entry.runtime_data.bot),
                self._async_handle_event,
            )
        )

    @callback
    def _async_handle_event(self, event_type: str, event_data: dict[str, Any]) -> None:
        """Handle the event."""
        self._trigger_event(event_type, event_data)
        self.async_write_ha_state()