"""Module for validators."""

from __future__ import annotations

from collections.abc import Iterable

from django.utils.translation import gettext_lazy as _

from apps.core.models.enums import CaseStage, CaseType

__all__: list[str] = [
    "APPLICABLE_TYPES",
    "is_applicable",
    "normalize_stages",
]

APPLICABLE_TYPES: set[str] = {
    str(CaseType.CIVIL),
    str(CaseType.CRIMINAL),
    str(CaseType.ADMINISTRATIVE),
    str(CaseType.LABOR),
    str(CaseType.INTL),
}


def is_applicable(case_type: str | None) -> bool:
    """检查案件类型是否适用阶段管理。"""
    return bool(case_type) and case_type in APPLICABLE_TYPES


def _allowed_stages() -> set[str]:
    """获取所有允许的案件阶段值。"""
    return {str(c[0]) for c in CaseStage.choices}


def normalize_stages(
    case_type: str | None,
    representation_stages: Iterable[str] | None,
    current_stage: str | None,
    strict: bool = False,
) -> tuple[list[str], str | None]:
    """规范化案件阶段数据。"""
    if not is_applicable(case_type):
        if strict and (representation_stages or current_stage):
            raise ValueError(_("stages_not_applicable"))
        return [], None
    rep = list(representation_stages or [])
    cur = current_stage or None
    allowed = _allowed_stages()
    invalid = set(rep) - allowed
    if invalid:
        raise ValueError(_("invalid_rep:%(stages)s") % {"stages": ",".join(sorted(invalid))})
    if cur:
        if cur not in allowed:
            raise ValueError(_("invalid_cur"))
        if rep and cur not in set(rep):
            raise ValueError(_("cur_not_in_rep"))
    return rep, cur