homelab/userbot/modules/amongus.py

# From CatUB
import asyncio
import re
from io import BytesIO
from random import choice, randint
from textwrap import wrap

import requests
from PIL import Image, ImageDraw, ImageFont
from pyrogram import Client, filters
from pyrogram.types import Message
from utils.misc import modules_help, prefix
from utils.scripts import edit_or_reply, reply_check


async def amongus_gen(text: str, clr: int) -> str:
    url = 'https://github.com/TgCatUB/CatUserbot-Resources/raw/master/Resources/Amongus/'
    font = ImageFont.truetype(
        BytesIO(
            requests.get(
                'https://github.com/TgCatUB/CatUserbot-Resources/raw/master/Resources/fonts/bold.ttf', timeout=10
            ).content
        ),
        60,
    )
    imposter = Image.open(BytesIO(requests.get(f'{url}{clr}.png', timeout=10).content))
    text_ = '\n'.join('\n'.join(wrap(part, 30)) for part in text.split('\n'))
    bbox = ImageDraw.Draw(Image.new('RGB', (1, 1))).multiline_textbbox((0, 0), text_, font, stroke_width=2)
    w, h = bbox[2], bbox[3]
    text = Image.new('RGBA', (w + 30, h + 30))
    ImageDraw.Draw(text).multiline_text((15, 15), text_, '#FFF', font, stroke_width=2, stroke_fill='#000')
    w = imposter.width + text.width + 10
    h = max(imposter.height, text.height)
    image = Image.new('RGBA', (w, h))
    image.paste(imposter, (0, h - imposter.height), imposter)
    image.paste(text, (w - text.width, 0), text)
    image.thumbnail((512, 512))
    output = BytesIO()
    output.name = 'imposter.webp'
    image.save(output, 'WebP')
    output.seek(0)
    return output


async def get_imposter_img(text: str) -> BytesIO:
    background = requests.get(
        f'https://github.com/TgCatUB/CatUserbot-Resources/raw/master/Resources/imposter/impostor{randint(1, 22)}.png',  # noqa: S311
        timeout=10,
    ).content
    font = requests.get(
        'https://github.com/TgCatUB/CatUserbot-Resources/raw/master/Resources/fonts/roboto_regular.ttf',
        timeout=10,
    ).content
    font = BytesIO(font)
    font = ImageFont.truetype(font, 30)
    image = Image.new('RGBA', (1, 1), (0, 0, 0, 0))
    draw = ImageDraw.Draw(image)
    bbox = ImageDraw.Draw(Image.new('RGB', (1, 1))).multiline_textbbox((0, 0), text, font, stroke_width=2)
    w, h = bbox[2], bbox[3]
    image = Image.open(BytesIO(background))
    x, y = image.size
    draw = ImageDraw.Draw(image)
    draw.multiline_text(((x - w) // 2, (y - h) // 2), text=text, font=font, fill='white', align='center')
    output = BytesIO()
    output.name = 'impostor.png'
    image.save(output, 'PNG')
    output.seek(0)
    return output


@Client.on_message(filters.command('amongus', prefix) & filters.me)
async def amongus_cmd(client: Client, message: Message):
    text = ' '.join(message.command[1:]) if len(message.command) > 1 else ''
    reply = message.reply_to_message
    await message.edit('tun tun tun...')
    if not text and reply:
        text = reply.text or reply.caption or ''

    clr = re.findall(r'-c\d+', text)
    try:
        clr = clr[0]
        clr = clr.replace('-c', '')
        text = text.replace(f'-c{clr}', '')
        clr = int(clr)
        if clr > 12 or clr < 1:
            clr = randint(1, 12)  # noqa: S311
    except IndexError:
        clr = randint(1, 12)  # noqa: S311

    if not text:
        if not reply:
            text = f'{message.from_user.first_name} Was a traitor!'
        else:
            text = f'{reply.from_user.first_name} Was a traitor!'

    imposter_file = await amongus_gen(text, clr)
    await message.delete()
    await client.send_sticker(
        message.chat.id,
        imposter_file,
        reply_to_message_id=reply_check(message),
    )


@Client.on_message(filters.command('imposter', prefix) & filters.me)
async def imposter_cmd(client: Client, message: Message):
    remain = randint(1, 2)  # noqa: S311
    imps = ["wasn't the impostor", 'was the impostor']

    if message.reply_to_message:
        user = message.reply_to_message.from_user
        text = f'{user.first_name} {choice(imps)}.'  # noqa: S311
    else:
        args = message.text.split()[1:]
        text = ' '.join(args) if args else f'{message.from_user.first_name} {choice(imps)}.'  # noqa: S311

    text += f'\n{remain} impostor(s) remain.'
    imposter_file = await get_imposter_img(text)
    await message.delete()
    await client.send_photo(
        message.chat.id,
        imposter_file,
        reply_to_message_id=reply_check(message),
    )


@Client.on_message(filters.command(['imp', 'impn'], prefix) & filters.me)
async def imp_animation(client: Client, message: Message):
    name = ' '.join(message.command[1:]) if len(message.command) > 1 else ''
    if not name:
        reply = message.reply_to_message
        name = reply.from_user.first_name if reply else message.from_user.first_name
    cmd = message.command[0].lower()

    text1 = await edit_or_reply(message, 'Uhmm... Something is wrong here!!')
    await asyncio.sleep(2)
    await text1.delete()

    stcr1 = await client.send_sticker(message.chat.id, 'CAADAQADRwADnjOcH98isYD5RJTwAg')
    text2 = await message.reply(f'<b>{message.from_user.first_name}:</b> I have to call discussion')
    await asyncio.sleep(3)
    await stcr1.delete()
    await text2.delete()

    stcr2 = await client.send_sticker(message.chat.id, 'CAADAQADRgADnjOcH9odHIXtfgmvAg')
    text3 = await message.reply(f'<b>{message.from_user.first_name}:</b> We have to eject the imposter or will lose')
    await asyncio.sleep(3)
    await stcr2.delete()
    await text3.delete()

    stcr3 = await client.send_sticker(message.chat.id, 'CAADAQADOwADnjOcH77v3Ap51R7gAg')
    text4 = await message.reply('<b>Others:</b> Where???')
    await asyncio.sleep(2)
    await text4.edit('<b>Others:</b> Who??')
    await asyncio.sleep(2)
    await text4.edit(f'<b>{message.from_user.first_name}:</b> Its {name}, I saw {name} using vent')
    await asyncio.sleep(3)
    await text4.edit(f'<b>Others:</b> Okay.. Vote {name}')
    await asyncio.sleep(2)
    await stcr3.delete()
    await text4.delete()

    stcr4 = await client.send_sticker(message.chat.id, 'CAADAQADLwADnjOcH-wxu-ehy6NRAg')
    event = await message.reply(f'{name} is ejected.......')

    # Ejection animation
    for _ in range(9):
        await asyncio.sleep(0.5)
        curr_pos = _ + 1
        spaces_before = 'ㅤ' * curr_pos
        await event.edit(f'{spaces_before}ඞ{"ㅤ" * (9 - curr_pos)}')

    await asyncio.sleep(0.5)
    await event.edit('ㅤㅤㅤㅤㅤㅤㅤㅤㅤ')
    await asyncio.sleep(0.2)
    await stcr4.delete()

    if cmd == 'imp':
        text = f". 　　　。　　　　•　 　ﾟ　　。 　　.\n .　　　 　　.　　　　　。　　 。　. 　\n\n  . 　　 。   　     ඞ         。 . 　　 • 　　　　•\n\n  ﾟ{name} was an Imposter.      。　. 　 　       。　.                                        。　. \n                                   　.          。　  　. \n　'         0 Impostor remains    　 。　.  　　.                。　.        。 　     .          。 　            .               .         .    ,      。\n　　ﾟ　　　.　　.    ,　 　。　 　. 　 .     。"
        sticker_id = 'CAADAQADLQADnjOcH39IqwyR6Q_0Ag'
    else:
        text = f". 　　　。　　　　•　 　ﾟ　　。 　　.\n .　　　 　　.　　　　　。　　 。　. 　\n\n  . 　　 。   　     ඞ         。 . 　　 • 　　　　•\n\n  ﾟ{name} was not an Imposter.      。　. 　 　       。　.                                        。　. \n                                   　.          。　  　. \n　'         1 Impostor remains    　 。　.  　　.                。　.        。 　     .          。 　            .               .         .    ,      。\n　　ﾟ　　　.　　.    ,　 　。　 　. 　 .     。"
        sticker_id = 'CAADAQADQAADnjOcH-WOkB8DEctJAg'

    await event.edit(text)
    await asyncio.sleep(4)
    await event.delete()
    await client.send_sticker(message.chat.id, sticker_id)


modules_help['amongus'] = {
    'amongus': 'Create Among Us themed sticker [text/reply] [-c1 to -c12 for colors]',
    'imposter': 'Create Among Us imposter image [username/reply]',
    'imp': 'Create Among Us ejection animation (imposter)',
    'impn': 'Create Among Us ejection animation (not imposter)',
}