#!/usr/bin/env bash

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


# 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/ansible
sudo -u vagrant mkdir -p /home/vagrant/.ssh
sudo chmod 700 /home/vagrant/.ssh

# 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 chmod 600 /home/vagrant/.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

cp /vagrant/ansible/* /home/vagrant/ansible/

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
    source "venv/bin/activate"
    if [ $? -ne 0 ]; then
        echo "Failed to activate virtual environment. Please check your Python installation."
        exit 1
    fi
    echo "Virtual environment created and activated."
    if [ -f "requirements.txt" ]; then
        echo "Installing dependencies from requirements.txt..."
        pip install --upgrade pip
        pip install -r requirements.txt
        if [ $? -ne 0 ]; 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"

if [ -d "$ANSIBLE_VENV_DIR" ]; then
    echo "Activating Ansible virtual environment..."
    source "$ANSIBLE_VENV_DIR/bin/activate"
else
    echo "Ansible virtual environment not found at $ANSIBLE_VENV_DIR. Please create it before running this script."
    exit 1
fi

echo ""

ansible --version

cp -r /vagrant/.vagrant/machines /home/vagrant/machines
cp -r ~/.vagrant/machines /home/vagrant/machines

chmod 600 /home/vagrant/machines/*/virtualbox/private_key
chmod 600 ~/machines/*/virtualbox/private_key
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
EOF
else
    echo "Provisioning block already present in $BASHRC"
fi

echo ""

ANSIBLE_HOST_KEY_CHECKING=False ansible --inventory-file ansible_inventory.ini  -m ping all | cat

if [ $? -ne 0 ]; then
    echo "Ansible ping failed. Please check your Vagrant VMs and network configuration."
    exit 1
fi

# install_keepalived.yaml
ANSIBLE_HOST_KEY_CHECKING=False ansible-playbook install_keepalived.yaml --inventory-file ansible_inventory.ini | cat
if [ $? -ne 0 ]; then
    echo "Ansible playbook failed. Please check your Vagrant VMs and network configuration."
    exit 1
fi
echo "Keepalived installation completed successfully."

# install_k3s_3node.yaml
ANSIBLE_HOST_KEY_CHECKING=False ansible-playbook install_k3s_3node.yaml --inventory-file ansible_inventory.ini | cat
if [ $? -ne 0 ]; then
    echo "Ansible playbook failed. Please check your Vagrant VMs and network configuration."
    exit 1
fi
echo "K3s installation completed successfully."