from defence360agent.contracts.plugins import MessageSink
from defence360agent.utils import (
    check_run_outside_sandbox,
    recurring_check,
    RecurringCheckStop,
)
from defence360agent.utils.resource_limits import is_lve_active, has_lvectl


class LveUtilsAutoInstaller(MessageSink):
    """
    Install lve-utils package on CL with LVE automatically
    (according to DEF-11452) to provide tools to limit CPU/IO.

    Used tools:
    /usr/sbin/lvectl - provided by lve-utils package
    /bin/lve_suwrapper - provided by lve-wrappers package
                         (which is a dependency of lve-utils)

    lve-utils package is installed by default on CL,
    but for some reason may not exist.
    """

    def __init__(self, *, check_period=3600):
        self._check_period = check_period
        self._task = None

    async def create_sink(self, loop):
        self._loop = loop
        self._task = self._loop.create_task(
            recurring_check(self._check_period)(
                self._install_lve_utils_if_needed
            )()
        )

    async def shutdown(self):
        if self._task is not None:
            self._task.cancel()
            await self._task
            self._task = None

    async def _install_lve_utils_if_needed(self):
        if not is_lve_active():  # kernel doesn't support lve or it is disabled
            # no point trying to install lve-utils
            raise RecurringCheckStop()
        # suppose that lve should be actived on CL only
        if not has_lvectl():  # utilities might have been removed
            # DEF-41613: yum install triggers RPM scriptlets whose LSM
            # transition on exec is blocked by the agent unit's NNP.
            await check_run_outside_sandbox(
                ["yum", "-y", "install", "lve-utils"]
            )