from typing import List, Optional, TYPE_CHECKING
import time
from src.chat.message_receive.chat_stream import ChatStream, get_chat_manager
from src.common.logger import get_logger
from src.manager.local_store_manager import local_storage
from src.person_info.relationship_builder_manager import RelationshipBuilder
from src.chat.express.expression_learner import ExpressionLearner
from src.plugin_system.base.component_types import ChatMode
from src.chat.planner_actions.action_manager import ActionManager
from src.chat.chat_loop.hfc_utils import CycleDetail

if TYPE_CHECKING:
    from .wakeup_manager import WakeUpManager
    from .energy_manager import EnergyManager

class HfcContext:
    def __init__(self, chat_id: str):
        """
        初始化HFC聊天上下文
        
        Args:
            chat_id: 聊天ID标识符
            
        功能说明:
        - 存储和管理单个聊天会话的所有状态信息
        - 包含聊天流、关系构建器、表达学习器等核心组件
        - 管理聊天模式、能量值、时间戳等关键状态
        - 提供循环历史记录和当前循环详情的存储
        - 集成唤醒度管理器，处理休眠状态下的唤醒机制
        
        Raises:
            ValueError: 如果找不到对应的聊天流
        """
        self.stream_id: str = chat_id
        self.chat_stream: Optional[ChatStream] = get_chat_manager().get_stream(self.stream_id)
        if not self.chat_stream:
            raise ValueError(f"无法找到聊天流: {self.stream_id}")

        self.log_prefix = f"[{get_chat_manager().get_stream_name(self.stream_id) or self.stream_id}]"
        
        self.relationship_builder: Optional[RelationshipBuilder] = None
        self.expression_learner: Optional[ExpressionLearner] = None
        
        self.loop_mode = ChatMode.NORMAL
        self.energy_value = 5.0
        self.sleep_pressure = 0.0
        self.was_sleeping = False # 用于检测睡眠状态的切换
        
        # 失眠状态
        self.is_in_insomnia: bool = False
        self.insomnia_end_time: float = 0.0
        self.last_wakeup_time: float = 0.0 # 被吵醒的时间
        
        self.last_message_time = time.time()
        self.last_read_time = time.time() - 10
        
        self.action_manager = ActionManager()
        
        self.running: bool = False
        
        self.history_loop: List[CycleDetail] = []
        self.cycle_counter = 0
        self.current_cycle_detail: Optional[CycleDetail] = None
        
        # 唤醒度管理器 - 延迟初始化以避免循环导入
        self.wakeup_manager: Optional['WakeUpManager'] = None
        self.energy_manager: Optional['EnergyManager'] = None

        self._load_context_state()

    def _get_storage_key(self) -> str:
        """获取当前聊天流的本地存储键"""
        return f"hfc_context_state_{self.stream_id}"

    def _load_context_state(self):
        """从本地存储加载状态"""
        state = local_storage[self._get_storage_key()]
        if state and isinstance(state, dict):
            self.energy_value = state.get("energy_value", 5.0)
            self.sleep_pressure = state.get("sleep_pressure", 0.0)
            self.is_in_insomnia = state.get("is_in_insomnia", False)
            self.insomnia_end_time = state.get("insomnia_end_time", 0.0)
            logger = get_logger("hfc_context")
            logger.info(f"{self.log_prefix} 成功从本地存储加载HFC上下文状态: {state}")
        else:
            logger = get_logger("hfc_context")
            logger.info(f"{self.log_prefix} 未找到本地HFC上下文状态，将使用默认值初始化。")

    def save_context_state(self):
        """将当前状态保存到本地存储"""
        state = {
            "energy_value": self.energy_value,
            "sleep_pressure": self.sleep_pressure,
            "is_in_insomnia": self.is_in_insomnia,
            "insomnia_end_time": self.insomnia_end_time,
            "last_wakeup_time": self.last_wakeup_time,
        }
        local_storage[self._get_storage_key()] = state
        logger = get_logger("hfc_context")
        logger.debug(f"{self.log_prefix} 已将HFC上下文状态保存到本地存储: {state}")