from __future__ import annotations

from yuxi.channels.models import MentionsInfo


def parse_mentions(message: dict, bot_user: str = "") -> MentionsInfo:
    annotations = message.get("annotations", [])
    mentioned_ids = [
        a.get("userMention", {}).get("user", {}).get("name", "") for a in annotations if a.get("type") == "USER_MENTION"
    ]
    is_at_bot = False
    for a in annotations:
        if a.get("type") != "USER_MENTION":
            continue
        um = a.get("userMention", {})
        if um.get("type") == "BOT":
            is_at_bot = True
            break
        if bot_user and um.get("user", {}).get("name", "") == bot_user:
            is_at_bot = True
            break
    return MentionsInfo(
        mentioned_user_ids=mentioned_ids,
        is_bot_mentioned=is_at_bot,
    )