# -*- coding: utf-8 -*-

# Module author: @ftgmodulesbyfl1yd, @GovnoCodules

from asyncio import sleep
from random import randint

from telethon import functions

from .. import loader, utils


@loader.tds
class FakeMod(loader.Module):
    """Imitates your actions"""

    strings = {"name": "Fake Actions"}

    async def typecmd(self, message):
        """Imitates typing"""
        activity_time = utils.get_args(message)
        await message.delete()
        if activity_time:
            try:
                async with message.client.action(message.chat_id, "typing"):
                    await sleep(int(activity_time[0]))
            except BaseException:
                return
        else:
            try:
                async with message.client.action(message.chat_id, "typing"):
                    await sleep(randint(30, 60))
            except BaseException:
                return

    async def voicecmd(self, message):
        """Imitates sending voices"""
        activity_time = utils.get_args(message)
        await message.delete()
        if activity_time:
            try:
                async with message.client.action(message.chat_id, "voice"):
                    await sleep(int(activity_time[0]))
            except BaseException:
                return
        else:
            try:
                async with message.client.action(message.chat_id, "voice"):
                    await sleep(randint(30, 60))
            except BaseException:
                return

    async def gamecmd(self, message):
        """Imitates your game activity"""
        activity_time = utils.get_args(message)
        await message.delete()
        if activity_time:
            try:
                async with message.client.action(message.chat_id, "game"):
                    await sleep(int(activity_time[0]))
            except BaseException:
                return
        else:
            try:
                async with message.client.action(message.chat_id, "game"):
                    await sleep(randint(30, 60))
            except BaseException:
                return

    async def videocmd(self, message):
        """Imitates sending video"""
        activity_time = utils.get_args(message)
        await message.delete()
        if activity_time:
            try:
                async with message.client.action(message.chat_id, "video"):
                    await sleep(int(activity_time[0]))
            except BaseException:
                return
        else:
            try:
                async with message.client.action(message.chat_id, "video"):
                    await sleep(randint(30, 60))
            except BaseException:
                return

    async def photocmd(self, message):
        """Imitates sending photo"""
        activity_time = utils.get_args(message)
        await message.delete()
        if activity_time:
            try:
                async with message.client.action(message.chat_id, "photo"):
                    await sleep(int(activity_time[0]))
            except BaseException:
                return
        else:
            try:
                async with message.client.action(message.chat_id, "photo"):
                    await sleep(randint(30, 60))
            except BaseException:
                return

    async def documentcmd(self, message):
        """Imitates sending document"""
        activity_time = utils.get_args(message)
        await message.delete()
        if activity_time:
            try:
                async with message.client.action(message.chat_id, "document"):
                    await sleep(int(activity_time[0]))
            except BaseException:
                return
        else:
            try:
                async with message.client.action(message.chat_id, "document"):
                    await sleep(randint(30, 60))
            except BaseException:
                return

    async def locationcmd(self, message):
        """Imitates sending location"""
        activity_time = utils.get_args(message)
        await message.delete()
        if activity_time:
            try:
                async with message.client.action(message.chat_id, "location"):
                    await sleep(int(activity_time[0]))
            except BaseException:
                return
        else:
            try:
                async with message.client.action(message.chat_id, "location"):
                    await sleep(randint(30, 60))
            except BaseException:
                return

    async def recordvideocmd(self, message):
        """Imitates recording video"""
        activity_time = utils.get_args(message)
        await message.delete()
        if activity_time:
            try:
                async with message.client.action(message.chat_id, "record-video"):
                    await sleep(int(activity_time[0]))
            except BaseException:
                return
        else:
            try:
                async with message.client.action(message.chat_id, "record-video"):
                    await sleep(randint(30, 60))
            except BaseException:
                return

    async def recordvoicecmd(self, message):
        """Imitates recording voice"""
        activity_time = utils.get_args(message)
        await message.delete()
        if activity_time:
            try:
                async with message.client.action(message.chat_id, "record-audio"):
                    await sleep(int(activity_time[0]))
            except BaseException:
                return
        else:
            try:
                async with message.client.action(message.chat_id, "record-audio"):
                    await sleep(randint(30, 60))
            except BaseException:
                return

    async def recordroundcmd(self, message):
        """Imitates recording round video"""
        activity_time = utils.get_args(message)
        await message.delete()
        if activity_time:
            try:
                async with message.client.action(message.chat_id, "record-round"):
                    await sleep(int(activity_time[0]))
            except BaseException:
                return
        else:
            try:
                async with message.client.action(message.chat_id, "record-round"):
                    await sleep(randint(30, 60))
            except BaseException:
                return

    async def scrncmd(self, message):
        """Screenshot notification (Only PM)"""
        a = 1
        r = utils.get_args(message)
        if r and r[0].isdigit():
            a = int(r[0])
        for _ in range(a):
            await message.client(
                functions.messages.SendScreenshotNotificationRequest(
                    peer=message.to_id, reply_to_msg_id=message.id
                )
            )
        await message.delete()