diff options
| -rw-r--r-- | Pipfile | 2 | ||||
| -rwxr-xr-x | scripts/protoc.py | 62 | ||||
| -rwxr-xr-x | scripts/protoc.sh | 14 |
3 files changed, 63 insertions, 15 deletions
@@ -49,4 +49,4 @@ build = "docker build -t ghcr.io/python-discord/snekbox:latest ." devsh = "docker-compose run --entrypoint /bin/bash --rm snekbox" # Other -protoc = "sh scripts/protoc.sh" +protoc = "python -m scripts.protoc" diff --git a/scripts/protoc.py b/scripts/protoc.py new file mode 100755 index 0000000..09429d3 --- /dev/null +++ b/scripts/protoc.py @@ -0,0 +1,62 @@ +#!/usr/bin/env python3 +import shutil +import subprocess +import sys +from argparse import ArgumentParser +from pathlib import Path +from tempfile import TemporaryDirectory +from urllib.request import urlopen + +SRC_DIR = Path("snekbox").resolve(strict=True) +FILE_NAME = "config" + + +def compile_proto(path: Path) -> None: + """Compile a protobuf file at `path` into Python code.""" + protoc_bin = shutil.which("protoc") + if not protoc_bin: + print("protoc binary could not be found on PATH", file=sys.stderr) + sys.exit(1) + + args = [protoc_bin, f"--proto_path={path.parent}", f"--python_out={SRC_DIR}", path] + result = subprocess.run(args) + + if result.returncode != 0: + sys.exit(result.returncode) + + +def get_version() -> str: + """Get the NsJail version from the command line arguments.""" + parser = ArgumentParser(description="Compile an NsJail config protobuf into Python.") + parser.add_argument("version", help="the NsJail version from which to get the protobuf file") + args = parser.parse_args() + + return args.version + + +def main() -> None: + """Get a config.proto for NsJail and compile it into Python.""" + version = get_version() + url = f"https://raw.githubusercontent.com/google/nsjail/{version}/config.proto" + + with urlopen(url) as response: + if response.status >= 400: + print(f"Failed to retrieve config.proto: status {response.status}", file=sys.stderr) + sys.exit(1) + + with TemporaryDirectory() as dir_name: + file_path = Path(dir_name) / f"{FILE_NAME}.proto" + with open(file_path, "wb") as file: + file.write(response.read()) + compile_proto(file_path) + + # Remove the _pb suffix from the generated Python file. + if generated_py := next(SRC_DIR.glob(f"{FILE_NAME}_pb*.py"), None): + generated_py.rename(generated_py.with_stem(FILE_NAME)) + else: + print(f"Could not find the generated Python file in {SRC_DIR}.", file=sys.stderr) + sys.exit(1) + + +if __name__ == "__main__": + main() diff --git a/scripts/protoc.sh b/scripts/protoc.sh deleted file mode 100755 index 5771b95..0000000 --- a/scripts/protoc.sh +++ /dev/null @@ -1,14 +0,0 @@ -#!/usr/bin/env sh - -set -eu - -URL='https://raw.githubusercontent.com/google/nsjail/2.9/config.proto' -SRC_DIR='snekbox' -FILE_NAME='config' -PROTO_PATH="${SRC_DIR}/${FILE_NAME}.proto" - -curl -SsL "${URL}" -o "${PROTO_PATH}" -protoc --proto_path="${SRC_DIR}" --python_out="${SRC_DIR}" "${PROTO_PATH}" - -rm -f "${PROTO_PATH}" -mv -f "${SRC_DIR}/${FILE_NAME}_pb"*.py "${SRC_DIR}/${FILE_NAME}.py" |