from apscheduler.schedulers.asyncio import AsyncIOScheduler
from apscheduler.triggers.interval import IntervalTrigger
from banban.service.pending_voice_message import pending_voice_message_service
from utils.logger import session_logger as logger

class TaskScheduler:
    _instance = None

    def __init__(self):
        self._scheduler = AsyncIOScheduler(timezone="Asia/Shanghai")

    @classmethod
    def get_instance(cls) -> "TaskScheduler":
        if cls._instance is None:
            cls._instance = cls()
        return cls._instance

    @classmethod
    def reset_instance(cls):
        if cls._instance is not None:
            cls._instance.shutdown()
            cls._instance = None

    def start(self):
        if not self._scheduler.running:
            self._scheduler.start()
            logger.system_info("", "定时任务调度器已启动")

    def shutdown(self):
        if self._scheduler.running:
            self._scheduler.shutdown(wait=False)
            logger.system_info("", "定时任务调度器已关闭")

    async def _execute_task(self):
        try:
            pending_devices = await pending_voice_message_service.list_devices_with_pending()
            device_ids = list(dict.fromkeys(pending_devices))
            if not device_ids:
                logger.warning("", "", "无待收听留言，跳过本次执行")
                return
            for device_id in device_ids:
                sent = await pending_voice_message_service.notify_device_pending_message(
                    target_device_id=device_id,
                )
                if sent:
                    logger.info(device_id, "", f"[定时任务] 发送待收听留言提醒成功: device={device_id}")
                else:
                    logger.info(device_id, "", f"[定时任务] 待收听留言提醒未发送: device={device_id}")
        except Exception as e:
            logger.error("", "", f"[定时任务] 执行失败: {e}")

    def add_interval_task(self, interval_seconds: int = 600):
        task_id = "Voice_Message_Task"
        task_name = "Voice_Message_Task"
        job = self._scheduler.add_job(
            self._execute_task,
            trigger=IntervalTrigger(seconds=interval_seconds),
            id=task_id,
            name=task_name,
            replace_existing=True,
        )

        logger.system_info("", f"[定时任务] 已添加: {task_id} | 间隔={interval_seconds}s | 下次执行={job.next_run_time}")