22

✏️ Editing: strategy.py

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

MR SKB