"""
Avatar 信息展示模块

将信息格式化逻辑从 Avatar 类中分离，作为独立函数提供。
"""
from __future__ import annotations

from typing import TYPE_CHECKING, Optional, List

if TYPE_CHECKING:
    from src.classes.avatar.core import Avatar

from src.classes.battle import get_base_strength
from src.classes.relation.relation import get_relation_label
from src.classes.emotions import EMOTION_EMOJIS, EmotionType
from src.utils.config import CONFIG


def _get_effects_text(avatar: "Avatar") -> str:
    """获取格式化的效果文本"""
    from src.i18n import t
    from src.classes.effect import format_effects_to_text
    breakdown = avatar.get_effect_breakdown()
    effect_parts = []
    for source_name, effects in breakdown:
        desc_str = format_effects_to_text(effects)
        if desc_str:
            effect_parts.append(f"[{source_name}] {desc_str}")
    return "\n".join(effect_parts) if effect_parts else t("None")


def get_avatar_info(avatar: "Avatar", detailed: bool = False) -> dict:
    """
    获取 avatar 的信息，返回 dict；根据 detailed 控制信息粒度。
    """
    from src.i18n import t
    region = avatar.tile.region if avatar.tile is not None else None
    from src.classes.relation.relation import get_relations_strs
    relation_lines = get_relations_strs(avatar, max_lines=8)
    relations_info = t("relation_separator").join(relation_lines) if relation_lines else t("None")
    magic_stone_info = str(avatar.magic_stone)

    from src.classes.sect import get_sect_info_with_rank
    
    if detailed:
        weapon_info = t("{weapon_name}, Proficiency: {proficiency}%", 
                       weapon_name=avatar.weapon.get_detailed_info(), 
                       proficiency=f"{avatar.weapon_proficiency:.1f}") if avatar.weapon else t("None")
        auxiliary_info = avatar.auxiliary.get_detailed_info() if avatar.auxiliary else t("None")
        sect_info = get_sect_info_with_rank(avatar, detailed=True)
        alignment_info = avatar.alignment.get_detailed_info() if avatar.alignment is not None else t("Unknown")
        region_info = region.get_detailed_info() if region is not None else t("None")
        root_info = avatar.root.get_detailed_info()
        technique_info = avatar.technique.get_detailed_info() if avatar.technique is not None else t("None")
        cultivation_info = avatar.cultivation_progress.get_detailed_info()
        personas_info = ", ".join([p.get_detailed_info() for p in avatar.personas]) if avatar.personas else t("None")
        materials_info = t("material_separator").join([f"{mat.get_detailed_info()}x{quantity}" for mat, quantity in avatar.materials.items()]) if avatar.materials else t("None")
        appearance_info = avatar.appearance.get_detailed_info(avatar.gender)
        spirit_animal_info = avatar.spirit_animal.get_info() if avatar.spirit_animal is not None else t("None")
    else:
        weapon_info = avatar.weapon.get_info() if avatar.weapon is not None else t("None")
        auxiliary_info = avatar.auxiliary.get_info() if avatar.auxiliary is not None else t("None")
        sect_info = get_sect_info_with_rank(avatar, detailed=False)
        region_info = region.get_info() if region is not None else t("None")
        alignment_info = avatar.alignment.get_info() if avatar.alignment is not None else t("Unknown")
        root_info = avatar.root.get_info()
        technique_info = avatar.technique.get_info() if avatar.technique is not None else t("None")
        cultivation_info = avatar.cultivation_progress.get_info()
        personas_info = ", ".join([p.get_detailed_info() for p in avatar.personas]) if avatar.personas else t("None")
        materials_info = t("material_separator").join([f"{mat.get_info()}x{quantity}" for mat, quantity in avatar.materials.items()]) if avatar.materials else t("None")
        appearance_info = avatar.appearance.get_info()
        spirit_animal_info = avatar.spirit_animal.get_info() if avatar.spirit_animal is not None else t("None")

    info_dict = {
        t("Name"): avatar.name,
        t("Gender"): str(avatar.gender),
        t("Age"): str(avatar.age),
        t("HP"): str(avatar.hp),
        t("Spirit Stones"): magic_stone_info,
        t("Relations"): relations_info,
        t("Sect"): sect_info,
        t("Alignment"): alignment_info,
        t("Region"): region_info,
        t("Spirit Root"): root_info,
        t("Technique"): technique_info,
        t("Realm"): cultivation_info,
        t("Traits"): personas_info,
        t("Materials"): materials_info,
        t("Appearance"): appearance_info,
        t("Weapon"): weapon_info,
        t("Auxiliary"): auxiliary_info,
        t("Emotion"): t(avatar.emotion.value),
        t("Long-term Goal"): avatar.long_term_objective.content if avatar.long_term_objective else t("None"),
        t("Short-term Goal"): avatar.short_term_objective if avatar.short_term_objective else t("None"),
    }
    
    if detailed:
        info_dict[t("Current Effects")] = _get_effects_text(avatar)

    # 绰号：仅在存在时显示
    if avatar.nickname is not None:
        info_dict[t("Nickname")] = avatar.nickname.value
    # 灵兽：仅在存在时显示
    if avatar.spirit_animal is not None:
        info_dict[t("Spirit Animal")] = spirit_animal_info
    return info_dict


def get_avatar_structured_info(avatar: "Avatar") -> dict:
    """
    获取结构化的角色信息，用于前端展示和交互。
    """
    # 基础信息
    from src.i18n import t
    emoji = EMOTION_EMOJIS.get(avatar.emotion, EMOTION_EMOJIS[EmotionType.CALM])
    
    info = {
        "id": avatar.id,
        "name": avatar.name,
        "gender": str(avatar.gender),
        "age": avatar.age.age,
        "lifespan": avatar.age.max_lifespan,
        "realm": avatar.cultivation_progress.get_info(),
        "level": avatar.cultivation_progress.level,
        "hp": {"cur": avatar.hp.cur, "max": avatar.hp.max},
        "alignment": str(avatar.alignment) if avatar.alignment else t("Unknown"),
        "magic_stone": avatar.magic_stone.value,
        "base_battle_strength": int(get_base_strength(avatar)),
        "emotion": {
            "name": t(avatar.emotion.value),
            "emoji": emoji,
            "desc": t(avatar.emotion.value)
        },
        "thinking": avatar.thinking,
        "short_term_objective": avatar.short_term_objective,
        "long_term_objective": avatar.long_term_objective.content if avatar.long_term_objective else "",
        "nickname": avatar.nickname.value if avatar.nickname else None,
        "nickname_reason": avatar.nickname.reason if avatar.nickname else None,
        "is_dead": avatar.is_dead,
        "death_info": avatar.death_info,
        "action_state": t("Performing {action}", action=avatar.current_action_name)
    }

    # 1. 特质 (Personas)
    info["personas"] = [p.get_structured_info() for p in avatar.personas]
    
    # 2. 功法 (Technique)
    if avatar.technique:
        info["technique"] = avatar.technique.get_structured_info()
    else:
        info["technique"] = None
        
    # 3. 宗门 (Sect)
    if avatar.sect:
        sect_info = avatar.sect.get_structured_info()
        if avatar.sect_rank:
            from src.classes.sect_ranks import get_rank_display_name
            sect_info["rank"] = get_rank_display_name(avatar.sect_rank, avatar.sect)
        else:
            sect_info["rank"] = t("Disciple")
        info["sect"] = sect_info
    else:
        info["sect"] = None
        
    # 补充：阵营详情
    from src.classes.alignment import alignment_info_msg_ids
    info["alignment"] = str(avatar.alignment) if avatar.alignment else t("Unknown")
    if avatar.alignment:
        desc_id = alignment_info_msg_ids.get(avatar.alignment, "")
        info["alignment_detail"] = {
            "name": str(avatar.alignment),
            "desc": t(desc_id) if desc_id else "",
        }

    # 4. 装备 (Weapon & Auxiliary)
    if avatar.weapon:
        w_info = avatar.weapon.get_structured_info()
        w_info["proficiency"] = f"{avatar.weapon_proficiency:.1f}%"
        info["weapon"] = w_info
    else:
        info["weapon"] = None
        
    if avatar.auxiliary:
        info["auxiliary"] = avatar.auxiliary.get_structured_info()
    else:
        info["auxiliary"] = None
        
    # 5. 材料 (Materials)
    materials_list = []
    for material, count in avatar.materials.items():
        m_info = material.get_structured_info()
        m_info["count"] = count
        materials_list.append(m_info)
    info["materials"] = materials_list
    
    # 6. 关系 (Relations)
    relations_list = []
    for other, relation in avatar.relations.items():
        relations_list.append({
            "target_id": other.id,
            "name": other.name,
            "relation": get_relation_label(relation, avatar, other),
            "relation_type": relation.value,
            "realm": other.cultivation_progress.get_info(),
            "sect": other.sect.name if other.sect else t("Rogue Cultivator")
        })
    info["relations"] = relations_list
    
    # 7. 外貌
    info["appearance"] = avatar.appearance.get_info()
    
    # 8. 灵根
    from src.classes.root import format_root_cn
    root_str = format_root_cn(avatar.root)
    info["root"] = root_str
    info["root_detail"] = {
         "name": root_str,
         "desc": t("Contains elements: {elements}", elements=t("element_separator").join(str(e) for e in avatar.root.elements)),
         "effect_desc": avatar.root.effect_desc
    }
    
    # 9. 灵兽
    if avatar.spirit_animal:
         info["spirit_animal"] = avatar.spirit_animal.get_structured_info()

    # 当前效果
    info[t("Current Effects")] = _get_effects_text(avatar)

    return info


def get_avatar_expanded_info(
    avatar: "Avatar", 
    co_region_avatars: Optional[List["Avatar"]] = None,
    other_avatar: Optional["Avatar"] = None,
    detailed: bool = False
) -> dict:
    """
    获取角色的扩展信息，包含基础信息、观察到的角色和事件历史。
    
    Args:
        avatar: 目标角色
        co_region_avatars: 同区域的其他角色列表，用于"观察到的角色"字段
        other_avatar: 另一个角色，如果提供则返回两人共同经历的事件，否则返回单人事件
        detailed: 是否返回详细信息
    """
    from src.i18n import t
    info = get_avatar_info(avatar, detailed=detailed)

    observed: list[str] = []
    if co_region_avatars:
        for other in co_region_avatars[:8]:
            observed.append(t("{name}, Realm: {realm}", name=other.name, realm=other.cultivation_progress.get_info()))

    # 历史事件改为从全局事件管理器分类查询
    em = avatar.world.event_manager
    major_limit = CONFIG.social.major_event_context_num
    minor_limit = CONFIG.social.minor_event_context_num
    
    # 根据是否提供 other_avatar 决定获取单人事件还是双人共同事件
    if other_avatar is not None:
        major_events = em.get_major_events_between(avatar.id, other_avatar.id, limit=major_limit)
        minor_events = em.get_minor_events_between(avatar.id, other_avatar.id, limit=minor_limit)
    else:
        major_events = em.get_major_events_by_avatar(avatar.id, limit=major_limit)
        minor_events = em.get_minor_events_by_avatar(avatar.id, limit=minor_limit)
    
    major_list = [str(e) for e in major_events]
    minor_list = [str(e) for e in minor_events]

    info[t("Nearby Avatars")] = observed
    info[t("Major Events")] = major_list
    info[t("Recent Events")] = minor_list
    return info


def get_other_avatar_info(from_avatar: "Avatar", to_avatar: "Avatar") -> str:
    """
    仅显示几个字段：名字、绰号、境界、关系、宗门、阵营、外貌、功法、武器、辅助装备、HP
    """
    from src.i18n import t
    nickname = to_avatar.nickname.value if to_avatar.nickname else t("None")
    sect = to_avatar.sect.name if to_avatar.sect else t("Rogue Cultivator")
    tech = to_avatar.technique.get_info() if to_avatar.technique else t("None")
    weapon = to_avatar.weapon.get_info() if to_avatar.weapon else t("None")
    aux = to_avatar.auxiliary.get_info() if to_avatar.auxiliary else t("None")
    alignment = to_avatar.alignment
    
    # 关系可能为空
    relation = from_avatar.get_relation(to_avatar) or t("None")

    return t(
        "{name}, Nickname: {nickname}, Realm: {realm}, Relation: {relation}, Sect: {sect}, Alignment: {alignment}, Appearance: {appearance}, Technique: {technique}, Weapon: {weapon}, Auxiliary: {aux}, HP: {hp}",
        name=to_avatar.name,
        nickname=nickname,
        realm=to_avatar.cultivation_progress.get_info(),
        relation=relation,
        sect=sect,
        alignment=alignment,
        appearance=to_avatar.appearance.get_info(),
        technique=tech,
        weapon=weapon,
        aux=aux,
        hp=to_avatar.hp
    )


def get_avatar_desc(avatar: "Avatar", detailed: bool = False) -> str:
    """
    获取角色的文本描述。
    detailed=True 时包含详细的效果来源分析。
    """
    from src.i18n import t
    # 基础描述
    lines = [t("【{name}】 {gender} {age} years old", name=avatar.name, gender=avatar.gender, age=avatar.age)]
    lines.append(t("Realm: {realm}", realm=avatar.cultivation_progress.get_info()))
    if avatar.sect:
        lines.append(t("Identity: {identity}", identity=avatar.get_sect_str()))
    
    if detailed:
        lines.append(t("\n--- Current Effects Detail ---"))
        breakdown = avatar.get_effect_breakdown()
        
        from src.classes.effect import format_effects_to_text
        
        if not breakdown:
            lines.append(t("No additional effects"))
        else:
            for source_name, effects in breakdown:
                # 使用现有的 format_effects_to_text 将字典转为中文描述
                desc_str = format_effects_to_text(effects)
                if desc_str:
                    lines.append(f"[{source_name}]: {desc_str}")
                
    return "\n".join(lines)