# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence.
"""
Enumerations for APE
"""

from enum import Enum

class TYPE(Enum):
    """
    Node type enumeration
    """

    CCTV = 1
    SWITCH = 2
    COMPUTER = 3
    LINK = 4
    MONITOR = 5
    PRINTER = 6
    LOP = 7
    RTU = 8
    ACTUATOR = 9
    SERVER = 10

class PRIORITY(Enum):
    """
    Node priority enumeration
    """

    P1 = 1
    P2 = 2
    P3 = 3
    P4 = 4
    P5 = 5

class HARDWARE_STATE(Enum):
    """
    Node hardware state enumeration
    """

    ON = 1
    OFF = 2
    RESETTING = 3

class SOFTWARE_STATE(Enum):
    """
    O/S or Service state enumeration
    """

    GOOD = 1
    PATCHING = 2
    COMPROMISED = 3
    OVERWHELMED = 4

class NODE_POL_TYPE(Enum):
    """
    Node Pattern of Life type enumeration
    """

    OPERATING = 1
    OS = 2
    SERVICE = 3

class PROTOCOL(Enum):
    """
    Service protocol enumeration
    """

    LDAP = 0
    FTP = 1
    HTTPS = 2
    SMTP = 3
    RTP = 4
    IPP = 5
    TCP = 6
    NONE = 7

class ACTION_TYPE(Enum):
    """
    Action type enumeration
    """

    NODE = 0
    ACL = 1