import os
from dataclasses import asdict

from loguru import logger

import streamlabsio

logger.enable('streamlabsio')


def on_streamlabs_event(event, data):
    match event:
        case 'donation':
            print(f'{data.name} donated {data.amount}! With message: {data.message}')


def on_twitch_event(event, data):
    event_message = {
        'follow': 'Received follow from {name}',
        'bits': '{name} donated {amount} bits! With message: {message}',
        'subscription': '{name} just subscribed for {months} months!',
        'raid': '{name} just raided with {raiders} raiders!',
        'host': '{name} just hosted with {viewers} viewers!',
    }

    if event in event_message:
        print(event_message[event].format(**asdict(data)))


def on_youtube_event(event, data):
    event_message = {
        'follow': 'Received follow from {name}',
        'superchat': '{name} donated {display_string} with a superchat! With comment: {comment}',
        'subscription': '{name} just subscribed for {months} months!',
    }

    if event in event_message:
        print(event_message[event].format(**asdict(data)))


def main():
    with streamlabsio.connect(token=os.getenv('STREAMLABS_TOKEN')) as client:
        client.obs.on('streamlabs', on_streamlabs_event)
        client.obs.on('twitch_account', on_twitch_event)
        client.obs.on('youtube_account', on_youtube_event)

        client.sio.sleep(30)


if __name__ == '__main__':
    main()