#!/usr/bin/env bash
set -e


# "OpenELEC_DEV" ; An automated development build updater script for OpenELEC nightly builds
#
# Copyright (c) February 2012, Eric Andrew Bixler
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#	* Redistributions of source code must retain the above copyright
#	  notice, this list of conditions and the following disclaimer.
#	* Redistributions in binary form must reproduce the above copyright
#	  notice, this list of conditions and the following disclaimer in the
#	  documentation and/or other materials provided with the distribution.
#	* Neither the name of the <organization> nor the names of its contributors
#     may be used to endorse or promote products derived from this software
#     without specific prior written permission.


# THIS SOFTWARE IS PROVIDED BY Eric Andrew Bixler ''AS IS'' AND ANY EXPRESS
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL Eric Andrew Bixler BE LIABLE FOR ANY DIRECT,
# INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.


###### stable/beta release check

stable ()
{
currentsys=$(cat /etc/version | cut -f 1 -d'-')
if [ "$currentsys" != "devel" ] ;
then
	echo
	echo "Your Version:"
	echo "Release: $currentsys"
	echo
	echo "You're currently on a Stable release of OpenELEC."
	echo "To use this script, you first need to manually"
	echo "migrate OpenELEC to a development build."
	echo "Be aware that development builds are inherently"
	echo "unstable and should not be used in production"
	echo "enviroments."
	echo "Usability of your HTPC may be adversly effected."
	echo
	echo "Once you are running a development build, you may"
	echo "keep your system up-to-date by re-running this script"
	echo
	echo "Visit the following locations to get started if you"
	echo "accept the risks of using pre-release software:"
	echo "---------------------------------------------------"
	echo
	echo "A How-To for Manually Updating OpenELEC:"
	echo "http://wiki.openelec.tv/index.php?title=Updating_OpenELEC"
	echo
	echo "Location of development builds:"
	echo "http://sources.openelec.tv/tmp/image/"
	echo
	echo "---------------------------------------------------"
	echo
	echo "Exiting Now."
	echo
	exit 1
fi
}

stable


###### we've already been updated; need to remove update indicator from out last script update run

if [ -f /tmp/update_in_progress ] ;
then
	rm -f /tmp/update_in_progress
fi


###### temporarily disable script updates ?

Allow_Update="YES"


###### script version

VERSION="40"


###### if no options specified; we continue as normal

options_found="0"


###### restart scanning of argv, less 1; we want to see if there are more then one option characters passed.

OPTIND="1"


###### default image location

mode="http://sources.openelec.tv/tmp/image"


###### set the temporary file location based on what device we are using...(the rPi does not have enough RAM to download the image to /dev/shm

arch=$(cat /etc/arch)
if [ "$arch" = "RPi.arm" ] ;
then
	echo "RaspberryPi Detected"
	echo
	temploc="/storage/downloads/xbmc-update"
	dkernel="KERNEL"
	dsystem="SYSTEM"
	akernel="kernel.img"
	asystem="SYSTEM"
	dkmd5="KERNEL.md5"
	dsmd5="SYSTEM.md5"
	mode="http://openelec.thestateofme.com/dev_builds/"
else
	echo "Device Detected: `cat /etc/arch | sed 's/\./ /g' | awk '{print $1}'`"
	echo "OpenELEC_Dev: v$VERSION"
	echo
	temploc="/dev/shm/xbmc-update"
	dkernel="KERNEL"
	dsystem="SYSTEM"
	akernel=$dkernel
	asystem=$dsystem
	dkmd5="KERNEL.md5"
	dsmd5="SYSTEM.md5"
fi


###### going to check for avaliable RAM, and if there isnt more then 200MB free; just use the harddisk; this will override the variable set just above

ram_mb=$((`cat /proc/meminfo | sed -n 2p | awk '{print $2}'`/1024))
if [ "$ram_mb" -lt "200" ] ;
then
	temploc="/storage/downloads/xbmc-update"
	unset ram_mb
fi


###### options

while getopts ":craospilqzvbh--:help" opt ;
do
	case $opt in

	c)
		options_found=1
		# quick check to see if we're up-to-date
		mkdir -p $temploc
		arch=$(cat /etc/arch)
		curl --silent $mode/ | grep $arch | sed -e ';s/.*<a href="//' -e 's/">.*//' > $temploc/temp
		if [ $(wc -l $temploc/temp | cut -c -1) -gt "1" ] ;
		then
			cat $temploc/temp | tail -1 > $temploc/temp2
		else
			mv $temploc/temp $temploc/temp2
		fi
		if [[ -z `cat $temploc/temp2` ]] ;
		then
			echo
 			echo "There are either no available builds for your architecture at this time, or"
			echo "the only build avaliable, is the same build revision you are currently on."
			echo "Unable to check remote revision number."
			echo "Please check again later. You may also check manually for yourself here:"
			echo "http://sources.openelec.tv/tmp/image/"
			echo
			echo "Exiting Now."
 			rm -rf $temploc
 			unset arch
			exit 1
		fi
		PAST=$(cat /etc/version | tail -c 6 | tr -d 'r')
		PRESENT=$(cat $temploc/temp2 | tail -c 15 | cut -c 0-5)
		if [ `echo $PRESENT | sed 's/.\{4\}$//'` == "-" ] ;
		then
			# this is for coming from revisions 9999 and lower
			PRESENT=$(cat $temploc/temp2 | tail -c 15 | sed 's/.\{8\}$//' | tr -d "\-r"})
		else
			# this is for coming from revisions 10000 and higher
			PRESENT=$(cat $temploc/temp2 | tail -c 15 | sed 's/.\{8\}$//' | tr -d "r")
		fi
		if [ "$PRESENT" -gt "$PAST" ] ;
		then
			echo
			echo "Updates are Available"
			echo "Local:   $PAST"
			echo "Remote:  $PRESENT"
		else
			echo
			echo "No Updates Available"
		fi
		rm -rf $temploc
		unset arch
		unset PAST
		unset PRESENT
		;;

	r)
		options_found=1
		# displays the remote build number
		mkdir -p $temploc
		arch=$(cat /etc/arch)
		curl --silent $mode/ | grep $arch | sed -e ';s/.*<a href="//' -e 's/">.*//' > $temploc/temp
		if [ $(wc -l $temploc/temp | cut -c -1) -gt "1" ] ;
		then
			cat $temploc/temp | tail -1 > $temploc/temp2
		else
			mv $temploc/temp $temploc/temp2
		fi
		if [[ -z `cat $temploc/temp2` ]] ;
		then
			echo
			echo "There are either no available builds for your architecture at this time, or"
			echo "the only build avaliable, is the same build revision you are currently on."
			echo "Unable to display remote revision number."
			echo "Please check again later. You may also check manually for yourself here:"
			echo "http://sources.openelec.tv/tmp/image/"
			echo
			echo "Exiting Now."
			rm -rf $temploc
			unset arch
			exit 1
		fi
		echo
		echo "Newest Remote Release for [ $arch ] is ---> `cat $temploc/temp2 | tail -c 15 | sed 's/.\{8\}$//' | tr -d "\-r"`"
		rm -rf $temploc
		unset arch
		;;

	a)
		options_found=1
		# show all remotely available builds for your architecture, and build date
		arch=$(cat /etc/arch)
		mkdir -p $temploc
		curl --silent $mode/ | grep $arch | sed -e ';s/.*<a href="//' -e 's/">.*//' > $temploc/temp3
		echo
		echo "New Builds Available for your Architecture:  ($arch)"
		echo "-------------------------------------------"
		echo
		if [[ -z `cat $temploc/temp3` ]] ;
		then
			echo
			echo "There are no available builds for your architecture at this time."
			echo "Please check again later."
			echo
			echo "Exiting Now."
			rm -rf $temploc
			unset arch
			exit 1
		fi
		for i in `cat $temploc/temp3`
		do
		echo -n "$i  --->  Compiled On: "; echo -n "$i" | cut -f 4-4 -d'-' | sed 's/......$//;s/./& /4' | sed 's/./& /7' | awk '{ print "[ "$2"/"$3"/"$1" ]" }'
		done
		rm -rf $temploc
		unset arch
		;;

	o)
		options_found=1
		# show all old archived builds for your architecture, as well as compilation date
		arch=$(cat /etc/arch)
		mkdir -p $temploc
		curl --silent $mode/old/ | grep $arch | sed -e ';s/.*<a href="//' -e 's/">.*//' > $temploc/temp
		echo
		echo "Archival Builds Avaliable for your Architecture:  ($arch)"
		echo "------------------------------------------------"
		echo
		if [[ -z `cat $temploc/temp` ]] ;
		then
			echo
			echo "There are no archived builds for your architecture at this time."
			echo "Please check again later."
			echo
			echo "Exiting Now."
			rm -rf $temploc
			exit 1
		fi
		for i in `cat $temploc/temp`
		do
			echo -n "$i  --->  Compiled On: "; echo -n "$i" | cut -f 4-4 -d'-' | sed 's/......$//;s/./& /4' | sed 's/./& /7' | awk '{ print "[ "$2"/"$3"/"$1" ]" }'
		done
		rm -rf $temploc
		unset arch
		;;

	i)
		options_found=1
		# check to see if the appropriate files are in the right place, for a reboot
		SYS_KERN=$(ls /storage/.update/* 2> /dev/null | wc -l)
		if [ "$SYS_KERN" = "4" ] ;
		then
			echo
			echo "A KERNEL, & SYSTEM image are already in place."
			echo "Please reboot your HTPC when possible"
			echo "to complete the update."
			echo
			echo "Would you like to reboot now (y/n) ?"
			read -n1 -p "==| " reb
			echo
			if [[ "$reb" != "Y" ]] && [[ "$reb" != "y" ]] && [[ "$reb" != "N" ]] && [[ "$reb" != "n" ]] ;
			then
				echo
				echo "Unrecognized Input."
				echo "Please answer (y/n)"
				echo "Exiting Now."
				echo
				rm -rf $temploc
				unset reb
				exit 1
			elif [[ "$reb" = "Y" || "$reb" = "y" ]] ;
			then
				sleep 1
				echo
				echo "Rebooting..."
				unset reb
				sync
				sleep 1
				/usr/sbin/reboot
				exit 0
			elif [[ "$reb" = "N" || "$reb" = "n" ]] ;
			then
				sleep 1
				echo
				echo "Exiting Now."
				echo
				unset reb
				exit 0
			fi
		else
			echo
			echo "No KERNEL/SYSTEM images are in-place."
		fi
		;;

	s)
		options_found=1
		# checking for a script update, and notifying. no actual update going on here.
		rsvers=$(curl --silent https://raw.github.com/sethyx/OpenELEC_Dev/master/openelec-nightly_latest.sh | grep "VERSION=" | grep -v grep | sed 's/[^0-9]*//g')
		if [ "$rsvers" -gt "$VERSION" ] ;
		then
			echo
			echo "*---| Script Update Avaliable."
			echo "*---| Current Version: $VERSION"
			echo "*---| New Version: $rsvers"
			echo
			echo "----> Run Without Options to Update"
		else
			echo
			echo "No Script Updates Available at this Time."
			echo "Check Back Later."
		fi
		;;

	l)
		options_found=1
		# whats our current revision
		echo
		echo "My Local Build: `cat /etc/version | tail -c 6 | tr -d 'r'`"
		;;

	q)
		options_found=1
		# supress output -- intentionally undocumented (needed only for GUI interaction)
		echo() { :; }
		options_found=0
		update_yes=1
		;;

	z)
		options_found=1
		spinner() {
		proc=$1
		while [ -d /proc/$proc ] ;
		do
			echo -ne '/' ; sleep 0.05
			echo -ne "\033[0K\r"
			echo -ne '-' ; sleep 0.05
			echo -ne "\033[0K\r"
			echo -ne '\' ; sleep 0.05
			echo -ne "\033[0K\r"
			echo -ne '|' ; sleep 0.05
			echo -ne "\033[0K\r"
		done
		return 0
		}
		trap ctrl_c 2
		ctrl_c ()
		{
		echo -ne "\n\n"
		if [ -d $temploc ] ;
		then
			rm -rf $temploc
		fi
		echo "User aborted process."
		echo -ne "SIGINT Interrupt caught"
		echo -ne "\nTemporary files removed\n"
		exit 1
		}
		arch=$(cat /etc/arch)
		# roll back or forward to a version of our choosing
		while true; do
			echo
			echo "Are you sure you want to switch to a newer/older build (y/n) ?"
			read -n1 -p "==| " alt
			alt=$alt
		if [[ $alt != "Y" ]] && [[ $alt != "y" ]] && [[ $alt != "N" ]] && [[ $alt != "n" ]] ;
		then
			echo
			echo
			echo "Unrecognized Input."
			sleep 2
			echo "Please answer (y/n)"
			continue
		elif [[ $alt = "Y" || $alt = "y" ]] ;
		then
			echo
			echo
			echo -ne "Please Wait...\033[0K\r"
			mkdir -p $temploc
			curl --silent $mode/ | grep $arch | sed -e ';s/.*<a href="//' -e 's/">.*//' > $temploc/temp
			curl --silent $mode/old/ | grep $arch | sed -e ';s/.*<a href="//' -e 's/">.*//' >> $temploc/temp
			echo -ne "\033[0K\r"
			echo
			echo "Builds Available for your Architecture: ($arch)"
			cat $temploc/temp | sort -n > $temploc/temp3
			rm $temploc/temp
			echo "---------------------------------------"
			echo
			if [[ -z `cat $temploc/temp3` ]] ;
			then
				echo "There are either no available builds for your architecture at this time, or"
				echo "the only build avaliable, is the same build revision you are currently on."
				echo "Please check again later. You may also check manually for yourself here:"
				echo "http://sources.openelec.tv/tmp/image/"
				echo
				echo "Exiting Now."
				rm -rf $temploc
				unsetv
				exit 1
				fi
				for i in `cat $temploc/temp3`
				do
					echo -n "Build: " ; echo -n "$i" | cut -f 5-5 -d'-' | sed '$s/........$//' | tr -d "r" ; echo -n "-----> Compiled On: " ; echo -n "$i" | cut -f 4-4 -d'-' | sed 's/......$//;s/./& /4' | sed 's/./& /7' | awk '{ print "[ "$2"/"$3"/"$1" ]" }' ; echo
				done
				echo "----------------------------------"
				echo
				numbers=$(cat $temploc/temp3)
				for i in $numbers; do
					echo $i | tail -c 15 | sed 's/.\{8\}$//' | tr -d "\-r" >> $temploc/numbers
				done
				list=$(cat $temploc/numbers)
				count=$(cat $temploc/numbers | sed -n 1p | wc -m)
				fbrevd=$(($count-1))
				while true; do
					echo "Enter the build number you want from the list above (Ex: "`head -1 $temploc/numbers`")"
					echo "Note: This list may be longer then your scrollback buffer will show."
					read -p "==| " fbrev
					count2=$(echo -n $fbrev | wc -m)
				if (( "$count2" < "$fbrevd" )) ;
				then
					echo
					echo "Error: [ $fbrev ] is not a valid build."
					echo "It does not exist; or is not a numerical value as expected."
					echo "Please check your entry, and try again."
					echo
				continue
				fi
				if ! [[ "$list" =~ "$fbrev" ]] ;
				then
					echo
					echo "Error: [ $fbrev ] is not a valid build number."
					echo "It does not exist, or is not a numerical value"
					echo "Please double check your entry."
					echo "not in list"
					continue
				fi
				break
				done
				unset numbers
				unset list
				fn=$(grep "$fbrev" $temploc/temp3 | awk '{print $1}')
				echo
				echo "Downloading:"
				echo -ne "Please Wait..\033[0K\r"
				fe=$(curl --silent $mode/$fn --head | head -n1 | wc -m)
				if [ "$fe" = "17" ] ;
				then
					wget -O $temploc/$fn $mode/$fn
				else
					wget -O $temploc/$fn $mode/old/$fn
				fi
				echo -ne "\033[0K\r"
				echo "Done!"
				rm $temploc/temp3
				extract="$temploc/$fn"
				echo
				echo
				echo "Extracting Files:"
				echo -ne "Please Wait...\033[0K\r"
				tar -xjf $extract -C $temploc &
				echo -ne "\033[0K\r"
				pid=$!
				spinner $pid
				echo "Done!"
				unset pid
				sleep 2
				echo
				###### Move KERNEL & SYSTEM  and respective md5's to /storage/.update/
				echo "Moving Images to /storage/.update"
				echo -ne "Please Wait...\033[0K\r"
				mv $temploc/OpenELEC-*/target/* /storage/.update &
				pid=$!
				spinner $pid
				echo -ne "\033[0K\r"
				echo "Done!"
				unset pid
				sleep 2
				###### Compare md5-sums
				sysmd5=$(cat /storage/.update/$dsmd5 | awk '{print $1}')
				kernmd5=$(cat /storage/.update/$dkmd5 | awk '{print $1}')
				kernrom=$(md5sum /storage/.update/$dkernel | awk '{print $1}')
				sysrom=$(md5sum /storage/.update/$dsystem | awk '{print $1}')
				if [ "$sysmd5" = "$sysrom" ] ;
				then
					echo
					echo "md5 ==> SYSTEM: OK!"
					sys_return=0
					sleep 2
				else
					sys_return=1
				echo "---   WARNING   ---"
				echo "SYSTEM md5 MISMATCH!"
				echo "--------------------"
				echo "There is an integrity problem with the SYSTEM package"
				echo "Notify one of the developers on the Forums or IRC that"
				echo "the SYSTEM image of $fn.tar.bz2 is corrupt"
				echo
				echo "IRC Webchat:"
				echo "http://webchat.freenode.net/?channels=openelec"
				sleep 3
				rm -f /storage/.update/$dsystem
				rm -f /storage/.update/$dsmd5
				rm -rf $temploc
				sync
				fi
				if [ "$kernmd5" = "$kernrom" ] ;
				then
					echo "md5 ==> KERNEL: OK!"
					kern_return=0
				else
				kern_return=1
				echo "---   WARNING   ---"
				echo "KERNEL md5 MISMATCH!"
				echo "--------------------"
				echo "There is an integrity problem with the KERNEL package"
				echo "Notify one of the developers on the Forums or IRC that"
				echo "the KERNEL image of $fn.tar.bz2 is corrupt"
				echo
				echo "IRC Webchat:"
				echo "http://webchat.freenode.net/?channels=openelec"
				sleep 3
				rm -f /storage/.update/$dkernel
				rm -f /storage/.update/$dkmd5
				rm -rf $temploc
				sync
				fi
				return=$(($kern_return+$sys_return))
				if [[ "$return" = "2" ]] ;
				then
					echo "md5 Mismatch Detected."
					echo "Update Terminated."
					rm -rf $temploc
					unsetv
					exit 1
				fi
				###### some feedback
				sleep 2
				echo "File Integrity: PASSED!"
				echo
				sleep 1
				echo -ne "Continuing...\033[0K\r"
				sleep 2
				echo -ne "\033[0K\r"
				###### ask if we want to reboot now
				while true; do
					echo
					echo "Update Preperation Complete."
					sleep 2
					echo "You must reboot to complete the update."
					echo "Would you like to reboot now (y/n) ?"
					read -n1 -p "==| " reb
					reb=$reb
				if [[ "$reb" != "Y" ]] && [[ "$reb" != "y" ]] && [[ "$reb" != "N" ]] && [[ "$reb" != "n" ]] ;
				then
					echo
					echo
					echo "Unrecognized Input."
					echo "Please answer (y/n)"
					echo
					continue
				elif [[ "$reb" = "Y" || "$reb" = "y" || "$reb" = "Yes" || "$reb" = "yes" ]] ;
				then
					sleep 1
					echo
					echo
					echo "Rebooting..."
					rm -rf $temploc
					sync
					sleep 1
					/usr/sbin/reboot
					exit 0
				elif [[ "$reb" = "N" || "$reb" = "n" || "$reb" = "No" || "$reb" = "no" ]] ;
				then
					sleep 1
					echo
					echo
					echo "Please reboot to complete the update."
					echo "Exiting."
					rm -rf $temploc
					exit 0
				fi
				done
				## everything went well: we're done !
				rm -rf $temploc
				exit 0
				elif [[ $alt = "N" || $alt = "n" || $alt = "No" || $alt = "no" ]] ;
				then
					echo
					echo
					echo "User aborted process."
					sleep 2
					echo "Exiting."
					rm -rf $temploc
					echo
					exit 0
				fi
				done
		;;

	v)
		options_found=1
		;;

	b)
		# reboot
		options_found=1
		echo "Rebooting..."
		/usr/sbin/reboot
		;;

	h|help)
		options_found=1
		# options avaliable and usage.
		echo
		echo "Usage:  $0 [-iozacrlsbvh]"
		echo
		echo "-i                   check if SYSTEM & KERNEL are already in-place; suggest reboot."
		echo "-o                   list all avaliable archived builds for your architecture."
		echo "-z                   roll back or forward to a version of your choosing."
		echo "-a                   list all current builds for your architecture."
		echo "-c                   quick check to see if we're up-to-date."
		echo "-r                   check the remote build revision."
		echo "-l                   check our local build revision."
		echo "-s                   check for new script version"
		echo "-b                   reboot OpenELEC"
		echo "-v                   script version."
		echo "-h/--help            help."
		echo
		exit 0
		;;

	\?)
		# terminate if invalid option is used
		echo "Invalid option used: -$OPTARG" >&2
		echo "Run with -h/--help"
		echo
		exit 1
		;;
	esac
done


###### allows multiple options to be calculated and displayed

shift $(($OPTIND - 1))


###### if options are specified, we wont proceede any further, unless -z is passed

if [ "$options_found" -ge "1" ] ;
then
	exit 0
fi


###### removes temporary files that have been created if the user prematurly aborts the update process

trap ctrl_c 2
ctrl_c ()
{
echo -ne "\n\n"
echo "User aborted process."
echo -ne "SIGINT Interrupt caught"
echo -ne "\nTemporary files removed\n"
if [ -d $temploc ] ;
then
	rm -rf $temploc
fi
unsetv
exit 1
}


###### for cleanup purposes, we're removing some enviroment variables we've set, after the script is run or aborted

unsetv ()
{
unset options_found
unset kern_return
unset currentsys
unset update_yes
unset sys_return
unset SYS_KERN
unset PRESENT
unset VERSION
unset kernmd5
unset kernrom
unset temploc
unset akernel
unset asystem
unset dkernel
unset dsystem
unset sysrom
#unset OPTIND
unset FOLDER
unset branch
unset sysmd5
unset rsvers
unset status
unset ALLOW
unset dkmd5
unset dsmd5
unset PAST
unset mode
unset port
unset pass
unset user
unset arch
unset reb
unset alt
unset pid
unset yn
}


###### some visual feedback for long operations; especially useful on the RPi

spinner() {
proc=$1
while [ -d /proc/$proc ];do
echo -ne '/' ; sleep 0.05
echo -ne "\033[0K\r"
echo -ne '-' ; sleep 0.05
echo -ne "\033[0K\r"
echo -ne '\' ; sleep 0.05
echo -ne "\033[0K\r"
echo -ne '|' ; sleep 0.05
echo -ne "\033[0K\r"
done
return 0
}


###### check that were actually running a devel build already; otherwise cancel the opertation, with an explanation of why (this function is defined above)

stable

###### making sure github is alive and ready to update the script if nessessary.

s_update ()
{
echo -ne "Please Wait...\033[0K\r"
sleep 2
echo -ne "\033[0K\r"
echo -ne "Checking Script Update Server State...\033[0K\r"
ping -qc 3 raw.github.com > /dev/null &
outcome="$?"
pid=$!
spinner $pid
unset pid
echo -ne "\033[0K\r"
if [ "$outcome" = "0" ] ;
then
	echo -ne "Update Server Active.\033[0K\r"
	sleep 2
	echo -ne "\033[0K\r"
	###### check if a script update is in progress
	if [ ! -f /tmp/update_in_progress ] ;
	then
		###### update_in_progress does not exist :: first run
		###### checking script version; auto updating and re-running new version
		mkdir -p $temploc
		curl --silent https://raw.github.com/sethyx/OpenELEC_Dev/master/openelec-nightly_latest.sh > $temploc/tempscript
		Allow=$(grep "Allow_Update=" $temploc/tempscript | grep -v grep | sed 's/=/ /g' | awk '{print $2}' | awk '{gsub(/[[:punct:]]/,"")}1')
		rsvers=$(grep "VERSION=" $temploc/tempscript | grep -v grep | sed 's/[^0-9]*//g')
		while true; do
		if [ "$rsvers" -gt "$VERSION" ] ;
		then
			if [ "$Allow" = "NO" ] ;
			then
				echo "| Script Updates are temporarily disabled |"
				echo "| Continuing..."
				echo
				break
			fi
			echo
			echo "*---| Script Update Available."
			echo "*---| Current Version: $VERSION"
			echo "*---| New Version: $rsvers"
			echo
			echo "Commit/Change Log:"
			echo "------------------"
			echo "https://github.com/wavrunrx/OpenELEC_Dev/commits/master"
			echo
			echo
			while true; do
			echo "Would you like to update the script now (y/n) ?"
			read -n1 -p "==| " supdate
			supdate=$supdate
			if [[ $supdate != "Y" ]] && [[ $supdate != "y" ]] && [[ $supdate != "N" ]] && [[ $supdate != "n" ]] ;
			then
				echo
				echo
				echo "Unrecognized Input."
				sleep 1
				echo "Please answer (y/n)"
				echo
				continue
			elif [[ $supdate = "Y" || $supdate = "y" ]] ;
			then
				echo
				echo
				echo
				echo "*---| Updating OpenELEC_DEV:"
				echo -ne "      Please Wait...\033[0K\r"
				sleep 1
				curl --silent -fksSL -A "`curl -V | head -1 | awk '{print $1, $2, $3}'`" http://bit.ly/TOf3qf > `dirname $0`/openelec-nightly_$rsvers.sh &
				pid=$!
				spinner $pid
				unset pid
				echo -ne "\033[0K\r"
				echo "Done !"
				echo
				echo
				sleep 1
				echo "BEGIN: OpenELEC_DEV v$rsvers Now"
				echo "---------------------------"
				echo
				echo
				echo
				###### indicate update in progress to next script instance
				touch /tmp/update_in_progress
				###### indicate no update check nessessary to next script instance since we've just finished a check
				touch /tmp/no_display
				###### remove update indication flag
			 	rm -f /tmp/update_in_progress
				###### swapping old script with new
				rm -f `dirname $0`/openelec-nightly_latest.sh
				mv `dirname $0`/openelec-nightly_$rsvers.sh `dirname $0`/openelec-nightly_latest.sh
				chmod 755 `dirname $0`/openelec-nightly_latest.sh
				###### run a new version of update script
				sh `dirname $0`/openelec-nightly_latest.sh
				###### exit old script
				exit
			elif [[ $supdate = "N" || $supdate = "n" ]] ;
			then
				echo
				echo
				echo
				break
			fi
			done
		else
			echo -ne "Script Update Not Avaliable.\033[0K\r"
			sleep 2
			echo -ne "\033[0K\r"
			echo -ne "Continuing...\033[0K\r"
			sleep 2
			echo -ne "\033[0K\r"
			break
		fi
		done
	fi
else
	echo
	echo "* Script Update Server Not Responding."
	echo "* Try again later."
	echo "  -------------------------------"
	echo -ne "Continuing...\033[0K\r"
	sleep 2
	echo -ne "\033[0K\r"
	echo
fi
}


###### if 'no_display' exists, we are going to skip the update check; otherwise, we check

if [ ! -f /tmp/no_display ] ;
then
	s_update
fi


###### remove no update check nessessary indicator so next time we run the script we will again check for updates

rm -f /tmp/no_display


###### create the .update directory for OpenELEC

mkdir -p /storage/.update


###### checking for a previous run :: if SYSTEM & KERNEL files are still in ~/.update then we havent rebooted since we last ran.
###### this check prevents us from redownloading the update package.

while true; do
	SYS_KERN=$(ls /storage/.update/* 2> /dev/null | wc -l)
	if [ "$SYS_KERN" = "4" ] ;
	then
		echo
		echo
		echo "KERNEL & SYSTEM are already in place."
		echo "You must reboot to complete the update."
		echo "Would you like to reboot now (y/n) ?"
		read -n1 -p "==| " reb
	if [[ $reb != "Y" ]] && [[ $reb != "y" ]] && [[ $reb != "N" ]] && [[ $reb != "n" ]] ;
	then
		echo
		echo
		echo "Unrecognized Input."
		sleep 1
		echo "Please answer (y/n)"
		continue
	elif [[ $reb = "Y" || $reb = "y" ]] ;
	then
		echo
		echo
		echo
		echo "Rebooting..."
		rm -rf $temploc
		unsetv
		sync
		sleep 2
		/usr/sbin/reboot
	elif [[ $reb = "N" || $reb = "n" ]] ;
	then
		echo
		echo
		echo "Please reboot to complete the update."
		sleep 1
		echo "Exiting."
		rm -rf $temploc
		unsetv
		exit 0
		fi
	fi
	break
done


###### delete the temporary working directory; create if doesnt exist

if [ -d "$temploc" ] ;
then
	rm -rf $temploc
	mkdir -p $temploc
else
	mkdir -p $temploc
fi


###### if there are no builds avaliable on the server for your specific architecture, we are going to notify you, and gracefully exit
###### also captures remote filename & extension to be used at later times

arch=$(cat /etc/arch)
curl --silent $mode/ | grep $arch | sed -e ';s/.*<a href="//' -e 's/">.*//' > $temploc/temp
if [[ -z `cat $temploc/temp` ]] ;
then
        echo "There are either no available builds for your architecture at this time, or"
        echo "the only build avaliable is the same revision you are already on."
        echo "Please check again later."
        echo
        echo "Exiting Now."
        rm -rf $temploc
        unsetv
        exit 1
fi


###### remove all but the newest build from our list

if [ $(wc -l $temploc/temp | cut -c -1) -gt "1" ] ;
then
	cat $temploc/temp | tail -n 1 > $temploc/temp2
else
	mv $temploc/temp $temploc/temp2
fi

if [ -f $temploc/temp ] ;
then
	rm $temploc/temp
fi


###### i386, or x86_64 ?
instruction_set=$(cat /etc/arch | sed 's/\./ /g' | awk '{print $2}')

## filename, no extension
FOLDER=$(cat $temploc/temp2 | sed 's/.tar.bz2//g')

## capture local build revision
PAST=$(cat /etc/version | awk '{gsub(/[[:punct:]]/," ")}1' | awk '{print $3}' | tr -d 'r')

## capture remote build revision (allows infinite revision growth)
PRESENT=$(cat $temploc/temp2 | awk '{gsub(/[[:punct:]]/," ")}1' | awk '{print $6}' | tr -d 'r')


###### set $PRESENT properly for 64-bit machines

if [ "$instruction_set" = "x86_64" ] ;
then
	PRESENT=$(cat $temploc/temp2 | awk '{gsub(/[[:punct:]]/," ")}1' | awk '{print $7}' | tr -d 'r')
fi


###### checking to make sure we are actually running an official development build. if we dont check this; the comparison routine will freak out if our local
###### build is larger then the largest (newest) build on the official openelec snapshot server.

if [ "$PRESENT" -lt "$PAST" ] ;
then
	echo
	echo "You are currently using an unofficial development build of OpenELEC."
	echo "This isn't supported, and will yield unexpected results if we continue."
	echo "Your build is a higher revision then the highest available on the official"
	echo "snapshot server as seen here: http://sources.openelec.tv/tmp/image/"
	echo "((RaspberryPi, see here: http://openelec.thestateofme.com/dev_builds/ ))"
	echo "In order to use this update script, you *MUST* be using an official"
	echo "build, that was obtained from the aforementioned snapshot server."
	echo
	echo "Local:  $PAST"
	echo "Remote: $PRESENT"
	echo
	sleep 2
	echo "Exiting Now."
	echo
	rm -rf $temploc
	unsetv
	exit 1
fi


###### this is only comes into play if the option -q is passed. if so, we supress output if an update isnt available. if one is, we dont care, and want to exit (this will be used for the addon gui sometime in the future)

if [ "$update_yes" = "1" ] ;
then
	exit 0
fi


###### variables used for GUI notifications

## xbmc webserver port
port=$(cat /storage/.xbmc/userdata/guisettings.xml | grep "<webserverport>" | sed 's/[^0-9]*//g')

## xbmc webserver password
pass=$(cat /storage/.xbmc/userdata/guisettings.xml | grep "<webserverpassword>" | grep -Eio "[a-z]+" | sed -n 2p)

## xbmc webserver username
user=$(cat /storage/.xbmc/userdata/guisettings.xml | grep "<webserverusername>" | grep -Eio "[a-z]+" | sed -n 2p)


###### compare local and remote revisions; decide if we have updates ready to donwload

if [ "$PRESENT" -gt "$PAST" ] ;
then
	echo
	echo "### WARNING:"
	echo "### UPDATING TO OR FROM DEVELOPMENT BUILDS MAY HAVE POTENTIALLY UNPREDICTABLE"
	echo "### EFFECTS ON THE STABILITY AND OVERALL USABILITY OF YOUR SYSTEM. SINCE NEW"
	echo "### CODE IS LARGELY UNTESTED, DO NOT EXPECT SUPPORT ON ANY ISSUES YOU MAY"
	echo "### ENCOUNTER. IF SUPPORT WERE TO BE OFFERED, IT WILL BE LIMITED TO"
	echo "### DEVELOPMENT LEVEL DEBUGGING."
	echo
	echo
	echo -ne "Please Wait...\033[0K\r"
	sleep 2
	echo -ne "\033[0K\r"
	echo ">>>| OpenELEC"
	echo "Updates Are Available."
	echo "Local:   $PAST          Compiled: `cat /etc/version | cut -f 2-2 -d'-' | sed 's/......$//;s/./& /4' | sed 's/./& /7' | awk '{ print "[ "$2"/"$3"/"$1" ]" }'`"
	echo "Remote:  $PRESENT          Compiled: `echo $FOLDER | cut -f 4-4 -d'-' | sed 's/......$//;s/./& /4' | sed 's/./& /7' | awk '{ print "[ "$2"/"$3"/"$1" ]" }'`"
	if [ "$arch" = "RPi.arm" ] ;
	then
		echo
		echo "RPi Build Source:"
		echo "http://openelec.thestateofme.com/dev_builds/"
	else
		echo
		echo "`cat /etc/arch | sed 's/\./ /g' | awk '{print $1}'` Build Source:"
		echo "http://sources.openelec.tv/tmp/image"
	fi
	#curl -v -H "Content-type: application/json" -u $user:$pass -X POST -d '{"id":1,"jsonrpc":"2.0","method":"GUI.ShowNotification","params":{"title":"OpenELEC_Dev","message":"Update Found ! Remote Build: $PRESENT","displaytime":8000}}' http://localhost:$port/jsonrpc
	echo
	## The remote build is newer then our local build. Asking for input.
	echo "Would you like to update (y/n) ?"
	read -n1 -p "==| " yn
	if [[ $yn != "Y" ]] && [[ $yn != "y" ]] && [[ $yn != "N" ]] && [[ $yn != "n" ]] ;
	then
		echo
		echo
		echo "Unrecognized Input."
		sleep 2
		echo "Please answer (y/n)"
		echo "Exiting."
		echo
		rm -rf $temploc
		unsetv
		exit 1
	elif [[ $yn = "Y" || $yn = "y" ]] ;
	then
		sleep .5
		echo
		echo
		echo "Downloading Image:"
		wget $mode/`cat $temploc/temp2` -P "$temploc"
		echo "Done!"
		extract="$temploc/$FOLDER.tar.bz2"
		sleep 1
	elif [[ $yn = "N" || $yn = "n" ]] ;
	then
		echo
		echo
		echo "User aborted process."
		sleep 2
		echo "Exiting."
		echo
		rm -rf $temploc
		unsetv
		exit 0
	fi
else
	## remote build is not newer then what we've got already. Exit.
	echo -ne "\033[0K\r"
	echo
	echo ">>>| OpenELEC"
	echo "No Updates Available."
	echo "Local:   $PAST          Compiled: `cat /etc/version | cut -f 2-2 -d'-' | sed 's/......$//;s/./& /4' | sed 's/./& /7' | awk '{ print "[ "$2"/"$3"/"$1" ]" }'`"
	echo "Remote:  $PRESENT          Compiled: `echo $FOLDER | cut -f 4-4 -d'-' | sed 's/......$//;s/./& /4' | sed 's/./& /7' | awk '{ print "[ "$2"/"$3"/"$1" ]" }'`"
	echo
	echo "You are on the latest build for your platform [`cat /etc/arch`]"
	echo "Check again later."
	echo
	rm -rf $temploc
	unsetv
	exit 0
fi


###### extract SYSTEM & KERNEL images to the proper location for update

echo
echo "Extracting Files:"
tar -xjf $extract -C $temploc &
pid=$!
spinner $pid
echo "Done!"
unset pid
sleep 2


###### Move KERNEL & SYSTEM  and respective md5's to /storage/.update/
echo
echo "Moving Images To: /storage/.update"
echo -ne "Please Wait...\033[0K\r"
mv $temploc/OpenELEC-*/target/* /storage/.update &
pid=$!
spinner $pid
echo -ne "\033[0K\r"
echo "Done!"
unset pid
sleep 2


###### Compare md5-sums

sysmd5=$(cat /storage/.update/$dsmd5 | awk '{print $1}')
kernmd5=$(cat /storage/.update/$dkmd5 | awk '{print $1}')
kernrom=$(md5sum /storage/.update/$dkernel | awk '{print $1}')
sysrom=$(md5sum /storage/.update/$dsystem | awk '{print $1}')

echo "Data Integrity Check:"
if [ "$sysmd5" = "$sysrom" ] ;
then
	echo
	echo "md5 ==> SYSTEM: OK!"
	sys_return=0
	sleep 2
else
	sys_return=1
	echo "---   WARNING   ---"
	echo "SYSTEM md5 MISMATCH!"
	echo "--------------------"
	echo "There is an integrity problem with the SYSTEM package"
	echo "Notify one of the developers in the Forums or IRC that"
	echo "the SYSTEM image of $fn.tar.bz2 is corrupt"
	echo
	echo "IRC Webchat:"
	echo "http://webchat.freenode.net/?channels=openelec"
	sleep 3
	rm -f /storage/.update/$dsystem
	rm -f /storage/.update/$dsmd5
	rm -rf $temploc
	sync
fi

if [ "$kernmd5" = "$kernrom" ] ;
then
	echo "md5 ==> KERNEL: OK!"
	kern_return=0
else
	kern_return=1
	echo "---   WARNING   ---"
	echo "KERNEL md5 MISMATCH!"
	echo "--------------------"
	echo "There is an integrity problem with the KERNEL package"
	echo "Notify one of the developers in the Forums or IRC that"
	echo "the SYSTEM image of $fn.tar.bz2 is corrupt"
	echo
	echo "IRC Webchat:"
	echo "http://webchat.freenode.net/?channels=openelec"
	sleep 3
	rm -f /storage/.update/$dkernel
	rm -f /storage/.update/$dkmd5
	rm -rf $temploc
	sync
fi


###### the system rom is evaluated first.
###### if an error is found, the process is terminated and we wouldnt know if the kernel image was broken as well.
######
###### here we know that if the sum of $kern_return, and $sys_return is over "1", that one or both of the images are broken, and we've already been
###### notified which one it was above. Exit.

return=$(($kern_return+$sys_return))
if [[ "$return" = "2" ]] ;
then
	echo "md5 Mismatch Detected."
	echo "Update Terminated."
	rm -rf $temploc
	unsetv
	exit 1
fi

sleep 1
echo "File Integrity Check: PASSED!"
echo
echo -ne "Continuing...\033[0K\r"
sleep 2
echo -ne "\033[0K\r"
echo


###### remove old backup builds

rm -rf /storage/downloads/OpenELEC_r*


###### make sure 'downloads' exists; doesnt get created untill the "Downloads" smb share is accessed for the first time.

mkdir -p /storage/downloads


###### create a backup of our current, and new build for easy access if needed for a emergency rollback

echo "Creating a backup of your PREVIOUS [ SYSTEM & KERNEL ] images."
echo -ne "Please Wait...\033[0K\r"
mkdir /storage/downloads/OpenELEC_r$PAST
cp /flash/$akernel /storage/downloads/OpenELEC_r$PAST/$dkernel
cp /flash/$asystem /storage/downloads/OpenELEC_r$PAST/$dsystem
chmod +x /storage/downloads/OpenELEC_r$PAST/$dkernel
chmod +x /storage/downloads/OpenELEC_r$PAST/$dsystem
md5sum /storage/downloads/OpenELEC_r$PAST/$dkernel > /storage/downloads/OpenELEC_r$PAST/$dkmd5 &
pid=$!
spinner $pid
unset pid
md5sum /storage/downloads/OpenELEC_r$PAST/$dsystem > /storage/downloads/OpenELEC_r$PAST/$dsmd5 &
pid=$!
spinner $pid
unset pid
echo -ne "\033[0K\r"
echo
echo "     Important Notice"
echo "--------------------------"
echo "     In the need of an emergency rollback:"
echo "-->  A backup copy of your *PREVIOUS* SYSTEM & KERNEL images [ revision $PAST ]"
echo "     have been created here:  /storage/downloads/OpenELEC_r$PAST"
echo
echo
echo "Creating a backup of your NEW [ SYSTEM & KERNEL ] images."
echo -ne "Please Wait...\033[0K\r"
mkdir -p /storage/downloads/OpenELEC_r$PRESENT
sleep 1
cp /storage/.update/$dkernel /storage/.update/$dsystem /storage/.update/$dkmd5 /storage/.update/$dsmd5 /storage/downloads/OpenELEC_r$PRESENT &
pid=$!
spinner $pid
unset pid
echo -ne "\033[0K\r"
echo
echo "     Important Notice"
echo "--------------------------"
echo "     In the need of an emergency rollback:"
echo "-->  A backup copy of your *NEW* SYSTEM & KERNEL images [ revision $PRESENT ]"
echo "     have been created here:  /storage/downloads/OpenELEC_r$PRESENT"
echo
sleep 5


###### ask if we want to reboot now

echo
echo
echo "Update Preperation Complete !"
sleep 2
while true; do
echo
echo "You must reboot to finish the update."
echo "Would you like to reboot now (y/n) ?"
read -n1 -p "==| " reb
echo
if [[ "$reb" != "Y" ]] && [[ "$reb" != "y" ]] && [[ "$reb" != "N" ]] && [[ "$reb" != "n" ]] ;
then
	echo
	echo "Unrecognized Input."
	echo "Please answer (y/n)"
	echo
	continue
elif [[ "$reb" = "Y" || "$reb" = "y" ]] ;
then
	sleep 1
	echo
	echo "Rebooting..."
	rm -rf $temploc
	sync
	/usr/sbin/reboot
elif [[ "$reb" = "N" || "$reb" = "n" ]] ;
then
	sleep 1
	echo
	echo "User aborted process."
	echo "Please reboot to complete the update."
	echo "Exiting."
	rm -rf $temploc
	unsetv
	exit 1
fi
done


## everything went well: we're done !

exit 0