#!/bin/bash

# Options
REPO="https://github.com/borcean/ansible-silverblue-kiosk.git"
BRANCH=main
VAULT_FILE=/root/.ansible_vault_key
INVENTORY="https://raw.githubusercontent.com/borcean/ansible-silverblue-kiosk/"$BRANCH"/hosts"
REQUIREMENTS="https://raw.githubusercontent.com/borcean/ansible-silverblue-kiosk/"$BRANCH"/requirements.yml"

confirm() {
    PROMPT="$1"
    while true; do
        read -r -p "$PROMPT" CHOICE
            if [[ $CHOICE =~ ^[Yy]$ ]]; then
                return 0
            elif [[ "$CHOICE" =~ ^[Nn]$ ]]; then
                return 1
            fi
    done
}

check_hostname () {

    HOSTNAME="$(hostnamectl --static)"

    if [ "$(wget -qO- "$INVENTORY" | grep -m 1 "$HOSTNAME")" == "$HOSTNAME" ]; then
        echo -e "Host "$HOSTNAME" found in inventory."
    else
        echo -e "Host "$HOSTNAME" not found in inventory."
        if confirm "Change hostname now?  y/n: "; then
            read -p "New hostname: " NEW_HOSTNAME
            hostnamectl set-hostname "$NEW_HOSTNAME"
            check_hostname
        fi
    fi
}

# Check if run as root
if [ "$(id -u)" != "0" ]; then
   echo "Must be run as root, exiting..." 1>&2
   exit 1
fi

# Check if hostname is in inventory
check_hostname

# Wait for rpm-ostree to be idle before installing updates
echo "Waiting for rpm-ostree idle state"
until rpm-ostree status | grep "State: idle"; do sleep 1; echo -n "." ; done;

# Install Ansible
OS=$(awk '/^ID=/' /etc/*-release | awk -F'=' '{ print tolower($2) }' | sed 's/"//g')

if [[ "$OS" == fedora ]]; then
    if ! command -v ansible &> /dev/null; then
        rpm-ostree install --apply-live -y ansible gnome-kiosk-script-session
    fi
    if ! rpm -qa | grep python3-rpm &> /dev/null; then
        rpm-ostree install python3-rpm
    fi
else
    if ! confirm "Unsupported distro detected. Only Fedora Silverblue is supported."; then
        exit
    fi
fi

# Clean Ansible cache
rm -rf /root/.ansible/

# Ansible pull command
echo -e "\n"
ansible-pull -U "$REPO" -C "$BRANCH"


# Offer restart after ansible pull finished
if confirm "Reboot system now?  y/n: "; then
    systemctl reboot
fi