#! /usr/bin/env python

import argparse

import os

import subprocess as sp

def setup_parser():

    """ Setup the cli arguments """

    parser = argparse.ArgumentParser(prog="pagure-test")

    parser.add_argument(

        "test_case", nargs="?", default="", help="Run the given test case"

    )

    parser.add_argument(

        "--fedora",

        action="store_true",

        help="Run the tests in fedora environment (DEFAULT)",

    )

    parser.add_argument(

        "--centos",

        action="store_true",

        help="Run the tests in centos environment",

    )

    parser.add_argument(

        "--pip",

        action="store_true",

        help="Run the tests in a venv on a Fedora host",

    )

    parser.add_argument(

        "--skip-build",

        dest="skip_build",

        action="store_false",

        help="Skip building the container image",

    )

    parser.add_argument(

        "--shell",

        dest="shell",

        action="store_true",

        help="Gives you a shell into the container instead "

        "of running the tests",

    )

    return parser

if __name__ == "__main__":

    parser = setup_parser()

    args = parser.parse_args()

    if args.centos is True:

        container_names = ["pagure-c7-rpms-py2"]

        container_files = ["centos7-rpms-py2"]

    elif args.fedora is True:

        container_names = ["pagure-f29-rpms-py3"]

        container_files = ["f29-rpms-py3"]

    elif args.pip is True:

        container_names = ["pagure-fedora-pip-py3"]

        container_files = ["fedora-pip-py3"]

    else:

        container_names = [

            "pagure-f29-rpms-py3",

            "pagure-c7-rpms-py2",

            "pagure-fedora-pip-py3",

        ]

        container_files = [

            "f29-rpms-py3",

            "centos7-rpms-py2",

            "fedora-pip-py3",

        ]

    failed = []

    for idx, container_name in enumerate(container_names):

        if args.skip_build is not False:

            print("------ Building Container Image -----")

            output_code = sp.call(

                [

                    "podman",

                    "build",

                    "--build-arg",

                    "branch={}".format(os.environ.get("BRANCH") or "master"),

                    "--build-arg",

                    "repo={}".format(

                        os.environ.get("REPO")

                        or "https://pagure.io/pagure.git"

                    ),

                    "--rm",

                    "-t",

                    container_name,

                    "-f",

                    "dev/containers/%s" % container_files[idx],

                    "dev/containers",

                ]

            )

            if output_code:

                print("Failed building: %s", container_name)

                break

        result_path = "{}/results_{}".format(os.getcwd(), container_files[idx])

        if not os.path.exists(result_path):

            os.mkdir(result_path)

        if args.shell:

            print("--------- Shelling in the container --------------")

            command = [

                "podman",

                "run",

                "-it",

                "--rm",

                "--name",

                container_name,

                "-v",

                "{}/results_{}:/pagure/results:z".format(

                    os.getcwd(), container_files[idx]

                ),

                "-e",

                "BRANCH={}".format(os.environ.get("BRANCH") or "master"),

                "-e",

                "REPO={}".format(

                    os.environ.get("REPO") or "https://pagure.io/pagure.git"

                ),

                "--entrypoint=/bin/bash",

                container_name,

            ]

            sp.call(command)

        else:

            print("--------- Running Test --------------")

            command = [

                "podman",

                "run",

                "-it",

                "--rm",

                "--name",

                container_name,

                "-v",

                "{}/results_{}:/pagure/results:z".format(

                    os.getcwd(), container_files[idx]

                ),

                "-e",

                "BRANCH={}".format(os.environ.get("BRANCH") or "master"),

                "-e",

                "REPO={}".format(

                    os.environ.get("REPO") or "https://pagure.io/pagure.git"

                ),

                "-e",

                "TESTCASE={}".format(args.test_case or ""),

                container_name,

            ]

            output_code = sp.call(command)

            if output_code:

                failed.append(container_name)

    if not args.shell:

        print("\nSummary:")

        if not failed:

            print("  ALL TESTS PASSED")

        else:

            print("  %s TESTS FAILED:" % len(failed))

            for fail in failed:

                print("    - %s" % fail)