from flask import Blueprint, render_template, session, redirect, url_for
from models import User, Finding

profile_bp = Blueprint('profile', __name__, url_prefix='/profile')

@profile_bp.route('/')
def my_findings():
    # Check if user is logged in
    if not session.get('loggedin'):
        return redirect(url_for('login.login'))

    user_id = session.get('id')
    user = User.query.get(user_id)
    if not user:
        return redirect(url_for('login.login'))

    # Get all findings by this user, exclude content_preview
    findings = Finding.query.filter_by(found_by=user_id).all()

    return render_template('profile.html', user=user, findings=findings)

@profile_bp.route('/get/<int:id>', methods=['GET'])
def view_profile(id):
    user = User.query.get(id)
    if not user:
        return "User not found. Please try again later.", 
    findings = Finding.query.filter_by(found_by=id).all()
    return render_template('view_profile.html', user=user, findings=findings)