From ae89b71bd783666d749c0a749f64990df0b83be9 Mon Sep 17 00:00:00 2001
From: Marco Ochse <t3chn0m4g3@gmail.com>
Date: Wed, 10 Jan 2018 17:13:48 +0000
Subject: [PATCH] tweak update script

---
 update.sh | 32 ++++++++++++++++++++++++++++++--
 1 file changed, 30 insertions(+), 2 deletions(-)

diff --git a/update.sh b/update.sh
index c2ab370f..011041e4 100755
--- a/update.sh
+++ b/update.sh
@@ -3,9 +3,8 @@
 ###################################################
 # Do not change any contents of this script!
 ###################################################
-
-# Some vars
 myCONFIGFILE="/opt/tpot/etc/tpot.yml"
+myCOMPOSEPATH="/opt/tpot/etc/compose"
 myRED=""
 myGREEN=""
 myWHITE=""
@@ -84,7 +83,36 @@ function fuSELFUPDATE () {
       fi
     else
       echo "###### Update script is already up-to-date."
+
+      # Better safe than sorry
+      echo "###### Creating backup and storing it in /home/tsec"
+      tar cvfz /home/tsec/tpot_backup.tgz /opt/tpot
+
+      echo "###### Pulling updates"
       git pull --force
+
+      echo "###### Getting the current install flavor"
+      myFLAVOR=$(head $myCONFIGFILE -n 1 | awk '{ print $3 }' | tr -d :'()':)
+
+      echo "###### Updating compose file"
+      case $myFLAVOR in
+        HP)
+          echo "###### Restoring HONEYPOT flavor installation."
+          cp $myCOMPOSEPATH/hp.yml $myCONFIGFILE
+        ;;
+        Industrial)
+          echo "###### Restoring INDUSTRIAL flavor installation."
+          cp $myCOMPOSEPATH/industrial.yml $myCONFIGFILE
+        ;;
+        Standard)
+          echo "###### Restoring TPOT flavor installation."
+          cp $myCOMPOSEPATH/tpot.yml $myCONFIGFILE
+        ;;
+        Everything)
+          echo "###### Restoring EVERYTHING flavor installation."
+          cp $myCOMPOSEPATH/all.yml $myCONFIGFILE
+        ;;
+      esac
   fi
 }