from __future__ import annotations

from enum import Enum


class HumidifierMode(Enum):
    HIGH = 1
    MEDIUM = 2
    LOW = 3
    QUIET = 4
    TARGET_HUMIDITY = 5
    SLEEP = 6
    AUTO = 7
    DRYING_FILTER = 8

    @classmethod
    def get_modes(cls) -> list[str]:
        return [mode.name.lower() for mode in cls]


class HumidifierWaterLevel(Enum):
    EMPTY = 0
    LOW = 1
    MEDIUM = 2
    HIGH = 3

    @classmethod
    def get_levels(cls) -> list[str]:
        return [level.name.lower() for level in cls]


class HumidifierAction(Enum):
    OFF = 0
    HUMIDIFYING = 1
    DRYING = 2


OVER_HUMIDIFY_PROTECTION_MODES = {
    HumidifierMode.QUIET,
    HumidifierMode.LOW,
    HumidifierMode.MEDIUM,
    HumidifierMode.HIGH,
}

TARGET_HUMIDITY_MODES = {
    HumidifierMode.SLEEP,
    HumidifierMode.TARGET_HUMIDITY,
}