????

Your IP : 3.137.220.187

Current Path : /proc/self/root/opt/imunify360/venv/lib64/python3.11/site-packages/im360/subsys/
Upload File :
Current File : //proc/self/root/opt/imunify360/venv/lib64/python3.11/site-packages/im360/subsys/webshield_mode.py

from pathlib import Path
from enum import StrEnum, unique


_CONF_PATH = Path("/usr/share/imunify360-webshield/modularity_mode")


@unique
class Mode(StrEnum):
    STANDALONE = "standalone"
    APACHE = "apache"
    NGINX = "nginx"

    @classmethod
    def get(cls):
        try:
            mode = _CONF_PATH.read_text().strip()
            if mode == "apache":
                return cls.APACHE
            if mode == "nginx":
                return cls.NGINX
            if mode == "standalone":
                return cls.STANDALONE
            raise TypeError(f"Unknown mode: {mode}")
        except FileNotFoundError:
            return cls.STANDALONE

    @classmethod
    def wants_redirect(cls, mode) -> bool:
        if mode is cls.STANDALONE or mode is cls.APACHE:
            return True
        return False


def get_module_based_ports() -> set[int]:
    # For now - hardcoded values. In the future maybe remake to a config file.
    return {80, 443}