import os
from enum import IntEnum
from math import log

import obsws_python as obs

LEVELTYPE = IntEnum(
    "LEVELTYPE",
    "VU POSTFADER PREFADER",
    start=0,
)

DEVICE = "Desktop Audio"


def on_input_mute_state_changed(data):
    """An input's mute state has changed."""
    if data.input_name == DEVICE:
        print(f"{DEVICE} mute toggled")


def on_input_volume_meters(data):
    """volume level update every 50 milliseconds"""

    def fget(x):
        return round(20 * log(x, 10), 1) if x > 0 else -200.0

    for device in data.inputs:
        name = device["inputName"]
        if name == DEVICE and device["inputLevelsMul"]:
            left, right = device["inputLevelsMul"]
            print(
                f"{name} [L: {fget(left[LEVELTYPE.POSTFADER])}, R: {fget(right[LEVELTYPE.POSTFADER])}]",
            )


def main():
    host = os.getenv("OBSWS_HOST", "localhost")
    port = int(os.getenv("OBSWS_PORT", 4455))
    password = os.getenv("OBSWS_PASSWORD", "")

    with obs.EventClient(
        host=host,
        port=port,
        password=password,
        subs=(obs.Subs.LOW_VOLUME | obs.Subs.INPUTVOLUMEMETERS),
    ) as client:
        client.callback.register([on_input_volume_meters, on_input_mute_state_changed])

        while _ := input("Press <Enter> to exit\n"):
            pass


if __name__ == "__main__":
    main()