from flask import Blueprint, render_template, redirect, url_for
from flask_login import login_required, current_user

main_bp = Blueprint("main", __name__)

""" def login_required(view):
    @wraps(view)
    def wrapped_view(**kwargs):
        if "user_id" not in session:
            return redirect(url_for("auth.login"))
        return view(**kwargs)
    return wrapped_view
     """

@main_bp.route("/", methods=["GET", "POST"])
def index():
    if current_user.is_authenticated:
        return redirect(url_for("main.home"))
    return render_template("index.html")

@main_bp.route("/home", methods=["GET", "POST"])
@login_required
def home():
    print(f"Current user: {current_user.username}")
    return render_template("home.html")


@main_bp.route("/play", methods=["GET"])
@login_required
def play():
    return render_template("play.html")


@main_bp.route("/friends", methods=["GET"])
@login_required
def friends():
    return render_template("friends.html")