#!/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 -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
    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."
    cp /vagrant/ansible/requirements.txt .
    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"

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..."
    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

if [ $? -ne 0 ]; then
    echo "Ansible is not installed or not found in the virtual environment. Please check your installation."
    exit 1
fi

cp -r /vagrant/ansible/* /home/vagrant/ansible/

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 
echo -------------------------
echo 

su - vagrant
id

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

ssh-add ~/.ssh/vm*_key

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 

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

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

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

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