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