from __future__ import annotations

import logging

from .graph import MSTeamsGraphClient

logger = logging.getLogger(__name__)


async def search_users(
    graph_client: MSTeamsGraphClient,
    query: str,
    *,
    limit: int = 10,
) -> list[dict]:
    path = f'/users?$search="displayName:{query}"&$top={limit}&$select=id,displayName,mail,userPrincipalName'
    try:
        data = await graph_client.fetch_json(path)
        return _parse_users(data)
    except Exception as e:
        logger.warning("User search failed for query '%s': %s", query, e)
        return []


async def get_user(
    graph_client: MSTeamsGraphClient,
    user_id: str,
) -> dict | None:
    path = f"/users/{user_id}?$select=id,displayName,mail,userPrincipalName"
    try:
        data = await graph_client.fetch_json(path)
        return _parse_single_user(data)
    except Exception as e:
        logger.warning("Failed to get user %s: %s", user_id, e)
        return None


async def get_aad_object_id(
    graph_client: MSTeamsGraphClient,
    identifier: str,
) -> str | None:
    users = await search_users(graph_client, identifier)
    for user in users:
        if (
            user.get("displayName", "").lower() == identifier.lower()
            or user.get("mail", "").lower() == identifier.lower()
        ):
            return user.get("id")
    return None


def _parse_single_user(data: dict) -> dict:
    return {
        "id": data.get("id", ""),
        "display_name": data.get("displayName", ""),
        "mail": data.get("mail", ""),
        "user_principal_name": data.get("userPrincipalName", ""),
    }


def _parse_users(data: dict) -> list[dict]:
    users = data.get("value", [])
    return [_parse_single_user(u) for u in users]