from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, SubmitField, BooleanField
from wtforms.validators import DataRequired, Length, EqualTo


class RegistrationForm(FlaskForm):
    alias = StringField('Alias',
                        validators=[DataRequired(),
                                    Length(min=2, max=20)])
    password = PasswordField('Password', validators=[DataRequired()])
    password_confirm = PasswordField(
        'Confirm Password', validators=[DataRequired(),
                                        EqualTo('password')])
    submit = SubmitField('Create Alias')


class LoginForm(FlaskForm):
    alias = StringField('Alias',
                        validators=[DataRequired(),
                                    Length(min=2, max=20)])
    password = PasswordField('Password', validators=[DataRequired()])
    remember = BooleanField('Remember Alias')
    submit = SubmitField('Login Alias')


class NewMessage(FlaskForm):
    recipient = StringField('Recipient', validators=[DataRequired()])
    message = StringField('message', validators=[DataRequired()])