from flask import Blueprint, render_template, redirect, url_for
from flask_login import current_user, login_user, logout_user
from forms import LoginForm, ChangePasswordForm
from models import User
from application.utils import default_return, is_valid_timezone
from application import db

bp = Blueprint(
    "auth",
    __name__,
    template_folder="templates",
)


@bp.route("/login", methods=["GET", "POST"])
def login():
    if current_user.is_authenticated:
        return default_return()

    form = LoginForm()
    if form.validate_on_submit():
        assert form.timezone.data
        user = User.query.filter_by(username=form.username.data).first()
        if user and user.check_password(password=form.password.data):
            # User found and password correct
            tz = form.timezone.data
            if is_valid_timezone(tz):
                user.set_timezone(tz)
                db.session.commit()
            login_user(user)  # Log in the user
            return default_return()
        else:
            pass
            # invalid user
    return render_template("login.html", form=form)


@bp.route("/change_pass", methods=["GET", "POST"])
def change_pass():
    if not current_user.is_authenticated:
        return redirect(url_for("auth.login"))

    form = ChangePasswordForm()
    if form.validate_on_submit():
        cur_check = current_user.check_password(
            password=form.current_password.data
        )
        eq_check = form.new_password.data == form.confirm_password.data
        if cur_check and eq_check:
            current_user.change_password(form.new_password.data)
            current_user.set_pw_change(False)
            db.session.commit()
            logout_user()
            return default_return()
    return render_template("new_change_password.html", form=form)


@bp.route("/logout")
def logout():
    if not current_user.is_authenticated:
        return redirect(url_for("auth.login"))

    logout_user()
    return redirect(url_for("index"))