"""Authentication-related API schemas."""

from ninja import Schema, ModelSchema
from typing import Optional

from users.models import User


class UserRegistrationSchema(Schema):
    username: str
    email: str
    password: str
    first_name: Optional[str] = None
    last_name: Optional[str] = None
    university: Optional[str] = None
    student_id: Optional[str] = None
    year_of_study: Optional[int] = None
    major: Optional[str] = None

class UserLoginSchema(Schema):
    email: str
    password: str

class UserProfileSchema(ModelSchema):
    profile_picture: Optional[str] = None
    student_id: Optional[str] = None
    major: Optional[str] = None
    university: Optional[str] = None

    class Meta:
        model = User
        fields = [
            'id',
            'username',
            'email',
            'first_name',
            'last_name',
            'student_id',
            'year_of_study',
            'major',
            'university',
            'bio',
            'date_joined',
        'is_email_verified',
        'is_active',
        'is_staff',
        'is_superuser',
        'is_deleted',
        'deleted_at',
        ]

    @staticmethod
    def resolve_major(obj):
        return obj.get_major_display()

    @staticmethod
    def resolve_university(obj):
        return obj.get_university_display()

    @staticmethod
    def resolve_profile_picture(obj, context):
        """
        Resolves the absolute URL for the profile picture.
        `context` contains the request object, which is needed for build_absolute_uri.
        """
        request = context['request']
        if obj.profile_picture and hasattr(obj.profile_picture, 'url'):
            return request.build_absolute_uri(obj.profile_picture.url)
        return None


class UserListSchema(ModelSchema):
    major: Optional[str] = None
    university: Optional[str] = None

    class Meta:
        model = User
        fields = [
            'id',
            'username',
            'email',
            'first_name',
            'last_name',
            'is_active',
            'is_staff',
            'is_superuser',
            'date_joined',
            'major',
            'university',
        ]

    @staticmethod
    def resolve_full_name(obj):
        return obj.get_full_name()

    @staticmethod
    def resolve_major(obj):
        return obj.get_major_display()

    @staticmethod
    def resolve_university(obj):
        return obj.get_university_display()

class UserUpdateSchema(Schema):
    first_name: Optional[str] = None
    last_name: Optional[str] = None
    bio: Optional[str] = None
    year_of_study: Optional[int] = None
    major: Optional[str] = None
    university: Optional[str] = None
    student_id: Optional[str] = None

class TokenSchema(Schema):
    access_token: str
    refresh_token: str
    token_type: str = "bearer"

class TokenRefreshIn(Schema):
    refresh_token: str

class PasswordResetRequestSchema(Schema):
    email: str

class PasswordResetConfirmSchema(Schema):
    token: str
    new_password: str

class UsernameCheckSchema(Schema):
    exists: bool