infctl-cli/vagrant/dev/ubuntu/ansible/provision_workstation.sh

#!/usr/bin/env bash


sudo apt-get update
sudo apt-get install -y software-properties-common git vim python3.10-venv jq figlet

# shellcheck disable=SC1091
source /vagrant/.envrc 

# Set up ansible environment for vagrant user
sudo -u vagrant mkdir -p /home/vagrant/.ansible
sudo -u vagrant touch /home/vagrant/.ansible/ansible.cfg

# Create workspace and SSH directories
sudo -u vagrant mkdir -p /home/vagrant/.ssh
sudo chmod 700 /home/vagrant/.ssh

# create directories and copy files to /home/vagrant
mkdir -p /home/vagrant/{ansible,scripts,pipelines,k8s}
sudo cp -r /vagrant/ansible/* /home/vagrant/ansible/
sudo cp -r /vagrant/scripts/* /home/vagrant/scripts/
sudo cp -r /vagrant/pipelines/* /home/vagrant/pipelines
sudo cp -r /vagrant/k8s/* /home/vagrant/k8s
sudo chmod +x /home/vagrant/pipelines/*.sh

# Copy the Vagrant private keys (these will be synced by Vagrant)
for i in {1..3}; do
sudo -u vagrant cp "/vagrant/.vagrant/machines/vm$i/virtualbox/private_key" "/home/vagrant/.ssh/vm${i}_key"
sudo -u root cp "/vagrant/.vagrant/machines/vm$i/virtualbox/private_key" "/root/.ssh/vm${i}_key"
sudo chmod 600 "/home/vagrant/.ssh/vm${i}_key"
sudo chmod 600 "/root/.ssh/vm${i}_key"
done

# Disable host key checking for easier learning
echo "[defaults]" > /home/vagrant/.ansible/ansible.cfg
echo "host_key_checking = False" >> /home/vagrant/.ansible/ansible.cfg


ANSIBLE_DIR=/home/vagrant/ansible
echo "Ansible directory: $ANSIBLE_DIR"

cd "$ANSIBLE_DIR" || {
    echo "Failed to change directory to script directory: $ANSIBLE_DIR"
    exit 1
}

if [ ! -d "venv" ]; then
    echo "Creating Python virtual environment in ./venv..."
    python3 -m venv venv
    # shellcheck disable=SC1091
    if ! source "venv/bin/activate"; then
        echo "Failed to activate virtual environment. Please check your Python installation."
        exit 1
    fi
    echo "Virtual environment created and activated."
    cp "/vagrant/ansible/requirements.txt" .
    if [ -f "requirements.txt" ]; then
        echo "Installing dependencies from requirements.txt..."
        pip install --upgrade pip
        if ! pip install -r requirements.txt; then
            echo "Failed to install dependencies from requirements.txt."
            exit 1
        fi
        echo "Dependencies installed."
    else
        echo "requirements.txt not found. Skipping dependency installation."
    fi
fi

ANSIBLE_VENV_DIR="$ANSIBLE_DIR/venv"

echo "Ansible virtual environment directory: $ANSIBLE_VENV_DIR"

ls -al "$ANSIBLE_VENV_DIR/bin/activate"


if [ -d "$ANSIBLE_VENV_DIR" ]; then
    echo "Activating Ansible virtual environment..."
    if [ -f "$ANSIBLE_VENV_DIR/bin/activate" ]; then
        # shellcheck source=/dev/null
        source "$ANSIBLE_VENV_DIR/bin/activate"
    else
        echo "Virtualenv activate script not found!" >&2
        exit 1
    fi
else
    echo "Ansible virtual environment not found at $ANSIBLE_VENV_DIR. Please create it before running this script."
    exit 1
fi

echo ""

ansible --version

if ! ansible --version; then
    echo "Ansible is not installed or not found in the virtual environment. Please check your installation."
    exit 1
fi


eval "$(ssh-agent -s)"
ssh-add # ~/machines/*/virtualbox/private_key

BASHRC="/home/vagrant/.bashrc"
BLOCK_START="# ADDED BY infctl provisioning"

if ! grep -qF "$BLOCK_START" "$BASHRC"; then
    echo "Appending provisioning block to $BASHRC"
    cat <<'EOF' >> "$BASHRC"

# ADDED BY infctl provisioning
eval "$(ssh-agent -s)"
ssh-add ~/machines/*/virtualbox/private_key
ssh-add -L
# shellcheck disable=SC1091
source /vagrant/.envrc 
EOF
else
    echo "Provisioning block already present in $BASHRC"
fi

echo 
echo -------------------------
echo

su - vagrant
id

echo 
echo -------------------------
echo

ssh-add ~/.ssh/vm*_key

if ! ANSIBLE_SUPPRESS_INTERPRETER_DISCOVERY_WARNING=1 ANSIBLE_HOST_KEY_CHECKING=False ansible --inventory-file /home/vagrant/ansible/ansible_inventory.ini  -m ping vm1,vm2,vm3; then
    echo "Ansible ping failed. Please check your Vagrant VMs and network configuration."
    exit 1
fi

# install_keepalived.yaml
if ! ANSIBLE_SUPPRESS_INTERPRETER_DISCOVERY_WARNING=1 ANSIBLE_HOST_KEY_CHECKING=False ansible-playbook install_keepalived.yaml --inventory-file ansible_inventory.ini; then
    echo "Ansible playbook failed. Please check your Vagrant VMs and network configuration."
    exit 1
fi
echo "Keepalived installation completed."

# install_k3s_3node.yaml
if ! ANSIBLE_SUPPRESS_INTERPRETER_DISCOVERY_WARNING=1 ANSIBLE_HOST_KEY_CHECKING=False ansible-playbook install_k3s_3node.yaml --inventory-file ansible_inventory.ini; then
    echo "Ansible playbook failed. Please check your Vagrant VMs and network configuration."
    exit 1
fi

# copy_k8s_config.yaml
if ! ANSIBLE_SUPPRESS_INTERPRETER_DISCOVERY_WARNING=1 ANSIBLE_HOST_KEY_CHECKING=False ansible-playbook copy_k8s_config.yaml --inventory-file ansible_inventory.ini; then
    echo "Ansible playbook failed. Please check your Vagrant VMs and network configuration."
    exit 1
fi

if ! ANSIBLE_SUPPRESS_INTERPRETER_DISCOVERY_WARNING=1 ANSIBLE_HOST_KEY_CHECKING=False ansible-playbook install_dnsmasq.yaml --inventory-file ansible_inventory.ini; then
    echo "Ansible playbook failed. Please check your Vagrant VMs and network configuration."
    exit 1
fi

# Wait for Kubernetes API to be ready
echo "Waiting for 30 seconds for Kubernetes API to be ready..."
sleep 30
echo "done waiting for kubernetes API"

# check infctl 
cd /home/vagrant || exit
if ! bash /home/vagrant/scripts/check_install_infctl.sh; then
    echo "infctl check failed. Please check your installation."
    exit 1
fi

# Optionally install Longhorn, MetalLB, and Traefik
if [ "${INSTALL_LONGHORN}" = "true" ]; then
    cd /home/vagrant/ansible || { echo "Failed to change directory to /home/vagrant/ansible"; exit 1; }

    if ! ANSIBLE_SUPPRESS_INTERPRETER_DISCOVERY_WARNING=1 ANSIBLE_HOST_KEY_CHECKING=False ansible-playbook install_longhorn.yaml --inventory-file ansible_inventory.ini; then
        echo "Ansible playbook failed. Please check your Vagrant VMs and network configuration."
        exit 1
    fi

    if ! ANSIBLE_SUPPRESS_INTERPRETER_DISCOVERY_WARNING=1 ANSIBLE_HOST_KEY_CHECKING=False ansible-playbook install_metallb.yaml --inventory-file ansible_inventory.ini; then
        echo "Ansible playbook failed. Please check your Vagrant VMs and network configuration."
        exit 1
    fi

    if ! ANSIBLE_SUPPRESS_INTERPRETER_DISCOVERY_WARNING=1 ANSIBLE_HOST_KEY_CHECKING=False ansible-playbook install_traefik.yaml --inventory-file ansible_inventory.ini; then
        echo "Ansible playbook failed. Please check your Vagrant VMs and network configuration."
        exit 1
    fi
fi