"""Third Reality plug devices."""

from typing import Final

from zigpy.quirks import CustomCluster
from zigpy.quirks.v2 import QuirkBuilder
from zigpy.quirks.v2.homeassistant import UnitOfTime
from zigpy.quirks.v2.homeassistant.number import NumberDeviceClass
import zigpy.types as t
from zigpy.zcl.foundation import BaseAttributeDefs, ZCLAttributeDef


class ThirdRealityPlugCluster(CustomCluster):
    """Third Reality's plug private cluster."""

    cluster_id = 0xFF03

    class AttributeDefs(BaseAttributeDefs):
        """Define the attributes of a private cluster."""

        # reset the accumulated power of the plug
        reset_summation_delivered: Final = ZCLAttributeDef(
            id=0x0000,
            type=t.uint8_t,
            manufacturer_code=0x1233,
        )

        # turn off delay
        on_to_off_delay: Final = ZCLAttributeDef(
            id=0x0001,
            type=t.uint16_t,
            manufacturer_code=0x1233,
        )

        # turn on delay
        off_to_on_delay: Final = ZCLAttributeDef(
            id=0x0002,
            type=t.uint16_t,
            manufacturer_code=0x1233,
        )


# single outlet plugs
(
    QuirkBuilder("Third Reality, Inc", "3RSP02028BZ")
    .applies_to("Third Reality, Inc", "3RSPE01044BZ")
    .applies_to("Third Reality, Inc", "3RSPU01080Z")
    .applies_to("Third Reality, Inc", "3RSP02064Z")
    .applies_to("Third Reality, Inc", "3RSPE02065Z")
    .replaces(ThirdRealityPlugCluster)
    .write_attr_button(
        attribute_name=ThirdRealityPlugCluster.AttributeDefs.reset_summation_delivered.name,
        attribute_value=0x01,  # 1 reset summation delivered
        cluster_id=ThirdRealityPlugCluster.cluster_id,
        translation_key="reset_summation_delivered",
        fallback_name="Reset summation delivered",
    )
    .number(
        attribute_name=ThirdRealityPlugCluster.AttributeDefs.on_to_off_delay.name,
        cluster_id=ThirdRealityPlugCluster.cluster_id,
        endpoint_id=1,
        min_value=0,
        max_value=65535,
        mode="box",
        unit=UnitOfTime.SECONDS,
        device_class=NumberDeviceClass.DURATION,
        translation_key="turn_off_delay",
        fallback_name="Turn off delay",
    )
    .number(
        attribute_name=ThirdRealityPlugCluster.AttributeDefs.off_to_on_delay.name,
        cluster_id=ThirdRealityPlugCluster.cluster_id,
        endpoint_id=1,
        min_value=0,
        max_value=65535,
        mode="box",
        unit=UnitOfTime.SECONDS,
        device_class=NumberDeviceClass.DURATION,
        translation_key="turn_on_delay",
        fallback_name="Turn on delay",
    )
    .add_to_registry()
)

# double outlet plugs
(
    QuirkBuilder("Third Reality, Inc", "3RDP01072Z")
    .applies_to("Third Reality, Inc", "3RWP01073Z")
    .replaces(ThirdRealityPlugCluster, endpoint_id=1)
    .replaces(ThirdRealityPlugCluster, endpoint_id=2)
    .write_attr_button(
        attribute_name=ThirdRealityPlugCluster.AttributeDefs.reset_summation_delivered.name,
        attribute_value=0x01,  # 1 reset summation delivered
        cluster_id=ThirdRealityPlugCluster.cluster_id,
        endpoint_id=1,
        translation_key="reset_summation_delivered_left",
        fallback_name="Reset left summation delivered",  # ep1 is left
    )
    .write_attr_button(
        attribute_name=ThirdRealityPlugCluster.AttributeDefs.reset_summation_delivered.name,
        attribute_value=0x01,
        cluster_id=ThirdRealityPlugCluster.cluster_id,
        endpoint_id=2,
        translation_key="reset_summation_delivered_right",
        fallback_name="Reset right summation delivered",  # ep2 is right
    )
    .number(
        attribute_name=ThirdRealityPlugCluster.AttributeDefs.on_to_off_delay.name,
        cluster_id=ThirdRealityPlugCluster.cluster_id,
        endpoint_id=1,
        min_value=0,
        max_value=65535,
        mode="box",
        unit=UnitOfTime.SECONDS,
        device_class=NumberDeviceClass.DURATION,
        translation_key="turn_off_delay_left",
        fallback_name="Turn off delay left",
    )
    .number(
        attribute_name=ThirdRealityPlugCluster.AttributeDefs.on_to_off_delay.name,
        cluster_id=ThirdRealityPlugCluster.cluster_id,
        endpoint_id=2,
        min_value=0,
        max_value=65535,
        mode="box",
        unit=UnitOfTime.SECONDS,
        device_class=NumberDeviceClass.DURATION,
        translation_key="turn_off_delay_right",
        fallback_name="Turn off delay right",
    )
    .number(
        attribute_name=ThirdRealityPlugCluster.AttributeDefs.off_to_on_delay.name,
        cluster_id=ThirdRealityPlugCluster.cluster_id,
        endpoint_id=1,
        min_value=0,
        max_value=65535,
        mode="box",
        unit=UnitOfTime.SECONDS,
        device_class=NumberDeviceClass.DURATION,
        translation_key="turn_on_delay_left",
        fallback_name="Turn on delay left",
    )
    .number(
        attribute_name=ThirdRealityPlugCluster.AttributeDefs.off_to_on_delay.name,
        cluster_id=ThirdRealityPlugCluster.cluster_id,
        endpoint_id=2,
        min_value=0,
        max_value=65535,
        mode="box",
        unit=UnitOfTime.SECONDS,
        device_class=NumberDeviceClass.DURATION,
        translation_key="turn_on_delay_right",
        fallback_name="Turn on delay right",
    )
    .add_to_registry()
)