123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310 |
- """Library to handle connection with Switchbot."""
- from __future__ import annotations
- import logging
- from collections.abc import Callable
- from functools import lru_cache
- from typing import Any, TypedDict
- from bleak.backends.device import BLEDevice
- from bleak.backends.scanner import AdvertisementData
- from .adv_parsers.blind_tilt import process_woblindtilt
- from .adv_parsers.bot import process_wohand
- from .adv_parsers.bulb import process_color_bulb
- from .adv_parsers.ceiling_light import process_woceiling
- from .adv_parsers.contact import process_wocontact
- from .adv_parsers.curtain import process_wocurtain
- from .adv_parsers.hub2 import process_wohub2
- from .adv_parsers.humidifier import process_wohumidifier
- from .adv_parsers.keypad import process_wokeypad
- from .adv_parsers.leak import process_leak
- from .adv_parsers.light_strip import process_wostrip
- from .adv_parsers.lock import process_wolock, process_wolock_pro
- from .adv_parsers.meter import process_wosensorth, process_wosensorth_c
- from .adv_parsers.motion import process_wopresence
- from .adv_parsers.plug import process_woplugmini
- from .adv_parsers.relay_switch import (
- process_worelay_switch_1,
- process_worelay_switch_1pm,
- )
- from .const import SwitchbotModel
- from .models import SwitchBotAdvertisement
- _LOGGER = logging.getLogger(__name__)
- SERVICE_DATA_ORDER = (
- "0000fd3d-0000-1000-8000-00805f9b34fb",
- "00000d00-0000-1000-8000-00805f9b34fb",
- )
- MFR_DATA_ORDER = (2409, 741, 89)
- class SwitchbotSupportedType(TypedDict):
- """Supported type of Switchbot."""
- modelName: SwitchbotModel
- modelFriendlyName: str
- func: Callable[[bytes, bytes | None], dict[str, bool | int]]
- manufacturer_id: int | None
- manufacturer_data_length: int | None
- SUPPORTED_TYPES: dict[str, SwitchbotSupportedType] = {
- "d": {
- "modelName": SwitchbotModel.CONTACT_SENSOR,
- "modelFriendlyName": "Contact Sensor",
- "func": process_wocontact,
- "manufacturer_id": 2409,
- },
- "H": {
- "modelName": SwitchbotModel.BOT,
- "modelFriendlyName": "Bot",
- "func": process_wohand,
- "manufacturer_id": 89,
- },
- "s": {
- "modelName": SwitchbotModel.MOTION_SENSOR,
- "modelFriendlyName": "Motion Sensor",
- "func": process_wopresence,
- "manufacturer_id": 2409,
- },
- "r": {
- "modelName": SwitchbotModel.LIGHT_STRIP,
- "modelFriendlyName": "Light Strip",
- "func": process_wostrip,
- "manufacturer_id": 2409,
- },
- "{": {
- "modelName": SwitchbotModel.CURTAIN,
- "modelFriendlyName": "Curtain 3",
- "func": process_wocurtain,
- "manufacturer_id": 2409,
- },
- "c": {
- "modelName": SwitchbotModel.CURTAIN,
- "modelFriendlyName": "Curtain",
- "func": process_wocurtain,
- "manufacturer_id": 2409,
- },
- "w": {
- "modelName": SwitchbotModel.IO_METER,
- "modelFriendlyName": "Indoor/Outdoor Meter",
- "func": process_wosensorth,
- "manufacturer_id": 2409,
- },
- "i": {
- "modelName": SwitchbotModel.METER,
- "modelFriendlyName": "Meter Plus",
- "func": process_wosensorth,
- "manufacturer_id": 2409,
- },
- "T": {
- "modelName": SwitchbotModel.METER,
- "modelFriendlyName": "Meter",
- "func": process_wosensorth,
- "manufacturer_id": 2409,
- },
- "4": {
- "modelName": SwitchbotModel.METER_PRO,
- "modelFriendlyName": "Meter",
- "func": process_wosensorth,
- "manufacturer_id": 2409,
- },
- "5": {
- "modelName": SwitchbotModel.METER_PRO_C,
- "modelFriendlyName": "Meter",
- "func": process_wosensorth_c,
- "manufacturer_id": 2409,
- },
- "v": {
- "modelName": SwitchbotModel.HUB2,
- "modelFriendlyName": "Hub 2",
- "func": process_wohub2,
- "manufacturer_id": 2409,
- },
- "g": {
- "modelName": SwitchbotModel.PLUG_MINI,
- "modelFriendlyName": "Plug Mini",
- "func": process_woplugmini,
- "manufacturer_id": 2409,
- },
- "j": {
- "modelName": SwitchbotModel.PLUG_MINI,
- "modelFriendlyName": "Plug Mini (JP)",
- "func": process_woplugmini,
- "manufacturer_id": 2409,
- },
- "u": {
- "modelName": SwitchbotModel.COLOR_BULB,
- "modelFriendlyName": "Color Bulb",
- "func": process_color_bulb,
- "manufacturer_id": 2409,
- },
- "q": {
- "modelName": SwitchbotModel.CEILING_LIGHT,
- "modelFriendlyName": "Ceiling Light",
- "func": process_woceiling,
- "manufacturer_id": 2409,
- },
- "n": {
- "modelName": SwitchbotModel.CEILING_LIGHT,
- "modelFriendlyName": "Ceiling Light Pro",
- "func": process_woceiling,
- "manufacturer_id": 2409,
- },
- "e": {
- "modelName": SwitchbotModel.HUMIDIFIER,
- "modelFriendlyName": "Humidifier",
- "func": process_wohumidifier,
- "manufacturer_id": 741,
- "manufacturer_data_length": 6,
- },
- "o": {
- "modelName": SwitchbotModel.LOCK,
- "modelFriendlyName": "Lock",
- "func": process_wolock,
- "manufacturer_id": 2409,
- },
- "$": {
- "modelName": SwitchbotModel.LOCK_PRO,
- "modelFriendlyName": "Lock Pro",
- "func": process_wolock_pro,
- "manufacturer_id": 2409,
- },
- "x": {
- "modelName": SwitchbotModel.BLIND_TILT,
- "modelFriendlyName": "Blind Tilt",
- "func": process_woblindtilt,
- "manufacturer_id": 2409,
- },
- "3": {
- "modelName": SwitchbotModel.LEAK,
- "modelFriendlyName": "Leak Detector",
- "func": process_leak,
- "manufacturer_id": 2409,
- },
- "y": {
- "modelName": SwitchbotModel.KEYPAD,
- "modelFriendlyName": "Keypad",
- "func": process_wokeypad,
- "manufacturer_id": 2409,
- },
- "<": {
- "modelName": SwitchbotModel.RELAY_SWITCH_1PM,
- "modelFriendlyName": "Relay Switch 1PM",
- "func": process_worelay_switch_1pm,
- "manufacturer_id": 2409,
- },
- ";": {
- "modelName": SwitchbotModel.RELAY_SWITCH_1,
- "modelFriendlyName": "Relay Switch 1",
- "func": process_worelay_switch_1,
- "manufacturer_id": 2409,
- },
- }
- _SWITCHBOT_MODEL_TO_CHAR = {
- model_data["modelName"]: model_chr
- for model_chr, model_data in SUPPORTED_TYPES.items()
- }
- MODELS_BY_MANUFACTURER_DATA: dict[int, list[tuple[str, SwitchbotSupportedType]]] = {
- mfr_id: [] for mfr_id in MFR_DATA_ORDER
- }
- for model_chr, model in SUPPORTED_TYPES.items():
- if "manufacturer_id" in model:
- mfr_id = model["manufacturer_id"]
- MODELS_BY_MANUFACTURER_DATA[mfr_id].append((model_chr, model))
- def parse_advertisement_data(
- device: BLEDevice,
- advertisement_data: AdvertisementData,
- model: SwitchbotModel | None = None,
- ) -> SwitchBotAdvertisement | None:
- """Parse advertisement data."""
- service_data = advertisement_data.service_data
- _service_data = None
- for uuid in SERVICE_DATA_ORDER:
- if uuid in service_data:
- _service_data = service_data[uuid]
- break
- _mfr_data = None
- _mfr_id = None
- for mfr_id in MFR_DATA_ORDER:
- if mfr_id in advertisement_data.manufacturer_data:
- _mfr_id = mfr_id
- _mfr_data = advertisement_data.manufacturer_data[mfr_id]
- break
- if _mfr_data is None and _service_data is None:
- return None
- try:
- data = _parse_data(
- _service_data,
- _mfr_data,
- _mfr_id,
- model,
- )
- except Exception as err: # pylint: disable=broad-except
- _LOGGER.exception(
- "Failed to parse advertisement data: %s: %s", advertisement_data, err
- )
- return None
- if not data:
- return None
- return SwitchBotAdvertisement(
- device.address, data, device, advertisement_data.rssi, bool(_service_data)
- )
- @lru_cache(maxsize=128)
- def _parse_data(
- _service_data: bytes | None,
- _mfr_data: bytes | None,
- _mfr_id: int | None = None,
- _switchbot_model: SwitchbotModel | None = None,
- ) -> dict[str, Any] | None:
- """Parse advertisement data."""
- _model = chr(_service_data[0] & 0b01111111) if _service_data else None
- if _switchbot_model and _switchbot_model in _SWITCHBOT_MODEL_TO_CHAR:
- _model = _SWITCHBOT_MODEL_TO_CHAR[_switchbot_model]
- if not _model and _mfr_id and _mfr_id in MODELS_BY_MANUFACTURER_DATA:
- for model_chr, model_data in MODELS_BY_MANUFACTURER_DATA[_mfr_id]:
- if model_data.get("manufacturer_data_length") == len(_mfr_data):
- _model = model_chr
- break
- if not _model:
- return None
- _isEncrypted = bool(_service_data[0] & 0b10000000) if _service_data else False
- data = {
- "rawAdvData": _service_data,
- "data": {},
- "model": _model,
- "isEncrypted": _isEncrypted,
- }
- type_data = SUPPORTED_TYPES.get(_model)
- if type_data:
- model_data = type_data["func"](_service_data, _mfr_data)
- if model_data:
- data.update(
- {
- "modelFriendlyName": type_data["modelFriendlyName"],
- "modelName": type_data["modelName"],
- "data": model_data,
- }
- )
- return data
|