#             █ █ ▀ █▄▀ ▄▀█ █▀█ ▀
#             █▀█ █ █ █ █▀█ █▀▄ █
#              © Copyright 2022
#           https://t.me/hikariatama
#
# 🔒      Licensed under the GNU AGPLv3
# 🌐 https://www.gnu.org/licenses/agpl-3.0.html

# meta pic: https://static.dan.tatar/youdbettersleep_icon.png
# meta banner: https://mods.hikariatama.ru/badges/youdbettersleep.jpg
# meta developer: @hikarimods
# scope: hikka_only
# scope: hikka_min 1.3.0

import re
import time

from telethon.tl.types import Message

from .. import loader, utils


def s2time(temp_time: str) -> int:
    seconds, minutes, hours = 0, 0, 0

    try:
        seconds = int(str(re.search("([0-9]+)s", temp_time).group(1)))
    except Exception:
        pass

    try:
        minutes = int(str(re.search("([0-9]+)m", temp_time).group(1))) * 60
    except Exception:
        pass

    try:
        hours = int(str(re.search("([0-9]+)h", temp_time).group(1))) * 60 * 60
    except Exception:
        pass

    return round(seconds + minutes + hours)


@loader.tds
class YouDBetterSleepMod(loader.Module):
    """Restricts user from sending messages while sleeping"""

    strings = {
        "name": "YouDBetterSleep",
        "no_time": "🚫 <b>You can't sleep forever, specify <time> argument</b>",
        "awake": "🥱 <b>Good morning</b>",
        "asleep": "😴 <b>Good night. Now I can't write messages for {}</b>",
        "disabled": (
            "😴 <b>I can't write messages, because my userbot wants me to sleep</b>"
        ),
    }

    strings_ru = {
        "no_time": "👾 <b>Ты не можешь спать вечно, укажи аргумент <время></b>",
        "awake": "🥱 <b>Доброе утро</b>",
        "asleep": (
            "😴 <b>Спокойной ночи. Я не могу писать сообщения на протяжении {}</b>"
        ),
        "disabled": (
            "😴 <b>Я не могу писать сообщения, так как мой юзербот хочет, чтобы я"
            " поспал</b>"
        ),
        "_cls_doc": "Запрещает писать во время сна",
    }

    @loader.command(ru_doc="<время> - Поспать")
    async def sleepcmd(self, message: Message):
        """<time> - Sleep for time"""
        args = utils.get_args_raw(message)

        t = s2time(args)

        if not args or t == 0:
            self.set("asleep", False)
            self.get("until", 0)
            await utils.answer(message, self.strings("awake"))
        else:
            self.set("asleep", True)
            self.set("until", t + time.time())
            await utils.answer(message, self.strings("asleep").format(args))

    @loader.tag("only_messages", "no_commands")
    async def watcher(self, message: Message):
        if not self.get("asleep", False) or not self.get("until", False):
            return

        if self.get("until", 0) <= time.time():
            self.set("until", 0)
            self.set("asleep", False)
            await self.inline.bot.send_message(
                self._tg_id,
                self.strings("awake"),
                parse_mode="HTML",
            )
            return

        if message.mentioned:
            await self._client.send_read_acknowledge(
                message.peer_id,
                message,
                clear_mentions=True,
            )
            await utils.answer(message, self.strings("disabled"))

        if message.out:
            await utils.answer(message, self.strings("disabled"))