helpers/update.sh

@@ -1,7 +1,7 @@
 #!/bin/sh
 
 ### update script for ab-nginx container and utility scripts
-# version 2.0.0
+# version 1.0.0
 # script by Asif Bacchus
 ###
 
@@ -23,63 +23,40 @@ okNotify() {
   printf "%s[OK]%s\n" "$ok" "$norm"
 }
 
-scriptHelp() {
-  textNewline
-  textblock "Update ${containerName} container and helper script files"
-  textblock "${bold}Usage: ${localScriptName} [parameters]${norm}"
-  textNewline
-  textblock "If run with no parameters, the script will update both the container and the helper script files, including this update script."
-  textblockHeader " parameters "
-  textblockParam "-h|-?|--help" "Display this help screen."
-  textblockParam "-c|--container|--container-only" "Update the docker container only."
-  textblockParam "-s|--scripts|--scripts-only" "Update the helper scripts (including this update script) only."
-  textNewline
-  exit 0
-}
-
-textblock() {
-  printf "%s\n" "$1" | fold -w "$width" -s
-}
-
-textblockHeader() {
-  printf "\n%s***%s***%s\n" "$header" "$1" "$norm"
-}
-
-textblockParam() {
-  printf "%s%-35s%s%s\n" "$info" "$1" "$2" "$norm"
-}
-
-textNewline() {
-  printf "\n"
-}
-
 ### text formatting presets
 if command -v tput >/dev/null 2>&1; then
-  bold=$(tput bold)
   err=$(tput bold)$(tput setaf 1)
   info=$(tput bold)$(tput setaf 6)
-  header=$(tput bold)$(tput setaf 5)
   norm=$(tput sgr0)
   ok=$(tput sgr0)$(tput setaf 2)
   warn=$(tput bold)$(tput setaf 3)
-  width=$(tput cols)
 else
-  bold=''
   err=''
   info=''
-  header=''
   norm=''
   ok=''
   warn=''
-  width=80
 fi
 
 ### pre-requisites
+
 # check if wget is installed
 if ! command -v wget >/dev/null 2>&1; then
   errMsg "Sorry, this script requires that 'wget' is installed in order to download updates. Exiting."
 fi
 
+# check if docker is installed
+if ! command -v docker >/dev/null 2>&1; then
+  errMsg "Sorry, it appears that docker is not installed on this machine! Exiting."
+fi
+
+# is user root or in the docker group?
+if [ ! "$(id -u)" -eq 0 ]; then
+  if ! id -Gn | grep docker >/dev/null; then
+    errMsg "You must either be root or in the 'docker' group to pull container updates."
+  fi
+fi
+
 # zero counters
 updatesAvailable=0
 downloadFailed=0
@@ -94,157 +71,114 @@ containerUpdatePath="docker.asifbacchus.app/$dockerNamespace/$containerName:late
 server="https://asifbacchus.app/updates/docker/$dockerNamespace/$containerName/"
 checksumFilename='checksums.sha256'
 
-# operation triggers
-doDockerUpdate=1
-doScriptUpdate=1
-
 # files to update
 localScriptName="$(basename "$0")"
 repoScriptName='update.sh'
 
-### process startup parameters
-while [ $# -gt 0 ]; do
-  case "$1" in
-  -h | -\? | --help)
-    # display inline help
-    scriptHelp
-    ;;
-  -s | --scripts | --scripts-only)
-    # update scripts only, skip docker container update
-    doDockerUpdate=0
-    ;;
-  -c | --container | --container-only)
-    # update docker container only, skip script update
-    doScriptUpdate=0
-    ;;
-  *)
-    printf "%s\nUnknown option: %s\n" "$err" "$1"
-    printf "%sUse '--help' for valid options%s\n\n" "$info" "$norm"
-    exit 1
-    ;;
-  esac
-  shift
-done
-
 ### update container
-if [ "$doDockerUpdate" -eq 1 ]; then
+printf "%s\n*** Updating %s container and service scripts ***\n\n%s" "$info" "$containerName" "$norm"
-  # check if docker is installed
-  if ! command -v docker >/dev/null 2>&1; then
-    errMsg "Sorry, it appears that docker is not installed on this machine! Exiting."
-  fi
 
-  # is user root or in the docker group?
+printf "Updating container:\n"
-  if [ ! "$(id -u)" -eq 0 ]; then
+if ! docker pull "$containerUpdatePath"; then
-    if ! id -Gn | grep docker >/dev/null; then
+  errMsg "There was an error updating the container. Try again later."
-      errMsg "You must either be root or in the 'docker' group to pull container updates."
+else
-    fi
+  okMsg "Container updated!"
-  fi
-
-  printf "%s\n*** Updating %s container ***\n\n%s" "$info" "$containerName" "$norm"
-  if ! docker pull "$containerUpdatePath"; then
-    errMsg "There was an error updating the container. Try again later."
-  else
-    okMsg "Container updated!"
-  fi
 fi
 
 ### update scripts
-if [ "$doScriptUpdate" -eq 1 ]; then
+printf "%sUpdating %s service scripts%s\n" "$info" "$containerName" "$norm"
-  printf "%s*** Updating %s service scripts ***%s\n" "$info" "$containerName" "$norm"
 
-  ## download latest checksums
+## download latest checksums
-  printf "Getting latest checksums... "
+printf "Getting latest checksums... "
-  if ! wget --quiet --tries=3 --timeout=10 -N "${server}${checksumFilename}"; then
+if ! wget --quiet --tries=3 --timeout=10 -N "${server}${checksumFilename}"; then
+  errNotify
+  errMsg "Unable to download checksums. Try again later."
+else
+  okNotify
+fi
+
+## check for updates to this script
+printf "Checking for updates to this script... "
+repoScriptChecksum=$(grep "$repoScriptName" "$checksumFilename" | grep -o '^\S*')
+localScriptChecksum=$(sha256sum "$localScriptName" | grep -o '^\S*')
+if [ "$localScriptChecksum" = "$repoScriptChecksum" ]; then
+  printf "[NONE]\n"
+else
+  printf "[AVAILABLE]\n"
+  printf "Getting updated script... "
+  # download updated script
+  if ! wget --quiet --tries=3 --timeout=10 -O "update.sh.tmp" "${server}${repoScriptName}"; then
     errNotify
-    errMsg "Unable to download checksums. Try again later."
+    # delete failed download as necessary
+    rm -f ./update.sh.tmp 2>/dev/null
+    errMsg "Unable to download script update. Try again later."
   else
-    okNotify
+    # verify download
-  fi
+    dlScriptChecksum=$(sha256sum "update.sh.tmp" | grep -o '^\S*')
-
+    if ! [ "$dlScriptChecksum" = "$repoScriptChecksum" ]; then
-  ## check for updates to this script
+      printf "[ERROR]\n"
-  printf "Checking for updates to this script... "
+      # delete corrupt download as necessary
-  repoScriptChecksum=$(grep "$repoScriptName" "$checksumFilename" | grep -o '^\S*')
-  localScriptChecksum=$(sha256sum "$localScriptName" | grep -o '^\S*')
-  if [ "$localScriptChecksum" = "$repoScriptChecksum" ]; then
-    printf "[NONE]\n"
-  else
-    printf "[AVAILABLE]\n"
-    printf "Getting updated script... "
-    # download updated script
-    if ! wget --quiet --tries=3 --timeout=10 -O "update.sh.tmp" "${server}${repoScriptName}"; then
-      errNotify
-      # delete failed download as necessary
       rm -f ./update.sh.tmp 2>/dev/null
-      errMsg "Unable to download script update. Try again later."
+      errMsg "Checksum mismatch! Try again later."
     else
-      # verify download
+      okNotify
-      dlScriptChecksum=$(sha256sum "update.sh.tmp" | grep -o '^\S*')
+      printf "\n%s*** This script has been updated. Please re-run it to load the updated version of this file. ***%s\n\n" "$warn" "$norm"
-      if ! [ "$dlScriptChecksum" = "$repoScriptChecksum" ]; then
+      # overwrite this script with updated script
-        printf "[ERROR]\n"
+      mv -f ./update.sh.tmp "$localScriptName"
-        # delete corrupt download as necessary
-        rm -f ./update.sh.tmp 2>/dev/null
-        errMsg "Checksum mismatch! Try again later."
-      else
-        okNotify
-        printf "\n%s*** This script has been updated. Please re-run it to load the updated version of this file. ***%s\n\n" "$warn" "$norm"
-        # overwrite this script with updated script
-        mv -f ./update.sh.tmp "$localScriptName"
-      fi
     fi
   fi
+fi
 
-  ## update files
+## update files
-  while IFS='  ' read -r field1 field2; do
+while IFS='  ' read -r field1 field2; do
-    printf "\nChecking '%s' for updates... " "$field2"
+  printf "\nChecking '%s' for updates... " "$field2"
-    updateFilename="$field2"
+  updateFilename="$field2"
-    repoFileChecksum="$field1"
+  repoFileChecksum="$field1"
-    if [ -f "$updateFilename" ]; then
+  if [ -f "$updateFilename" ]; then
       localFileChecksum=$(sha256sum "$updateFilename" | grep -o '^\S*')
-    else
+  else
       localFileChecksum=0
-    fi
+  fi
 
-    # update file if necessary
+  # update file if necessary
-    if ! [ "$localFileChecksum" = "$repoFileChecksum" ]; then
+  if ! [ "$localFileChecksum" = "$repoFileChecksum" ]; then
-      printf "[AVAILABLE]\n"
+    printf "[AVAILABLE]\n"
-      updatesAvailable=$((updatesAvailable + 1))
+    updatesAvailable=$((updatesAvailable + 1))
-      # download update
+    # download update
-      printf "Downloading updated '%s'... " "$updateFilename"
+    printf "Downloading updated '%s'... " "$updateFilename"
-      if ! wget --quiet --tries=3 --timeout=10 -O "$updateFilename.tmp" "${server}${updateFilename}"; then
+    if ! wget --quiet --tries=3 --timeout=10 -O "$updateFilename.tmp" "${server}${updateFilename}"; then
+      errNotify
+      downloadFailed=$((downloadFailed + 1))
+      # delete failed download file as necessary
+      rm -f "$updateFilename.tmp" 2>&1
+    else
+      okNotify
+      downloadSuccess=$((downloadSuccess + 1))
+      # verify download
+      printf "Verifying '%s'... " "$updateFilename"
+      localFileChecksum=$(sha256sum "$updateFilename.tmp" | grep -o '^\S*')
+      if ! [ "$localFileChecksum" = "$repoFileChecksum" ]; then
         errNotify
-        downloadFailed=$((downloadFailed + 1))
+        updateFailed=$((updateFailed + 1))
-        # delete failed download file as necessary
+        # delete corrupted download file as necessary
         rm -f "$updateFilename.tmp" 2>&1
       else
         okNotify
-        downloadSuccess=$((downloadSuccess + 1))
+        updateSuccess=$((updateSuccess + 1))
-        # verify download
+        # overwrite old version of file
-        printf "Verifying '%s'... " "$updateFilename"
+        mv -f "$updateFilename.tmp" "$updateFilename"
-        localFileChecksum=$(sha256sum "$updateFilename.tmp" | grep -o '^\S*')
-        if ! [ "$localFileChecksum" = "$repoFileChecksum" ]; then
-          errNotify
-          updateFailed=$((updateFailed + 1))
-          # delete corrupted download file as necessary
-          rm -f "$updateFilename.tmp" 2>&1
-        else
-          okNotify
-          updateSuccess=$((updateSuccess + 1))
-          # overwrite old version of file
-          mv -f "$updateFilename.tmp" "$updateFilename"
-        fi
       fi
-    else
-      printf "[NONE]\n"
     fi
-  done <"$checksumFilename"
+  else
-fi
+    printf "[NONE]\n"
+  fi
+done <"$checksumFilename"
+
 
 ### display results
-if [ "$doScriptUpdate" -eq 1 ]; then
+printf "\n%sResults:%s\n" "$info" "$norm"
-  printf "\n%s*** Results ***%s\n" "$info" "$norm"
+printf "\tUpdates: %s available\n" "$updatesAvailable"
-  printf "\tUpdates: %s available\n" "$updatesAvailable"
+printf "\tDownloads: %s%s successful%s, %s%s failed%s\n" "$ok" "$downloadSuccess" "$norm" "$err" "$downloadFailed" "$norm"
-  printf "\tDownloads: %s%s successful%s, %s%s failed%s\n" "$ok" "$downloadSuccess" "$norm" "$err" "$downloadFailed" "$norm"
+printf "\tUpdates: %s%s applied%s, %s%s failed%s\n" "$ok" "$updateSuccess" "$norm" "$err" "$updateFailed" "$norm"
-  printf "\tUpdates: %s%s applied%s, %s%s failed%s\n" "$ok" "$updateSuccess" "$norm" "$err" "$updateFailed" "$norm"
-fi
 
 exit 0