update.sh

#!/bin/bash

check_exit_status() {
    if [ $? -eq 0 ]; then
        echo
        echo "Success"
        echo
    else
        echo
        echo "[ERROR] Process Failed!"
        echo

        read -p "The last command exited with an error. Exit script? (yes/no) " answer

        if [ "$answer" == "yes" ]; then
            exit 1
        fi
    fi
}

greeting() {
    echo
    echo "Hello, $USER. Let's update this system."
    echo
}

update() {
    sudo apt-get update
    check_exit_status

    if [ $(dpkg-query -W -f='${Status}' locate 2>/dev/null | grep -c "ok installed") -eq 0 ]; then
        apt-get install locate -y
    fi

    if [ $(dpkg-query -W -f='${Status}' mlocate 2>/dev/null | grep -c "ok installed") -eq 0 ]; then
        apt-get install mlocate -y
    fi

    sudo apt-get upgrade -y
    check_exit_status

    sudo apt-get dist-upgrade -y
    check_exit_status
}

housekeeping() {
    sudo apt-get autoremove -y
    check_exit_status

    sudo apt-get autoclean -y
    check_exit_status

    sudo updatedb
    check_exit_status
}

leave() {
    echo
    echo "--------------------"
    echo "- Update Complete! -"
    echo "--------------------"
    echo
    exit
}

greeting
update
housekeeping
leave