from flask_wtf import FlaskForm
from flask_wtf.file import FileAllowed, FileRequired
from wtforms import StringField, PasswordField, SubmitField, BooleanField, FileField
from wtforms.validators import DataRequired
from flask_babel import lazy_gettext as _l

class LoginForm(FlaskForm):
    username = StringField(_l('Username'), validators=[DataRequired()])
    password = PasswordField(_l('Password'), validators=[DataRequired()])
    submit = SubmitField(_l('Sign In'))
    remember_me = BooleanField('Remember Me')


class SettingsForm(FlaskForm):
    # Access point settings
    ssid = StringField('SSID', validators=[DataRequired()])
    wifi_password = PasswordField(_l('WiFi Password'))
    disable_access_point = BooleanField(_l('Disable Access Point'))
    # Customisation settings
    butterbox_name = StringField(_l('Butterbox Name'), validators=[DataRequired()])
    butterbox_logo = FileField((_l('Butterbox Logo')), validators=[FileAllowed(['jpg', 'png', 'svg'], 'Images only!')])
    # Services settings
    disable_file_viewer = BooleanField(_l('Disable File Viewer'))
    disable_map_viewer = BooleanField(_l('Disable Map Viewer'))
    disable_chat = BooleanField(_l('Disable Chat'))
    disable_app_store = BooleanField(_l('Disable App Store'))
    # Access Settings
    admin_password = PasswordField(_l('Admin Password'))
    ssh_password = PasswordField(_l('SSH Password'))

    submit = SubmitField(_l('Submit'))
    apply_changes = SubmitField(_l('Apply Changes'))