from __future__ import annotations

from typing import Dict, List


class GroupProfileResolver:
    def __init__(self, config: Dict):
        self.config = config or {}
        self.default_profile = self.config.get("default", {}) or {}
        self.profiles = self.config.get("profiles", []) or []

    def resolve(self, room_id: str, group_name: str = "", group_memory_profile: Dict | None = None) -> Dict:
        group_name_lower = str(group_name or "").lower()
        for profile in self.profiles:
            room_ids = set(profile.get("room_ids", []) or [])
            keywords = [str(item).lower() for item in (profile.get("group_name_keywords", []) or [])]
            if room_id and room_id in room_ids:
                return self._normalize(profile, room_id, group_name, group_memory_profile or {})
            if group_name_lower and any(keyword and keyword in group_name_lower for keyword in keywords):
                return self._normalize(profile, room_id, group_name, group_memory_profile or {})
        return self._normalize(self.default_profile, room_id, group_name, group_memory_profile or {})

    @staticmethod
    def _normalize(profile: Dict, room_id: str, group_name: str, group_memory_profile: Dict) -> Dict:
        focus = list(profile.get("knowledge_focus", []))
        configured_domain = str(profile.get("knowledge_domain", "general") or "general")
        inferred_domain = str(group_memory_profile.get("inferred_domain", "general") or "general")
        inferred_style = group_memory_profile.get("style_profile", {}) or {}
        effective_domain = configured_domain
        if configured_domain in {"", "general", "casual"} and inferred_domain not in {"", "general"}:
            effective_domain = inferred_domain
        inferred_focus = list(group_memory_profile.get("focus_topics", []))
        merged_focus = []
        for item in focus + inferred_focus:
            if item and item not in merged_focus:
                merged_focus.append(item)
        interaction_tone = str(profile.get("interaction_tone", "自然群友感") or "自然群友感")
        humor_style = str(profile.get("humor_style", "轻微") or "轻微")
        sharpness_style = str(profile.get("sharpness_style", "轻微嘴硬，不刻薄") or "轻微嘴硬，不刻薄")
        expressiveness_style = str(profile.get("expressiveness_style", "克制") or "克制")
        address_style = str(profile.get("address_style", "低频称呼，默认直接接话") or "低频称呼，默认直接接话")
        if configured_domain in {"", "general", "casual"}:
            interaction_tone = inferred_style.get("interaction_tone", interaction_tone)
            humor_style = inferred_style.get("humor_style", humor_style)
            sharpness_style = inferred_style.get("sharpness_style", sharpness_style)
            expressiveness_style = inferred_style.get("expressiveness_style", expressiveness_style)
        return {
            "room_id": room_id,
            "group_name": group_name,
            "mode": profile.get("mode", "social"),
            "persona_overlay": profile.get("persona_overlay", ""),
            "interaction_tone": interaction_tone,
            "humor_style": humor_style,
            "sharpness_style": sharpness_style,
            "expressiveness_style": expressiveness_style,
            "address_style": address_style,
            "knowledge_domain": effective_domain,
            "configured_domain": configured_domain,
            "knowledge_focus": merged_focus,
            "reply_style": profile.get("reply_style", "自然短句"),
            "topic_boundaries": profile.get("topic_boundaries", []),
            "group_memory_domain": inferred_domain,
            "group_memory_summary": group_memory_profile.get("summary_text", ""),
            "group_memory_sample_count": group_memory_profile.get("message_sample_count", 0),
            "group_memory_style": inferred_style,
        }