"""
Utilities for working with Telegram user information
"""
from typing import Optional, Dict
from pyrogram import Client
import logging

logger = logging.getLogger(__name__)


def get_app_client() -> Optional[Client]:
    """
    Get Pyrogram client from executor.py.
    
    Uses a single source of client to avoid conflicts.
    
    Returns:
        Pyrogram client instance, or None if import fails
    """
    try:
        from bot.modules.task_scheduler.executor import get_app_client as get_executor_client
        return get_executor_client()
    except ImportError:
        logger.warning("Failed to import get_app_client from executor")
        return None


async def get_user_info(user_id: int) -> Optional[Dict[str, Optional[str]]]:
    """
    Get user information from Telegram API.
    
    Args:
        user_id: Telegram user ID
    
    Returns:
        Dictionary with user information:
        {
            "username": str or None,
            "first_name": str or None,
            "last_name": str or None
        }
        or None in case of error
    """
    app_client = get_app_client()
    
    if not app_client:
        logger.warning(f"Pyrogram client not set, cannot get user information for {user_id}")
        return None
    
    try:
        # Get user information through Pyrogram
        user = await app_client.get_users(user_id)
        
        if user:
            return {
                "username": user.username,
                "first_name": user.first_name,
                "last_name": user.last_name
            }
        else:
            logger.warning(f"User {user_id} not found in Telegram")
            return None
    except Exception as e:
        logger.error(f"Error getting user information for {user_id}: {e}", exc_info=True)
        return None


async def get_username_by_id(user_id: int) -> Optional[str]:
    """
    Get username by user ID.
    
    Args:
        user_id: Telegram user ID
    
    Returns:
        Username or None if not found
    """
    user_info = await get_user_info(user_id)
    if user_info:
        return user_info.get("username")
    return None