from flask import Blueprint, request, redirect, url_for, flash, render_template, current_app
from flask_login import current_user, login_user, login_required
import flask_login
from werkzeug.security import check_password_hash, generate_password_hash
from datetime import datetime

from app.auth.forms import LoginForm, RegisterForm
from app.database import User
from app import db

import ulid

bp = Blueprint('auth', __name__, url_prefix='/auth')

@bp.route("/login", methods=["GET", "POST"])
def login():
    if current_user.is_authenticated:
        flash("You are already logged in")
        return redirect(url_for("meta.home"))

    form = LoginForm(request.form)

    if request.method == 'POST' and form.validate():
        email = request.form.get('username')
        password = request.form.get('password')

        u = User.query.filter_by(email=email).first()

        if u is not None:
            if check_password_hash(u.password, password):
                if u.active:
                    login_user(u)

                    u.last_login = datetime.now()
                    db.session.commit()
    
                    flash("Logged in successfully")

                    return redirect(url_for("meta.home"))
                else:
                    flash("User is inactive. Contact an administrator")

            else:
                flash("Incorrect password")
        else:
            flash("Incorrect username")

    return render_template("login.html", form=form)

@bp.route("/register", methods=["GET", "POST"])
def register():
    if current_user.is_authenticated:
        flash("You are already logged in")
        return redirect(url_for("meta.home"))

    form = RegisterForm(request.form)

    if request.method == 'POST' and form.validate():
        email = request.form.get("username")
        pref_name = request.form.get("pref_name")
        password = request.form.get("password")
        pw_confirm = request.form.get("password_confirm")

        if password == pw_confirm:
            # Passwords match
            
            user = User(
                id=str(ulid.ulid()),
                email=email,
                password=generate_password_hash(password),
                pref_name=pref_name,
                last_login=datetime.now(),
                active=True
                )

            db.session.add(user)
            db.session.commit()
            flask_login.login_user(user)

            return redirect(url_for('meta.home'))
        else:
            flash("Passwords do not match")

    return render_template("register.html", form=form)


@bp.route("/logout")
@login_required
def logout():
    flask_login.logout_user()

    return redirect("/")

@bp.route("/profile")
@login_required
def profile():
    debug = current_app.config['DEBUG']
    peers = current_user.peers
    networks = current_user.networks
    return render_template("profile.html", debug=debug, peers=peers,
            nets=networks)