from pyrogram.types import Update
from pyrogram.raw.types import UpdateBotStopped

import app

class RawUpdateHandler:
    async def raw_update_handler(self: "app.App", _, update: Update, __, ___):
        if not isinstance(update, UpdateBotStopped):
            return
        
        user_id = update.user_id
        get_user = await self.database.get_user(user_id=user_id)
        
        if not get_user:
            return
        
        elif update.stopped:
            await self.database.update_user(get_user, block=1)
            
        elif not update.stopped:
            await self.database.update_user(get_user, block=0)