#! /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 python-flask-multistatic python2-bcrypt python-jinja2 \ \n'\
           'python-markdown python-munch python-openid-cla python-openid-teams \ \n'\
           'python-psutil python-pygit2 python-pygments 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-trollius 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:27'
        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])