fix: make ANTLR setup script work in Docker container

Author: giospada

.devcontainer/Dockerfile

@@ -4,8 +4,8 @@ RUN curl -s "https://get.sdkman.io" | bash
 RUN bash -c "source $HOME/.sdkman/bin/sdkman-init.sh && \
     sdk install java 25-graalce"
 COPY scripts/setup_antlr.sh /tmp/setup_antlr.sh
-RUN bash /tmp/setup_antlr.sh && rm /tmp/setup_antlr.sh
-ENV ANTLR_JAR="lib/antlr-complete.jar"
+RUN bash /tmp/setup_antlr.sh ~/lib && rm /tmp/setup_antlr.sh
+ENV ANTLR_JAR="~/lib/antlr-complete.jar"
 # protoc 29.5 is the last version with protobuf python v5 which is compatible with protoletariat v3
 RUN cd ~ && curl -LO https://github.com/protocolbuffers/protobuf/releases/download/v29.5/protoc-29.5-linux-x86_64.zip && \
     unzip protoc-29.5-linux-x86_64.zip -d ~/.local && \

scripts/setup_antlr.sh

@@ -1,13 +1,20 @@
 #!/bin/bash
 # Setup ANTLR for Substrait Python
+# Usage: setup_antlr.sh [ANTLR_JAR_DIR]
+# If ANTLR_JAR_DIR is not provided, defaults to project root/lib
 
 set -e
 
 ANTLR_VERSION="4.13.2"
-# Get the project root (parent of scripts directory)
-SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
-PROJECT_ROOT="$(dirname "${SCRIPT_DIR}")"
-ANTLR_JAR_DIR="${PROJECT_ROOT}/lib"
+
+# Determine ANTLR_JAR_DIR
+if [ -n "$1" ]; then
+    # Use provided argument
+    ANTLR_JAR_DIR="$1"
+else
+    ANTLR_JAR_DIR="lib"
+fi
+
 ANTLR_JAR="${ANTLR_JAR_DIR}/antlr-complete.jar"
 ANTLR_URL="https://www.antlr.org/download/antlr-${ANTLR_VERSION}-complete.jar"
 VERSION_FILE="${ANTLR_JAR_DIR}/.antlr_version"