Backend/farm_alerts/views.py

from drf_spectacular.utils import OpenApiExample, extend_schema
from rest_framework import serializers, status
from rest_framework.permissions import IsAuthenticated
from rest_framework.response import Response
from rest_framework.views import APIView

from config.swagger import code_response
from external_api_adapter import request as external_api_request
from farm_hub.models import FarmHub

from .serializers import AlertTrackerAIResponseSerializer, FarmAlertsTrackerRequestSerializer
from .services import AlertService, build_tracker_context, build_tracker_response


class FarmAlertsBaseView(APIView):
    permission_classes = [IsAuthenticated]

    @staticmethod
    def _extract_result(adapter_data):
        if not isinstance(adapter_data, dict):
            return {}

        data = adapter_data.get("data")
        if isinstance(data, dict) and isinstance(data.get("result"), dict):
            return data["result"]
        if isinstance(data, dict):
            return data

        result = adapter_data.get("result")
        if isinstance(result, dict):
            return result

        return adapter_data

    @staticmethod
    def _error_response(adapter_response):
        response_data = (
            adapter_response.data
            if isinstance(adapter_response.data, dict)
            else {"message": str(adapter_response.data)}
        )
        return Response(
            {"code": adapter_response.status_code, "msg": "error", "data": response_data},
            status=adapter_response.status_code,
        )

    @staticmethod
    def _get_farm(request, farm_uuid):
        if not farm_uuid:
            raise serializers.ValidationError({"farm_uuid": ["This field is required."]})
        try:
            return FarmHub.objects.get(farm_uuid=farm_uuid, owner=request.user)
        except FarmHub.DoesNotExist as exc:
            raise serializers.ValidationError({"farm_uuid": ["Farm not found."]}) from exc


class AlertTrackerView(FarmAlertsBaseView):
    @extend_schema(
        tags=["Farm Alerts"],
        request=FarmAlertsTrackerRequestSerializer,
        examples=[
            OpenApiExample(
                "Tracker Request",
                value={
                    "farm_uuid": "11111111-1111-1111-1111-111111111111",
                    "alerts": [
                        {
                            "alert_id": "soil-moisture-001",
                            "level": "warning",
                            "title": "افت رطوبت خاک",
                            "message": "رطوبت خاک کمتر از حد مطلوب گزارش شده است.",
                            "suggested_action": "آبیاری اصلاحی بررسی شود.",
                            "source_metric_type": "moisture",
                        }
                    ],
                },
                request_only=True,
            )
        ],
        responses={200: code_response("FarmAlertsTrackerResponse", data=AlertTrackerAIResponseSerializer())},
    )
    def post(self, request):
        request_serializer = FarmAlertsTrackerRequestSerializer(data=request.data)
        request_serializer.is_valid(raise_exception=True)

        farm = self._get_farm(request, request_serializer.validated_data["farm_uuid"])
        incoming_alerts = request_serializer.validated_data.get("alerts", [])
        AlertService.persist_incoming_alerts(farm=farm, alerts=incoming_alerts)

        tracker_payload = build_tracker_context(farm=farm, alerts=incoming_alerts)
        adapter_response = external_api_request(
            "ai",
            "/api/farm-alerts/tracker/",
            method="POST",
            payload=tracker_payload,
        )
        if adapter_response.status_code >= 400:
            return self._error_response(adapter_response)

        payload = self._extract_result(adapter_response.data)
        response_data = build_tracker_response(farm=farm, adapter_payload=payload)
        serializer = AlertTrackerAIResponseSerializer(instance=response_data)
        return Response({"code": 200, "msg": "success", "data": serializer.data}, status=status.HTTP_200_OK)