#!/usr/bin/env python3

from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, BooleanField, SubmitField
from wtforms.validators import DataRequired, Email, Optional, Regexp, Length, EqualTo, ValidationError
from app.models import Contributor, EmailWhiteList
from zxcvbn import zxcvbn


class DisableTotp(FlaskForm):
    submit = SubmitField('Disable 2FA')


class GetTotp(FlaskForm):
    totp_code = StringField('6-Digit Code?', validators=[DataRequired(), Length(min=6, max=6, message="6 Digits")], render_kw={'autofocus': True})
    submit = SubmitField('OK')


class ConfirmTotp(FlaskForm):
    totp_code = StringField('6-Digit Code?', validators=[DataRequired(), Length(min=6, max=6, message="Rescan And Try Again")], render_kw={'autofocus': True})
    submit = SubmitField('Enable 2FA')


class LoginForm(FlaskForm):
    username = StringField('Username', validators=[Optional()], render_kw={'autofocus': True})
    email = StringField('Email', validators=[Optional(), Email()])
    password = PasswordField('Password', validators=[DataRequired()])
    remember_me = BooleanField('Remember Me')
    submit = SubmitField('Sign In')


class RegistrationForm(FlaskForm):
    username = StringField('Username', validators=[DataRequired(), Regexp('^[a-zA-Z0-9]+$', message='letters and digits only (no spaces)')], render_kw={'autofocus': True})
    email = StringField('Email', validators=[DataRequired(), Email()])
    password = PasswordField('Password', validators=[DataRequired(), Length(min=15, )])
    password2 = PasswordField('Repeat Password', validators=[DataRequired(), EqualTo('password')])
    submit = SubmitField('Register')

    def validate_password(self, password):
        if zxcvbn(password.data)['score'] < 3:
            raise ValidationError('needs to be stronger')

    def validate_username(self, username):
        user = Contributor.query.filter_by(name=username.data).first()
        if user is not None:
            raise ValidationError('Please use a different username.')

    def validate_email(self, email):
        white_listed_user = EmailWhiteList.query.filter_by(email=email.data).first()
        if white_listed_user is None:
            raise ValidationError('This email address is not authorized.')
        user = Contributor.query.filter_by(email=email.data).first()
        if user is not None:
            raise ValidationError('Please use a different email address.')


class ChangePassword(FlaskForm):
    password = PasswordField('Confirm Password', validators=[DataRequired()], render_kw={'autofocus': True})
    new_password = PasswordField('New Password', validators=[DataRequired(), Length(min=15, )])
    new_password2 = PasswordField('Repeat New Password', validators=[DataRequired(), EqualTo('new_password')])
    submit = SubmitField('Save')

    def validate_password(self, password):
        if zxcvbn(password.data)['score'] < 3:
            raise ValidationError('needs to be stronger')


class ResetPasswordForm(FlaskForm):
    password = PasswordField('Password', validators=[DataRequired(), Length(min=15, )], render_kw={'autofocus': True})
    password2 = PasswordField('Repeat Password', validators=[DataRequired(), EqualTo('password')])
    submit = SubmitField('Request Password Reset')

    def validate_password(self, password):
        if zxcvbn(password.data)['score'] < 3:
            raise ValidationError('needs to be stronger')


class EditProfile(FlaskForm):
    username = StringField('Username', validators=[DataRequired(), Regexp('^[a-zA-Z0-9]+$', message='letters and digits only (no spaces)')], render_kw={'autofocus': True})
    email = StringField('Email', validators=[Optional(), Email()])
    password = PasswordField('Confirm Password', validators=[DataRequired()])
    submit = SubmitField('Update Name/Email')

    def validate_username(self, username):
        from flask_login import current_user
        if username.data != current_user.name:
            user = Contributor.query.filter_by(name=username.data).first()
            if user is not None:
                raise ValidationError('Please pick a different username.')

    def validate_email(self, email):
        from flask_login import current_user
        if email.data != current_user.email:
            user = Contributor.query.filter_by(email=email.data).first()
            if user is not None:
                raise ValidationError('Please pick a different email address.')


class ResetPasswordRequestForm(FlaskForm):
    email = StringField('Email', validators=[DataRequired(), Email()], render_kw={'autofocus': True})
    submit = SubmitField('Request Password Reset')