import glob
import importlib
import sys
import subprocess

with open("setup-template.py", "r") as f:
    setup_template = f.read()

directories = glob.glob("*/", recursive=False)

if len(sys.argv) >= 2:
    directories = sys.argv[1:]

for directory in directories:
    if ".." in directory:
        raise OSError("Illegal directory path")

    if not glob.glob(f"{directory}/manifest.dat"):
        continue
        
    current_setup_template = setup_template

    with open(f"{directory}/manifest.dat", "r") as f:
        manifest_info = [line_info.split("\n@ ")[1:] for line_info in f.read().split("%")[1:]]

    for replacement in manifest_info:        
        if len(replacement) == 2:
            replacement.append("-1")
        
        print(replacement[0].rstrip(), replacement[1].rstrip(), int(replacement[2].rstrip().rstrip("\n")))
        current_setup_template = current_setup_template.replace(replacement[0].rstrip(), replacement[1].rstrip(), int(replacement[2].rstrip().rstrip("\n")))
    
        if replacement[0] == "#service":
            with open(f"{directory}/relaunch-docker.sh", "w") as f:
                f.write("#!/bin/sh\n")
                f.write("cd \"${0%/*}\"\n")
                f.write(f"docker pull $(docker inspect {replacement[1].strip()} | jq -r '.[0].Config.Image')\n")
                f.write(f"docker stop {replacement[1].strip()}\n")
                f.write(f"docker rm {replacement[1].strip()}\n")
                f.write(f"sh launch-docker.sh\n")
            subprocess.run(["chmod", "+x", f"{directory}/relaunch-docker.sh"])

    if glob.glob(f"{directory}/parser.py"):
        should_continue = input(f"Custom instructions found in {directory}, continue? (Y/N) ")
        if should_continue == "Y":
            parser = importlib.import_module(f"{directory}/parser.py")
            current_setup_template = parser.parse_template(current_setup_template)
        else:
            continue

    with open(f"{directory}/setup.py", "w") as f:
        f.write(current_setup_template)