ForcePilot/backend/package/yuxi/channel/extensions/msteams/graph_members.py

from __future__ import annotations

import logging

from .graph import MSTeamsGraphClient

logger = logging.getLogger(__name__)


async def get_chat_members(
    graph_client: MSTeamsGraphClient,
    chat_id: str,
    *,
    limit: int = 50,
) -> list[dict]:
    path = f"/chats/{chat_id}/members?$top={limit}"
    try:
        data = await graph_client.fetch_json(path)
        return _parse_members(data)
    except Exception as e:
        logger.warning("Failed to get members for chat %s: %s", chat_id, e)
        return []


async def add_chat_members(
    graph_client: MSTeamsGraphClient,
    chat_id: str,
    user_ids: list[str],
    *,
    roles: list[str] | None = None,
) -> bool:
    if not user_ids:
        return False

    for user_id in user_ids:
        body = {
            "@odata.type": "#microsoft.graph.aadUserConversationMember",
            "roles": roles or ["owner"],
            "user@odata.bind": f"https://graph.microsoft.com/v1.0/users/{user_id}",
        }
        path = f"/chats/{chat_id}/members"
        try:
            await graph_client.fetch_json(path, method="POST", body=body)
        except Exception as e:
            logger.warning("Failed to add member %s to chat %s: %s", user_id, chat_id, e)
            return False

    return True


async def remove_chat_member(
    graph_client: MSTeamsGraphClient,
    chat_id: str,
    membership_id: str,
) -> bool:
    path = f"/chats/{chat_id}/members/{membership_id}"
    try:
        await graph_client.fetch_json(path, method="DELETE")
        return True
    except Exception as e:
        logger.warning("Failed to remove member %s from chat %s: %s", membership_id, chat_id, e)
        return False


async def get_team_members(
    graph_client: MSTeamsGraphClient,
    team_id: str,
    *,
    limit: int = 50,
) -> list[dict]:
    path = f"/teams/{team_id}/members?$top={limit}"
    try:
        data = await graph_client.fetch_json(path)
        return _parse_members(data)
    except Exception as e:
        logger.warning("Failed to get members for team %s: %s", team_id, e)
        return []


async def get_channel_members(
    graph_client: MSTeamsGraphClient,
    team_id: str,
    channel_id: str,
    *,
    limit: int = 50,
) -> list[dict]:
    path = f"/teams/{team_id}/channels/{channel_id}/members?$top={limit}"
    try:
        data = await graph_client.fetch_json(path)
        return _parse_members(data)
    except Exception as e:
        logger.warning("Failed to get members for channel %s in team %s: %s", channel_id, team_id, e)
        return []


def _parse_members(data: dict) -> list[dict]:
    members = data.get("value", [])
    result = []
    for m in members:
        result.append(
            {
                "id": m.get("id", ""),
                "display_name": m.get("displayName", ""),
                "user_id": m.get("userId", ""),
                "email": m.get("email", ""),
                "roles": m.get("roles", []),
                "tenant_id": m.get("tenantId", ""),
                "visible_history_start": m.get("visibleHistoryStartDateTime", ""),
            }
        )
    return result