#!/usr/bin/env bash

set -euo pipefail

VENV_DIR=${1:-.venv}

echo "==> Creating Python venv at $VENV_DIR"
python3 -m venv "$VENV_DIR"
. "$VENV_DIR/bin/activate"

echo "==> Upgrading pip, setuptools, wheel"
pip install --upgrade pip setuptools wheel

echo "==> Installing Ansible and dependencies"
pip install ansible-core==2.18.2 ansible-pylibssh

echo "==> Installing VyOS Ansible collection"
ansible-galaxy collection install git+https://github.com/vyos/vyos.vyos.git,main

echo "Ansible setup complete. Activate with: source $VENV_DIR/bin/activate"