"""Enumerations used throughout POLYGUARD-RL."""

from __future__ import annotations

from enum import Enum


class Difficulty(str, Enum):
    EASY = "easy"
    MEDIUM = "medium"
    HARD = "hard"


class SubEnvironment(str, Enum):
    DDI = "DDI"
    BANDIT_MINING = "BANDIT_MINING"
    REGIMEN_RISK = "REGIMEN_RISK"
    PRECISION_DOSING = "PRECISION_DOSING"
    LONGITUDINAL_DEPRESCRIBING = "LONGITUDINAL_DEPRESCRIBING"
    WEB_SEARCH_MISSING_DATA = "WEB_SEARCH_MISSING_DATA"
    ALTERNATIVE_SUGGESTION = "ALTERNATIVE_SUGGESTION"
    NEW_DRUG_DECOMPOSITION = "NEW_DRUG_DECOMPOSITION"


class DecisionMode(str, Enum):
    REGIMEN_OPT = "REGIMEN_OPT"
    DOSE_OPT = "DOSE_OPT"
    REVIEW = "REVIEW"
    ABSTAIN_REVIEW = "ABSTAIN_REVIEW"


class ActionType(str, Enum):
    KEEP_REGIMEN = "KEEP_REGIMEN"
    STOP_DRUG = "STOP_DRUG"
    SUBSTITUTE_WITHIN_CLASS = "SUBSTITUTE_WITHIN_CLASS"
    RECOMMEND_ALTERNATIVE = "RECOMMEND_ALTERNATIVE"
    REDUCE_DOSE_BUCKET = "REDUCE_DOSE_BUCKET"
    INCREASE_DOSE_BUCKET = "INCREASE_DOSE_BUCKET"
    TAPER_INITIATE = "TAPER_INITIATE"
    TAPER_CONTINUE = "TAPER_CONTINUE"
    DOSE_HOLD = "DOSE_HOLD"
    ORDER_MONITORING_AND_WAIT = "ORDER_MONITORING_AND_WAIT"
    FETCH_EXTERNAL_EVIDENCE = "FETCH_EXTERNAL_EVIDENCE"
    DECOMPOSE_NEW_DRUG = "DECOMPOSE_NEW_DRUG"
    REQUEST_SPECIALIST_REVIEW = "REQUEST_SPECIALIST_REVIEW"
    REQUEST_PHARMACIST_REVIEW = "REQUEST_PHARMACIST_REVIEW"


class DoseBucket(str, Enum):
    LOW = "LOW"
    MEDIUM = "MEDIUM"
    HIGH = "HIGH"
    HOLD = "HOLD"
    NA = "NA"


class CoordinationMode(str, Enum):
    SEQUENTIAL = "sequential_pipeline"
    SUPERVISOR_ROUTED = "supervisor_routed"
    REPLAN_ON_VETO = "replan_on_veto"
    LIGHT_DEBATE = "lightweight_debate"