import asyncio
from logging import getLogger
from pyrogram.errors import ChatWriteForbidden, MessageNotModified, FloodWait

LOGGER = getLogger(__name__)

# handler for TG function, so need write exception in every code


async def kirimPesan(msg, text: str, disable_web_page_preview=True, reply_markup=None):
    try:
        return await msg.reply(text)
    except FloodWait as e:
        LOGGER.warning(str(e))
        await asyncio.sleep(e.value)
        return await kirimPesan(text)
    except ChatWriteForbidden:
        return await msg.leave()
    except Exception as e:
        LOGGER.error(str(e))


async def editPesan(msg, text: str, disable_web_page_preview=True, reply_markup=None):
    try:
        return await msg.edit(text)
    except FloodWait as e:
        LOGGER.warning(str(e))
        await asyncio.sleep(e.value)
        return await editPesan(msg, text)
    except MessageNotModified:
        return
    except Exception as e:
        LOGGER.error(str(e))


async def hapusPesan(msg):
    try:
        return await msg.delete()
    except FloodWait as e:
        LOGGER.warning(str(e))
        await asyncio.sleep(e.value)
        return await hapusPesan(msg)
    except Exception as e:
        LOGGER.error(str(e))