from flask import (
    Blueprint, session, request, url_for, render_template, redirect,
    flash, g
)
from werkzeug.security import check_password_hash, generate_password_hash
from .db import get_db
import functools

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

def get_next():
    return request.args.get('next',url_for('index'))

@bp.route("/login",methods=('GET','POST'))
def login():
    if request.method == "POST":
        username = request.form["username"]
        password = request.form["password"]
        db = get_db()
        err = None
        user = db.execute(
            "SELECT password FROM users WHERE username = ?;",(username,)
        ).fetchone()
        if not username:
            err = "username required"
        elif not password:
            err = "password required"
        elif user is None or not check_password_hash(user['password'], password):
            err = "invalid login"

        if err is None:
            session.clear()
            session['user'] = username
            if 'keep_logged_in' in request.form:
                session['keep_logged_in']=True
                session.permanent = True
            flash("logged in successfully")
            return redirect(get_next())

        flash(err)
        
    return render_template("auth/login.html")


@bp.route("/register", methods=("GET","POST"))
def register():
    if request.method == "POST":
        username = request.form["username"]
        password = request.form["password"]
        db = get_db()
        err = None
        if not username:
            err = "Username required"
        elif not password:
            err = "Password required"
        elif db.execute(
            "SELECT 1 FROM users WHERE username = ?;", (username,)
        ).fetchone() is not None:
            err = f"User {username} is already registered."
        elif len(username) > 20:
            err = "username can't be longer than 20 characters"
        elif not username.isalnum():
            err = "username must be alphanumeric"

        if err is None:
            db.execute(
                "INSERT INTO users (username, password, joined) VALUES (?,?,current_timestamp);",
                (username,generate_password_hash(password))
            )
            db.commit()
            session['user'] = username
            if 'keep_logged_in' in request.form:
                session['keep_logged_in'] = True
                session.permanent = True
            flash("successfully created account")
            return redirect(get_next())

        flash(err)
            
    return render_template("auth/register.html")

@bp.route("/logout")
def logout():
    session.clear()
    flash("logged out successfully")
    return redirect(get_next())

@bp.before_app_request
def load_user():
    username = session.get("user")
    if session.get("keep_logged_in",False):
        session.permanent = True
    if username is None:
        g.user = None
        g.user_info = None
    else:
        row = get_db().execute(
            "SELECT * FROM users WHERE username = ?;", (username,)
        ).fetchone()
        if row is None:
            g.user = None
            g.user_info = None
        else:
            g.user = row['username']
            g.user_info = row
        

def login_required(view):
    @functools.wraps(view)
    def wrapped(**kwargs):
        print(g.user)
        if g.user is None:
            return redirect(url_for("auth.login"))
        return view(**kwargs)
    return wrapped

@bp.route("/cool")
def cool():
    user = session.get("user")
    if user is None:
        return "you are not logged in"
    else:
        return f"you are logged in as {user}"

@bp.route("/cooler")
@login_required
def cooler():
    return "bee"