standard-build/scripts/setup-generator.py

52 lines
2.0 KiB
Python

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)