from flask import current_app
from flask_login import current_user
from flask_wtf import FlaskForm
from wtforms import (BooleanField, PasswordField, SelectField, StringField,
                     SubmitField, ValidationError)
from wtforms.validators import DataRequired, Email, EqualTo, Length, Regexp


class ChangePasswordForm(FlaskForm):
    password = PasswordField('Old password', validators=[DataRequired()])
    new_password = PasswordField(
        'New password',
        validators=[DataRequired(), EqualTo('password_confirmation',
                                            message='Passwords must match.')]
    )
    new_password2 = PasswordField(
        'Confirm new password', validators=[DataRequired()])
    submit = SubmitField('Change password')

    def __init__(self, user=current_user, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.user = user

    def validate_current_password(self, field):
        if not self.user.verify_password(field.data):
            raise ValidationError('Invalid password.')


class EditGeneralSettingsForm(FlaskForm):
    dark_mode = BooleanField('Dark mode')
    email = StringField('E-Mail',
                        validators=[DataRequired(), Length(1, 254), Email()])
    username = StringField(
        'Benutzername',
        validators=[DataRequired(),
                    Length(1, 64),
                    Regexp(current_app.config['ALLOWED_USERNAME_REGEX'],
                           message='Usernames must have only letters, numbers,'
                                   ' dots or underscores')]
    )
    submit = SubmitField('Submit')

    def __init__(self, user=current_user, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.user = user
        self.email.data = self.email.data or user.email
        self.dark_mode.data = self.dark_mode.data or user.setting_dark_mode
        self.username.data = self.username.data or user.username

    def validate_email(self, field):
        if (field.data != self.user.email
                and User.query.filter_by(email=field.data).first()):
            raise ValidationError('Email already registered.')

    def validate_username(self, field):
        if (field.data != self.user.username
                and User.query.filter_by(username=field.data).first()):
            raise ValidationError('Username already in use.')


class EditNotificationSettingsForm(FlaskForm):
    job_status_mail_notifications = SelectField(
        'Job status mail notifications',
        choices=[('', 'Choose your option'),
                 ('all', 'Notify on all status changes'),
                 ('end', 'Notify only when a job ended'),
                 ('none', 'No status update notifications')],
        validators=[DataRequired()])
    job_status_site_notifications = SelectField(
        'Job status site notifications',
        choices=[('', 'Choose your option'),
                 ('all', 'Notify on all status changes'),
                 ('end', 'Notify only when a job ended'),
                 ('none', 'No status update notifications')],
        validators=[DataRequired()])
    submit = SubmitField('Save settings')

    def __init__(self, user=current_user, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.user = user
        self.job_status_mail_notifications.data = \
            self.job_status_mail_notifications.data \
            or user.setting_job_status_mail_notifications
        self.job_status_site_notifications.data = \
            self.job_status_site_notifications.data \
            or user.setting_job_status_site_notifications