"""
宗门等级系统
定义宗门内的职位等级及其与修仙境界的映射关系
"""
from enum import Enum
from functools import total_ordering
from typing import Optional, TYPE_CHECKING

if TYPE_CHECKING:
    from src.classes.cultivation import Realm
    from src.classes.sect import Sect


@total_ordering
class SectRank(Enum):
    """
    宗门职位等级
    从高到低：掌门 > 长老 > 内门弟子 > 外门弟子
    """
    Patriarch = "patriarch"      # 掌门
    Elder = "elder"              # 长老
    InnerDisciple = "inner"      # 内门弟子
    OuterDisciple = "outer"      # 外门弟子
    
    def __lt__(self, other):
        if not isinstance(other, SectRank):
            return NotImplemented
        # 数字越小职位越高，所以比较时反过来
        return RANK_ORDER[self] > RANK_ORDER[other]
    
    def __le__(self, other):
        if not isinstance(other, SectRank):
            return NotImplemented
        return RANK_ORDER[self] >= RANK_ORDER[other]
    
    def __gt__(self, other):
        if not isinstance(other, SectRank):
            return NotImplemented
        # 数字越小职位越高，所以比较时反过来
        return RANK_ORDER[self] < RANK_ORDER[other]
    
    def __ge__(self, other):
        if not isinstance(other, SectRank):
            return NotImplemented
        return RANK_ORDER[self] <= RANK_ORDER[other]


# 宗门职位顺序（数字越小职位越高）
RANK_ORDER = {
    SectRank.Patriarch: 0,
    SectRank.Elder: 1,
    SectRank.InnerDisciple: 2,
    SectRank.OuterDisciple: 3,
}


# 默认职位名称（可被宗门自定义覆盖）
DEFAULT_RANK_NAMES = {
    SectRank.Patriarch: "掌门",
    SectRank.Elder: "长老",
    SectRank.InnerDisciple: "内门弟子",
    SectRank.OuterDisciple: "外门弟子",
}


def get_rank_from_realm(realm: "Realm") -> SectRank:
    """
    根据修仙境界映射为宗门职位
    
    映射规则：
    - 练气 → 外门弟子
    - 筑基 → 内门弟子
    - 金丹 → 长老
    - 元婴 → 掌门（需要额外检查唯一性）
    
    Args:
        realm: 修仙境界
        
    Returns:
        对应的宗门职位
    """
    from src.classes.cultivation import Realm
    
    mapping = {
        Realm.Qi_Refinement: SectRank.OuterDisciple,
        Realm.Foundation_Establishment: SectRank.InnerDisciple,
        Realm.Core_Formation: SectRank.Elder,
        Realm.Nascent_Soul: SectRank.Patriarch,
    }
    return mapping.get(realm, SectRank.OuterDisciple)


def get_rank_display_name(rank: SectRank, sect: Optional["Sect"] = None) -> str:
    """
    获取职位的显示名称（支持宗门自定义）
    
    Args:
        rank: 宗门职位
        sect: 宗门对象（可选，如果提供则使用其自定义名称）
        
    Returns:
        职位的显示名称
    """
    if sect is not None:
        custom_name = sect.get_rank_name(rank)
        if custom_name:
            return custom_name
    return DEFAULT_RANK_NAMES.get(rank, "弟子")


def should_auto_promote(old_realm: "Realm", new_realm: "Realm") -> bool:
    """
    判断境界突破后是否应该自动晋升宗门职位
    
    Args:
        old_realm: 旧境界
        new_realm: 新境界
        
    Returns:
        是否应该晋升
    """
    if old_realm == new_realm:
        return False
    
    from src.classes.cultivation import Realm
    
    # 检查境界是否提升
    old_rank = get_rank_from_realm(old_realm)
    new_rank = get_rank_from_realm(new_realm)
    
    # 只有当新境界对应的职位更高时才晋升（职位枚举中 > 表示更高）
    return new_rank > old_rank


def check_and_promote_sect_rank(avatar: "Avatar", old_realm: "Realm", new_realm: "Realm") -> None:
    """
    检查境界突破后是否需要晋升宗门职位，并执行晋升
    
    Args:
        avatar: 要检查的角色
        old_realm: 旧境界
        new_realm: 新境界
    """
    # 无宗门或无职位，不需要晋升
    if avatar.sect is None or avatar.sect_rank is None:
        return
    
    # 检查是否应该晋升
    if not should_auto_promote(old_realm, new_realm):
        return
    
    new_rank = get_rank_from_realm(new_realm)
    
    # 如果新职位是掌门，需要检查该宗门是否已有掌门
    if new_rank == SectRank.Patriarch:
        if sect_has_patriarch(avatar):
            # 已有掌门，只能晋升为长老
            new_rank = SectRank.Elder
    
    # 执行晋升
    avatar.sect_rank = new_rank


def sect_has_patriarch(avatar: "Avatar") -> bool:
    """
    检查当前宗门是否已有掌门（不包括自己）
    
    Args:
        avatar: 要检查的角色
        
    Returns:
        是否已有其他掌门
    """
    if avatar.sect is None:
        return False
    
    # 从world中查找同宗门的其他avatar
    for other in avatar.world.avatar_manager.avatars.values():
        if other is avatar:
            continue
        if other.sect == avatar.sect and other.sect_rank == SectRank.Patriarch:
            return True
    
    return False