# Импорт сторонних модулей
from pyrogram.types import Message
from pyrogram.enums import MessageEntityType

from . import client


async def get_telegram_id(username: str) -> int:
    """Возвращает Telegram ID"""
    telegram_user = await client.get_users(username)

    return telegram_user.id


async def get_player_data(message: Message, idx: int) -> tuple():
    """..."""
    if message.entities[idx].type == MessageEntityType.TEXT_LINK:
        player_url = message.entities[idx].url
        player_telegram_id = await get_telegram_id(player_url.replace('http://t.me/', ''))
    elif message.entities[idx].type == MessageEntityType.TEXT_MENTION:
        player_url = None
        player_telegram_id = message.entities[idx].user.id
    else:
        player_url = None
        player_telegram_id = None

    return (player_url, player_telegram_id)