import streamlabsio


def on_youtube_event(event, data):
    print(f"{event}: {data.attrs()}")


def on_twitch_event(event, data):
    if event == "follow":
        print(f"Received follow from {data.name}")
    elif event == "bits":
        print(f"{data.name} donated {data.amount} bits! With message: {data.message}")
    elif event == "donation":
        print(
            f"{data.name} donated {data.formatted_amount}! With message: {data.message}"
        )


def main():
    # read token from config.toml
    with streamlabsio.connect() as client:
        client.obs.on("streamlabs", on_twitch_event)
        client.obs.on("twitch_account", on_twitch_event)
        client.obs.on("youtube_account", on_youtube_event)

        # run for 30 seconds then disconnect client from server
        client.sio.sleep(30)


if __name__ == "__main__":
    main()