dietpi-update

#!/bin/bash
#Load the whole script into memory.
{
	#////////////////////////////////////
	# DietPi Update
	#
	#////////////////////////////////////
	# Created by Daniel Knight / daniel.knight@dietpi.com / dietpi.com
	#
	#////////////////////////////////////
	#
	# Info:
	# - Updates DietPi from Git
	# - Uses patch_file for online pactching
	#
	# Usage:
	# - dietpi-update   = Normal
	# - dietpi-update 1 = forced update
	# - dietpi-update 2 = Check for updates. print server_version to /DietPi/dietpi/.update_available (-1=new image required)
	#////////////////////////////////////

	#Force en_GB Locale for whole script. Prevents incorrect parsing with non-english locales.
	LANG=en_GB.UTF-8

	INPUT=0

	if [[ $1 =~ ^-?[0-9]+$ ]]; then
		INPUT=$1
	fi

	#/////////////////////////////////////////////////////////////////////////////////////
	#Exit Paths
	#/////////////////////////////////////////////////////////////////////////////////////
	#Exit path for non-root logins.
	if (( $UID != 0 )); then
		clear
		echo -e "\n ERROR: You do not have root privileges.\n Please login as root, or, use:\n sudo dietpi-update\n"

		exit
	fi

	#/////////////////////////////////////////////////////////////////////////////////////
	#GIT Branch | master / testing
	#/////////////////////////////////////////////////////////////////////////////////////
	GITBRANCH=$(cat /DietPi/dietpi.txt | grep -m1 '^gitbranch=' | sed 's/.*=//')
	GITFORKOWNER=$(cat /DietPi/dietpi.txt | grep -m1 '^gitforkowner=' | sed 's/.*=//')

	#/////////////////////////////////////////////////////////////////////////////////////
	#UPDATE Vars
	#/////////////////////////////////////////////////////////////////////////////////////
	DIETPIUPDATE_VERSION_CURRENT=5 # Version of dietpi-update / set server_version line two to value++ and obsolete previous dietpi-update scripts
	DIETPIUPDATE_VERSION_REQUIRED=0

	FILEPATH_TEMP="/tmp/dietpi-update"

	SERVER_ONLINE=0
	UPDATE_AVAILABLE=0
	UPDATE_REQUIRESNEWIMAGE=0
	RUN_UPDATE=0

	VERSION_CURRENT=0
	VERSION_SERVER=0

	Get_Client_Version(){
		VERSION_CURRENT=$(cat /DietPi/dietpi/.version)
	}

	Get_Server_Version(){

		#Get server version file
		curl -k -L https://raw.githubusercontent.com/${GITFORKOWNER:-Fourdee}/DietPi/"$GITBRANCH"/dietpi/server_version > "$FILEPATH_TEMP"/server_version
		if (( $? == 0 )); then
			SERVER_ONLINE=1

			#Get server Version info
			VERSION_SERVER=$(sed -n 1p "$FILEPATH_TEMP"/server_version)
			DIETPIUPDATE_VERSION_REQUIRED=$(sed -n 2p "$FILEPATH_TEMP"/server_version)

		fi
	}

	Check_Update_Available(){

		#Check if server_version is a valid interger.
		if [[ $VERSION_SERVER =~ ^-?[0-9]+$ ]]; then

			#Update Requires new image?
			if (( $DIETPIUPDATE_VERSION_CURRENT < $DIETPIUPDATE_VERSION_REQUIRED )); then

				UPDATE_REQUIRESNEWIMAGE=1

			#Update available
			elif (( $VERSION_CURRENT < $VERSION_SERVER )); then

				UPDATE_AVAILABLE=1
				echo -e ""
				/DietPi/dietpi/func/dietpi-notify 0 "Update available"
				/DietPi/dietpi/func/dietpi-notify 2 "Current Version : $VERSION_CURRENT"
				/DietPi/dietpi/func/dietpi-notify 2 "Server Version  : $VERSION_SERVER"

			fi

		else

			/DietPi/dietpi/func/dietpi-notify 1 "Obtained server version value from github is invalid. Please try running dietpi-update again."
			exit

		fi

	}

	#/////////////////////////////////////////////////////////////////////////////////////
	# MENUS
	#/////////////////////////////////////////////////////////////////////////////////////
	WHIP_BACKTITLE='- DietPi-Update -'
	WHIP_TITLE=$WHIP_BACKTITLE
	CHOICE=0
	OPTION=0

	Menu_Update(){

		whiptail --title "$WHIP_TITLE" --yesno "Update available   : https://goo.gl/G9ikqn
Installed version  : $VERSION_CURRENT
Latest version     : $VERSION_SERVER

---------------------------------Notice-------------------------------------
The benefit of DietPi is we use standard linux configurations and commands. The downside is we can't possibly accommodate or predict, every modifcation to Linux configurations files by the end user, during the update.

Although we test the updates thoroughly, if you have made custom changes to Linux configuration files, outside of the DietPi programs, we highly recommend you create a backup of your system, before updating.

You can create a system backup, by running:
 - dietpi-backup
----------------------------------------------------------------------------

Do you wish to continue and update DietPi to v$VERSION_SERVER?" --yes-button "Ok" --no-button "Exit" --defaultno --backtitle "$WHIP_BACKTITLE" 24 80
		CHOICE=$?
		#Confirm action
		if (( $CHOICE == 0 )); then
			RUN_UPDATE=1
		fi
	}

	#/////////////////////////////////////////////////////////////////////////////////////
	# Update DietPi
	#/////////////////////////////////////////////////////////////////////////////////////

	Run_Update(){

		#git clone Zip method (no need to install GIT)
		curl -k -L https://github.com/${GITFORKOWNER:-Fourdee}/DietPi/archive/"$GITBRANCH".zip > "$FILEPATH_TEMP"/update.zip
		if (( $? == 0 )); then

			unzip "$FILEPATH_TEMP"/update.zip -d "$FILEPATH_TEMP"/

			#Remove setting files from git that are not to be updated
			rm "$FILEPATH_TEMP"/DietPi-"$GITBRANCH"/dietpi/.* &> /dev/null

			#Remove folders of "non-critical scripts" before updating them. (eg: so we dont need to patch for /conf/* file removals)
			rm -R /DietPi/dietpi/conf
			rm -R /DietPi/dietpi/func
			rm -R /DietPi/dietpi/misc

			#Copy downloaded DietPi to Ramdisk
			cp -Rf "$FILEPATH_TEMP"/DietPi-"$GITBRANCH"/dietpi /DietPi/

			#Allow execute of all DietPi scripts and patch file
			chmod -R +x /DietPi

			#Run Patch file
			while (( $VERSION_CURRENT < $VERSION_SERVER )); do

				/DietPi/dietpi/func/dietpi-notify 3 DietPi-Update "Updating DietPi"
				/DietPi/dietpi/func/dietpi-notify 2 "Current Version : $VERSION_CURRENT"
				/DietPi/dietpi/func/dietpi-notify 2 "Server Version  : $VERSION_SERVER\n"

				#Run patch file
				/DietPi/dietpi/func/dietpi-notify 2 "Patching $VERSION_CURRENT to $(( $VERSION_CURRENT + 1 ))"
				/DietPi/dietpi/patch_file "$VERSION_CURRENT" "$VERSION_SERVER"

				#Update Local Version ID
				((VERSION_CURRENT++))
				echo -e "$VERSION_CURRENT" > /DietPi/dietpi/.version

				/DietPi/dietpi/func/dietpi-notify 0 "Patch $VERSION_CURRENT completed\n"

				Get_Client_Version

			done

			#Remove Patch files.
			rm /DietPi/dietpi/patch_file &> /dev/null
			rm /DietPi/dietpi/server_version &> /dev/null

			#Remove temp
			rm -R "$FILEPATH_TEMP" &> /dev/null

		#Unable to download file.
		else

			/DietPi/dietpi/func/dietpi-notify 1 "Download failed, unable to run update. Please try running dietpi-update again."
			exit

		fi

	}

	#/////////////////////////////////////////////////////////////////////////////////////
	# Main Loop
	#/////////////////////////////////////////////////////////////////////////////////////
	#----------------------------------------------------------------
	#Inform user
	/DietPi/dietpi/func/dietpi-notify 3 DietPi-Update "Checking for DietPi updates"
	#----------------------------------------------------------------
	#Create temp directory used in dietpi-update
	rm -R "$FILEPATH_TEMP" &> /dev/null
	mkdir -p "$FILEPATH_TEMP" &> /dev/null
	#----------------------------------------------------------------
	#Get versions
	Get_Client_Version
	Get_Server_Version
	#----------------------------------------------------------------
	#Check if update is available
	Check_Update_Available
	#----------------------------------------------------------------
	#Check for updates only. Send result to global file for use by dietpi-banner update available notifications.
	if (( $INPUT == 2 )); then

		#Clear previous .update_available file
		rm /DietPi/dietpi/.update_available &> /dev/null

		#Write update available version to file.
		if (( $UPDATE_AVAILABLE == 1 )); then

			echo -e "$VERSION_SERVER" > /DietPi/dietpi/.update_available

		elif (( $UPDATE_REQUIRESNEWIMAGE == 1 )); then

			echo -e "-1" > /DietPi/dietpi/.update_available

		fi

	#----------------------------------------------------------------
	#Run menus / prompts / updates
	else

		#----------------------------------------------------------------
		#Server offline
		if (( $SERVER_ONLINE == 0 )); then

			whiptail --title "Error - Server Offline" --msgbox "http://github.com is either offline, or, unable to connect \n \n No updates applied." 12 60

		#Update requires new DietPi image
		elif (( $UPDATE_REQUIRESNEWIMAGE == 1 )); then

			#Cannot update, Image required
			whiptail --title "New image required" --msgbox " Your version of DietPi is now obsolete and cannot be updated. \n\n Please download the latest DietPi image:\n - http://dietpi.com/downloads/images \n\n - Current Version : $VERSION_CURRENT \n - Server Version  : $VERSION_SERVER \n " 13 70

			/DietPi/dietpi/func/dietpi-notify 1 "Your version of DietPi is now obsolete and cannot be updated."
			echo -e "Please download the latest DietPi image:\n - http://dietpi.com/download \n\n - Current Version : $VERSION_CURRENT \n - Server Version  : $VERSION_SERVER \n "

		#Update available
		elif (( $UPDATE_AVAILABLE == 1 )); then

			#Forced update
			if (( $INPUT == 1 )); then

				/DietPi/dietpi/func/dietpi-notify 0 "Updates have been found and are being applied, please wait..."
				/DietPi/dietpi/func/dietpi-notify 2 "Current Version : $VERSION_CURRENT"
				/DietPi/dietpi/func/dietpi-notify 2 "Server Version  : $VERSION_SERVER"

				/DietPi/dietpi/dietpi-funtime 0

				RUN_UPDATE=1

			#Ask for update
			else

				Menu_Update

			fi

		#No Updates
		else

			echo -e ""
			/DietPi/dietpi/func/dietpi-notify 0 "No updates required, your DietPi installation is up to date.\n"
			/DietPi/dietpi/func/dietpi-notify 2 "Current Version : $VERSION_CURRENT"
			/DietPi/dietpi/func/dietpi-notify 2 "Server Version  : $VERSION_SERVER"
			sleep 2

		fi

		#----------------------------------------------------------------
		#Run Update
		if (( $RUN_UPDATE == 1 )); then

			#Stop Services
			/DietPi/dietpi/dietpi-services stop

			#Run update and patcher
			Run_Update

			#.update file stage (only used on 1st run of dietpi-software to check/apply updates, 0 tells dietpi-software to reboot)
			echo 0 > /DietPi/dietpi/.update_stage

			#Remove update_available file
			rm /DietPi/dietpi/.update_available &> /dev/null

			#Done
			/DietPi/dietpi/func/dietpi-notify 3 DietPi-Update "Completed"
			/DietPi/dietpi/func/dietpi-notify 2 "Current Version : $VERSION_CURRENT"
			/DietPi/dietpi/func/dietpi-notify 2 "Server Version  : $VERSION_SERVER"
			/DietPi/dietpi/func/dietpi-notify 0 "Update completed"
			echo -e ""
			echo -e "Please reboot your system now, using the command \e[31;49;1mreboot\e[0m"

		fi

		#----------------------------------------------------------------
		#Desktop Run, exit key prompt
		if (( $(ps aux | grep -ci -m1 '[l]xpanel') == 1 )); then

			read -p "Press any key to exit DietPi-Update..."

		fi

	fi

	#----------------------------------------------------------------
	#Clear temp files and folders
	rm -R "$FILEPATH_TEMP" &> /dev/null
	#----------------------------------------------------------------
	exit
	#----------------------------------------------------------------
}