from enum import IntEnum, StrEnum


class AlertType(StrEnum):
    BOOT = "BOOT"
    ERROR = "ERROR"
    TEST = "TEST"
    REMIND = "REMIND"

    CPU = "CPU"
    NET = "NET"
    RAM = "RAM"
    TEMP = "TEMP"

    VULN = "VULN"
    # LOGIN = "LOGIN"
    # SMART = "SMART"  # TODO
    # RAID = "RAID"
    # DISKS = "DISKS"
    # UPS = "UPS"
    # UPDATE = "UPDATE"


class Severity(IntEnum):
    OK = 0  # should only be used when persistent alerts resolve
    INFO = 1
    WARNING = 2
    CRITICAL = 3


SEVERITY_TO_EMOJI = {
    Severity.OK: "🟢",
    Severity.INFO: "ℹ️",
    Severity.WARNING: "⚠️",
    Severity.CRITICAL: "🆘",
}