Mini Shell

from __future__ import absolute_import

import os
import subprocess
import logging
from glob import iglob

from typing import Iterable

from clwpos.optimization_features import (
    Feature,
    OBJECT_CACHE_FEATURE,
    CDN_FEATURE
)

_DEFAULT_SCRIPT_ENV = {
    "CLSHARE": "/usr/share/cloudlinux",
    "WPOS_REQ_CRON_FILE": "/etc/cron.d/clwpos_req_cron",
    "CLWPOS_COLLECT_INFORMATION_CRON": "/etc/cron.d/clwpos_collect_information_cron.py"
}

_OBJECT_CACHING_ENV = {
    "CLWPOS_REDIS_EXTENSION_INSTALLER": "/etc/cron.d/clwpos_redis_extension_installer",
    "CLWPOS_CLEANER_CRON": "/etc/cron.d/clwpos_cleaner_cron",
    "CLWPOS_BILLING_CRON": "/etc/cron.d/clwpos_billing_cron",
}

_CDN_ENV = {
    "CLWPOS_BILLING_CRON": "/etc/cron.d/clwpos_billing_cron",
}

_CRON_ENV_BY_FEATURE = {
    OBJECT_CACHE_FEATURE: _OBJECT_CACHING_ENV,
    CDN_FEATURE: _CDN_ENV
}


def install_cron_files(features: Iterable[Feature], wait_child_process=False):
    """
    This script installs cron files based on the features
    that are currently enabled.

    Please note that already installed cron files will not be removed if
    new list has missing feature against previous one.
    """
    features = tuple(features)

    if wait_child_process:
        exec_func = subprocess.run
    else:
        exec_func = subprocess.Popen

    if not features:
        logging.info('There are no features active, skip installing cron files.')
        return

    env_params = _DEFAULT_SCRIPT_ENV.copy()
    for feature in features:
        feature_env = _CRON_ENV_BY_FEATURE.get(feature)
        if feature_env is None:
            continue
        logging.info('Feature %s is active, installing cron files', feature)
        env_params.update(feature_env)

    exec_func(
        ["/usr/share/cloudlinux/add_clwpos_crons.sh"],
        stdout=subprocess.DEVNULL,
        stderr=subprocess.DEVNULL,
        env=env_params
    )


def clean_crons(prefix: str):
    """Remove all cron files starting from prefix"""
    crondir = '/etc/cron.d'
    for _f in iglob(f'{crondir}/{prefix}*'):
        os.unlink(_f)


def clean_clwpos_crons():
    """Remove all WPOS cron files"""
    clean_crons(prefix='clwpos_')