Logic/Modules/Backend/auth/views.py

import secrets

from django.conf import settings
from django.contrib.auth import authenticate
from django.core.cache import cache
from django.core.signing import BadSignature, SignatureExpired, TimestampSigner
from django.db import IntegrityError
from rest_framework import serializers, status
from rest_framework.permissions import AllowAny
from rest_framework.response import Response
from rest_framework.views import APIView
from drf_spectacular.utils import extend_schema, extend_schema_view
from rest_framework_simplejwt.tokens import AccessToken

from account.models import User
from config.swagger import code_response
from .serializers import (
    AuthUserSerializer,
    LoginSerializer,
    RegisterSerializer,
    RequestOTPSerializer,
    VerifyOTPSerializer,
)
from .sms_service import send_otp_sms


OTP_TTL_SECONDS = 300
OTP_SIGNER = TimestampSigner(salt="auth.otp")


def _auth_user_to_data(user):
    if user is None or not getattr(user, "pk", None):
        return None
    return {
        "id": user.id,
        "username": getattr(user, "username", "") or "",
        "email": getattr(user, "email", "") or "",
        "first_name": getattr(user, "first_name", "") or "",
        "last_name": getattr(user, "last_name", "") or "",
        "phone_number": getattr(user, "phone_number", "") or "",
    }


def _issue_token(user):
    return str(AccessToken.for_user(user))


@extend_schema_view(
    post=extend_schema(
        tags=["Authentication"],
        request=RegisterSerializer,
        responses={
            201: code_response("RegisterResponse", data=AuthUserSerializer(), token=True),
            400: code_response("RegisterErrorResponse"),
        },
    ),
)
class RegisterView(APIView):
    permission_classes = [AllowAny]

    def post(self, request):
        serializer = RegisterSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        data = serializer.validated_data

        try:
            user = User.objects.create_user(
                username=data["username"],
                email=data["email"],
                phone_number=data["phone_number"],
                password=data["password"],
                first_name=data.get("first_name", ""),
                last_name=data.get("last_name", ""),
            )
        except IntegrityError as exc:
            msg = str(exc).lower()
            if "username" in msg:
                detail = "A user with this username already exists."
            elif "email" in msg:
                detail = "A user with this email already exists."
            elif "phone_number" in msg:
                detail = "A user with this phone number already exists."
            else:
                detail = "A user with these credentials already exists."
            return Response({"code": 400, "msg": detail}, status=status.HTTP_400_BAD_REQUEST)

        return Response(
            {
                "code": 201,
                "msg": "success",
                "data": _auth_user_to_data(user),
                "token": _issue_token(user),
            },
            status=status.HTTP_201_CREATED,
        )


@extend_schema_view(
    post=extend_schema(
        tags=["Authentication"],
        request=LoginSerializer,
        responses={
            200: code_response("LoginResponse", data=AuthUserSerializer(), token=True),
            401: code_response("LoginErrorResponse"),
        },
    ),
)
class LoginView(APIView):
    permission_classes = [AllowAny]

    def post(self, request):
        serializer = LoginSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)

        identifier = serializer.validated_data["identifier"]
        password = serializer.validated_data["password"]
        user = authenticate(request, username=identifier, password=password)

        if user is None:
            return Response({"code": 401, "msg": "Invalid credentials."}, status=status.HTTP_401_UNAUTHORIZED)

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


@extend_schema_view(
    post=extend_schema(
        tags=["Authentication"],
        request=RequestOTPSerializer,
        responses={
            200: code_response(
                "RequestOtpResponse",
                extra_fields={
                    "token": serializers.CharField(),
                    "sms_warning": serializers.CharField(required=False),
                    "debug_otp": serializers.CharField(required=False),
                },
            ),
            400: code_response("RequestOtpErrorResponse"),
        },
    ),
)
class AuthenticationView(APIView):
    permission_classes = [AllowAny]

    def post(self, request):
        if "verify-otp" in request.path:
            return self._verify_otp(request)
        return self._request_otp(request)

    def _request_otp(self, request):
        serializer = RequestOTPSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)

        phone_number = serializer.validated_data["phone_number"].strip()
        otp_code = f"{secrets.randbelow(1_000_000):06d}"
        cache.set(f"otp_code:{phone_number}", otp_code, timeout=OTP_TTL_SECONDS)
        otp_token = OTP_SIGNER.sign(phone_number)
        sms_sent = send_otp_sms(phone_number, otp_code)

        payload = {"code": 200, "msg": "success", "token": otp_token}
        if not sms_sent:
            payload["sms_warning"] = "SMS delivery failed; OTP stored server-side."
        if settings.DEBUG:
            payload["debug_otp"] = otp_code
        return Response(payload, status=status.HTTP_200_OK)

    def _verify_otp(self, request):
        serializer = VerifyOTPSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)

        token = serializer.validated_data["token"]
        otp_code = serializer.validated_data["otp_code"].strip()

        try:
            phone_number = OTP_SIGNER.unsign(token, max_age=OTP_TTL_SECONDS)
        except (BadSignature, SignatureExpired):
            return Response({"code": 400, "msg": "Token is invalid or expired."}, status=status.HTTP_400_BAD_REQUEST)

        cached_otp = cache.get(f"otp_code:{phone_number}")
        if cached_otp is None or cached_otp != otp_code:
            return Response({"code": 400, "msg": "OTP code is invalid or expired."}, status=status.HTTP_400_BAD_REQUEST)

        cache.delete(f"otp_code:{phone_number}")
        user, created = User.objects.get_or_create(
            phone_number=phone_number,
            defaults={
                "username": phone_number,
                "email": f"{phone_number}@otp.local",
            },
        )

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