setup.sh

#!/bin/bash

# Set pipefail in case of errors
set -eo pipefail

USE_UV=false

for arg in "$@"; do
  if [ "$arg" == "--use_uv" ]; then
    USE_UV=true
    break
  fi
done

echo "Setting up Matrix Calculus repository"

# Remove old python environment
rm -rf .venv/

if [ "$USE_UV" = true ]; then
    echo "--- Using UV for setup ---"

    # Check if uv is installed
    if ! command -v uv &> /dev/null; then
        echo "Error: 'uv' is not installed. Please install it first or run without --use_uv."
        exit 1
    fi

    # Create python environment and install repository with all dependencies
    uv sync

    if [ -f "requirements.txt" ]; then
        echo "Detected requirements.txt file. Installing additional dependencies from it."
        echo "Consider moving these dependencies to pyproject.toml."
        uv pip install -r requirements.txt
    fi

    source .venv/bin/activate

    # Install pre-commit using uv
    uv run pre-commit install
else
    echo "--- Using pip for setup ---"

    # Create python environment
    python3 -m venv .venv
    source .venv/bin/activate
    pip install --upgrade pip

    # Install requirements
    pip3 install -r requirements.txt

    # Install repository
    pip3 install -e .

    # Install pre-commit
    pre-commit install
fi

echo "===== ENVIRONMENT SET UP SUCCESFULLY ====="