import requests
from loguru import logger
from storage import Storage

s = Storage()

_TOKEN = '6767909836:AAFpsqtWeBNIBgSSi2_19rltEHOF0mrvTg0'
_URL = 'https://api.telegram.org/bot'


def _log(message):
    if message.record.get('level').name == "ERROR":
        icon = "❌ "
    elif message.record.get('level').name == "SUCCESS":
        icon = "✅ "
    else:
        icon = "ℹ️ "
    for chat_id in s.get_users():
        r = requests.post("{0}{1}/sendMessage".format(_URL, _TOKEN), {
            "chat_id": int(chat_id),
            "disable_notification": True,
            "text": icon + message
        })

        if r.status_code >= 400:
            logger.error("Failed to send message: {0} {1}".format(r.status_code, r.text))


def _filter_info_only(record):
    return record.get('level').name == "INFO" or record.get('level').name == "SUCCESS"


logger.add(_log, format="{message}", filter=_filter_info_only)
logger.add(_log, format=" {message}", filter=lambda record: record.get('level').name == "ERROR")