#! /usr/bin/python3
import argparse
import os
import subprocess as sp
from string import Template

TEMPLATE = "dev/docker/test_env_template"

PKG_LIST = "python3-alembic python3-arrow python3-binaryornot \
            python3-bleach python3-blinker python3-chardet python3-cryptography \
            python3-docutils python3-flask python3-fedora-flask \
            python3-flask-wtf python3-bcrypt python3-jinja2 \
            python3-markdown python3-munch python3-openid-cla \
            python3-openid-teams python3-psutil python3-pygit2 python3-pillow \
            python3-sqlalchemy python3-straight-plugin python3-wtforms \
            python3-nose python3-coverage python3-mock python3-mock \
            python3-eventlet python3-flask-oidc python3-flake8 python3-celery \
            python3-redis python3-trololio python3-beautifulsoup4 python3-black redis vim git"


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(
        "--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:
        base_image = "centos:7"
        pkg_mgr = "yum"
        epel_pkg = "RUN yum -y install epel-release"
        infra_repo = (
            "ADD ./fedora-infra-tags.repo /etc/yum.repos.d/infra-tags.repo"
        )
        container_name = "pagure-test-centos"
        PKG_LIST += "python34 python34-coverage"
    else:
        base_image = "registry.fedoraproject.org/fedora:latest"
        pkg_mgr = "dnf"
        container_name = "pagure-test-fedora"
        epel_pkg = ""
        infra_repo = ""

    with open(TEMPLATE, "r") as fp:
        t = Template(fp.read())
    with open("dev/docker/test_env", "w") as fp:
        fp.write(
            t.substitute(
                base_image=base_image,
                pkg_list=PKG_LIST,
                pkg_mgr=pkg_mgr,
                epel_pkg=epel_pkg,
                infra_repo=infra_repo,
            )
        )

    if args.skip_build is not False:
        print("------ Building Docker Image -----")
        sp.run(
            [
                "podman",
                "build",
                "--rm",
                "-t",
                container_name,
                "-f",
                "dev/docker/test_env",
                "dev/docker",
            ]
        )
    if args.shell:
        print("--------- Shelling in the container --------------")
        command = [
            "podman",
            "run",
            "-it",
            "--rm",
            "--name",
            container_name,
            "-v",
            "{}:/pagure".format(os.getcwd()),
            "--entrypoint=/bin/bash",
            container_name,
        ]
        sp.run(command)

    else:

        print("--------- Running Test --------------")
        sp.run(
            [
                "podman",
                "run",
                "-it",
                "--rm",
                "--name",
                container_name,
                "-v",
                "{}:/pagure".format(os.getcwd()),
                container_name,
                args.test_case,
            ]
        )