1#@!#!123s

: / opt / imunify360 / venv / lib / python3.11 / site-packages / im360 / internals /
Filename : strategy.py
back
from pathlib import Path
from im360.subsys import csf

CSF_COOP_STRATEGY = Path("/var/imunify360/.csf_coop")
PRIMARY_IDS_STRATEGY = Path("/var/imunify360/.primary_ids")


class Strategy:
    PRIMARY_IDS_STRATEGY = "PRIMARY_IDS"
    CSF_COOP_STRATEGY = "CSF_COOP"
    UNKNOWN = "UNKNOWN"

    current = UNKNOWN

    @classmethod
    async def detect(cls):
        csf_running = await csf.is_running()
        if csf_running:
            return cls.CSF_COOP_STRATEGY
        else:
            return cls.PRIMARY_IDS_STRATEGY

    @classmethod
    def save(cls, new_strategy):
        if new_strategy is cls.CSF_COOP_STRATEGY:
            CSF_COOP_STRATEGY.touch()
            PRIMARY_IDS_STRATEGY.unlink(missing_ok=True)
        elif new_strategy is cls.PRIMARY_IDS_STRATEGY:
            PRIMARY_IDS_STRATEGY.touch()
            CSF_COOP_STRATEGY.unlink(missing_ok=True)
        else:
            PRIMARY_IDS_STRATEGY.unlink(missing_ok=True)
            CSF_COOP_STRATEGY.unlink(missing_ok=True)

    @classmethod
    def get(cls):
        if CSF_COOP_STRATEGY.exists():
            return cls.CSF_COOP_STRATEGY
        elif PRIMARY_IDS_STRATEGY.exists():
            return cls.PRIMARY_IDS_STRATEGY
        return cls.UNKNOWN