from flask import Blueprint, render_template, request, redirect, url_for
from flask_login import login_required, current_user
from ..extensions import db
from ..models import User, Profile, Notification

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

@bp.route("/<int:user_id>")
def profile(user_id):
    user = User.query.get_or_404(user_id)
    return render_template("users/profile.html", user=user)

@bp.route("/me/edit", methods=["GET", "POST"])
@login_required
def edit_profile():
    profile = Profile.query.filter_by(user_id=current_user.id).first()
    if request.method == "POST":
        profile.bio = request.form.get("bio")
        profile.grade = request.form.get("grade")
        profile.class_name = request.form.get("class_name")
        db.session.commit()
        return redirect(url_for("users.profile", user_id=current_user.id))
    return render_template("users/edit.html", profile=profile)

@bp.route("/me/notifications")
@login_required
def notifications():
    items = Notification.query.filter_by(user_id=current_user.id).order_by(Notification.created_at.desc()).limit(50).all()
    return render_template("users/notifications.html", items=items)