from flask_babelex import lazy_gettext
from flask_wtf import FlaskForm
from wtforms import (
    StringField,
    SubmitField,
    DecimalField,
    FormField,
)
from wtforms.fields.html5 import EmailField, TelField, URLField
from wtforms.validators import DataRequired, Optional, Regexp
from wtforms.widgets.html5 import ColorInput
from project.models import Location, Image
from project.forms.common import FileImageForm


class AdminUnitLocationForm(FlaskForm):
    street = StringField(lazy_gettext("Street"), validators=[Optional()])
    postalCode = StringField(lazy_gettext("Postal code"), validators=[DataRequired()])
    city = StringField(lazy_gettext("City"), validators=[DataRequired()])
    state = StringField(lazy_gettext("State"), validators=[Optional()])
    latitude = DecimalField(
        lazy_gettext("Latitude"), places=16, validators=[Optional()]
    )
    longitude = DecimalField(
        lazy_gettext("Longitude"), places=16, validators=[Optional()]
    )


class BaseAdminUnitForm(FlaskForm):
    name = StringField(lazy_gettext("Name"), validators=[DataRequired()])
    short_name = StringField(
        lazy_gettext("Short name"),
        description=lazy_gettext(
            "The short name is used to create a unique identifier for your events"
        ),
        validators=[
            DataRequired(),
            Regexp(
                r"^\w+$",
                message=lazy_gettext(
                    "Short name must contain only letters numbers or underscore"
                ),
            ),
        ],
    )
    url = URLField(lazy_gettext("Link URL"), validators=[Optional()])
    email = EmailField(lazy_gettext("Email"), validators=[Optional()])
    phone = TelField(lazy_gettext("Phone"), validators=[Optional()])
    fax = TelField(lazy_gettext("Fax"), validators=[Optional()])
    logo = FormField(FileImageForm, lazy_gettext("Logo"), default=lambda: Image())
    location = FormField(AdminUnitLocationForm, default=lambda: Location())

    def populate_obj(self, obj):
        for name, field in self._fields.items():
            if name == "location" and not obj.location:
                obj.location = Location()
            elif name == "logo" and not obj.logo:
                obj.logo = Image()
            field.populate_obj(obj, name)


class CreateAdminUnitForm(BaseAdminUnitForm):
    submit = SubmitField(lazy_gettext("Create admin unit"))


class UpdateAdminUnitForm(BaseAdminUnitForm):
    submit = SubmitField(lazy_gettext("Update settings"))


class UpdateAdminUnitWidgetForm(FlaskForm):
    widget_font = StringField(lazy_gettext("Font"), validators=[Optional()])
    widget_background_color = StringField(
        lazy_gettext("Background Color"),
        default="#ffffff",
        widget=ColorInput(),
        validators=[Optional()],
    )
    widget_primary_color = StringField(
        lazy_gettext("Primary Color"),
        default="#007bff",
        widget=ColorInput(),
        validators=[Optional()],
    )
    widget_link_color = StringField(
        lazy_gettext("Link Color"),
        default="#007bff",
        widget=ColorInput(),
        validators=[Optional()],
    )
    submit = SubmitField(lazy_gettext("Update settings"))