import flask

from donut import auth_utils
from donut.modules.account import blueprint, helpers


@blueprint.route("/request")
def request_account():
    """Provides a form to request an account."""
    return flask.render_template("request_account.html")


@blueprint.route("/request/submit", methods=["POST"])
def request_account_submit():
    """Handles an account creation request."""
    uid = flask.request.form.get("uid", None)
    last_name = flask.request.form.get("last_name", None)
    if uid is None or last_name is None:
        flask.flash("Invalid request.")
        return flask.redirect(flask.url_for("account.request_account"))

    success, error_msg = helpers.handle_request_account(uid, last_name)
    if success:
        flask.flash(
            "An email has been sent with a link to create your account.")
        return flask.redirect(flask.url_for("home"))
    else:
        flask.flash(error_msg)
        return flask.redirect(flask.url_for("account.request_account"))


@blueprint.route("/create/<create_account_key>")
def 
(create_account_key):
    """Checks the key. If valid, displays the create account page."""
    user_id = auth_utils.check_create_account_key(create_account_key)
    if user_id is None:
        flask.current_app.logger.warn(
            f'Invalid create_account_key: {create_account_key}')
        flask.flash("Invalid request. Please check your link and try again.")
        return flask.redirect(flask.url_for("home"))

    user_data = helpers.get_user_data(user_id)
    if user_data is None:
        flask.flash("An unexpected error occurred. Please contact DevTeam.")
        return flask.redirect(flask.url_for("home"))
    return flask.render_template(
        "create_account.html", user_data=user_data, key=create_account_key)


@blueprint.route("/create/<create_account_key>/submit", methods=["POST"])
def create_account_submit(create_account_key):
    """Handles a create account request."""
    user_id = auth_utils.check_create_account_key(create_account_key)
    if user_id is None:
        # Key is invalid.
        flask.current_app.logger.warn(
            f'Invalid create_account_key: {create_account_key}')
        flask.flash("Someone's been naughty.")
        return flask.redirect(flask.url_for("home"))
    username = flask.request.form.get("username", None)
    password = flask.request.form.get("password", None)
    password2 = flask.request.form.get("password2", None)
    if username is None \
        or password is None \
        or password2 is None:
        flask.current_app.logger.warn(
            f'Invalid create account form for user ID {user_id}')
        flask.flash("Invalid request.")
        return flask.redirect(flask.url_for("home"))

    if helpers.handle_create_account(user_id, username, password, password2):
        flask.session['username'] = username
        flask.current_app.logger.info(
            f'Created account with username {username} for user ID {user_id}')
        flask.flash("Account successfully created.")
        return flask.redirect(flask.url_for("home"))
    else:
        # Flashes already handled.
        return flask.redirect(
            flask.url_for(
                "account.create_account",
                create_account_key=create_account_key))

create_account