setup_ubuntu20.04.sh

#!/bin/bash

SCRIPT_DIR=$(readlink -f "$(dirname "$0")")

# Parse args
noninteractive=0
while getopts c OPT; do
    case $OPT in
    "c")
        noninteractive=1
        echo "run setup with a noninteractive mode"
        ;;
    "*") ;;

    esac
done

# Confirm to start installation
if [ $noninteractive -eq 0 ]; then
    # Prevent root execution in interactive mode
    if [ "$(id -u)" -eq 0 ]; then
        echo "Do not run setup as root!" >&2
        exit 1
    fi

    echo "Setting up the build environment take up to 45 minutes."
    read -rp ">  Are you sure you want to run setup? [y/N] " answer
else
    answer="y"
fi

# Use "Super Cow Powers"!
if [[ $answer == [cC] ]]; then
    if ! (command -v cowsay >/dev/null 2>&1); then
        sudo apt-get install -y cowsay
    fi

    answer="y"
fi

# Run ansible if confirmed
case $answer in
[yY]*)
    # Install ansible
    if ! (command -v ansible-playbook >/dev/null 2>&1); then
        sudo apt-get install -y ansible
    fi

    # Set options
    if [ $noninteractive -eq 0 ]; then
        options=("--ask-become-pass")
    else
        options=("--extra-vars" "yn_gpu=y")
    fi

    # Run ansible
    if ansible-playbook "$SCRIPT_DIR/ansible/playbook.yml" -e WORKSPACE_ROOT="$SCRIPT_DIR" "${options[@]}"; then
        echo -e "\e[32mCompleted.\e[0m"
        exit 0
    else
        echo -e "\e[31mFailed.\e[0m"
        exit 1
    fi
    ;;
*)
    echo -e "\e[33mCancelled.\e[0m"
    exit 1
    ;;
esac