import logging
import random
import string

from django.contrib.auth import get_user_model
from django.db import transaction
from django.utils import timezone
from django_redis import get_redis_connection
from drf_spectacular.utils import extend_schema_serializer
from rest_framework import serializers

from core.serializers.base import BaseModelSerializer

from apps.users.tasks import send_verification_sms
from apps.users.utils import record_login_attempt

User = get_user_model()

logger = logging.getLogger(__name__)


class UserProfilePictureSerializer(BaseModelSerializer):
    class Meta:
        model = User
        fields = BaseModelSerializer.Meta.fields + ("profile_picture",)


class UserListSerializer(BaseModelSerializer):
    full_name = serializers.CharField(read_only=True)

    class Meta:
        model = User
        fields = BaseModelSerializer.Meta.fields + (
            "mobile",
            "full_name",
            "profile_picture",
        )


class RegisterSerializer(serializers.Serializer):
    mobile = serializers.CharField(max_length=11)
    code = serializers.CharField(max_length=6)
    password = serializers.CharField(write_only=True)
    re_password = serializers.CharField(write_only=True)
    first_name = serializers.CharField(max_length=100, required=False, allow_blank=True)
    last_name = serializers.CharField(max_length=100, required=False, allow_blank=True)

    def validate(self, data):
        mobile = data.get("mobile", "")
        password = data.get("password", "")
        re_password = data.get("re_password", "")

        if not (mobile.isdigit() and len(mobile) == 11):
            raise serializers.ValidationError({"mobile": "فرمت شماره موبایل نادرست است."})

        if password != re_password:
            raise serializers.ValidationError({"password": "رمز عبور مطابقت ندارد."})

        return data


@extend_schema_serializer(component_name="UsersSendOTP")
class SendOTPSerializer(serializers.Serializer):
    mobile = serializers.CharField(max_length=11)
    mode = serializers.ChoiceField(choices=["register", "login", "forget_password"])

    def validate_mobile(self, value):
        """
        Normalize and validate Iranian mobile numbers (example: 09XXXXXXXXX).
        """
        if not value.isdigit() or len(value) != 11 or not value.startswith("09"):
            raise serializers.ValidationError("شماره موبایل معتبر نیست.")
        return value


@extend_schema_serializer(component_name="UsersLoginOtp")
class LoginOtpSerializer(serializers.Serializer):
    mobile = serializers.CharField(max_length=11)
    code = serializers.CharField(max_length=6)

    def validate_mobile(self, value):
        if not (value.isdigit() and len(value) == 11):
            raise serializers.ValidationError("فرمت شماره موبایل نادرست است.")
        return value


class LoginSerializer(serializers.Serializer):
    mobile = serializers.CharField(max_length=11)
    password = serializers.CharField(write_only=True)

    def validate_mobile(self, value):
        if not (value.isdigit() and len(value) == 11):
            raise serializers.ValidationError("فرمت شماره موبایل نادرست است.")
        return value


class ResetPasswordSerializer(serializers.Serializer):
    mobile = serializers.CharField(max_length=11)
    code = serializers.CharField(max_length=6)
    password = serializers.CharField(write_only=True)
    re_password = serializers.CharField(write_only=True)

    def validate(self, data):
        if data.get("password") != data.get("re_password"):
            raise serializers.ValidationError({"password": "رمز عبور مطابقت ندارد."})
        return data


class ChangePasswordSerializer(serializers.Serializer):
    old_password = serializers.CharField(required=True, write_only=True)
    new_password = serializers.CharField(required=True, write_only=True)
    re_password = serializers.CharField(required=True, write_only=True)

    def validate(self, data):
        if data.get("new_password") != data.get("re_password"):
            raise serializers.ValidationError({"new_password": "رمز عبور جدید و تکرار آن مطابقت ندارند."})
        return data


class LogoutSerializer(serializers.Serializer):
    refresh = serializers.CharField()


class TokenPairSerializer(serializers.Serializer):
    access = serializers.CharField()
    refresh = serializers.CharField()


class RegisterWithPasswordSerializer(serializers.Serializer):
    mobile = serializers.CharField()
    password = serializers.CharField()


class UserProfileSerializer(BaseModelSerializer):
    full_name = serializers.ReadOnlyField()
    age = serializers.ReadOnlyField()

    class Meta:
        model = User
        fields = BaseModelSerializer.Meta.fields + (
            "mobile", "email", "first_name", "last_name", 
            "description", "profile_picture", "birth_date", 
            "is_verified", "full_name", "age"
        )
        read_only_fields = BaseModelSerializer.Meta.fields + ("mobile", "is_verified")


class UserSearchSerializer(serializers.ModelSerializer):
    class Meta:
        model = User
        fields = (
            'id', 
            'first_name', 
            'last_name', 
            'mobile', 
            'profile_picture',
        )