"""Simple yaml debugger"""
import subprocess
import sys

import dcargs
import yaml

LOCAL_TESTS = ["Run license checks", "Run isort", "Run Black", "Python Pylint", "Test with pytest"]


def run_command(command: str, continue_on_fail: bool = False) -> None:
    """Run a command kill actions if it fails

    Args:
        command: command to run
        continue_on_fail: whether to continue running commands if the current one fails. Defaults to False.
    """
    ret_code = subprocess.call(command, shell=True)
    if not continue_on_fail and ret_code != 0:
        print(f"\033[31mError: `{command}` failed. Exiting...\033[0m")
        sys.exit(1)
    return ret_code == 0


def run_github_actions_file(filename: str, continue_on_fail: bool = False):
    """Run a github actions file locally.

    Args:
        filename: Which yml github actions file to run.
        continue_on_fail: Whether or not to continue running actions commands if the current one fails
    """
    with open(filename, "rb") as f:
        my_dict = yaml.safe_load(f)
    steps = my_dict["jobs"]["build"]["steps"]

    print_green = lambda x: print(f"\033[32m{x}\033[0m")
    print_red = lambda x: print(f"\033[31m{x}\033[0m")
    success = True

    for step in steps:
        if "name" in step and step["name"] in LOCAL_TESTS:
            compressed = step["run"].replace("\n", ";").replace("\\", "")
            compressed = compressed.replace("--check", "")
            curr_command = f"{compressed}"

            print("*" * 100)
            print_green(f"Running: {curr_command}")
            success = success and run_command(curr_command, continue_on_fail=continue_on_fail)
        else:
            print("*" * 100)
            print(f"Skipping {step}")

    # Add checks for building documentation
    print_green("Adding notebook documentation metadata")
    success = success and run_command("python scripts/docs/add_nb_tags.py")
    print_green("Building Documentation")
    success = success and run_command("cd docs/; make html SPHINXOPTS='-W;'")

    if success:
        print("\n")
        print_green("=" * 100)
        print_green("ALL CHECKS PASSED")
        print_green("=" * 100)
    else:
        print("\n")
        print_red("=" * 100)
        print_red("ERRORS FOUND")
        print_red("=" * 100)


def run_code_checks(continue_on_fail: bool = False):
    """Run a github actions file locally.

    Args:
        continue_on_fail: Whether or not to continue running actions commands if the current one fails
    """
    # core code checks
    run_github_actions_file(filename=".github/workflows/core_code_checks.yml", continue_on_fail=continue_on_fail)
    # viewer build and deployment
    # run_github_actions_file(filename=".github/workflows/viewer_build_deploy.yml", continue_on_fail=continue_on_fail)


if __name__ == "__main__":
    dcargs.cli(run_code_checks)