from django import forms
from django.contrib.auth.forms import PasswordResetForm as BasePasswordResetForm
from django.contrib.auth.forms import SetPasswordForm as BaseSetPasswordForm
from django.contrib.auth.tokens import default_token_generator
from django.core.exceptions import ValidationError
from django.utils.text import gettext_lazy as _

from web.mixins import StylingMixin


class PasswordResetForm(StylingMixin, BasePasswordResetForm):
    styled_fields = [
        "email",
    ]
    placeholder_for_field = {
        "email": _("pablo@shoppy.com"),
    }


class SetPasswordForm(StylingMixin, BaseSetPasswordForm):
    styled_fields = [
        "new_password1",
        "new_password2",
    ]
    placeholder_for_field = {
        "new_password1": "********",
        "new_password2": "********",
    }

    token = forms.CharField(
        required=True,
        widget=forms.HiddenInput(),
        validators=[],
    )

    def validate_token(self):
        is_valid = default_token_generator.check_token(
            self.user, self.data.get("token")
        )

        if not is_valid:
            raise ValidationError(_("El token no es válido o ha expirado"))

    def is_valid(self):
        try:
            self.validate_token()
        except ValidationError as e:
            self.add_error("token", str(e))
            return False
        return super().is_valid()