"""
Context processors for global template variables
"""
from django.db.models import Q


def sidebar_counts(request):
    """
    Provide counts for sidebar badges.
    
    Returns counts for:
    - Active complaints
    - Pending feedback
    - Open PX actions
    - Provisional users (PX Admin only)
    """
    if not request.user.is_authenticated:
        return {}
    
    from apps.complaints.models import Complaint
    from apps.feedback.models import Feedback
    from apps.px_action_center.models import PXAction
    
    user = request.user
    
    # Filter based on user role
    if user.is_px_admin():
        complaint_count = Complaint.objects.filter(
            status__in=['open', 'in_progress']
        ).count()
        feedback_count = Feedback.objects.filter(
            status__in=['submitted', 'reviewed']
        ).count()
        action_count = PXAction.objects.filter(
            status__in=['open', 'in_progress']
        ).count()
        # Count provisional users for PX Admin
        from apps.accounts.models import User
        provisional_user_count = User.objects.filter(
            is_provisional=True,
            acknowledgement_completed=False
        ).count()
    elif user.hospital:
        complaint_count = Complaint.objects.filter(
            hospital=user.hospital,
            status__in=['open', 'in_progress']
        ).count()
        feedback_count = Feedback.objects.filter(
            hospital=user.hospital,
            status__in=['submitted', 'reviewed']
        ).count()
        action_count = PXAction.objects.filter(
            hospital=user.hospital,
            status__in=['open', 'in_progress']
        ).count()
        provisional_user_count = 0
    else:
        complaint_count = 0
        feedback_count = 0
        action_count = 0
        provisional_user_count = 0
    
    return {
        'complaint_count': complaint_count,
        'feedback_count': feedback_count,
        'action_count': action_count,
        'provisional_user_count': provisional_user_count,
    }