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

from app.models.resource import Resource
from app.models.user import User

main_bp = Blueprint('main', __name__)


@main_bp.route('/')
@login_required
def index():
    if current_user.is_admin:
        return redirect(url_for('admin.dashboard'))
    return redirect(url_for('resources.list_resources'))


@main_bp.route('/profile')
@login_required
def profile():
    total = Resource.query.filter_by(user_id=current_user.id).count()
    by_type = {}
    for t in ('text', 'image', 'audio', 'video'):
        by_type[t] = Resource.query.filter_by(
            user_id=current_user.id, resource_type=t).count()
    return render_template('user/profile.html', total=total, by_type=by_type)