"""
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

    # Source Users only see their own created complaints
    if user.is_source_user():
        complaint_count = Complaint.objects.filter(created_by=user, status__in=["open", "in_progress"]).count()
        return {
            "complaint_count": complaint_count,
            "feedback_count": 0,
            "action_count": 0,
            "current_hospital": None,
            "is_px_admin": False,
            "is_source_user": True,
        }

    # Filter based on user role and tenant_hospital
    if user.is_px_admin():
        # PX Admins use their selected hospital from session
        hospital = getattr(request, "tenant_hospital", None)
        if hospital:
            complaint_count = Complaint.objects.filter(hospital=hospital, status__in=["open", "in_progress"]).count()
            feedback_count = Feedback.objects.filter(hospital=hospital, status__in=["submitted", "reviewed"]).count()
            action_count = PXAction.objects.filter(hospital=hospital, status__in=["open", "in_progress"]).count()
        else:
            complaint_count = 0
            feedback_count = 0
            action_count = 0
        # 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

    return {
        "complaint_count": complaint_count,
        "feedback_count": feedback_count,
        "action_count": action_count,
        "current_hospital": getattr(request, "tenant_hospital", None),
        "is_px_admin": request.user.is_authenticated and request.user.is_px_admin(),
        "is_source_user": False,
    }


def hospital_context(request):
    """
    Provide current hospital context to templates.

    This ensures hospital information is available in the header for all pages.
    """
    if not request.user.is_authenticated:
        return {}

    hospital = getattr(request, "tenant_hospital", None)

    # Get list of hospitals for PX Admin switcher
    hospitals_list = []
    if request.user.is_px_admin():
        from apps.organizations.models import Hospital

        hospitals_list = list(Hospital.objects.filter(status="active").order_by("name").values("id", "name", "code"))

    # Source user context
    is_source_user = request.user.is_source_user()
    source_user_profile = getattr(request, "source_user_profile", None)

    return {
        "current_hospital": hospital,
        "is_px_admin": request.user.is_px_admin(),
        "is_source_user": is_source_user,
        "source_user_profile": source_user_profile,
        "hospitals_list": hospitals_list,
        "show_hospital_selector": False,
    }