UPDATE

farm_alerts/views.py

@@ -1,13 +1,20 @@
-from rest_framework import status
+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 AlertTimelineSerializer, AlertTrackerSerializer
+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):
@@ -37,39 +44,60 @@ class FarmAlertsBaseView(APIView):
             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=request.data,
+            payload=tracker_payload,
         )
         if adapter_response.status_code >= 400:
             return self._error_response(adapter_response)
 
         payload = self._extract_result(adapter_response.data)
-        serializer = AlertTrackerSerializer(data=payload)
-        serializer.is_valid(raise_exception=True)
-        return Response({"code": 200, "msg": "success", "data": serializer.validated_data}, status=status.HTTP_200_OK)
-
-
-class AlertTimelineView(FarmAlertsBaseView):
-    def post(self, request):
-        adapter_response = external_api_request(
-            "ai",
-            "/api/farm-alerts/timeline/",
-            method="POST",
-            payload=request.data,
-        )
-        if adapter_response.status_code >= 400:
-            return self._error_response(adapter_response)
-
-        payload = self._extract_result(adapter_response.data)
-        serializer = AlertTimelineSerializer(data=payload)
-        serializer.is_valid(raise_exception=True)
-        return Response(
-            {"code": 200, "msg": "success", "data": serializer.validated_data},
-            status=status.HTTP_200_OK,
-        )
+        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)