from typing import Any
import time

from app.config import settings
from app.mqtt.client import MqttService
from app.services.course_service import CourseService

SYNC_COURSE_TASK_ACTION = "SYNC_COURSE_TASK"


class InitTaskDispatcher:
    def __init__(self, mqtt: MqttService, course_service: CourseService) -> None:
        self.mqtt = mqtt
        self.course_service = course_service

    def dispatch_device_init(self, device_id: str) -> list[dict[str, Any]]:
        tasklist = self.course_service.get_tasklist()
        commands: list[dict[str, Any]] = []

        for task in tasklist.get("tasks", []):
            action = task["action"]
            payload = dict(task.get("payload", {}))
            channel = task.get("channel", "ctl")

            if action == "SET_CLOCK" and payload.get("ts") == 0:
                payload["ts"] = int(time.time())

            cmd = self._send(device_id, action, payload, channel, task.get("ttl", 60))
            commands.append(cmd)

        return commands

    def dispatch_base_courses_sync(self, device_id: str) -> list[dict[str, Any]]:
        commands: list[dict[str, Any]] = []
        for payload in self.course_service.build_base_course_sync_payloads(settings.base_url):
            cmd = self.mqtt.publish_task(device_id, SYNC_COURSE_TASK_ACTION, payload, ttl=7200)
            commands.append(cmd)
        return commands

    def dispatch_course_init(self, device_id: str, course_id: str) -> list[dict[str, Any]]:
        payload = self.course_service.build_course_task_sync_payload(course_id, settings.base_url)
        cmd = self.mqtt.publish_task(device_id, SYNC_COURSE_TASK_ACTION, payload, ttl=7200)
        return [cmd]

    def _send(self, device_id: str, action: str, payload: dict[str, Any], channel: str, ttl: int) -> dict[str, Any]:
        if channel == "task":
            return self.mqtt.publish_task(device_id, action, payload, ttl)
        return self.mqtt.publish_ctl(device_id, action, payload, ttl)