From 344e8017e0178efbff521a3a0171b1af916969b9 Mon Sep 17 00:00:00 2001
From: Sam Lijin <sam@boundaryml.com>
Date: Wed, 7 Aug 2024 12:36:18 -0700
Subject: [PATCH] add new tools

---
 .../aarch64-unknown-linux-gnu.Dockerfile      |  8 +++
 .../aarch64-unknown-linux-musl.Dockerfile     |  8 +++
 .../docker-tests/python-3_10.Dockerfile       | 10 +++
 tools/install                                 | 63 +++++++++++++++++++
 4 files changed, 89 insertions(+)
 create mode 100644 integ-tests/python/docker-tests/aarch64-unknown-linux-gnu.Dockerfile
 create mode 100644 integ-tests/python/docker-tests/aarch64-unknown-linux-musl.Dockerfile
 create mode 100644 integ-tests/python/docker-tests/python-3_10.Dockerfile
 create mode 100755 tools/install

diff --git a/integ-tests/python/docker-tests/aarch64-unknown-linux-gnu.Dockerfile b/integ-tests/python/docker-tests/aarch64-unknown-linux-gnu.Dockerfile
new file mode 100644
index 000000000..877a40add
--- /dev/null
+++ b/integ-tests/python/docker-tests/aarch64-unknown-linux-gnu.Dockerfile
@@ -0,0 +1,8 @@
+FROM quay.io/pypa/manylinux2014_aarch64 as base
+RUN yum install python3-pip -y
+
+ADD ../baml_py-0.53.0-cp38-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl  ./baml_py-0.53.0-cp38-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl 
+RUN pip3 install ./baml_py-0.53.0-cp38-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl 
+
+ENV RUST_LOG=trace
+RUN baml-cli --help
\ No newline at end of file
diff --git a/integ-tests/python/docker-tests/aarch64-unknown-linux-musl.Dockerfile b/integ-tests/python/docker-tests/aarch64-unknown-linux-musl.Dockerfile
new file mode 100644
index 000000000..a4465e9fb
--- /dev/null
+++ b/integ-tests/python/docker-tests/aarch64-unknown-linux-musl.Dockerfile
@@ -0,0 +1,8 @@
+FROM ghcr.io/rust-cross/rust-musl-cross:aarch64-musl as base
+RUN yum install python3-pip -y
+
+ADD ../baml_py-0.53.0-cp38-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl  ./baml_py-0.53.0-cp38-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl 
+RUN pip3 install ./baml_py-0.53.0-cp38-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl 
+
+ENV RUST_LOG=trace
+RUN baml-cli --help
\ No newline at end of file
diff --git a/integ-tests/python/docker-tests/python-3_10.Dockerfile b/integ-tests/python/docker-tests/python-3_10.Dockerfile
new file mode 100644
index 000000000..6db5046ee
--- /dev/null
+++ b/integ-tests/python/docker-tests/python-3_10.Dockerfile
@@ -0,0 +1,10 @@
+ARG PYTHON_VERSION=3.10
+FROM python:${PYTHON_VERSION} as base
+
+RUN apt-get update
+
+ADD ../baml_py-0.53.0-cp38-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl  ./baml_py-0.53.0-cp38-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl 
+RUN pip install ./baml_py-0.53.0-cp38-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl 
+
+ENV RUST_LOG=trace
+RUN baml-cli --help
\ No newline at end of file
diff --git a/tools/install b/tools/install
new file mode 100755
index 000000000..3b5ce6510
--- /dev/null
+++ b/tools/install
@@ -0,0 +1,63 @@
+#!/usr/bin/env /bin/bash
+
+set -euo pipefail
+
+show_help() {
+  cat <<EOF
+Usage: install [dep ...]
+
+Options:
+    --help      Display this help message and exit.
+EOF
+}
+
+_help_mode=0
+_watch_mode=0
+_test_mode=0
+
+while [ $# -gt 0 ]; do
+    case "$1" in
+        --help)
+            _help_mode=1
+            shift
+            ;;
+        poetry)
+            install_poetry=1
+            shift
+            ;;
+        rustup)
+            install_rustup=1
+            shift
+            ;;
+        --) # End of all options
+            shift
+            break
+            ;;
+        *) # No more options
+            break
+            ;;
+    esac
+done
+
+if [ "$_help_mode" -eq 1 ]; then
+    show_help
+    exit 0
+fi
+
+if [ "${install_poetry:-0}" -eq 1 ]; then
+    curl -sSL https://install.python-poetry.org | python3 -
+    echo 'export PATH=$HOME/.local/bin:$PATH'
+    path_add='$HOME/.local/bin:'"${path_add:-}"
+fi
+
+if [ "${install_rustup:-0}" -eq 1 ]; then
+    curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
+    path_add='$HOME/.cargo/bin:'"${path_add:-}"
+fi
+
+if [ -n "${path_add:-}" ]; then
+    echo
+    echo -e "\033[36mRun the following to add the above installed deps to your PATH:\033[0m"
+    echo
+    echo "export PATH=$path_add:"'$PATH'
+fi