Sindbad~EG File Manager

# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2018 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# https://cloudlinux.com/docs/LICENCE.TXT
#
from __future__ import absolute_import
import logging
from clconfigure import task, run

STATE_MASKED = 'masked'
STATE_STOPPED = 'stopped'

STATE_UNMASKED = 'unmasked'
STATE_RUNNING = 'running'

STATUS_STOPPED = 'inactive'
STATUS_RUNNING = 'active'


def _get_service_state(service, kind='is-active'):
    """
    Get information about systemctl service state
    (either activity or enability)
    """
    result = run(['systemctl', kind, service + '.service'])

    state = result.stdout.strip()
    return state


@task("Changing service '{service_name}' state to '{desired_state}'")
def set_service_state(desired_state, service_name):
    """
    Brings service to given state. May be executed more that once,
    does't crash on future calls
    """
    current_state = _get_service_state(service_name, kind='is-enabled')
    current_status = _get_service_state(service_name)

    logging.debug("Checking service '%s' state... package is in state='%s' running='%s'"
                  "", service_name, current_state, current_status)

    action = None
    if desired_state == STATE_MASKED and current_state != STATE_MASKED:
        action = 'mask'
    elif desired_state == STATE_UNMASKED and current_state == STATE_MASKED:
        action = 'unmask'
    elif desired_state == STATE_STOPPED and current_status != STATUS_STOPPED:
        action = 'stop'
    elif desired_state == STATE_RUNNING and current_status != STATUS_RUNNING:
        action = 'start'

    if action is None:
        logging.debug("No actions needed for service '%s'", service_name)
    else:
        run(['systemctl', action, service_name + '.service'])
        current_state = _get_service_state(service_name)
        current_status = _get_service_state(service_name)
        logging.debug("Checking service '%s' state again... service now is in state='%s' running='%s'"
                      "", service_name, current_state, current_status)