Mini Shell

# -*- coding: utf-8 -*-

# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2021 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT

import logging
import os
import time

from xray.apiclient import get_client
from xray.internal.exceptions import XRayError
from xray.internal.utils import read_sys_id
from xray.manager import initialize_manager
from xray.reconfiguration.system_id_shift import (
    remove_xray_inis,
    clear_dbm_file
)

storage = '/usr/share/alt-php-xray/running_tasks_list'
marker = '/usr/share/alt-php-xray/migrated'
migration_log = '/var/log/alt-php-xray/migration.log'

logger = logging.getLogger('migrate')


def running_tasks() -> list:
    """
    Get list of running tasks and save it into file
    """
    client = get_client('lists')(system_id=read_sys_id())
    try:
        tasks = client.get_task_list()
        logger.info('Loaded tasks: %s', tasks)
    except XRayError:
        return list()
    running = [t['tracing_task_id'] for t in tasks['result'] if
               t['status'] == 'running']
    logger.info('Running tasks: %s', running)
    with open(storage, 'w') as f:
        f.write('\n'.join(running))
    return running


def stop_tasks() -> None:
    """
    Stop running tasks.
    Try to remove ini and clear dbm storage after
    """
    manager = initialize_manager(read_sys_id())
    for task in running_tasks():
        logger.info('Stopping task %s', task)
        try:
            manager.stop(task)
        except XRayError:
            pass
        time.sleep(1)
    # clear if some garbage remains
    remove_xray_inis()
    clear_dbm_file()


def continue_tasks() -> None:
    """"""
    manager = initialize_manager(read_sys_id())
    try:
        with open(storage) as f:
            tasks = [l for l in f.read().split('\n') if l]
    except OSError:
        return

    logger.info('Saved task list loaded: %s', tasks)
    os.unlink(storage)
    for task in tasks:
        logger.info('Continue task %s', task)
        try:
            manager.continue_(task)
        except XRayError:
            pass
        time.sleep(1)
    # leave marker
    open(marker, 'w').close()