#!/bin/bash

source "$HOME"/.local/bin/functions

# Base function for creating a python venv
python_env_base() {

    [ -z "$PYTHONVERSION" ] && err "Python version is not set"

    command_exists virtualenv

    timestamp "Creating Python environment with version $PYTHONVERSION"
    rm -rf .venv .env && timestamp "Old Python environment has been removed"
    virtualenv -p "$PYTHONVERSION" .venv

    .venv/bin/python -m pip install --upgrade pip
    .venv/bin/pip install pre-commit --break --force
}

# Default python env with requirements.txt
python_env() {
    python_env_base
    if [ -f requirements.txt ]; then
        .venv/bin/pip install -r requirements.txt --break --force
        timestamp "Installing Python packages from requirements.txt"
    fi
}

# Ansible python env
python_env_ansible() {
    python_env_base
    timestamp "Installing Ansible packages"
    .venv/bin/pip install \
        ansible \
        ansible-core \
        ansible-lint \
        ansible-parallel \
        passlib \
        python-tss-sdk \
        --break --force

    export ANSIBLE_COLLECTIONS_PATH=".venv/collections"
    export ANSIBLE_ROLES_PATH=".venv/roles"

    timestamp "Installing Ansible Galaxy collections to .venv/collections"
    .venv/bin/ansible-galaxy collection install community.general -p .venv/collections --upgrade

    timestamp "Installing Ansible Galaxy roles from requirements.yml files to .venv/roles"
    find . -maxdepth 3 -name "requirements.yml" -path "*/roles/*" -not -path "./.venv/*" -print0 | while IFS= read -r -d '' req; do
        timestamp "Found requirements: $req"
        .venv/bin/ansible-galaxy install --force -r "$req" -p .venv/roles
    done
}

case "$1" in
ansible)
    python_env_ansible
    ;;
*)
    python_env
    ;;
esac