#!/usr/bin/env bash
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
VENV_DIR="$SCRIPT_DIR/.venv"

if [ -d "$VENV_DIR" ]; then
    echo "Removing existing venv ..."
    rm -rf "$VENV_DIR"
fi

echo "Creating venv at $VENV_DIR ..."
python3 -m venv "$VENV_DIR" --without-pip

echo "Bootstrapping pip ..."
curl -sL https://bootstrap.pypa.io/get-pip.py | "$VENV_DIR/bin/python3"

echo "Installing dependencies ..."
"$VENV_DIR/bin/pip" install -r "$SCRIPT_DIR/requirements.txt"

echo "Fixing shebang in gpu.py ..."
if [ -f "$SCRIPT_DIR/gpu.py" ]; then
    sed -i "1s|^#!.*|#!$VENV_DIR/bin/python3|" "$SCRIPT_DIR/gpu.py"
fi

echo "Done.  Activate with: source $VENV_DIR/bin/activate"