import json
import logging
from typing import TYPE_CHECKING

from awscrt import mqtt5
from awsiot import mqtt5_client_builder  # type: ignore[import-untyped]

from pyhon import const
from pyhon.appliance import HonAppliance

if TYPE_CHECKING:
    from pyhon import HonAPI

_LOGGER = logging.getLogger(__name__)

appliances: list[HonAppliance] = []


def on_lifecycle_stopped(lifecycle_stopped_data: mqtt5.LifecycleStoppedData) -> None:
    _LOGGER.info("Lifecycle Stopped: %s", str(lifecycle_stopped_data))


def on_lifecycle_connection_success(
    lifecycle_connect_success_data: mqtt5.LifecycleConnectSuccessData,
) -> None:
    _LOGGER.info(
        "Lifecycle Connection Success: %s", str(lifecycle_connect_success_data)
    )


def on_publish_received(data: mqtt5.PublishReceivedData) -> None:
    if not (data and data.publish_packet and data.publish_packet.payload):
        return
    payload = json.loads(data.publish_packet.payload.decode())
    topic = data.publish_packet.topic
    if topic and "appliancestatus" in topic:
        appliance = next(
            a for a in appliances if topic in a.info["topics"]["subscribe"]
        )
        for parameter in payload["parameters"]:
            appliance.attributes["parameters"][parameter["parName"]].update(parameter)
        appliance.notify()
        _LOGGER.debug("%s - %s", topic, payload)
    else:
        _LOGGER.debug("%s - %s", topic, payload)


async def create_mqtt_client(api: "HonAPI") -> mqtt5.Client:
    client: mqtt5.Client = mqtt5_client_builder.websockets_with_custom_authorizer(
        endpoint=const.AWS_ENDPOINT,
        auth_authorizer_name=const.AWS_AUTHORIZER,
        auth_authorizer_signature=await api.load_aws_token(),
        auth_token_key_name="token",
        auth_token_value=api.auth.id_token,
        client_id=const.MOBILE_ID,
        on_lifecycle_stopped=on_lifecycle_stopped,
        on_lifecycle_connection_success=on_lifecycle_connection_success,
        on_publish_received=on_publish_received,
    )
    client.start()
    return client


def subscribe(client: mqtt5.Client, appliance: HonAppliance) -> None:
    for topic in appliance.info.get("topics", {}).get("subscribe", []):
        client.subscribe(mqtt5.SubscribePacket([mqtt5.Subscription(topic)])).result(10)
        _LOGGER.info("Subscribed to topic %s", topic)


async def start(api: "HonAPI", app: list[HonAppliance]) -> mqtt5.Client:
    client = await create_mqtt_client(api)
    global appliances  # pylint: disable=global-statement
    appliances = app
    for appliance in appliances:
        subscribe(client, appliance)
    return client