games.py

import asyncio
import json
import random

import requests
from userbot import catub
from userbot.core.managers import edit_delete, edit_or_reply
from userbot.helpers.utils import reply_id

plugin_category = "fun"

game_code = ["ttt", "ttf", "ex", "cf", "rps", "rpsls", "rr", "c", "pc"]
game_name = [
    "Tic-Tac-Toe",
    "Tic-Tac-Four",
    "Elephant XO",
    "Connect Four",
    "Rock-Paper-Scissors",
    "Rock-Paper-Scissors-Lizard-Spock",
    "Russian Roulette",
    "Checkers",
    "Pool Checkers",
]

game = dict(zip(game_code, game_name))
category = ["classic", "kids", "party", "hot", "mixed"]


async def get_task(mode, choice):
    url = "https://psycatgames.com/api/tod-v2/"
    data = {
        "id": "truth-or-dare",
        "language": "en",
        "category": category[choice],
        "type": mode,
    }
    headers = {
        "referer": "https://psycatgames.com/app/truth-or-dare/?utm_campaign=tod_website&utm_source=tod_en&utm_medium=website"
    }
    response = requests.post(url, headers=headers, data=json.dumps(data))
    result = response.json()["results"]
    return random.choice(result)


@catub.cat_cmd(
    pattern="(task|truth|dare)(?: |$)([1-5]+)?$",
    command=("task", plugin_category),
    info={
        "header": "Get a random truth or dare task.",
        "description": "if no input is given then task will be from classic or kids category. if you want specific category then give input.",
        "note": "U need to give input as 1 for classic , 2 for kids , 3 for party , 4 for hot and 5 for mixed. if you want to give multiple catgories then use numbers without space like 12",
        "example": [
            "{tr}task",
            "{tr}task 2",
            "{tr}task 123",
        ],
    },
)
async def truth_dare_task(event):
    "Get a random task either truth or dare."
    taskmode = event.pattern_match.group(1)
    if taskmode == "task":
        catevent = await edit_or_reply(event, "Getting a random task for you.....")
        taskmode = random.choice(["truth", "dare"])
    else:
        catevent = await edit_or_reply(
            event, f"Getting a random {taskmode} task for you....."
        )
    category = event.pattern_match.group(2)
    category = int(random.choice(category)) if category else random.choice([1, 2])
    try:
        task = await get_task(taskmode, category)
        if taskmode == "truth":
            await catevent.edit(f"**The truth task for you is**\n`{task}`")
        else:
            await catevent.edit(f"**The dare task for you is**\n`{task}`")
    except Exception as e:
        await edit_delete(catevent, f"**Error while getting task**\n`{e}`", 7)


@catub.cat_cmd(
    command=("truth", plugin_category),
    info={
        "header": "Get a random truth task.",
        "description": "if no input is given then task will be from classic or kids category. if you want specific category then give input.",
        "note": "U need to give input as 1 for classic , 2 for kids , 3 for party , 4 for hot and 5 for mixed. if you want to give multiple catgories then use numbers without space like 12",
        "example": [
            "{tr}truth",
            "{tr}truth 2",
            "{tr}truth 123",
        ],
    },
)
async def truth_task(event):
    "Get a random truth task."
    # just to show in help menu as seperate


@catub.cat_cmd(
    command=("dare", plugin_category),
    info={
        "header": "Get a random dare task.",
        "description": "if no input is given then task will be from classic or kids category. if you want specific category then give input.",
        "note": "U need to give input as 1 for classic , 2 for kids , 3 for party , 4 for hot and 5 for mixed. if you want to give multiple catgories then use numbers without space like 12",
        "example": [
            "{tr}dare",
            "{tr}dare 2",
            "{tr}dare 123",
        ],
    },
)
async def dare_task(event):
    "Get a random dare task."
    # just to show in help menu as seperate


@catub.cat_cmd(
    pattern="game(?:\s|$)([\s\S]*)",
    command=("game", plugin_category),
    info={
        "header": "Play inline games",
        "description": "Start an inline game by inlinegamebot",
        "Game code & Name": game,
        "usage": "{tr}game <game code>",
        "examples": "{tr}game ttt ",
    },
)
async def igame(event):
    "Fun game by inline"
    reply_to_id = await reply_id(event)
    input_str = event.pattern_match.group(1)
    game_list = "".join(
        f"**{i}.** `{item}` :- __{game[item]}__\n"
        for i, item in enumerate(game, start=1)
    )
    if not input_str:
        await edit_delete(
            event, f"**Available Game Codes & Names :-**\n\n{game_list}", time=60
        )
        return
    if input_str not in game_code:
        catevent = await edit_or_reply(event, "`Give me a correct game code...`")
        await asyncio.sleep(1)
        await edit_delete(
            catevent, f"**Available Game Codes & Names :-**\n\n{game_list}", time=60
        )
    else:
        await edit_or_reply(
            event,
            f"**Game code `{input_str}` is selected for game:-** __{game[input_str]}__",
        )
        await asyncio.sleep(1)
        bot = "@inlinegamesbot"
        results = await event.client.inline_query(bot, input_str)
        await results[game_code.index(input_str)].click(
            event.chat_id, reply_to=reply_to_id
        )
        await event.delete()