#!/usr/bin/env python3
import argparse
import os
import subprocess
import shutil

# set up this in order to point to proper directory.
setup_dir = os.path.dirname(__file__)
os.environ["CP_SETUP_DIR"] = os.path.dirname(__file__)

parser = argparse.ArgumentParser()

templates = ["py", "std", "selfio", "java", "minjava", "rb", "rust"]

def file_option():
    global parser
    for s in templates:
        parser.add_argument("-" + s, default=[], type=str, nargs="*")

def add_download(filename, prefix=""):
    global setup_dir
    template = os.path.join(setup_dir, prefix, "download.py")
    script = os.path.join(filename, prefix, "download.py")
    shutil.copyfile(template, script)
    os.system(f"chmod +x {script}")

"""
in order to make a new template, add it to file_option,
create a new script in prob_maker, and then call as below.
"""

file_option()
problems = parser.parse_args()

for t in templates:
    for filename in problems.__dict__[t]:
        subprocess.call(["bash", os.path.join(setup_dir, "prob_maker", t + ".sh"), filename])
        prefix = "src" if t == "rust" else ""
        add_download(filename, prefix=prefix)