# -*- coding: utf-8 -*-
import mastodon
from pubsub import pub

class StreamListener(mastodon.StreamListener):

    def __init__(self, session_name, user_id):
        self.session_name = session_name
        self.user_id = user_id
        super(StreamListener, self).__init__()

    def on_update(self, status):
        pub.sendMessage("mastodon.status_received", status=status, session_name=self.session_name)

    def on_status_update(self, status):
        pub.sendMessage("mastodon.status_updated", status=status, session_name=self.session_name)

    def on_conversation(self, conversation):
        pub.sendMessage("mastodon.conversation_received", conversation=conversation, session_name=self.session_name)

    def on_notification(self, notification):
        pub.sendMessage("mastodon.notification_received", notification=notification, session_name=self.session_name)

    def on_unknown_event(self, event, payload):
        log.error("Unknown event: {} with payload as {}".format(event, payload))