"""
绰号生成模块
为满足条件的角色生成修仙界绰号
"""
from typing import Optional, List, TYPE_CHECKING

if TYPE_CHECKING:
    from src.classes.avatar import Avatar
    from src.classes.world import World

from src.classes.event import Event
from src.utils.config import CONFIG
from src.utils.llm import get_prompt_and_call_llm_async
from src.run.log import get_logger

logger = get_logger().logger


def can_get_nickname(avatar: "Avatar") -> bool:
    """
    检查角色是否满足获得绰号的条件
    
    条件：
    1. 尚未拥有绰号（nickname为None）
    2. 长期事件数量 >= major_event_threshold
    3. 短期事件数量 >= minor_event_threshold
    
    Args:
        avatar: 要检查的角色
        
    Returns:
        是否满足条件
    """
    # 已有绰号，不再生成
    if avatar.nickname is not None:
        return False
    
    # 检查事件数量
    em = avatar.world.event_manager
    major_threshold = CONFIG.nickname.major_event_threshold
    minor_threshold = CONFIG.nickname.minor_event_threshold
    
    major_events = em.get_major_events_by_avatar(avatar.id)
    minor_events = em.get_minor_events_by_avatar(avatar.id)
    
    major_count = len(major_events)
    minor_count = len(minor_events)
    
    # AND逻辑：两个条件都要满足
    return major_count >= major_threshold and minor_count >= minor_threshold


async def generate_nickname(avatar: "Avatar") -> Optional[str]:
    """
    为角色生成绰号
    
    调用LLM基于角色信息和事件历史生成合适的绰号
    
    Args:
        avatar: 要生成绰号的角色
        
    Returns:
        生成的绰号，失败则返回None
    """
    try:
        # 获取 expanded_info（包含详细信息和事件历史）
        expanded_info = avatar.get_expanded_info(detailed=True)
        
        # 准备模板参数
        template_path = CONFIG.paths.templates / "nickname.txt"
        infos = {
            "avatar_info": expanded_info,
        }
        
        # 调用LLM并自动解析JSON
        response_data = await get_prompt_and_call_llm_async(template_path, infos, mode="fast")
        
        nickname = response_data.get("nickname", "").strip()
        thinking = response_data.get("thinking", "")
        
        if not nickname:
            logger.warning(f"为角色 {avatar.name} 生成绰号失败：返回空绰号")
            return None
        
        logger.info(f"为角色 {avatar.name} 生成绰号：{nickname}")
        logger.debug(f"绰号生成思考过程：{thinking}")
        
        return nickname
        
    except Exception as e:
        logger.error(f"生成绰号时出错：{e}")
        return None


async def process_avatar_nickname(avatar: "Avatar") -> Optional[Event]:
    """
    处理单个角色的绰号生成
    
    检查角色是否满足条件，满足则生成绰号并返回对应事件
    
    Args:
        avatar: 要处理的角色
        
    Returns:
        生成的事件，如果不满足条件或生成失败则返回None
    """
    if not can_get_nickname(avatar):
        return None
    
    nickname = await generate_nickname(avatar)
    if not nickname:
        return None
    
    avatar.nickname = nickname
    # 生成事件：角色获得绰号
    event = Event(
        avatar.world.month_stamp,
        f"{avatar.name}在修仙界中闯出名号，被人称为'{nickname}'。",
        related_avatars=[avatar.id],
        is_major=True
    )
    return event