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, Follow

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

@bp.route("/<int:user_id>")
def profile(user_id):
    user = User.query.get_or_404(user_id)
    followers_count = Follow.query.filter_by(followee_id=user.id).count()
    following_count = Follow.query.filter_by(follower_id=user.id).count()
    is_following = False
    try:
        from flask_login import current_user
        if current_user.is_authenticated:
            is_following = Follow.query.filter_by(follower_id=current_user.id, followee_id=user.id).first() is not None
    except Exception:
        pass
    return render_template("users/profile.html", user=user, followers_count=followers_count, following_count=following_count, is_following=is_following)

@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)