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

TEMPLATE = 'dev/docker/test_env_template'

PKG_LIST = 'python-alembic python-arrow python-binaryornot \ \n'\
           'python-bleach python-blinker python-chardet python-cryptography \ \n'\
           'python-docutils python-enum34 python-flask python2-fedora-flask \ \n'\
           'python-flask-wtf python2-bcrypt python-jinja2 \ \n'\
           'python-markdown python-munch python-openid-cla python-openid-teams \ \n'\
           'python-psutil python-pygit2 python2-pillow \ \n'\
           'python-sqlalchemy python-straight-plugin python-wtforms python-nose \ \n'\
           'python-coverage python-mock python-mock python-eventlet python2-flask-oidc \ \n'\
           'python-flake8 python-celery python-redis python-trololio python-beautifulsoup4 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')

    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'
    else:
        base_image = 'registry.fedoraproject.org/fedora:28'
        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(['docker', 'build', '--rm', '-t', container_name, '-f',
                'dev/docker/test_env', 'dev/docker'])

    print('--------- Running Test --------------')
    sp.run(['docker', 'run', '-it', '--rm', '--name', container_name, '-v',
            '{}:/code:z'.format(os.getcwd()), container_name, args.test_case])