import logging

from yuxi.channel.extensions.bluebubbles.config import BlueBubblesConfigAdapter

logger = logging.getLogger(__name__)

config_adapter = BlueBubblesConfigAdapter()


class BlueBubblesSecurity:
    def __init__(self):
        self._config_adapter = config_adapter

    async def check_dm_access(self, config: dict, peer_id: str) -> bool:
        channel_config = self._config_adapter._get_channel_config(config)
        dm_policy = channel_config.get("dmPolicy", "pairing")

        if dm_policy == "disabled":
            return False

        if dm_policy == "open":
            return True

        allow_from = channel_config.get("allowFrom", [])
        if peer_id in allow_from:
            return True

        if dm_policy == "pairing":
            return False

        return False

    async def check_group_access(self, config: dict, chat_guid: str, sender_handle: str) -> bool:
        channel_config = self._config_adapter._get_channel_config(config)
        group_policy = channel_config.get("groupPolicy", "allowlist")

        if group_policy == "disabled":
            return False

        if group_policy == "open":
            return True

        group_allow_from = channel_config.get("groupAllowFrom", [])
        if sender_handle in group_allow_from:
            return True

        groups = channel_config.get("groups", {})
        if "*" in groups:
            return True

        if chat_guid in groups:
            return True

        return False

    def resolve_dm_policy(self, config: dict) -> dict:
        channel_config = self._config_adapter._get_channel_config(config)
        return {
            "mode": channel_config.get("dmPolicy", "pairing"),
            "allow_from": channel_config.get("allowFrom", []),
        }