#!/bin/bash


RED="33[1;31m"
GREEN="33[1;32m"
YELLOW="33[1;33m"
NOCOLOR="33[0m"


function show_success {

  echo -e "${GREEN}Success.${NOCOLOR}n"

}


function show_error_and_exit {

  echo -e "${RED}An error occured.${NOCOLOR}n"

  exit "$1"

}

function error_handler {

  if [[ $1 -ne 0 ]];
  then

    show_error_and_exit "$2"

  else

    show_success

  fi

}


echo -e "n${GREEN}Step 1: configure packages${NOCOLOR}"
echo -e "${YELLOW}dpkg --configure -a${NOCOLOR}"

sudo dpkg --configure -a

error_handler $?  1


echo -e "${GREEN}Step 2: fix broken dependencies${NOCOLOR}"
echo -e "${YELLOW}apt-get install --fix-broken${NOCOLOR}"

sudo apt-get install --fix-broken

error_handler $?  2


echo -e "${GREEN}Step 3: update cache${NOCOLOR}"
echo -e "${YELLOW}apt-get update${NOCOLOR}"

sudo apt-get update

error_handler $?  3


echo -e "${GREEN}Step 4: upgrade packages${NOCOLOR}"
echo -e "${YELLOW}apt-get upgrade${NOCOLOR}"

sudo apt-get upgrade

error_handler $?  4


echo -e "${GREEN}Step 5: upgrade distribution${NOCOLOR}"
echo -e "${YELLOW}apt-get dist-upgrade${NOCOLOR}"

sudo apt-get dist-upgrade

error_handler $?  5


echo -e "${GREEN}Step 6: remove unused packages${NOCOLOR}"
echo -e "${YELLOW}apt-get --purge autoremove${NOCOLOR}"

sudo apt-get --purge autoremove

error_handler $?  6


echo -e "${GREEN}Step 7: clean up${NOCOLOR}"
echo -e "${YELLOW}apt-get autoclean${NOCOLOR}"

sudo apt-get autoclean

error_handler $?  7

#!/bin/bash

set -e

run () {
    local task=$1
    local desc=$2
    shift 2

    #local rc

    # More portable would be to use tput instead of literal escape codes
    # Avoid uppercase for non-system variables
    local red="33[1;31m"
    local green="33[1;32m"
    local yellow="33[1;33m"
    local nocolor="33[0m"

    printf "${green}Step $task: $desc.${nocolor}n"
    printf "${yellow}$@${nocolor}n"

    if sudo "$@"; then
        printf "${green}Success.${nocolor}n"
    else
        # The fix to capture the failed command's exit code
        # was removed by the OP in an edit of the code
        # but I'm recording it here for posterity.
        #rc=$?
        #printf "${red}Failure: $task${nocolor}n"
        #return $rc
        printf "${red}Failure: $task${nocolor}n"
        return $task
    fi
}

while IFS=: read exit cmd doco; do
    run $exit "$doco" $cmd || exit
done <<____HERE
    1:dpkg --configure -a          :configure packages
    2:apt-get install --fix-broken :fix broken dependencies
    3:apt-get update               :update cache
    4:apt-get upgrade              :upgrade packages
    5:apt-get dist-upgrade         :upgrade distribution
    6:apt-get --purge autoremove   :remove unused packages
    7:apt-get autoclean            :clean up
____HERE