from django.shortcuts import get_object_or_404 from drf_spectacular.utils import extend_schema from rest_framework.permissions import IsAuthenticated from rest_framework.response import Response from rest_framework.throttling import ScopedRateThrottle from rest_framework.views import APIView from .models import BugReport, Ticket, TicketMessage from .serializers import ( BugReportCreateRequestSerializer, BugReportCreateResponseSerializer, TicketCreateRequestSerializer, TicketCreateResponseSerializer, TicketDetailResponseSerializer, TicketMessageCreateRequestSerializer, TicketMessageCreateResponseSerializer, ) class GuestAwareSupportThrottle(ScopedRateThrottle): """ ⚠️ اضافه شده: همون مشکلِ GuestAwareAIThrottle (ai_tutor/views.py)، GuestAwareFamilyThrottle (family/views.py) و GuestAwareDiscountThrottle/ GuestAwarePurchaseThrottle (discount/views.py، subscription/views.py) اینجا هم هست. ScopedRateThrottle استاندارد برای درخواستِ احرازهویت‌شده از request.user.pk به‌عنوانِ کلیدِ throttle استفاده می‌کنه؛ چون GuestSessionView با هر deviceId جدید یک کاربرِ مهمانِ تازه (pk جدید) می‌سازه، یک مهاجم می‌تونست با ساختنِ پی‌درپیِ حساب‌های مهمان، برای هر حساب یک bucket تازه از throttle_scope 'support_create' بگیره. چون هر سه Endpointِ این اپ (BugReportCreateView، TicketCreateView، TicketMessageCreateView) این scope رو به‌اشتراک می‌ذارن و هرکدوم واقعاً یک رکورد در دیتابیس می‌سازن (BugReport، Ticket، یا TicketMessage)، چرخشِ حساب مهمان می‌تونست محدودیتِ ۳/ساعتِ در نظر گرفته‌شده رو چند برابر کنه و باعثِ spam بشه. این کلاسِ throttle یک‌بار اینجا تعریف شده و بینِ هر سه Endpoint به‌اشتراک گذاشته می‌شه (به‌جایِ تکرارِ یک کلاسِ مشابه برای هرکدوم)، دقیقاً چون هر سه از یک throttle_scope مشترک استفاده می‌کنن. فقط برای کاربرهای مهمان (is_guest=True) کلیدِ cache از pk کاربر به IP کلاینت (get_ident) تغییر می‌کنه، پس همه‌ی حساب‌های مهمانِ ساخته‌شده از یک IP یک bucket مشترک دارن و چرخشِ حساب دیگه bucket تازه نمی‌ده. رفتارِ throttle برای کاربرهای غیرمهمان (ثبت‌نام‌شده) دقیقاً مثلِ قبل (کلید بر اساسِ pk) باقی می‌مونه؛ منطقِ ساختِ رکورد/ownership/serializer در هر سه ویو دست‌نخورده‌ست. """ def get_cache_key(self, request, view): user = request.user if user and user.is_authenticated and getattr(user, 'is_guest', False): ident = f'guest-ip:{self.get_ident(request)}' elif user and user.is_authenticated: ident = user.pk else: ident = self.get_ident(request) return self.cache_format % {'scope': self.scope, 'ident': ident} class BugReportCreateView(APIView): """POST /api/support/bug-reports — ثبت گزارش باگ""" permission_classes = [IsAuthenticated] # بدونِ idempotency/throttle، یک کاربر می‌تونست تعدادِ نامحدودی BugReport # بسازه؛ GuestAwareSupportThrottle برای مهمان‌ها کلید رو بر اساسِ IP # می‌سازه تا چرخشِ حساب bucket تازه نده. throttle_scope = 'support_create' throttle_classes = [GuestAwareSupportThrottle] @extend_schema( request=BugReportCreateRequestSerializer, responses=BugReportCreateResponseSerializer, tags=['Support (support)'], ) def post(self, request): serializer = BugReportCreateRequestSerializer(data=request.data) serializer.is_valid(raise_exception=True) v = serializer.validated_data bug = BugReport.objects.create( user_id=request.user.id, title=v['title'], description=v['description'], screenshot_url=v.get('screenshotUrl', ''), ) data = {'bugReportId': str(bug.pk), 'status': bug.status} return Response(BugReportCreateResponseSerializer(data).data) class TicketCreateView(APIView): """POST /api/support/tickets — ایجاد تیکت""" permission_classes = [IsAuthenticated] # مثلِ BugReportCreateView: هیچ idempotency ای نیست، پس این endpoint هم # همون scope=support_create رو به‌اشتراک می‌ذاره تا محدودیتِ ساختِ # رکوردهای پشتیبانی، مستقل از این‌که کدوم نوع رکورده، رعایت بشه. # ⚠️ اصلاح شده: از همون GuestAwareSupportThrottleِ مشترک استفاده می‌شه # تا چرخشِ حساب مهمان bucket تازه ندهد. throttle_scope = 'support_create' throttle_classes = [GuestAwareSupportThrottle] @extend_schema( request=TicketCreateRequestSerializer, responses=TicketCreateResponseSerializer, tags=['Support (support)'], ) def post(self, request): serializer = TicketCreateRequestSerializer(data=request.data) serializer.is_valid(raise_exception=True) v = serializer.validated_data ticket = Ticket.objects.create( user_id=request.user.id, subject=v['subject'], category=v['category'], priority=v.get('priority', 'normal'), ) data = {'ticketId': str(ticket.pk), 'status': ticket.status} return Response(TicketCreateResponseSerializer(data).data) class TicketDetailView(APIView): """ GET /api/support/tickets/{id} — جزئیات و پیام‌های تیکت فقط تیکت متعلق به همون کاربر برمی‌گرده (فیلتر با user_id، دقیقاً همون الگوی محدودسازی به مالک که در progress/views.py برای UserStats و در ai_tutor/views.py برای Conversation استفاده شده). """ permission_classes = [IsAuthenticated] @extend_schema(responses=TicketDetailResponseSerializer, tags=['Support (support)']) def get(self, request, ticket_id): ticket = get_object_or_404(Ticket, pk=ticket_id, user_id=request.user.id) messages = [ {'id': str(m.pk), 'sender': m.sender, 'text': m.text} for m in ticket.messages.all() ] data = { 'ticketId': str(ticket.pk), 'subject': ticket.subject, 'status': ticket.status, 'messages': messages, } return Response(TicketDetailResponseSerializer(data).data) class TicketMessageCreateView(APIView): """ POST /api/support/tickets/{id}/messages — ارسال پیام در تیکت این Endpoint سمت کاربره، پس sender همیشه 'user' ثبت می‌شه. پاسخ پشتیبانی (sender='support') از پنل ادمین ثبت می‌شه و طبق سند معماری از همون‌جا رویداد TicketReplied برای Notifications ساطع می‌شه (پیاده‌سازی در support/admin.py، نه اینجا). """ permission_classes = [IsAuthenticated] # مثلِ دو Endpoint بالا: بدونِ throttle، یک کاربر می‌تونست تعدادِ # نامحدودی پیام در یک تیکت بسازه. # ⚠️ اصلاح شده: از همون GuestAwareSupportThrottleِ مشترک استفاده می‌شه # تا چرخشِ حساب مهمان bucket تازه ندهد. throttle_scope = 'support_create' throttle_classes = [GuestAwareSupportThrottle] @extend_schema( request=TicketMessageCreateRequestSerializer, responses=TicketMessageCreateResponseSerializer, tags=['Support (support)'], ) def post(self, request, ticket_id): ticket = get_object_or_404(Ticket, pk=ticket_id, user_id=request.user.id) serializer = TicketMessageCreateRequestSerializer(data=request.data) serializer.is_valid(raise_exception=True) v = serializer.validated_data message = TicketMessage.objects.create(ticket=ticket, sender='user', text=v['text']) data = {'messageId': str(message.pk), 'success': True} return Response(TicketMessageCreateResponseSerializer(data).data)