Merge branch 'master' of https://github.com/MoFox-Studio/MoFox_Bot

src/chat/chat_loop/cycle_processor.py

@@ -133,7 +133,7 @@ class CycleProcessor:
             await stop_typing()
 
         # 在一轮动作执行完毕后，增加睡眠压力
-        if self.context.energy_manager and global_config.wakeup_system.enable_insomnia_system:
+        if self.context.energy_manager and global_config.sleep_system.enable_insomnia_system:
             if action_type not in ["no_reply", "no_action"]:
                 self.context.energy_manager.increase_sleep_pressure()
         

src/chat/chat_loop/energy_manager.py

@@ -98,7 +98,7 @@ class EnergyManager:
 
             if is_sleeping:
                 # 睡眠中：减少睡眠压力
-                decay_per_10s = global_config.wakeup_system.sleep_pressure_decay_rate / 6
+                decay_per_10s = global_config.sleep_system.sleep_pressure_decay_rate / 6
                 self.context.sleep_pressure -= decay_per_10s
                 self.context.sleep_pressure = max(self.context.sleep_pressure, 0)
                 self._log_sleep_pressure_change("睡眠压力释放")
@@ -145,7 +145,7 @@ class EnergyManager:
         """
         在执行动作后增加睡眠压力
         """
-        increment = global_config.wakeup_system.sleep_pressure_increment
+        increment = global_config.sleep_system.sleep_pressure_increment
         self.context.sleep_pressure += increment
         self.context.sleep_pressure = min(self.context.sleep_pressure, 100.0) # 设置一个100的上限
         self._log_sleep_pressure_change("执行动作，睡眠压力累积")

src/chat/chat_loop/heartFC_chat.py

@@ -8,7 +8,7 @@ from src.config.config import global_config
 from src.person_info.relationship_builder_manager import relationship_builder_manager
 from src.chat.express.expression_learner import expression_learner_manager
 from src.plugin_system.base.component_types import ChatMode
-from src.schedule.schedule_manager import schedule_manager
+from src.schedule.schedule_manager import schedule_manager, SleepState
 from src.plugin_system.apis import message_api
 
 from .hfc_context import HfcContext
@@ -196,30 +196,14 @@ class HeartFChatting:
         - FOCUS模式：直接处理所有消息并检查退出条件
         - NORMAL模式：检查进入FOCUS模式的条件，并通过normal_mode_handler处理消息
         """
-        is_sleeping = schedule_manager.is_sleeping(self.wakeup_manager)
-
-        # --- 失眠状态管理 ---
-        if self.context.is_in_insomnia and time.time() > self.context.insomnia_end_time:
-            # 失眠状态结束
-            self.context.is_in_insomnia = False
-            await self.proactive_thinker.trigger_goodnight_thinking()
-
-        if is_sleeping and not self.context.was_sleeping:
-            # 刚刚进入睡眠状态，进行一次入睡检查
-            if self.wakeup_manager and self.wakeup_manager.check_for_insomnia():
-                # 触发失眠
-                self.context.is_in_insomnia = True
-                duration = global_config.wakeup_system.insomnia_duration_minutes * 60
-                self.context.insomnia_end_time = time.time() + duration
-                
-                # 判断失眠原因并触发思考
-                reason = "random"
-                if self.context.sleep_pressure < global_config.wakeup_system.sleep_pressure_threshold:
-                    reason = "low_pressure"
-                await self.proactive_thinker.trigger_insomnia_thinking(reason)
+        # --- 核心状态更新 ---
+        await schedule_manager.update_sleep_state(self.wakeup_manager)
+        current_sleep_state = schedule_manager.get_current_sleep_state()
+        is_sleeping = current_sleep_state == SleepState.SLEEPING
+        is_in_insomnia = current_sleep_state == SleepState.INSOMNIA
 
         # 核心修复：在睡眠模式(包括失眠)下获取消息时，不过滤命令消息，以确保@消息能被接收
-        filter_command_flag = not is_sleeping
+        filter_command_flag = not (is_sleeping or is_in_insomnia)
         
         recent_messages = message_api.get_messages_by_time_in_chat(
             chat_id=self.context.stream_id,
@@ -239,18 +223,18 @@ class HeartFChatting:
             self.context.last_read_time = time.time()
             
             # 处理唤醒度逻辑
-            if is_sleeping:
+            if current_sleep_state in [SleepState.SLEEPING, SleepState.PREPARING_SLEEP, SleepState.INSOMNIA]:
                 self._handle_wakeup_messages(recent_messages)
-                # 再次检查睡眠状态，因为_handle_wakeup_messages可能会触发唤醒
-                current_is_sleeping = schedule_manager.is_sleeping(self.wakeup_manager)
                 
-                if not self.context.is_in_insomnia and current_is_sleeping:
-                    # 仍然在睡眠，跳过本轮的消息处理
+                # 再次获取最新状态，因为 handle_wakeup 可能导致状态变为 WOKEN_UP
+                current_sleep_state = schedule_manager.get_current_sleep_state()
+                
+                if current_sleep_state == SleepState.SLEEPING:
+                    # 只有在纯粹的 SLEEPING 状态下才跳过消息处理
                     return has_new_messages
-                else:
-                    # 从睡眠中被唤醒，需要继续处理本轮消息
+                
+                if current_sleep_state == SleepState.WOKEN_UP:
                     logger.info(f"{self.context.log_prefix} 从睡眠中被唤醒，将处理积压的消息。")
-                    self.context.last_wakeup_time = time.time()
 
             # 根据聊天模式处理新消息
             if self.context.loop_mode == ChatMode.FOCUS:
@@ -273,12 +257,12 @@ class HeartFChatting:
         
         # --- 重新入睡逻辑 ---
         # 如果被吵醒了，并且在一定时间内没有新消息，则尝试重新入睡
-        if schedule_manager._is_woken_up and not has_new_messages:
-            re_sleep_delay = global_config.wakeup_system.re_sleep_delay_minutes * 60
+        if schedule_manager.get_current_sleep_state() == SleepState.WOKEN_UP and not has_new_messages:
+            re_sleep_delay = global_config.sleep_system.re_sleep_delay_minutes * 60
             # 使用 last_message_time 来判断空闲时间
             if time.time() - self.context.last_message_time > re_sleep_delay:
                 logger.info(f"{self.context.log_prefix} 已被唤醒且超过 {re_sleep_delay / 60} 分钟无新消息，尝试重新入睡。")
-                schedule_manager.reset_wakeup_state()
+                schedule_manager.reset_sleep_state_after_wakeup()
         
         # 保存HFC上下文状态
         self.context.save_context_state()

src/chat/chat_loop/hfc_context.py

@@ -46,11 +46,6 @@ class HfcContext:
         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
         
@@ -78,8 +73,6 @@ class HfcContext:
         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:
@@ -91,9 +84,6 @@ class HfcContext:
         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")

src/chat/chat_loop/wakeup_manager.py

@@ -31,22 +31,15 @@ class WakeUpManager:
         self.log_interval = 30
         
         # 从配置文件获取参数
-        wakeup_config = global_config.wakeup_system
-        self.wakeup_threshold = wakeup_config.wakeup_threshold
-        self.private_message_increment = wakeup_config.private_message_increment
-        self.group_mention_increment = wakeup_config.group_mention_increment
-        self.decay_rate = wakeup_config.decay_rate
-        self.decay_interval = wakeup_config.decay_interval
-        self.angry_duration = wakeup_config.angry_duration
-        self.enabled = wakeup_config.enable
-        self.angry_prompt = wakeup_config.angry_prompt
-        
-        # 失眠系统参数
-        self.insomnia_enabled = wakeup_config.enable_insomnia_system
-        self.sleep_pressure_threshold = wakeup_config.sleep_pressure_threshold
-        self.deep_sleep_threshold = wakeup_config.deep_sleep_threshold
-        self.insomnia_chance_low_pressure = wakeup_config.insomnia_chance_low_pressure
-        self.insomnia_chance_normal_pressure = wakeup_config.insomnia_chance_normal_pressure
+        sleep_config = global_config.sleep_system
+        self.wakeup_threshold = sleep_config.wakeup_threshold
+        self.private_message_increment = sleep_config.private_message_increment
+        self.group_mention_increment = sleep_config.group_mention_increment
+        self.decay_rate = sleep_config.decay_rate
+        self.decay_interval = sleep_config.decay_interval
+        self.angry_duration = sleep_config.angry_duration
+        self.enabled = sleep_config.enable
+        self.angry_prompt = sleep_config.angry_prompt
         
         self._load_wakeup_state()
 
@@ -144,7 +137,9 @@ class WakeUpManager:
             
         # 只有在休眠且非失眠状态下才累积唤醒度
         from src.schedule.schedule_manager import schedule_manager
-        if not schedule_manager.is_sleeping() or self.context.is_in_insomnia:
+        from src.schedule.sleep_manager import SleepState
+        current_sleep_state = schedule_manager.get_current_sleep_state()
+        if current_sleep_state != SleepState.SLEEPING:
             return False
             
         old_value = self.wakeup_value
@@ -220,39 +215,4 @@ class WakeUpManager:
             "wakeup_threshold": self.wakeup_threshold,
             "is_angry": self.is_angry,
             "angry_remaining_time": max(0, self.angry_duration - (time.time() - self.angry_start_time)) if self.is_angry else 0
-        }
-
-    def check_for_insomnia(self) -> bool:
-        """
-        在尝试入睡时检查是否会失眠
-        
-        Returns:
-            bool: 如果失眠则返回 True，否则返回 False
-        """
-        if not self.insomnia_enabled:
-            return False
-
-        import random
-        
-        pressure = self.context.sleep_pressure
-        
-        # 压力过高，深度睡眠，极难失眠
-        if pressure > self.deep_sleep_threshold:
-            return False
-            
-        # 根据睡眠压力决定失眠概率
-        from src.schedule.schedule_manager import schedule_manager
-        if pressure < self.sleep_pressure_threshold:
-            # 压力不足型失眠
-            if schedule_manager._is_in_voluntary_delay:
-                logger.debug(f"{self.context.log_prefix} 处于主动延迟睡眠期间，跳过压力不足型失眠判断。")
-            elif random.random() < self.insomnia_chance_low_pressure:
-                logger.info(f"{self.context.log_prefix} 睡眠压力不足 ({pressure:.1f})，触发失眠！")
-                return True
-        else:
-            # 压力正常，随机失眠
-            if random.random() < self.insomnia_chance_normal_pressure:
-                logger.info(f"{self.context.log_prefix} 睡眠压力正常 ({pressure:.1f})，触发随机失眠！")
-                return True
-        
-        return False
+        }

src/config/config.py

@@ -41,7 +41,7 @@ from src.config.official_configs import (
     WebSearchConfig,
     AntiPromptInjectionConfig,
     PluginsConfig,
-    WakeUpSystemConfig,
+    SleepSystemConfig,
     MonthlyPlanSystemConfig,
     CrossContextConfig,
     PermissionConfig,
@@ -390,7 +390,7 @@ class Config(ValidatedConfigBase):
     dependency_management: DependencyManagementConfig = Field(default_factory=lambda: DependencyManagementConfig(), description="依赖管理配置")
     web_search: WebSearchConfig = Field(default_factory=lambda: WebSearchConfig(), description="网络搜索配置")
     plugins: PluginsConfig = Field(default_factory=lambda: PluginsConfig(), description="插件配置")
-    wakeup_system: WakeUpSystemConfig = Field(default_factory=lambda: WakeUpSystemConfig(), description="唤醒度系统配置")
+    sleep_system: SleepSystemConfig = Field(default_factory=lambda: SleepSystemConfig(), description="睡眠系统配置")
     monthly_plan_system: MonthlyPlanSystemConfig = Field(default_factory=lambda: MonthlyPlanSystemConfig(), description="月层计划系统配置")
     cross_context: CrossContextConfig = Field(default_factory=lambda: CrossContextConfig(), description="跨群聊上下文共享配置")
     maizone_intercom: MaizoneIntercomConfig = Field(default_factory=lambda: MaizoneIntercomConfig(), description="Maizone互通组配置")

src/config/official_configs.py

@@ -529,16 +529,6 @@ class ScheduleConfig(ValidatedConfigBase):
 
     enable: bool = Field(default=True, description="启用")
     guidelines: Optional[str] = Field(default=None, description="指导方针")
-    enable_is_sleep: bool = Field(default=True, description="让AI会根据日程表睡觉和苏醒")
-    
-    enable_flexible_sleep: bool = Field(default=True, description="是否启用弹性睡眠")
-    flexible_sleep_pressure_threshold: float = Field(default=40.0, description="触发弹性睡眠的睡眠压力阈值，低于该值可能延迟入睡")
-    max_sleep_delay_minutes: int = Field(default=60, description="单日最大延迟入睡分钟数")
-    
-    enable_pre_sleep_notification: bool = Field(default=True, description="是否启用睡前消息")
-    pre_sleep_notification_groups: List[str] = Field(default_factory=list, description="接收睡前消息的群号列表, 格式: [\"platform:group_id1\", \"platform:group_id2\"]")
-    pre_sleep_prompt: str = Field(default="我准备睡觉了，请生成一句简短自然的晚安问候。", description="用于生成睡前消息的提示")
-
 
 
 class DependencyManagementConfig(ValidatedConfigBase):
@@ -617,10 +607,10 @@ class PluginsConfig(ValidatedConfigBase):
     centralized_config: bool = Field(default=True, description="是否启用插件配置集中化管理")
 
 
-class WakeUpSystemConfig(ValidatedConfigBase):
-    """唤醒度与失眠系统配置类"""
+class SleepSystemConfig(ValidatedConfigBase):
+    """睡眠系统配置类"""
 
-    enable: bool = Field(default=True, description="是否启用唤醒度系统")
+    enable: bool = Field(default=True, description="是否启用睡眠系统")
     wakeup_threshold: float = Field(default=15.0, ge=1.0, description="唤醒阈值，达到此值时会被唤醒")
     private_message_increment: float = Field(default=3.0, ge=0.1, description="私聊消息增加的唤醒度")
     group_mention_increment: float = Field(default=2.0, ge=0.1, description="群聊艾特增加的唤醒度")
@@ -640,6 +630,14 @@ class WakeUpSystemConfig(ValidatedConfigBase):
     sleep_pressure_increment: float = Field(default=1.5, ge=0.0, description="每次AI执行动作后，增加的睡眠压力值")
     sleep_pressure_decay_rate: float = Field(default=1.5, ge=0.0, description="睡眠时，每分钟衰减的睡眠压力值")
 
+    # --- 弹性睡眠与睡前消息 ---
+    enable_flexible_sleep: bool = Field(default=True, description="是否启用弹性睡眠")
+    flexible_sleep_pressure_threshold: float = Field(default=40.0, description="触发弹性睡眠的睡眠压力阈值，低于该值可能延迟入睡")
+    max_sleep_delay_minutes: int = Field(default=60, description="单日最大延迟入睡分钟数")
+    enable_pre_sleep_notification: bool = Field(default=True, description="是否启用睡前消息")
+    pre_sleep_notification_groups: List[str] = Field(default_factory=list, description="接收睡前消息的群号列表, 格式: [\"platform:group_id1\", \"platform:group_id2\"]")
+    pre_sleep_prompt: str = Field(default="我准备睡觉了，请生成一句简短自然的晚安问候。", description="用于生成睡前消息的提示")
+
 
 class MonthlyPlanSystemConfig(ValidatedConfigBase):
     """月度计划系统配置类"""

src/plugin_system/base/base_event.py

@@ -9,7 +9,7 @@ class HandlerResult:
     
     所有事件处理器必须返回此类的实例
     """
-    def __init__(self, success: bool, continue_process: bool, message: Any = {}, handler_name: str = ""):
+    def __init__(self, success: bool, continue_process: bool, message: Any = None, handler_name: str = ""):
         self.success = success
         self.continue_process = continue_process
         self.message = message
@@ -40,6 +40,19 @@ class HandlerResultsCollection:
         """获取continue_process为False的handler结果"""
         return [result for result in self.results if not result.continue_process]
     
+    def get_message_result(self) -> Any:
+        """获取handler的message
+        
+        当只有一个handler的结果时，直接返回那个handler结果中的message字段
+        否则用字典的形式{handler_name:message}返回
+        """
+        if len(self.results) == 0:
+            return {}
+        elif len(self.results) == 1:
+            return self.results[0].message
+        else:
+            return {result.handler_name: result.message for result in self.results}
+    
     def get_handler_result(self, handler_name: str) -> Optional[HandlerResult]:
         """获取指定handler的结果"""
         for result in self.results:
@@ -70,8 +83,8 @@ class BaseEvent:
     def __init__(
             self, 
             name: str, 
-            allowed_subscribers: List[str]=[],  
-            allowed_triggers: List[str]=[]
+            allowed_subscribers: List[str] = None,
+            allowed_triggers: List[str] = None
         ):
         self.name = name
         self.enabled = True

src/plugin_system/core/plugin_hot_reload.py

@@ -348,7 +348,6 @@ class PluginHotReloadManager:
 
     def _force_clear_plugin_modules(self, plugin_name: str):
         """强制清理插件相关的模块缓存"""
-        import sys
         
         # 找到所有相关的模块名
         modules_to_remove = []

src/plugin_system/core/plugin_manager.py

@@ -1,7 +1,6 @@
 import asyncio
 import os
 import traceback
-import sys
 import importlib
 
 from typing import Dict, List, Optional, Tuple, Type, Any

src/plugins/built_in/maizone_refactored/actions/read_feed_action.py

@@ -6,9 +6,9 @@ from typing import Tuple
 
 from src.common.logger import get_logger
 from src.plugin_system import BaseAction, ActionActivationType, ChatMode
-from src.plugin_system.apis import person_api, generator_api
+from src.plugin_system.apis import generator_api
 from src.plugin_system.apis.permission_api import permission_api
-from ..services.manager import get_qzone_service, get_config_getter
+from ..services.manager import get_qzone_service
 
 logger = get_logger("MaiZone.ReadFeedAction")
 

src/plugins/built_in/maizone_refactored/actions/send_feed_action.py

@@ -6,9 +6,9 @@ from typing import Tuple
 
 from src.common.logger import get_logger
 from src.plugin_system import BaseAction, ActionActivationType, ChatMode
-from src.plugin_system.apis import person_api, generator_api
+from src.plugin_system.apis import generator_api
 from src.plugin_system.apis.permission_api import permission_api
-from ..services.manager import get_qzone_service, get_config_getter
+from ..services.manager import get_qzone_service
 
 logger = get_logger("MaiZone.SendFeedAction")
 

src/plugins/built_in/plugin_management/plugin.py

@@ -3,15 +3,12 @@ import asyncio
 from typing import List, Tuple, Type
 from src.plugin_system import (
     BasePlugin,
-    BaseCommand,
-    CommandInfo,
     ConfigField,
     register_plugin,
     plugin_manage_api,
     component_manage_api,
     ComponentInfo,
     ComponentType,
-    send_api,
 )
 from src.plugin_system.base.plus_command import PlusCommand
 from src.plugin_system.base.command_args import CommandArgs

src/schedule/schedule_manager.py

@@ -1,23 +1,24 @@
 import orjson
 import asyncio
-import random
 from datetime import datetime, time, timedelta
-from typing import Optional, List, Dict, Any
+from typing import Optional, List, Dict, Any, TYPE_CHECKING
 from lunar_python import Lunar
 from pydantic import BaseModel, ValidationError, validator
 
 from src.common.database.sqlalchemy_models import Schedule, get_db_session
 from src.common.database.monthly_plan_db import (
     get_smart_plans_for_daily_schedule,
-    update_plan_usage  # 保留兼容性
+    update_plan_usage,  # 保留兼容性
 )
 from src.config.config import global_config, model_config
 from src.llm_models.utils_model import LLMRequest
 from src.common.logger import get_logger
 from json_repair import repair_json
 from src.manager.async_task_manager import AsyncTask, async_task_manager
-from src.manager.local_store_manager import local_storage
-from src.plugin_system.apis import send_api, generator_api
+from .sleep_manager import SleepManager, SleepState
+
+if TYPE_CHECKING:
+    from src.chat.chat_loop.wakeup_manager import WakeUpManager
 
 
 logger = get_logger("schedule_manager")
@@ -31,81 +32,85 @@ DEFAULT_SCHEDULE_GUIDELINES = """
 另外，请保证充足的休眠时间来处理和整合一天的数据。
 """
 
+
 class ScheduleItem(BaseModel):
     """单个日程项的Pydantic模型"""
+
     time_range: str
     activity: str
-    
-    @validator('time_range')
+
+    @validator("time_range")
     def validate_time_range(cls, v):
         """验证时间范围格式"""
-        if not v or '-' not in v:
+        if not v or "-" not in v:
             raise ValueError("时间范围必须包含'-'分隔符")
-        
+
         try:
-            start_str, end_str = v.split('-', 1)
+            start_str, end_str = v.split("-", 1)
             start_str = start_str.strip()
             end_str = end_str.strip()
-            
+
             # 验证时间格式
             datetime.strptime(start_str, "%H:%M")
             datetime.strptime(end_str, "%H:%M")
-            
+
             return v
         except ValueError as e:
             raise ValueError(f"时间格式无效，应为HH:MM-HH:MM格式: {e}") from e
-    
-    @validator('activity')
+
+    @validator("activity")
     def validate_activity(cls, v):
         """验证活动描述"""
         if not v or not v.strip():
             raise ValueError("活动描述不能为空")
         return v.strip()
 
+
 class ScheduleData(BaseModel):
     """完整日程数据的Pydantic模型"""
+
     schedule: List[ScheduleItem]
-    
-    @validator('schedule')
+
+    @validator("schedule")
     def validate_schedule_completeness(cls, v):
         """验证日程是否覆盖24小时"""
         if not v:
             raise ValueError("日程不能为空")
-        
+
         # 收集所有时间段
         time_ranges = []
         for item in v:
             try:
-                start_str, end_str = item.time_range.split('-', 1)
+                start_str, end_str = item.time_range.split("-", 1)
                 start_time = datetime.strptime(start_str.strip(), "%H:%M").time()
                 end_time = datetime.strptime(end_str.strip(), "%H:%M").time()
                 time_ranges.append((start_time, end_time))
             except ValueError:
                 continue
-        
+
         # 检查是否覆盖24小时
         if not cls._check_24_hour_coverage(time_ranges):
             raise ValueError("日程必须覆盖完整的24小时")
-        
+
         return v
-    
+
     @staticmethod
     def _check_24_hour_coverage(time_ranges: List[tuple]) -> bool:
         """检查时间段是否覆盖24小时"""
         if not time_ranges:
             return False
-        
+
         # 将时间转换为分钟数进行计算
         def time_to_minutes(t: time) -> int:
             return t.hour * 60 + t.minute
-        
+
         # 创建覆盖情况数组 (1440分钟 = 24小时)
         covered = [False] * 1440
-        
+
         for start_time, end_time in time_ranges:
             start_min = time_to_minutes(start_time)
             end_min = time_to_minutes(end_time)
-            
+
             if start_min <= end_min:
                 # 同一天内的时间段
                 for i in range(start_min, end_min):
@@ -117,30 +122,19 @@ class ScheduleData(BaseModel):
                     covered[i] = True
                 for i in range(0, end_min):
                     covered[i] = True
-        
+
         # 检查是否所有分钟都被覆盖
         return all(covered)
 
+
 class ScheduleManager:
     def __init__(self):
         self.today_schedule: Optional[List[Dict[str, Any]]] = None
         self.llm = LLMRequest(model_set=model_config.model_task_config.schedule_generator, request_type="schedule")
         self.max_retries = -1  # 无限重试，直到成功生成标准日程表
         self.daily_task_started = False
-        self.last_sleep_log_time = 0
-        self.sleep_log_interval = 35  # 日志记录间隔，单位秒
         self.schedule_generation_running = False  # 防止重复生成任务
-
-        # 弹性睡眠相关状态
-        self._is_preparing_sleep: bool = False
-        self._sleep_buffer_end_time: Optional[datetime] = None
-        self._total_delayed_minutes_today: int = 0
-        self._last_sleep_check_date: Optional[datetime.date] = None
-        self._last_fully_slept_log_time: float = 0
-        self._is_in_voluntary_delay: bool = False # 新增：标记是否处于主动延迟睡眠状态
-        self._is_woken_up: bool = False # 新增：标记是否被吵醒
-        
-        self._load_sleep_state()
+        self.sleep_manager = SleepManager(self)
 
     async def start_daily_schedule_generation(self):
         """启动每日零点自动生成新日程的任务"""
@@ -165,10 +159,10 @@ class ScheduleManager:
                 schedule_record = session.query(Schedule).filter(Schedule.date == today_str).first()
                 if schedule_record:
                     logger.info(f"从数据库加载今天的日程 ({today_str})。")
-                    
+
                     try:
                         schedule_data = orjson.loads(str(schedule_record.schedule_data))
-                        
+
                         # 使用Pydantic验证日程数据
                         if self._validate_schedule_with_pydantic(schedule_data):
                             self.today_schedule = schedule_data
@@ -197,15 +191,15 @@ class ScheduleManager:
         if self.schedule_generation_running:
             logger.info("日程生成任务已在运行中，跳过重复启动")
             return
-            
+
         # 创建异步任务进行日程生成，不阻塞主程序
         asyncio.create_task(self._async_generate_and_save_schedule())
         logger.info("已启动异步日程生成任务")
-        
+
     async def _async_generate_and_save_schedule(self):
         """异步生成并保存日程的内部方法"""
         self.schedule_generation_running = True
-        
+
         try:
             now = datetime.now()
             today_str = now.strftime("%Y-%m-%d")
@@ -217,7 +211,7 @@ class ScheduleManager:
             festivals = lunar.getFestivals()
             other_festivals = lunar.getOtherFestivals()
             all_festivals = festivals + other_festivals
-            
+
             festival_block = ""
             if all_festivals:
                 festival_text = "、".join(all_festivals)
@@ -228,33 +222,28 @@ class ScheduleManager:
             used_plan_ids = []
             if global_config.monthly_plan_system and global_config.monthly_plan_system.enable:
                 # 使用新的智能抽取逻辑
-                avoid_days = getattr(global_config.monthly_plan_system, 'avoid_repetition_days', 7)
+                avoid_days = getattr(global_config.monthly_plan_system, "avoid_repetition_days", 7)
                 # 使用新的智能抽取逻辑
-                avoid_days = getattr(global_config.monthly_plan_system, 'avoid_repetition_days', 7)
+                avoid_days = getattr(global_config.monthly_plan_system, "avoid_repetition_days", 7)
                 sampled_plans = get_smart_plans_for_daily_schedule(
-                    current_month_str,
-                    max_count=3,
-                    avoid_days=avoid_days
+                    current_month_str, max_count=3, avoid_days=avoid_days
                 )
 
                 # 如果计划耗尽，则触发补充生成
                 if not sampled_plans:
                     logger.info("可用的月度计划已耗尽或不足，尝试进行补充生成...")
                     from mmc.src.schedule.monthly_plan_manager import monthly_plan_manager
+
                     success = await monthly_plan_manager.generate_monthly_plans(current_month_str)
                     if success:
                         logger.info("补充生成完成，重新抽取月度计划...")
                         sampled_plans = get_smart_plans_for_daily_schedule(
-                            current_month_str,
-                            max_count=3,
-                            avoid_days=avoid_days
+                            current_month_str, max_count=3, avoid_days=avoid_days
                         )
                     else:
                         logger.warning("月度计划补充生成失败。")
-                
+
                 if sampled_plans:
-                    used_plan_ids = [plan.id for plan in sampled_plans]  # SQLAlchemy 对象的 id 属性会自动返回实际值
-                    
                     plan_texts = "\n".join([f"- {plan.plan_text}" for plan in sampled_plans])
                     monthly_plans_block = f"""
 **我这个月的一些小目标/计划 (请在今天的日程中适当体现)**:
@@ -294,33 +283,33 @@ class ScheduleManager:
 
 请你扮演我，以我的身份和口吻，为我生成一份完整的24小时日程表。
 """
-            
+
             # 无限重试直到生成成功的标准日程表
             attempt = 0
             while True:
                 attempt += 1
                 try:
                     logger.info(f"正在生成日程 (第 {attempt} 次尝试)")
-                    
+
                     # 构建当前尝试的prompt，增加压力提示
                     prompt = base_prompt
                     if attempt > 1:
                         failure_hint = f"""
 
 **重要提醒 (第{attempt}次尝试)**:
-- 前面{attempt-1}次生成都失败了，请务必严格按照要求生成完整的24小时日程
+- 前面{attempt - 1}次生成都失败了，请务必严格按照要求生成完整的24小时日程
 - 确保JSON格式正确，所有时间段连续覆盖24小时
 - 时间格式必须为HH:MM-HH:MM，不能有时间间隙或重叠
 - 不要输出任何解释文字，只输出纯JSON数组
 - 确保输出完整，不要被截断
 """
                         prompt += failure_hint
-                    
+
                     response, _ = await self.llm.generate_response_async(prompt)
-                    
+
                     # 尝试解析和验证JSON（项目内置的反截断机制会自动处理截断问题）
                     schedule_data = orjson.loads(repair_json(response))
-                    
+
                     # 使用Pydantic验证生成的日程数据
                     if self._validate_schedule_with_pydantic(schedule_data):
                         # 验证通过，保存到数据库
@@ -329,46 +318,49 @@ class ScheduleManager:
                             existing_schedule = session.query(Schedule).filter(Schedule.date == today_str).first()
                             if existing_schedule:
                                 # 更新现有日程
-                                session.query(Schedule).filter(Schedule.date == today_str).update({
-                                    Schedule.schedule_data: orjson.dumps(schedule_data).decode('utf-8'),
-                                    Schedule.updated_at: datetime.now()
-                                })
+                                session.query(Schedule).filter(Schedule.date == today_str).update(
+                                    {
+                                        Schedule.schedule_data: orjson.dumps(schedule_data).decode("utf-8"),
+                                        Schedule.updated_at: datetime.now(),
+                                    }
+                                )
                             else:
                                 # 创建新日程
                                 new_schedule = Schedule(
-                                    date=today_str,
-                                    schedule_data=orjson.dumps(schedule_data).decode('utf-8')
+                                    date=today_str, schedule_data=orjson.dumps(schedule_data).decode("utf-8")
                                 )
                                 session.add(new_schedule)
                             session.commit()
-                        
+
                         # 美化输出
                         schedule_str = f"✅ 经过 {attempt} 次尝试，成功生成并保存今天的日程 ({today_str})：\n"
                         for item in schedule_data:
-                            schedule_str += f"  - {item.get('time_range', '未知时间')}: {item.get('activity', '未知活动')}\n"
+                            schedule_str += (
+                                f"  - {item.get('time_range', '未知时间')}: {item.get('activity', '未知活动')}\n"
+                            )
                         logger.info(schedule_str)
-                        
+
                         self.today_schedule = schedule_data
-                        
+
                         # 成功生成日程后，更新使用过的月度计划的统计信息
                         if used_plan_ids and global_config.monthly_plan_system:
                             logger.info(f"更新使用过的月度计划 {used_plan_ids} 的统计信息。")
                             update_plan_usage(used_plan_ids, today_str)  # type: ignore
-                                
+
                         # 成功生成，退出无限循环
                         break
-                        
+
                     else:
                         logger.warning(f"第 {attempt} 次生成的日程验证失败，继续重试...")
                         # 添加短暂延迟，避免过于频繁的请求
                         await asyncio.sleep(2)
-                        
+
                 except Exception as e:
                     logger.error(f"第 {attempt} 次生成日程失败: {e}")
                     logger.info("继续重试...")
                     # 添加短暂延迟，避免过于频繁的请求
                     await asyncio.sleep(3)
-                    
+
         finally:
             self.schedule_generation_running = False
             logger.info("日程生成任务结束")
@@ -386,12 +378,12 @@ class ScheduleManager:
             try:
                 time_range = event.get("time_range")
                 activity = event.get("activity")
-                
+
                 if not time_range or not activity:
                     logger.warning(f"日程事件缺少必要字段: {event}")
                     continue
-                    
-                start_str, end_str = time_range.split('-')
+
+                start_str, end_str = time_range.split("-")
                 start_time = datetime.strptime(start_str.strip(), "%H:%M").time()
                 end_time = datetime.strptime(end_str.strip(), "%H:%M").time()
 
@@ -406,249 +398,21 @@ class ScheduleManager:
                 continue
         return None
 
-    def is_sleeping(self, wakeup_manager: Optional["WakeUpManager"] = None) -> bool:
-        """
-        通过关键词匹配、唤醒度、睡眠压力等综合判断是否处于休眠时间。
-        新增弹性睡眠机制，允许在压力低时延迟入睡，并在入睡前发送通知。
-        """
-        # --- 基础检查 ---
-        if not global_config.schedule.enable_is_sleep:
-            return False
-        if not self.today_schedule:
-            return False
+    def get_current_sleep_state(self) -> SleepState:
+        """获取当前的睡眠状态"""
+        return self.sleep_manager.get_current_sleep_state()
 
-        now = datetime.now()
-        today = now.date()
+    def is_sleeping(self) -> bool:
+        """检查当前是否处于正式休眠状态"""
+        return self.sleep_manager.is_sleeping()
 
-        # --- 每日状态重置 ---
-        if self._last_sleep_check_date != today:
-            logger.info(f"新的一天 ({today})，重置弹性睡眠状态。")
-            self._total_delayed_minutes_today = 0
-            self._is_preparing_sleep = False
-            self._sleep_buffer_end_time = None
-            self._last_sleep_check_date = today
-            self._is_in_voluntary_delay = False
-            self._save_sleep_state()
-
-        # --- 检查是否在“准备入睡”的缓冲期 ---
-        if self._is_preparing_sleep and self._sleep_buffer_end_time:
-            if now >= self._sleep_buffer_end_time:
-                current_timestamp = now.timestamp()
-                if current_timestamp - self._last_fully_slept_log_time > 45:
-                    logger.info("睡眠缓冲期结束，正式进入休眠状态。")
-                    self._last_fully_slept_log_time = current_timestamp
-                return True
-            else:
-                remaining_seconds = (self._sleep_buffer_end_time - now).total_seconds()
-                logger.debug(f"处于入睡缓冲期，剩余 {remaining_seconds:.1f} 秒。")
-                return False
-
-        # --- 判断当前是否为理论上的睡眠时间 ---
-        is_in_theoretical_sleep, activity = self._is_in_theoretical_sleep_time(now.time())
-
-        if not is_in_theoretical_sleep:
-            # 如果不在理论睡眠时间，确保重置准备状态
-            if self._is_preparing_sleep:
-                logger.info("已离开理论休眠时间，取消“准备入睡”状态。")
-                self._is_preparing_sleep = False
-                self._sleep_buffer_end_time = None
-                self._is_in_voluntary_delay = False
-                self._is_woken_up = False # 离开睡眠时间，重置唤醒状态
-                self._save_sleep_state()
-            return False
-
-        # --- 处理唤醒状态 ---
-        if self._is_woken_up:
-            current_timestamp = now.timestamp()
-            if current_timestamp - self.last_sleep_log_time > self.sleep_log_interval:
-                logger.info(f"在休眠活动 '{activity}' 期间，但已被唤醒，保持清醒状态。")
-                self.last_sleep_log_time = current_timestamp
-            return False
-
-        # --- 核心：弹性睡眠逻辑 ---
-        if global_config.schedule.enable_flexible_sleep and not self._is_preparing_sleep:
-            # 首次进入理论睡眠时间，触发弹性判断
-            logger.info(f"进入理论休眠时间 '{activity}'，开始弹性睡眠判断...")
-            
-            # 1. 获取睡眠压力
-            sleep_pressure = wakeup_manager.context.sleep_pressure if wakeup_manager else 999
-            pressure_threshold = global_config.schedule.flexible_sleep_pressure_threshold
-            
-            # 2. 判断是否延迟
-            if sleep_pressure < pressure_threshold and self._total_delayed_minutes_today < global_config.schedule.max_sleep_delay_minutes:
-                delay_minutes = 15  # 每次延迟15分钟
-                self._total_delayed_minutes_today += delay_minutes
-                self._sleep_buffer_end_time = now + timedelta(minutes=delay_minutes)
-                self._is_in_voluntary_delay = True # 标记进入主动延迟
-                logger.info(f"睡眠压力 ({sleep_pressure:.1f}) 低于阈值 ({pressure_threshold})，延迟入睡 {delay_minutes} 分钟。今日已累计延迟 {self._total_delayed_minutes_today} 分钟。")
-            else:
-                # 3. 计算5-10分钟的入睡缓冲
-                self._is_in_voluntary_delay = False # 非主动延迟
-                buffer_seconds = random.randint(5 * 60, 10 * 60)
-                self._sleep_buffer_end_time = now + timedelta(seconds=buffer_seconds)
-                logger.info(f"睡眠压力正常或已达今日最大延迟，将在 {buffer_seconds / 60:.1f} 分钟内入睡。")
-
-            # 4. 发送睡前通知
-            if global_config.schedule.enable_pre_sleep_notification:
-                asyncio.create_task(self._send_pre_sleep_notification())
-
-            self._is_preparing_sleep = True
-            self._save_sleep_state()
-            return False  # 进入准备阶段，但尚未正式入睡
-
-        # --- 经典模式或已在弹性睡眠流程中 ---
-        current_timestamp = now.timestamp()
-        if current_timestamp - self.last_sleep_log_time > self.sleep_log_interval:
-            logger.info(f"当前处于休眠活动 '{activity}' 中 (经典模式)。")
-            self.last_sleep_log_time = current_timestamp
-        return True
+    async def update_sleep_state(self, wakeup_manager: Optional["WakeUpManager"] = None):
+        """更新睡眠状态"""
+        await self.sleep_manager.update_sleep_state(wakeup_manager)
 
     def reset_sleep_state_after_wakeup(self):
-        """被唤醒后重置睡眠状态"""
-        if self._is_preparing_sleep or self.is_sleeping():
-            logger.info("被唤醒，重置所有睡眠准备状态，恢复清醒！")
-            self._is_preparing_sleep = False
-            self._sleep_buffer_end_time = None
-            self._is_in_voluntary_delay = False
-            self._is_woken_up = True  # 标记为已被唤醒
-            self._save_sleep_state()
-
-    def _is_in_theoretical_sleep_time(self, now_time: time) -> (bool, Optional[str]):
-        """检查当前时间是否落在日程表的任何一个睡眠活动中"""
-        sleep_keywords = ["休眠", "睡觉", "梦乡"]
-        
-        for event in self.today_schedule:
-            try:
-                activity = event.get("activity", "").strip()
-                time_range = event.get("time_range")
-
-                if not activity or not time_range:
-                    continue
-
-                if any(keyword in activity for keyword in sleep_keywords):
-                    start_str, end_str = time_range.split('-')
-                    start_time = datetime.strptime(start_str.strip(), "%H:%M").time()
-                    end_time = datetime.strptime(end_str.strip(), "%H:%M").time()
-
-                    if start_time <= end_time:  # 同一天
-                        if start_time <= now_time < end_time:
-                            return True, activity
-                    else:  # 跨天
-                        if now_time >= start_time or now_time < end_time:
-                            return True, activity
-            except (ValueError, KeyError, AttributeError) as e:
-                logger.warning(f"解析日程事件时出错: {event}, 错误: {e}")
-                continue
-        
-        return False, None
-
-    async def _send_pre_sleep_notification(self):
-        """异步生成并发送睡前通知"""
-        try:
-            groups = global_config.schedule.pre_sleep_notification_groups
-            prompt = global_config.schedule.pre_sleep_prompt
-
-            if not groups:
-                logger.info("未配置睡前通知的群组，跳过发送。")
-                return
-            
-            if not prompt:
-                logger.warning("睡前通知的prompt为空，跳过发送。")
-                return
-
-            # 为防止消息风暴，稍微延迟一下
-            await asyncio.sleep(random.uniform(5, 15))
-
-            for group_id_str in groups:
-                try:
-                    # 格式 "platform:group_id"
-                    parts = group_id_str.split(":")
-                    if len(parts) != 2:
-                        logger.warning(f"无效的群组ID格式: {group_id_str}")
-                        continue
-                    
-                    platform, group_id = parts
-                    
-                    # 使用与 ChatStream.get_stream_id 相同的逻辑生成 stream_id
-                    import hashlib
-                    key = "_".join([platform, group_id])
-                    stream_id = hashlib.md5(key.encode()).hexdigest()
-
-                    logger.info(f"正在为群组 {group_id_str} (Stream ID: {stream_id}) 生成睡前消息...")
-                    
-                    # 调用 generator_api 生成回复
-                    success, reply_set, _ = await generator_api.generate_reply(
-                        chat_id=stream_id,
-                        extra_info=prompt,
-                        request_type="schedule.pre_sleep_notification"
-                    )
-
-                    if success and reply_set:
-                        # 提取文本内容并发送
-                        reply_text = "".join([content for msg_type, content in reply_set if msg_type == "text"])
-                        if reply_text:
-                            logger.info(f"向群组 {group_id_str} 发送睡前消息: {reply_text}")
-                            await send_api.text_to_stream(text=reply_text, stream_id=stream_id)
-                        else:
-                            logger.warning(f"为群组 {group_id_str} 生成的回复内容为空。")
-                    else:
-                        logger.error(f"为群组 {group_id_str} 生成睡前消息失败。")
-
-                    await asyncio.sleep(random.uniform(2, 5)) # 避免发送过快
-
-                except Exception as e:
-                    logger.error(f"向群组 {group_id_str} 发送睡前消息失败: {e}")
-
-        except Exception as e:
-            logger.error(f"发送睡前通知任务失败: {e}")
-
-    def _save_sleep_state(self):
-        """将当前弹性睡眠状态保存到本地存储"""
-        try:
-            state = {
-                "is_preparing_sleep": self._is_preparing_sleep,
-                "sleep_buffer_end_time_ts": self._sleep_buffer_end_time.timestamp() if self._sleep_buffer_end_time else None,
-                "total_delayed_minutes_today": self._total_delayed_minutes_today,
-                "last_sleep_check_date_str": self._last_sleep_check_date.isoformat() if self._last_sleep_check_date else None,
-                "is_in_voluntary_delay": self._is_in_voluntary_delay,
-                "is_woken_up": self._is_woken_up,
-            }
-            local_storage["schedule_sleep_state"] = state
-            logger.debug(f"已保存睡眠状态: {state}")
-        except Exception as e:
-            logger.error(f"保存睡眠状态失败: {e}")
-
-    def _load_sleep_state(self):
-        """从本地存储加载弹性睡眠状态"""
-        try:
-            state = local_storage["schedule_sleep_state"]
-            if state and isinstance(state, dict):
-                self._is_preparing_sleep = state.get("is_preparing_sleep", False)
-                
-                end_time_ts = state.get("sleep_buffer_end_time_ts")
-                if end_time_ts:
-                    self._sleep_buffer_end_time = datetime.fromtimestamp(end_time_ts)
-                
-                self._total_delayed_minutes_today = state.get("total_delayed_minutes_today", 0)
-                self._is_in_voluntary_delay = state.get("is_in_voluntary_delay", False)
-                self._is_woken_up = state.get("is_woken_up", False)
-                
-                date_str = state.get("last_sleep_check_date_str")
-                if date_str:
-                    self._last_sleep_check_date = datetime.fromisoformat(date_str).date()
-
-                logger.info(f"成功从本地存储加载睡眠状态: {state}")
-        except Exception as e:
-            logger.warning(f"加载睡眠状态失败，将使用默认值: {e}")
-
-    def reset_wakeup_state(self):
-        """重置被唤醒的状态，允许重新尝试入睡"""
-        if self._is_woken_up:
-            logger.info("重置唤醒状态，将重新尝试入睡。")
-            self._is_woken_up = False
-            self._is_preparing_sleep = False # 允许重新进入弹性睡眠判断
-            self._sleep_buffer_end_time = None
-            self._save_sleep_state()
+        """被唤醒后，将状态切换到 WOKEN_UP"""
+        self.sleep_manager.reset_sleep_state_after_wakeup()
 
     def _validate_schedule_with_pydantic(self, schedule_data) -> bool:
         """使用Pydantic验证日程数据格式和完整性"""
@@ -669,22 +433,21 @@ class ScheduleManager:
         if not isinstance(schedule_data, list):
             logger.warning("日程数据不是列表格式")
             return False
-        
+
         for item in schedule_data:
             if not isinstance(item, dict):
                 logger.warning(f"日程项不是字典格式: {item}")
                 return False
-            
-            if 'time_range' not in item or 'activity' not in item:
+
+            if "time_range" not in item or "activity" not in item:
                 logger.warning(f"日程项缺少必要字段 (time_range 或 activity): {item}")
                 return False
-                
-            if not isinstance(item['time_range'], str) or not isinstance(item['activity'], str):
+
+            if not isinstance(item["time_range"], str) or not isinstance(item["activity"], str):
                 logger.warning(f"日程项字段类型不正确: {item}")
                 return False
-        
-        return True
 
+        return True
 
 
 class DailyScheduleGenerationTask(AsyncTask):
@@ -703,15 +466,17 @@ class DailyScheduleGenerationTask(AsyncTask):
                 midnight = datetime.combine(tomorrow, time.min)
                 sleep_seconds = (midnight - now).total_seconds()
 
-                logger.info(f"下一次日程生成任务将在 {sleep_seconds:.2f} 秒后运行 (北京时间 {midnight.strftime('%Y-%m-%d %H:%M:%S')})")
-                
+                logger.info(
+                    f"下一次日程生成任务将在 {sleep_seconds:.2f} 秒后运行 (北京时间 {midnight.strftime('%Y-%m-%d %H:%M:%S')})"
+                )
+
                 # 2. 等待直到零点
                 await asyncio.sleep(sleep_seconds)
 
                 # 3. 执行异步日程生成
                 logger.info("到达每日零点，开始异步生成新的一天日程...")
                 await self.schedule_manager.generate_and_save_schedule()
-                
+
             except asyncio.CancelledError:
                 logger.info("每日日程生成任务被取消。")
                 break
@@ -721,4 +486,4 @@ class DailyScheduleGenerationTask(AsyncTask):
                 await asyncio.sleep(300)
 
 
-schedule_manager = ScheduleManager()
+schedule_manager = ScheduleManager()

src/schedule/sleep_manager.py

@@ -0,0 +1,334 @@
+import asyncio
+import random
+from datetime import datetime, timedelta, date, time
+from enum import Enum, auto
+from typing import Optional, TYPE_CHECKING
+
+from src.common.logger import get_logger
+from src.config.config import global_config
+from src.manager.local_store_manager import local_storage
+from src.plugin_system.apis import send_api, generator_api
+
+if TYPE_CHECKING:
+    from src.chat.chat_loop.wakeup_manager import WakeUpManager
+
+logger = get_logger("sleep_manager")
+
+
+class SleepState(Enum):
+    """睡眠状态枚举"""
+    AWAKE = auto()           # 完全清醒
+    INSOMNIA = auto()        # 失眠（在理论睡眠时间内保持清醒）
+    PREPARING_SLEEP = auto() # 准备入睡（缓冲期）
+    SLEEPING = auto()        # 正在休眠
+    WOKEN_UP = auto()        # 被吵醒
+
+
+class SleepManager:
+    def __init__(self, schedule_manager):
+        self.schedule_manager = schedule_manager
+        self.last_sleep_log_time = 0
+        self.sleep_log_interval = 35  # 日志记录间隔，单位秒
+
+        # --- 统一睡眠状态管理 ---
+        self._current_state: SleepState = SleepState.AWAKE
+        self._sleep_buffer_end_time: Optional[datetime] = None
+        self._total_delayed_minutes_today: int = 0
+        self._last_sleep_check_date: Optional[date] = None
+        self._last_fully_slept_log_time: float = 0
+        self._re_sleep_attempt_time: Optional[datetime] = None # 新增：重新入睡的尝试时间
+        
+        self._load_sleep_state()
+
+    def get_current_sleep_state(self) -> SleepState:
+        """获取当前的睡眠状态"""
+        return self._current_state
+
+    def is_sleeping(self) -> bool:
+        """检查当前是否处于正式休眠状态"""
+        return self._current_state == SleepState.SLEEPING
+
+    async def update_sleep_state(self, wakeup_manager: Optional["WakeUpManager"] = None):
+        """
+        核心状态机：根据当前情况更新睡眠状态
+        """
+        # --- 基础检查 ---
+        if not global_config.sleep_system.enable or not self.schedule_manager.today_schedule:
+            if self._current_state != SleepState.AWAKE:
+                logger.debug("睡眠系统禁用或无日程，强制设为 AWAKE")
+                self._current_state = SleepState.AWAKE
+            return
+
+        now = datetime.now()
+        today = now.date()
+
+        # --- 每日状态重置 ---
+        if self._last_sleep_check_date != today:
+            logger.info(f"新的一天 ({today})，重置睡眠状态为 AWAKE。")
+            self._total_delayed_minutes_today = 0
+            self._current_state = SleepState.AWAKE
+            self._sleep_buffer_end_time = None
+            self._last_sleep_check_date = today
+            self._save_sleep_state()
+
+        # --- 判断当前是否为理论上的睡眠时间 ---
+        is_in_theoretical_sleep, activity = self._is_in_theoretical_sleep_time(now.time())
+
+        # ===================================
+        #  状态机核心逻辑
+        # ===================================
+
+        # 状态：清醒 (AWAKE)
+        if self._current_state == SleepState.AWAKE:
+            if is_in_theoretical_sleep:
+                logger.info(f"进入理论休眠时间 '{activity}'，开始进行睡眠决策...")
+                
+                # --- 合并后的失眠与弹性睡眠决策逻辑 ---
+                sleep_pressure = wakeup_manager.context.sleep_pressure if wakeup_manager else 999
+                pressure_threshold = global_config.sleep_system.flexible_sleep_pressure_threshold
+                
+                # 决策1：因睡眠压力低而延迟入睡（原弹性睡眠）
+                if sleep_pressure < pressure_threshold and self._total_delayed_minutes_today < global_config.sleep_system.max_sleep_delay_minutes:
+                    delay_minutes = 15
+                    self._total_delayed_minutes_today += delay_minutes
+                    self._sleep_buffer_end_time = now + timedelta(minutes=delay_minutes)
+                    self._current_state = SleepState.INSOMNIA
+                    logger.info(f"睡眠压力 ({sleep_pressure:.1f}) 低于阈值 ({pressure_threshold})，进入失眠状态，延迟入睡 {delay_minutes} 分钟。")
+                    
+                    # 发送睡前通知
+                    if global_config.sleep_system.enable_pre_sleep_notification:
+                        asyncio.create_task(self._send_pre_sleep_notification())
+                        
+                # 决策2：进入正常的入睡准备流程
+                else:
+                    buffer_seconds = random.randint(5 * 60, 10 * 60)
+                    self._sleep_buffer_end_time = now + timedelta(seconds=buffer_seconds)
+                    self._current_state = SleepState.PREPARING_SLEEP
+                    logger.info(f"睡眠压力正常或已达今日最大延迟，进入准备入睡状态，将在 {buffer_seconds / 60:.1f} 分钟内入睡。")
+                    
+                    # 发送睡前通知
+                    if global_config.sleep_system.enable_pre_sleep_notification:
+                        asyncio.create_task(self._send_pre_sleep_notification())
+
+                self._save_sleep_state()
+
+        # 状态：失眠 (INSOMNIA)
+        elif self._current_state == SleepState.INSOMNIA:
+            if not is_in_theoretical_sleep:
+                logger.info("已离开理论休眠时间，失眠结束，恢复清醒。")
+                self._current_state = SleepState.AWAKE
+                self._save_sleep_state()
+            elif self._sleep_buffer_end_time and now >= self._sleep_buffer_end_time:
+                logger.info("失眠状态下的延迟时间已过，重新评估是否入睡...")
+                sleep_pressure = wakeup_manager.context.sleep_pressure if wakeup_manager else 999
+                pressure_threshold = global_config.sleep_system.flexible_sleep_pressure_threshold
+
+                if sleep_pressure >= pressure_threshold or self._total_delayed_minutes_today >= global_config.sleep_system.max_sleep_delay_minutes:
+                    logger.info("睡眠压力足够或已达最大延迟，从失眠状态转换到准备入睡。")
+                    buffer_seconds = random.randint(5 * 60, 10 * 60)
+                    self._sleep_buffer_end_time = now + timedelta(seconds=buffer_seconds)
+                    self._current_state = SleepState.PREPARING_SLEEP
+                else:
+                    logger.info(f"睡眠压力({sleep_pressure:.1f})仍然较低，再延迟15分钟。")
+                    delay_minutes = 15
+                    self._total_delayed_minutes_today += delay_minutes
+                    self._sleep_buffer_end_time = now + timedelta(minutes=delay_minutes)
+                
+                self._save_sleep_state()
+
+        # 状态：准备入睡 (PREPARING_SLEEP)
+        elif self._current_state == SleepState.PREPARING_SLEEP:
+            if not is_in_theoretical_sleep:
+                logger.info("准备入睡期间离开理论休眠时间，取消入睡，恢复清醒。")
+                self._current_state = SleepState.AWAKE
+                self._sleep_buffer_end_time = None
+                self._save_sleep_state()
+            elif self._sleep_buffer_end_time and now >= self._sleep_buffer_end_time:
+                logger.info("睡眠缓冲期结束，正式进入休眠状态。")
+                self._current_state = SleepState.SLEEPING
+                self._last_fully_slept_log_time = now.timestamp()
+                self._save_sleep_state()
+
+        # 状态：休眠中 (SLEEPING)
+        elif self._current_state == SleepState.SLEEPING:
+            if not is_in_theoretical_sleep:
+                logger.info("理论休眠时间结束，自然醒来。")
+                self._current_state = SleepState.AWAKE
+                self._save_sleep_state()
+            else:
+                # 记录日志
+                current_timestamp = now.timestamp()
+                if current_timestamp - self.last_sleep_log_time > self.sleep_log_interval:
+                    logger.info(f"当前处于休眠活动 '{activity}' 中。")
+                    self.last_sleep_log_time = current_timestamp
+
+        # 状态：被吵醒 (WOKEN_UP)
+        elif self._current_state == SleepState.WOKEN_UP:
+            if not is_in_theoretical_sleep:
+                logger.info("理论休眠时间结束，被吵醒的状态自动结束。")
+                self._current_state = SleepState.AWAKE
+                self._re_sleep_attempt_time = None
+                self._save_sleep_state()
+            elif self._re_sleep_attempt_time and now >= self._re_sleep_attempt_time:
+                logger.info("被吵醒后经过一段时间，尝试重新入睡...")
+                
+                sleep_pressure = wakeup_manager.context.sleep_pressure if wakeup_manager else 999
+                pressure_threshold = global_config.sleep_system.flexible_sleep_pressure_threshold
+
+                if sleep_pressure >= pressure_threshold:
+                    logger.info("睡眠压力足够，从被吵醒状态转换到准备入睡。")
+                    buffer_seconds = random.randint(3 * 60, 8 * 60) # 重新入睡的缓冲期可以短一些
+                    self._sleep_buffer_end_time = now + timedelta(seconds=buffer_seconds)
+                    self._current_state = SleepState.PREPARING_SLEEP
+                    self._re_sleep_attempt_time = None
+                else:
+                    delay_minutes = 15
+                    self._re_sleep_attempt_time = now + timedelta(minutes=delay_minutes)
+                    logger.info(f"睡眠压力({sleep_pressure:.1f})仍然较低，暂时保持清醒，在 {delay_minutes} 分钟后再次尝试。")
+                
+                self._save_sleep_state()
+
+    def reset_sleep_state_after_wakeup(self):
+        """被唤醒后，将状态切换到 WOKEN_UP"""
+        if self._current_state in [SleepState.PREPARING_SLEEP, SleepState.SLEEPING, SleepState.INSOMNIA]:
+            logger.info("被唤醒，进入 WOKEN_UP 状态！")
+            self._current_state = SleepState.WOKEN_UP
+            self._sleep_buffer_end_time = None
+            
+            # 设置一个延迟，之后再尝试重新入睡
+            re_sleep_delay_minutes = getattr(global_config.sleep_system, 're_sleep_delay_minutes', 10)
+            self._re_sleep_attempt_time = datetime.now() + timedelta(minutes=re_sleep_delay_minutes)
+            logger.info(f"将在 {re_sleep_delay_minutes} 分钟后尝试重新入睡。")
+            
+            self._save_sleep_state()
+
+    def _is_in_theoretical_sleep_time(self, now_time: time) -> tuple[bool, Optional[str]]:
+        """检查当前时间是否落在日程表的任何一个睡眠活动中"""
+        sleep_keywords = ["休眠", "睡觉", "梦乡"]
+        if self.schedule_manager.today_schedule:
+            for event in self.schedule_manager.today_schedule:
+                try:
+                    activity = event.get("activity", "").strip()
+                    time_range = event.get("time_range")
+
+                    if not activity or not time_range:
+                        continue
+
+                    if any(keyword in activity for keyword in sleep_keywords):
+                        start_str, end_str = time_range.split('-')
+                        start_time = datetime.strptime(start_str.strip(), "%H:%M").time()
+                        end_time = datetime.strptime(end_str.strip(), "%H:%M").time()
+
+                        if start_time <= end_time:  # 同一天
+                            if start_time <= now_time < end_time:
+                                return True, activity
+                        else:  # 跨天
+                            if now_time >= start_time or now_time < end_time:
+                                return True, activity
+                except (ValueError, KeyError, AttributeError) as e:
+                    logger.warning(f"解析日程事件时出错: {event}, 错误: {e}")
+                    continue
+            
+        return False, None
+
+    async def _send_pre_sleep_notification(self):
+        """异步生成并发送睡前通知"""
+        try:
+            groups = global_config.sleep_system.pre_sleep_notification_groups
+            prompt = global_config.sleep_system.pre_sleep_prompt
+
+            if not groups:
+                logger.info("未配置睡前通知的群组，跳过发送。")
+                return
+            
+            if not prompt:
+                logger.warning("睡前通知的prompt为空，跳过发送。")
+                return
+
+            # 为防止消息风暴，稍微延迟一下
+            await asyncio.sleep(random.uniform(5, 15))
+
+            for group_id_str in groups:
+                try:
+                    # 格式 "platform:group_id"
+                    parts = group_id_str.split(":")
+                    if len(parts) != 2:
+                        logger.warning(f"无效的群组ID格式: {group_id_str}")
+                        continue
+                    
+                    platform, group_id = parts
+                    
+                    # 使用与 ChatStream.get_stream_id 相同的逻辑生成 stream_id
+                    import hashlib
+                    key = "_".join([platform, group_id])
+                    stream_id = hashlib.md5(key.encode()).hexdigest()
+
+                    logger.info(f"正在为群组 {group_id_str} (Stream ID: {stream_id}) 生成睡前消息...")
+                    
+                    # 调用 generator_api 生成回复
+                    success, reply_set, _ = await generator_api.generate_reply(
+                        chat_id=stream_id,
+                        extra_info=prompt,
+                        request_type="schedule.pre_sleep_notification"
+                    )
+
+                    if success and reply_set:
+                        # 提取文本内容并发送
+                        reply_text = "".join([content for msg_type, content in reply_set if msg_type == "text"])
+                        if reply_text:
+                            logger.info(f"向群组 {group_id_str} 发送睡前消息: {reply_text}")
+                            await send_api.text_to_stream(text=reply_text, stream_id=stream_id)
+                        else:
+                            logger.warning(f"为群组 {group_id_str} 生成的回复内容为空。")
+                    else:
+                        logger.error(f"为群组 {group_id_str} 生成睡前消息失败。")
+
+                    await asyncio.sleep(random.uniform(2, 5)) # 避免发送过快
+
+                except Exception as e:
+                    logger.error(f"向群组 {group_id_str} 发送睡前消息失败: {e}")
+
+        except Exception as e:
+            logger.error(f"发送睡前通知任务失败: {e}")
+
+    def _save_sleep_state(self):
+        """将当前睡眠状态保存到本地存储"""
+        try:
+            state = {
+                "current_state": self._current_state.name,
+                "sleep_buffer_end_time_ts": self._sleep_buffer_end_time.timestamp() if self._sleep_buffer_end_time else None,
+                "total_delayed_minutes_today": self._total_delayed_minutes_today,
+                "last_sleep_check_date_str": self._last_sleep_check_date.isoformat() if self._last_sleep_check_date else None,
+                "re_sleep_attempt_time_ts": self._re_sleep_attempt_time.timestamp() if self._re_sleep_attempt_time else None,
+            }
+            local_storage["schedule_sleep_state"] = state
+            logger.debug(f"已保存睡眠状态: {state}")
+        except Exception as e:
+            logger.error(f"保存睡眠状态失败: {e}")
+
+    def _load_sleep_state(self):
+        """从本地存储加载睡眠状态"""
+        try:
+            state = local_storage["schedule_sleep_state"]
+            if state and isinstance(state, dict):
+                state_name = state.get("current_state")
+                if state_name and hasattr(SleepState, state_name):
+                    self._current_state = SleepState[state_name]
+
+                end_time_ts = state.get("sleep_buffer_end_time_ts")
+                if end_time_ts:
+                    self._sleep_buffer_end_time = datetime.fromtimestamp(end_time_ts)
+                
+                re_sleep_ts = state.get("re_sleep_attempt_time_ts")
+                if re_sleep_ts:
+                    self._re_sleep_attempt_time = datetime.fromtimestamp(re_sleep_ts)
+
+                self._total_delayed_minutes_today = state.get("total_delayed_minutes_today", 0)
+                
+                date_str = state.get("last_sleep_check_date_str")
+                if date_str:
+                    self._last_sleep_check_date = datetime.fromisoformat(date_str).date()
+
+                logger.info(f"成功从本地存储加载睡眠状态: {state}")
+        except Exception as e:
+            logger.warning(f"加载睡眠状态失败，将使用默认值: {e}")