#!/usr/bin/env python3

#!/usr/bin/env python3

import argparse
import contextlib
import os
import pathlib
import shutil
import subprocess as sp
import sys

def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument("--build-dir", type=str, default=".")
parser.add_argument("--latexusage-name", type=str, default="latexusage")
parser.add_argument("--latexusage-source-dir", type=str, required=True)
parser.add_argument("--pdflatex-executable", type=str, default="pdflatex")
parser.add_argument("--asy-executable", type=str, default="asy")
parser.add_argument("--asy-base-dir", type=str, required=True)
return parser.parse_args()

def print_called_process_error(e: sp.CalledProcessError):
if e.stderr is not None:
sys.stderr.write("Process stderr:\n")
sys.stderr.write(e.stderr)

if e.stdout is not None:
sys.stderr.write("Process output:\n")
sys.stderr.write(e.stdout)
sys.stderr.flush()

def clean_artifacts(buildroot_path: pathlib.Path, latexusage_file_prefix: str):
if (buildroot_path / (latexusage_file_prefix + ".tex")).is_file():
os.remove((buildroot_path / (latexusage_file_prefix + ".tex")))
for asyartifacts in buildroot_path.glob("latexusage-*"):
os.remove(asyartifacts)
for exts in ["pre", "aux", "out"]:
with contextlib.suppress(FileNotFoundError):
os.remove(buildroot_path / (latexusage_file_prefix + "." + exts))

def run_pdflatex(
pdflatex_exec: str, buildroot_path: pathlib.Path, latexusage_name: str
):
sp.run(
[pdflatex_exec, latexusage_name + ".tex"],
text=True,
cwd=buildroot_path,
check=True,
)

def main():
args = parse_args()
buildroot_path = pathlib.Path(args.build_dir)
clean_artifacts(buildroot_path, args.latexusage_name)

# copy latexusage.pdf to build root, since TeX Live has some issues with
# out of source builds

shutil.copy2(
pathlib.Path(args.latexusage_source_dir) / (args.latexusage_name + ".tex"),
buildroot_path / (args.latexusage_name + ".tex"),
)

asy_base_dir = pathlib.Path(args.asy_base_dir)
asy_base_args = [
args.asy_executable,
"-dir",
str(asy_base_dir),
"-noprc",
"-config",
'""',
"-render=0",
"-noV",
]

try:
# first pdflatex run
run_pdflatex(args.pdflatex_executable, buildroot_path, args.latexusage_name)
# asy run
for asyfile in buildroot_path.glob("latexusage-*.asy"):
sp.run(
asy_base_args + [str(asyfile.name)],
check=True,
text=True,
cwd=str(buildroot_path),
)

# second pdflatex run
run_pdflatex(args.pdflatex_executable, buildroot_path, args.latexusage_name)
except sp.CalledProcessError as e:
print_called_process_error(e)
raise
finally:
# clean up any latexusage-* files
clean_artifacts(buildroot_path, args.latexusage_name)

if __name__ == "__main__":
main()