from telethon import events

from config import client
@events.register(events.NewMessage(outgoing=True , pattern=r"\.spam"))
async def spam(event):
    chat = await event.get_chat()
    if(event.is_reply):
        spamcount = event.message.message.split(" " , maxsplit=2)[1].strip()
        try:
            spamMsg = event.message.message.split(" " , maxsplit=2)[2].strip()
        except:
            spamMsg = NULL

        if spamMsg == NULL:
            msgTospam = await event.get_reply_message()
            for i in range(int(spamcount)):
                await client.send_message(chat , msgTospam)
        else:
            for i in range(int(spamcount)):
                await client.send_message(chat , spamMsg)
    else:
        # print(event.message.message.split(" " , maxsplit=2))
        try:
            spamcount = event.message.message.split(" " , maxsplit=2)[1].strip()
            spamMsg = event.message.message.split(" " , maxsplit=2)[2].strip()
            
            for i in range(int(spamcount)):
                # print(i)
                await client.send_message(chat , spamMsg)
            
        except:
            await client.edit_message(event.message , "Invalid Format")