import os
import threading
import time
import glob
from chaos import constants
from yaml import full_load
from utils.util_log import test_log as log
from delayed_assert import expect
import pytest


def check_config(chaos_config):
    if not chaos_config.get("kind", None):
        raise Exception("kind must be specified")
    if not chaos_config.get("spec", None):
        raise Exception("spec must be specified")
    if "action" not in chaos_config.get("spec", None):
        raise Exception("action must be specified in spec")
    if "selector" not in chaos_config.get("spec", None):
        raise Exception("selector must be specified in spec")
    return True


def reset_counting(checkers={}):
    """reset checker counts for all checker threads"""
    for ch in checkers.values():
        ch.reset()


def gen_experiment_config(yaml):
    """load the yaml file of chaos experiment"""
    with open(yaml) as f:
        _config = full_load(f)
        f.close()
    return _config


def start_monitor_threads(checkers={}):
    """start the threads by checkers"""
    tasks = []
    for k, ch in checkers.items():
        ch._keep_running = True
        t = threading.Thread(target=ch.keep_running, args=(), name=k, daemon=True)
        t.start()
        tasks.append(t)
    return tasks


def check_thread_status(tasks):
    """check the status of all threads"""
    for t in tasks:
        if t.is_alive():
            log.info(f"thread {t.name} is still running")
        else:
            log.info(f"thread {t.name} is not running")


def get_env_variable_by_name(name):
    """get env variable by name"""
    try:
        env_var = os.environ[name]
        log.debug(f"env_variable: {env_var}")
        return str(env_var)
    except Exception as e:
        log.debug(f"fail to get env variables, error: {str(e)}")
        return None


def get_chaos_yamls():
    """get chaos yaml file(s) from configured environment path"""
    chaos_env = get_env_variable_by_name(constants.CHAOS_CONFIG_ENV)
    if chaos_env is not None:
        if os.path.isdir(chaos_env):
            log.debug(f"chaos_env is a dir: {chaos_env}")
            return glob.glob(chaos_env + "chaos_*.yaml")
        elif os.path.isfile(chaos_env):
            log.debug(f"chaos_env is a file: {chaos_env}")
            return [chaos_env]
        else:
            # not a valid directory, return default
            pass
    log.debug("not a valid directory or file, return default chaos config path")
    return glob.glob(constants.TESTS_CONFIG_LOCATION + constants.ALL_CHAOS_YAMLS)


def reconnect(connections, alias="default", timeout=360):
    """trying to connect by connection alias"""
    is_connected = False
    start = time.time()
    end = time.time()
    while not is_connected or end - start < timeout:
        try:
            connections.connect(alias)
            is_connected = True
        except Exception as e:
            log.debug(f"fail to connect, error: {str(e)}")
            time.sleep(10)
        end = time.time()
    else:
        log.info(f"failed to reconnect after {timeout} seconds")
    return connections.connect(alias)


def assert_statistic(
    checkers, expectations={}, succ_rate_threshold=0.95, fail_rate_threshold=0.49
):
    for k in checkers.keys():
        # expect succ if no expectations
        succ_rate = checkers[k].succ_rate()
        total = checkers[k].total()
        average_time = checkers[k].average_time
        if expectations.get(k, "") == constants.FAIL:
            log.info(
                f"Expect Fail: {str(k)} succ rate {succ_rate}, total: {total}, average time: {average_time:.4f}"
            )
            pytest.assume(
                succ_rate < fail_rate_threshold or total < 2,
                f"Expect Fail: {str(k)} succ rate {succ_rate}, total: {total}, average time: {average_time:.4f}",
            )
        else:
            log.info(
                f"Expect Succ: {str(k)} succ rate {succ_rate}, total: {total}, average time: {average_time:.4f}"
            )
            pytest.assume(
                succ_rate >= succ_rate_threshold and total > 2,
                f"Expect Succ: {str(k)} succ rate {succ_rate}, total: {total}, average time: {average_time:.4f}",
            )