from __future__ import annotations

from typing import Any

from .rooms_config import get_room_config


class GroupMentionStrategy:
    def __init__(self, config: dict[str, Any]):
        self._config = config

    def should_require_mention(self, room_id: str) -> bool:
        room_cfg = get_room_config(room_id, self._config)
        return room_cfg.get("requireMention", self._config.get("requireMention", True))

    def is_at_mentioned(self, user_id: str, room_id: str, body: str, formatted_body: str) -> bool:
        return user_id in formatted_body or user_id in body

    def evaluate_room_policy(self, room_id: str, sender_id: str, body: str, formatted_body: str) -> bool:
        if not self.should_require_mention(room_id):
            return True

        if self.is_at_mentioned(self._config.get("user_id", ""), room_id, body, formatted_body):
            return True

        return False

    def get_room_mention_policy(self, room_id: str) -> dict[str, Any]:
        room_cfg = get_room_config(room_id, self._config)
        return {
            "require_mention": room_cfg.get("requireMention", self._config.get("requireMention", True)),
            "allowed_roles": room_cfg.get("allowedRoles", []),
            "mention_exempt_users": room_cfg.get("mentionExemptUsers", []),
        }


def build_room_mention_config(
    room_id: str,
    require_mention: bool = True,
    allowed_roles: list[str] | None = None,
    mention_exempt_users: list[str] | None = None,
) -> dict[str, Any]:
    result: dict[str, Any] = {"requireMention": require_mention}
    if allowed_roles:
        result["allowedRoles"] = allowed_roles
    if mention_exempt_users:
        result["mentionExemptUsers"] = mention_exempt_users
    return {room_id: result}