hcpy/hc2mqtt

#!/usr/bin/env python3
# Contact Bosh-Siemens Home Connect devices
# and connect their messages to the mqtt server
import json
import ssl
import sys
import time
from threading import Thread

import click
import click_config_file
import paho.mqtt.client as mqtt

from HCDevice import HCDevice
from HCSocket import HCSocket, now


@click.command()
@click.option("-d", "--devices_file", default="config/devices.json")
@click.option("-h", "--mqtt_host", default="localhost")
@click.option("-p", "--mqtt_prefix", default="homeconnect/")
@click.option("--mqtt_port", default=1883, type=int)
@click.option("--mqtt_username")
@click.option("--mqtt_password")
@click.option("--mqtt_ssl", is_flag=True)
@click.option("--mqtt_cafile")
@click.option("--mqtt_certfile")
@click.option("--mqtt_keyfile")
@click.option("--mqtt_clientname", default="hcpy")
@click_config_file.configuration_option()
def hc2mqtt(
    devices_file: str,
    mqtt_host: str,
    mqtt_prefix: str,
    mqtt_port: int,
    mqtt_username: str,
    mqtt_password: str,
    mqtt_ssl: bool,
    mqtt_cafile: str,
    mqtt_certfile: str,
    mqtt_keyfile: str,
    mqtt_clientname: str,
):
    click.echo(
        f"Hello {devices_file=} {mqtt_host=} {mqtt_prefix=} "
        f"{mqtt_port=} {mqtt_username=} {mqtt_password=} "
        f"{mqtt_ssl=} {mqtt_cafile=} {mqtt_certfile=} {mqtt_keyfile=} {mqtt_clientname=}"
    )

    with open(devices_file, "r") as f:
        devices = json.load(f)

    client = mqtt.Client(mqtt_clientname)

    if mqtt_username and mqtt_password:
        client.username_pw_set(mqtt_username, mqtt_password)

    if mqtt_ssl:
        if mqtt_cafile and mqtt_certfile and mqtt_keyfile:
            client.tls_set(
                ca_certs=mqtt_cafile,
                certfile=mqtt_certfile,
                keyfile=mqtt_keyfile,
                cert_reqs=ssl.CERT_REQUIRED,
            )
        else:
            client.tls_set(cert_reqs=ssl.CERT_NONE)

    client.connect(host=mqtt_host, port=mqtt_port, keepalive=70)

    for device in devices:
        mqtt_topic = mqtt_prefix + device["name"]
        print(now(), f"topic: {mqtt_topic}")
        thread = Thread(target=client_connect, args=(client, device, mqtt_topic))
        thread.start()

    client.loop_forever()


# Map their value names to easier state names
topics = {
    "InternalError": "Error",
    "FatalErrorOccured": "Error",
}
global dev
dev = {}


def client_connect(client, device, mqtt_topic):
    def on_message(client, userdata, msg):
        print(msg.topic)
        mqtt_state = msg.payload.decode()
        mqtt_topic = msg.topic.split("/")
        print(now(), f"received mqtt message {mqtt_state}")

        if len(mqtt_topic) >= 2:
            device_name = mqtt_topic[-2]
            topic = mqtt_topic[-1]
        else:
            raise Exception(f"Invalid mqtt topic {msg.topic}")

        try:
            msg = json.loads(mqtt_state)
            if topic == "set" and "uid" in msg:
                dev[device_name].get("/ro/values", 1, "POST", msg)
            elif topic == "activeProgram":
                dev[device_name].get("/ro/activeProgram", 1, "POST", msg)
            else:
                raise Exception(f"Payload {msg} is not correctly formatted")
        except Exception as e:
            print("ERROR", e, file=sys.stderr)

    host = device["host"]
    device_topics = topics
    active_program = False

    for value in device["features"]:
        # If the device has the ActiveProgram feature it allows programs to be started and scheduled via /ro/activeProgram
        if "BSH.Common.Root.ActiveProgram" == device["features"][value]["name"]:
            active_program = True
        if (
            "access" in device["features"][value]
            and "read" in device["features"][value]["access"].lower()
        ):
            name = device["features"][value]["name"].split(".")
            device_topics[name[-1]] = name[-1]
            device_topics[value] = name[
                -1
            ]  # sometimes the returned key is a digit, making translation possible

    state = {}
    for topic in device_topics:
        if not topic.isdigit():  # We only want the named topics
            state[device_topics[topic]] = None

    mqtt_set_topic = mqtt_topic + "/set"
    print(now(), device["name"], f"set topic: {mqtt_set_topic}")
    client.subscribe(mqtt_set_topic)

    if active_program:
        mqtt_active_program_topic = mqtt_topic + "/activeProgram"
        print(now(), device["name"], f"program topic: {mqtt_active_program_topic}")
        client.subscribe(mqtt_active_program_topic)

    client.on_message = on_message

    while True:
        try:
            print(now(), device["name"], f"connecting to {host}")
            ws = HCSocket(host, device["key"], device.get("iv", None))
            dev[device["name"]] = HCDevice(ws, device.get("features", None), device["name"])

            # ws.debug = True
            ws.reconnect()

            while True:
                msg = dev[device["name"]].recv()
                if msg is None:
                    break
                if len(msg) > 0:
                    print(now(), device["name"], msg)

                update = False
                for topic in device_topics:
                    value = msg.get(topic, None)
                    if value is None:
                        continue

                    new_topic = device_topics[topic]
                    state[new_topic] = value
                    update = True

                if not update:
                    continue

                msg = json.dumps(state)
                print(now(), device["name"], f"publish to {mqtt_topic} with {msg}")
                client.publish(mqtt_topic + "/state", msg)

        except Exception as e:
            print("ERROR", host, e, file=sys.stderr)

        time.sleep(5)


if __name__ == "__main__":
    hc2mqtt()