from django import forms
from django.contrib.auth.forms import UserCreationForm
from django.utils.translation import gettext_lazy as _

from django_recaptcha.fields import ReCaptchaField

from moku.models.user import User, UserSettings


class UserForm(UserCreationForm):
    captcha = ReCaptchaField(required=True)
    check = forms.BooleanField(required=True)

    class Meta(UserCreationForm.Meta):
        model = User
        fields = ("username", "email", "password1", "password2")
        labels = {
            "username": _("username"),
            "email": _("email address"),
        }
        help_texts = {
            "username": User._meta.get_field("username").help_text,
            "email": User._meta.get_field("email").help_text,
        }

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields["captcha"].error_messages = {
            "required": _("make sure you've ticked the captcha."),
        }
        self.fields["check"].error_messages = {
            "required": _("you must agree to the terms of use and privacy policy."),
        }


class UserSettingsForm(forms.ModelForm):
    class Meta:
        model = UserSettings
        fields = ("language",)
        labels = {
            "language": _("language"),
        }


class ProfileForm(forms.ModelForm):
    class Meta:
        model = User
        fields = ("avatar", "pronouns", "location", "bio")
        labels = {
            "avatar": _("avatar"),
            "pronouns": _("pronouns"),
            "location": _("location"),
            "bio": _("about me"),
        }