import inspect

import keyboard

import obsws_python as obs


class Observer:
    def __init__(self):
        self._client = obs.EventClient()
        self._client.callback.register(self.on_current_program_scene_changed)
        print(f"Registered events: {self._client.callback.get()}")

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_value, exc_traceback):
        self._client.disconnect()

    @property
    def event_identifier(self):
        return inspect.stack()[1].function

    def on_current_program_scene_changed(self, data):
        """The current program scene has changed."""
        print(f"{self.event_identifier}: {data.scene_name}")


def version():
    resp = req_client.get_version()
    print(
        f"Running OBS version:{resp.obs_version} with websocket version:{resp.obs_web_socket_version}"
    )


def set_scene(scene, *args):
    req_client.set_current_program_scene(scene)


if __name__ == "__main__":
    with obs.ReqClient() as req_client:
        with Observer() as observer:
            keyboard.add_hotkey("0", version)
            keyboard.add_hotkey("1", set_scene, args=("START",))
            keyboard.add_hotkey("2", set_scene, args=("BRB",))
            keyboard.add_hotkey("3", set_scene, args=("END",))

            print("press ctrl+enter to quit")
            keyboard.wait("ctrl+enter")