from flask import render_template, redirect, url_for, flash
from flask_login import login_required, current_user
from flask_wtf import FlaskForm
from wtforms import StringField, validators

from app.config import EMAIL_DOMAIN
from app.dashboard.base import dashboard_bp
from app.extensions import db
from app.log import LOG
from app.models import GenEmail
from app.utils import convert_to_id


class CustomAliasForm(FlaskForm):
    email = StringField("Email", validators=[validators.DataRequired()])


@dashboard_bp.route("/custom_alias", methods=["GET", "POST"])
@login_required
def custom_alias():
    # check if user has the right to create custom alias
    if not current_user.can_create_custom_email():
        # notify admin
        LOG.error("user %s tries to create custom alias", current_user)
        flash("ony premium user can choose custom alias", "warning")
        return redirect(url_for("dashboard.index"))

    form = CustomAliasForm()
    error = ""

    if form.validate_on_submit():
        email = form.email.data
        email = convert_to_id(email)
        if len(email) < 6:
            error = "email must be at least 6 letters"
        else:
            full_email = f"{email}@{EMAIL_DOMAIN}"
            # check if email already exists
            if GenEmail.get_by(email=full_email):
                error = "email already chosen, please choose another one"
            else:
                # create the new alias
                LOG.d("create custom alias %s for user %s", full_email, current_user)
                GenEmail.create(email=full_email, user_id=current_user.id)
                db.session.commit()

                flash(f"Email alias {full_email} has been created", "success")

                return redirect(url_for("dashboard.index"))

    return render_template("dashboard/custom_alias.html", form=form, error=error)