# Lint as: python2, python3
# Copyright 2015 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Facade to access the system-related functionality."""

import six
import os
import threading
import time

from autotest_lib.client.bin import utils


class SystemFacadeNativeError(Exception):
    """Error in SystemFacadeNative."""
    pass


class SystemFacadeNative(object):
    """Facede to access the system-related functionality.

    The methods inside this class only accept Python native types.

    """
    SCALING_GOVERNOR_MODES = [
            'performance',
            'powersave',
            'userspace',
            'ondemand',
            'conservative',
            'schedutil',
            'interactive', # deprecated since kernel v4.14
            'sched' # deprecated since kernel v4.14
            ]

    def __init__(self):
        self._bg_worker = None

    def set_scaling_governor_mode(self, index, mode):
        """Set mode of CPU scaling governor on one CPU.

        @param index: CPU index starting from 0.

        @param mode: Mode of scaling governor, accept 'interactive' or
                     'performance'.

        @returns: The original mode.

        """
        if mode not in self.SCALING_GOVERNOR_MODES:
            raise SystemFacadeNativeError('mode %s is invalid' % mode)

        governor_path = os.path.join(
                '/sys/devices/system/cpu/cpu%d' % index,
                'cpufreq/scaling_governor')
        if not os.path.exists(governor_path):
            raise SystemFacadeNativeError(
                    'scaling governor of CPU %d is not available' % index)

        original_mode = utils.read_one_line(governor_path)
        utils.open_write_close(governor_path, mode)

        return original_mode


    def get_cpu_usage(self):
        """Returns machine's CPU usage.

        Returns:
            A dictionary with 'user', 'nice', 'system' and 'idle' values.
            Sample dictionary:
            {
                'user': 254544,
                'nice': 9,
                'system': 254768,
                'idle': 2859878,
            }
        """
        return utils.get_cpu_usage()


    def compute_active_cpu_time(self, cpu_usage_start, cpu_usage_end):
        """Computes the fraction of CPU time spent non-idling.

        This function should be invoked using before/after values from calls to
        get_cpu_usage().
        """
        return utils.compute_active_cpu_time(cpu_usage_start,
                                                  cpu_usage_end)


    def get_mem_total(self):
        """Returns the total memory available in the system in MBytes."""
        return utils.get_mem_total()


    def get_mem_free(self):
        """Returns the currently free memory in the system in MBytes."""
        return utils.get_mem_free()

    def get_mem_free_plus_buffers_and_cached(self):
        """
        Returns the free memory in MBytes, counting buffers and cached as free.

        This is most often the most interesting number since buffers and cached
        memory can be reclaimed on demand. Note however, that there are cases
        where this as misleading as well, for example used tmpfs space
        count as Cached but can not be reclaimed on demand.
        See https://www.kernel.org/doc/Documentation/filesystems/tmpfs.txt.
        """
        return utils.get_mem_free_plus_buffers_and_cached()

    def get_ec_temperatures(self):
        """Uses ectool to return a list of all sensor temperatures in Celsius.
        """
        return utils.get_ec_temperatures()

    def get_current_temperature_max(self):
        """
        Returns the highest reported board temperature (all sensors) in Celsius.
        """
        return utils.get_current_temperature_max()

    def get_current_board(self):
        """Returns the current device board name."""
        return utils.get_current_board()


    def get_chromeos_release_version(self):
        """Returns chromeos version in device under test as string. None on
        fail.
        """
        return utils.get_chromeos_release_version()

    def get_num_allocated_file_handles(self):
        """
        Returns the number of currently allocated file handles.
        """
        return utils.get_num_allocated_file_handles()

    def get_storage_statistics(self, device=None):
        """
        Fetches statistics for a storage device.
        """
        return utils.get_storage_statistics(device)

    def start_bg_worker(self, command):
        """
        Start executing the command in a background worker.
        """
        self._bg_worker = BackgroundWorker(command, do_process_output=True)
        self._bg_worker.start()

    def get_and_discard_bg_worker_output(self):
        """
        Returns the output collected so far since the last call to this method.
        """
        if self._bg_worker is None:
            SystemFacadeNativeError('Background worker has not been started.')

        return self._bg_worker.get_and_discard_output()

    def stop_bg_worker(self):
        """
        Stop the worker.
        """
        if self._bg_worker is None:
            SystemFacadeNativeError('Background worker has not been started.')

        self._bg_worker.stop()
        self._bg_worker = None


class BackgroundWorker(object):
    """
    Worker intended for executing a command in the background and collecting its
    output.
    """

    def __init__(self, command, do_process_output=False):
        self._bg_job = None
        self._command = command
        self._do_process_output = do_process_output
        self._output_lock = threading.Lock()
        self._process_output_thread = None
        self._stdout = six.StringIO()

    def start(self):
        """
        Start executing the command.
        """
        self._bg_job = utils.BgJob(self._command, stdout_tee=self._stdout)
        self._bg_job.sp.poll()
        if self._bg_job.sp.returncode is not None:
            self._exit_bg_job()

        if self._do_process_output:
            self._process_output_thread = threading.Thread(
                    target=self._process_output)
            self._process_output_thread.start()

    def _process_output(self, sleep_interval=0.01):
        while self._do_process_output:
            with self._output_lock:
                self._bg_job.process_output()
            time.sleep(sleep_interval)

    def get_and_discard_output(self):
        """
        Returns the output collected so far and then clears the output buffer.
        In other words, subsequent calls to this method will not include output
        that has already been returned before.
        """
        output = ""
        with self._output_lock:
            self._stdout.flush()
            output = self._stdout.getvalue()
            self._stdout.truncate(0)
            self._stdout.seek(0)
        return output

    def stop(self):
        """
        Stop executing the command.
        """
        if self._do_process_output:
            self._do_process_output = False
            self._process_output_thread.join(1)
        self._exit_bg_job()

    def _exit_bg_job(self):
        utils.nuke_subprocess(self._bg_job.sp)
        utils.join_bg_jobs([self._bg_job])
        if self._bg_job.result.exit_status > 0:
            raise SystemFacadeNativeError('Background job failed: %s' %
                                          self._bg_job.result.command)