"""Duplicate therapy checks."""

from __future__ import annotations

from app.common.types import Medication
from app.knowledge.drug_catalog import DRUG_CLASSES


def has_duplicate_therapy(
    meds: list[Medication],
    target_drug: str | None,
    replacement_drug: str | None,
) -> bool:
    classes = [DRUG_CLASSES.get(m.drug) for m in meds if DRUG_CLASSES.get(m.drug)]
    if replacement_drug and DRUG_CLASSES.get(replacement_drug):
        classes.append(DRUG_CLASSES[replacement_drug])
    if target_drug and replacement_drug:
        # replacement within class is expected; duplicate only when 3+ active in same class.
        return any(classes.count(c) >= 3 for c in set(classes))
    return any(classes.count(c) >= 3 for c in set(classes))