"""Frient Smoke Alarm."""

from zigpy.quirks.v2 import QuirkBuilder
from zigpy.quirks.v2.homeassistant import EntityType
from zigpy.zcl.clusters.general import BinaryInput
from zigpy.zcl.clusters.security import IasWd, IasZone

from zhaquirks.develco import DevelcoIasZone, DevelcoPowerConfiguration
from zhaquirks.quirk_ids import SIREN_BASIC

(
    QuirkBuilder("frient A/S", "SMSZB-120")
    .applies_to("Develco Products A/S", "SMSZB-120")
    .replaces(DevelcoIasZone, endpoint_id=35)
    .replaces(DevelcoPowerConfiguration, endpoint_id=35)
    # The device only has basic siren features, so hint that to ZHA
    .exposes_feature(SIREN_BASIC)
    # Hide the default binary input sensor
    .prevent_default_entity_creation(
        endpoint_id=35,
        cluster_id=BinaryInput.cluster_id,
    )
    # The IAS Zone sensor should be primary
    .change_entity_metadata(
        endpoint_id=35,
        cluster_id=IasZone.cluster_id,
        new_primary=True,
    )
    # Not the siren
    .change_entity_metadata(
        endpoint_id=35,
        cluster_id=IasWd.cluster_id,
        new_primary=False,
        new_entity_category=EntityType.CONFIG,
    )
    .add_to_registry()
)