pySwitchbot/switchbot/adv_parsers/light_strip.py

"""Light strip adv parser."""

from __future__ import annotations

from ..helpers import _UNPACK_UINT16_BE


def process_wostrip(
    data: bytes | None, mfr_data: bytes | None
) -> dict[str, bool | int]:
    """Process WoStrip services data."""
    if mfr_data is None:
        return {}
    return {
        "sequence_number": mfr_data[6],
        "isOn": bool(mfr_data[7] & 0b10000000),
        "brightness": mfr_data[7] & 0b01111111,
        "delay": bool(mfr_data[8] & 0b10000000),
        "network_state": (mfr_data[8] & 0b01110000) >> 4,
        "color_mode": mfr_data[8] & 0b00001111,
    }


def process_light(
    data: bytes | None, mfr_data: bytes | None, cw_offset: int = 16
) -> dict[str, bool | int]:
    """Support for strip light 3 and floor lamp."""
    common_data = process_wostrip(data, mfr_data)
    if not common_data:
        return {}

    light_data = {"cw": _UNPACK_UINT16_BE(mfr_data, cw_offset)[0]}

    return common_data | light_data


def process_rgbic_light(
    data: bytes | None, mfr_data: bytes | None
) -> dict[str, bool | int]:
    """Support for RGBIC lights."""
    return process_light(data, mfr_data, cw_offset=10)