"""俱乐部上下文解析与查询辅助。"""
from jituan.constants import (
    CLUB_ID_DEFAULT,
    CLUB_HEADER,
    CLUB_SCOPE_HEADER,
    DATA_SCOPE_ALL,
    DATA_SCOPE_SINGLE,
)
from jituan.models import Club


def resolve_club_id_from_request(request):
    """
    从请求解析 club_id。
    优先级：已注入的 request.club_id > Header X-Club-Id > 默认 xq。
  旧客户端不传时恒为 xq，与现网行为一致。
    """
    injected = getattr(request, 'club_id', None)
    if injected:
        return injected
    header = request.META.get(f'HTTP_{CLUB_HEADER.upper().replace("-", "_")}', '')
    if not header:
        header = request.headers.get(CLUB_HEADER, '') if hasattr(request, 'headers') else ''
    club_id = (header or '').strip() or CLUB_ID_DEFAULT
    return club_id


def resolve_club_scope(request):
    """解析数据范围：ALL_CLUBS（集团汇总）或 SINGLE_CLUB。"""
    scope = getattr(request, 'club_scope', None)
    if scope:
        return scope
    header = request.META.get(f'HTTP_{CLUB_SCOPE_HEADER.upper().replace("-", "_")}', '')
    if not header:
        header = request.headers.get(CLUB_SCOPE_HEADER, '') if hasattr(request, 'headers') else ''
    if (header or '').strip().lower() == 'all':
        return DATA_SCOPE_ALL
    return DATA_SCOPE_SINGLE


def get_active_club(club_id):
    try:
        club = Club.query.get(club_id=club_id, status=1)
        return club
    except Club.DoesNotExist:
        return None


def filter_queryset_by_club(qs, request, club_field='club_id'):
    """
    按请求上下文过滤 QuerySet / FluentQuery。
    集团 ALL_CLUBS 视图不过滤；子公司 SINGLE_CLUB 按 club_id 过滤。
    """
    scope = resolve_club_scope(request)
    club_id = resolve_club_id_from_request(request)
    if scope == DATA_SCOPE_ALL:
        return qs
    if hasattr(qs, 'filter'):
        return qs.filter(**{club_field: club_id})
    return qs


def club_id_for_write(request):
    """写入业务流水时使用的 club_id（集团汇总视图写入仍须指定目标 club）。"""
    return resolve_club_id_from_request(request)


def orders_for_request(request):
    """按俱乐部过滤的订单 QuerySet。"""
    from orders.models import Order
    return filter_queryset_by_club(Order.query.all(), request, club_field='ClubID')


def filter_user_related_by_club(qs, request, user_club_path='user__ClubID'):
    """按 User.ClubID 过滤打手/管事/商家等关联查询。"""
    scope = resolve_club_scope(request)
    if scope == DATA_SCOPE_ALL:
        return qs
    return qs.filter(**{user_club_path: resolve_club_id_from_request(request)})


def filter_user_qs_by_club(qs, request):
    """直接过滤 User 查询集（字段 ClubID）。"""
    scope = resolve_club_scope(request)
    if scope == DATA_SCOPE_ALL:
        return qs
    return qs.filter(ClubID=resolve_club_id_from_request(request))