from uuid import UUID

from celery.result import AsyncResult
from drf_spectacular.utils import OpenApiResponse, extend_schema, inline_serializer
from rest_framework import serializers as drf_serializers
from rest_framework import status
from rest_framework.response import Response
from rest_framework.views import APIView

from .tasks import generate_dashboard_data_task


class DashboardDataGenerateView(APIView):
    @extend_schema(
        tags=["Dashboard Data"],
        summary="Generate dashboard data",
        request=inline_serializer(
            name="DashboardDataGenerateRequest",
            fields={
                "sensor_id": drf_serializers.UUIDField(required=False),
                "snesor_id": drf_serializers.UUIDField(required=False),
            },
        ),
        responses={
            202: inline_serializer(
                name="DashboardDataGenerateResponse",
                fields={
                    "code": drf_serializers.IntegerField(),
                    "msg": drf_serializers.CharField(),
                    "data": inline_serializer(
                        name="DashboardDataGenerateResponseData",
                        fields={
                            "task_id": drf_serializers.CharField(),
                            "status_url": drf_serializers.CharField(),
                        },
                    ),
                },
            ),
            400: OpenApiResponse(description="Invalid input"),
        },
    )
    def post(self, request):
        sensor_id = request.data.get("sensor_id") or request.data.get("snesor_id")
        if not sensor_id:
            return Response(
                {"code": 400, "msg": "پارامتر sensor_id الزامی است.", "data": None},
                status=status.HTTP_400_BAD_REQUEST,
            )
        try:
            sensor_id = str(UUID(str(sensor_id)))
        except (TypeError, ValueError):
            return Response(
                {"code": 400, "msg": "sensor_id باید UUID معتبر باشد.", "data": None},
                status=status.HTTP_400_BAD_REQUEST,
            )

        task = generate_dashboard_data_task.delay(sensor_id)
        return Response(
            {
                "code": 202,
                "msg": "dashboard task queued",
                "data": {
                    "task_id": task.id,
                    "status_url": f"/api/dashboard-data/{task.id}/status/",
                },
            },
            status=status.HTTP_202_ACCEPTED,
        )


class DashboardDataStatusView(APIView):
    @extend_schema(
        tags=["Dashboard Data"],
        summary="Dashboard task status",
    )
    def get(self, request, task_id):
        result = AsyncResult(task_id)
        data = {"task_id": task_id, "status": result.state}
        if result.state == "PENDING":
            data["message"] = "تسک در صف یا یافت نشد."
        elif result.state == "PROGRESS":
            data["progress"] = result.info
        elif result.state == "SUCCESS":
            data["result"] = result.result
        elif result.state == "FAILURE":
            data["error"] = str(result.result)

        return Response(
            {"code": 200, "msg": "success", "data": data},
            status=status.HTTP_200_OK,
        )