__init__.py

"""BrandMeister Event Feed"""
from dataclasses import asdict
from json import loads as json_loads

from socketio import AsyncClient

from homeassistant.const import EVENT_HOMEASSISTANT_STOP
from homeassistant.core import HomeAssistant
from .data import Callee, Caller, Call
from .trigger import Trigger

DOMAIN = "bm_feed"


async def async_setup(hass: HomeAssistant, config: dict):
    if DOMAIN in hass.data:
        sio = hass.data[DOMAIN]

    else:
        hass.data[DOMAIN] = sio = AsyncClient()

        async def setup():
            await sio.connect(
                url="https://api.brandmeister.network",
                socketio_path="lh/socket.io",
                transports="websocket",
            )
            await sio.wait()

        task = hass.loop.create_task(setup())

        async def teardown():
            await sio.disconnect()
            await task

        hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, teardown())

    triggers = (
        [Trigger(query) for query in config[DOMAIN]["triggers"]]
        if "triggers" in config[DOMAIN]
        else None
    )

    @sio.on("mqtt")
    async def _handler(data: dict):
        event = None
        if data["topic"] == "LH":
            payload = json_loads(data["payload"])
            if (
                payload["Event"] == "Session-Stop"
                and "Call" in payload["CallTypes"]
                and (
                    payload["LinkCall"]
                    + payload["SourceCall"]
                    + payload["DestinationCall"]
                )
                != ""
            ):
                event = (
                    "call",
                    Call(
                        Caller(
                            int(payload["SourceID"]),
                            str(payload["SourceCall"]),
                        ),
                        Callee(
                            int(payload["DestinationID"]),
                            str(payload["DestinationCall"]),
                            "Group" in payload["CallTypes"],
                        ),
                    ),
                )
        if event:
            if triggers is None:
                hass.bus.async_fire(f"{DOMAIN}/{event[0]}", event[1])
            else:
                triggered = [str(trigger) for trigger in triggers if trigger(event[1])]
                if triggered:
                    hass.bus.async_fire(
                        f"{DOMAIN}/{event[0]}",
                        {**asdict(event[1]), "triggers": triggered},
                    )

    return True