update-server.sh

#!/bin/bash
#import settings.
if [ -f ~/.sm5-server.rc ]; then
	. ~/.sm5-server.rc
else
	echo "configuration missing!" 
	exit 1
fi
_CPUS=($(nproc) - 1)
#verify settings import

#log to console if verbose and to syslog if not. 
log()
{
	_MESSAGE=$@
	if [ $VERBOSE = 1 ]; then
		echo "$_MESSAGE"
	fi
	logger piu-updater: $_MESSAGE
}

#usage display.
usage()
{
cat << EOF
usage: $0 options

This script will check for updates in stepmania, piuio, and the piu-dex-gw skins and build packages if they exist..

OPTIONS:
   -h      Show this message
   -s      Force a build of stepmania 
   -p      Force a build for piuio
   -t      Force a build on the themese directory
   -v      Verbose
   -c      Check only (implies -v)
EOF
}

#parse options. 
BUILD_SM=0
BUILD_PIUIO=0
BUILD_THEME=0
VERBOSE=0
CHECK_ONLY=0
declare -a BUILD_THEME_NAMES
while getopts .hsptvc. OPTION
do
	case $OPTION in
		h)
			usage
			exit 1
			;;
		v)
                       VERBOSE=1
                        ;;
		s)
			BUILD_SM=1
			;;
		p)
			BUILD_PIUIO=1 
			;;
		t)
			BUILD_THEME=1
			;;
		c)
			VERBOSE=1
			CHECK_ONLY=1
			;;
		?)
			usage
			exit
			;;
	esac
done
#declare SM_VER 
declare SM_VER="xx"

#spinner class for long tasks 
spinner()
{
	local pid=$1
	local delay=0.75
	local spinstr='|/-\'
	while [ "$(ps a | awk '{print $1}' | grep $pid)" ]; do
		if [ $VERBOSE = 1 ]; then
			local temp=${spinstr#?}
			printf " [%c]  " "$spinstr"
			local spinstr=$temp${spinstr%"$temp"}
			sleep $delay
			printf "\b\b\b\b\b\b"
		fi
	done
	if [ $VERBOSE = 1 ]; then
		printf "    \b\b\b\b"
	fi
}
#cleanup directories  Takes 1 arg for directory.  Will keep the latest 5 .md5sum and tar.gz in the directory.
cleanup ()
{
	CLEAN_PATH=$1
	cd "$CLEAN_PATH"
	log "Cleaning up $CLEAN_PATH"
        ls -d -C1 -t $CLEAN_PATH/*.gz| awk 'NR>6'|xargs rm -f
        ls -d -C1 -t $CLEAN_PATH/*.md5sum| awk 'NR>6'|xargs rm -f

}
#custom script specifically for cleaning the themes directory.
clean_themes ()
{
        CLEAN_PATH=$1
        cd "$THEME_PATH"
        for THEME in **
        do
	        log "Cleaning up $THEME in $CLEAN_PATH"
        	ls -d -C1 -t $CLEAN_PATH/*$THEME*.gz| awk 'NR>6'|xargs rm -f
        	ls -d -C1 -t $CLEAN_PATH/*$THEME*.md5sum| awk 'NR>6'|xargs rm -f
	done


}
#basic function to update to the latest git, run make clean and make and if successful build the release package.
build_sm ()
{
	cd "$SM_PATH"
	log "Updating Stepmania sources"
	git pull > /dev/null
	log "Cleaning Stepmania source directory"
	make clean > /dev/null 2>&1 &
	spinner $!
	log "Configuring Stepmania"
	./autogen.sh 
	./configure   --with-static-png   --with-static-jpeg --with-static-zlib  --with-static-bzip  --with-static-vorbis --prefix=$SM_INSTALL_PATH > configure.out 2>&1 &
	spinner $!
	rm -rf $SM_INSTALL_PATH/stepmania\ 5
	log "Making Stepmania!"
	make -j$_CPUS > make.out 2>&1 &
	spinner $!
	if [ $? -eq 0 ]; then
		make install > make.install 2>&1 &
		spinner $!
		SM_VER=$(grep PACKAGE_VERSION= configure | awk -F\' '{print $(NF-1)}')
	        cd "$SM_INSTALL_PATH/stepmania-$SM_VER"
        	mkdir -p $SM_INSTALL_PATH/stepmania-$SM_VER/Packages > /dev/null
	        touch portable.ini
		bundle_sm
	else
		log "error in make!"
	fi
}

#build the release package and md5sum and update -current symlinks
bundle_sm ()
{
	if [ -f "$SM_INSTALL_PATH/stepmania-$SM_VER/stepmania" ]; then
		_NOW=$(date +%Y%m%d%H%M)
		cd "$SM_INSTALL_PATH/stepmania-$SM_VER"
		touch build-$_NOW
		log "Creating stepmania tar bundle."
		tar -czf $SM_REPO_PATH/stepmania-build-$_NOW.tar.gz * 
		ln -sf $SM_REPO_PATH/stepmania-build-$_NOW.tar.gz $SM_REPO_PATH/stepmania-build-current.tar.gz
		log "Creating stepmania md5sum"
		md5sum $SM_REPO_PATH/stepmania-build-$_NOW.tar.gz | awk '{ print $1 }' > $SM_REPO_PATH/stepmania-build-$_NOW.md5sum
		ln -sf $SM_REPO_PATH/stepmania-build-$_NOW.md5sum $SM_REPO_PATH/stepmania-build-current.md5sum
		cleanup $SM_REPO_PATH
	else
		log "stepmania failed to build."
	fi
#clean up installation path. 
	cd ~
	rm -rf "$SM_INSTALL_PATH/stepmania-$SM_VER"
}
bundle_piu_theme ()
{
        _NOW=$(date +%Y%m%d%H%M)
        cd "$THEME_PATH"
	for THEME in ${BUILD_THEME_NAMES[@]}
	do
		if [ -d  $THEME_PATH/$THEME/.git ]; then 
			cd "$THEME_PATH/$THEME"
			git pull > /dev/null
		fi
		cd "$THEME_PATH"/..
		log "Creating theme $THEME bundle"
#       		tar -cazf $THEME_REPO_PATH/piu-$THEME-theme-$_NOW.tar.gz $THEME
#		ln -sf $THEME_REPO_PATH/piu-$THEME-theme-$_NOW.tar.gz $THEME_REPO_PATH/piu-$THEME-theme-current.tar.gz
#		log "Creating theme $THEME md5sum"
#		md5sum $THEME_REPO_PATH/piu-$THEME-theme-$_NOW.tar.gz | awk '{ print $1 }' > $THEME_REPO_PATH/piu-$THEME-theme-$_NOW.md5sum
#		ln -sf $THEME_REPO_PATH/piu-$THEME-theme-$_NOW.md5sum $THEME_REPO_PATH/piu-$THEME-theme-current.md5sum
		zip -9 -r -q $THEME_REPO_PATH/$THEME.smzip Themes/$THEME 
	done
#	clean_themes $THEME_REPO_PATH

}

bundle_piuio ()
{
	_NOW=$(date +%Y%m%d%H%M)
	cd "$PIUIO_PATH"
	git pull > /dev/null
	log "Creating PIUIO bundle"
	cd ..
        tar -cazf $PIUIO_REPO_PATH/piuio-$_NOW.tar.gz piuio
        ln -sf $PIUIO_REPO_PATH/piuio-$_NOW.tar.gz $PIUIO_REPO_PATH/piuio-current.tar.gz
        log "Creating PIUIO md5sum"
        md5sum $PIUIO_REPO_PATH/piuio-$_NOW.tar.gz | awk '{ print $1 }' > $PIUIO_REPO_PATH/piuio-$_NOW.md5sum
        ln -sf $PIUIO_REPO_PATH/piuio-$_NOW.md5sum $PIUIO_REPO_PATH/piuio-current.md5sum
	
	cleanup $PIUIO_REPO_PATH
}

#check a git repository to see if it needs to be updated. 
#expects $1 to be the path to the git repository and $2 to be the name of the repository. 
check_git ()
{
	local GIT_CHECK=0
	cd "$1"
	git fetch
        LOCAL=$(git rev-parse HEAD)
        REMOTE=$(git rev-parse @{u})
        BASE=$(git merge-base @ @{u})
        if [ $LOCAL = $REMOTE ]; then
                log "$2 is current."
        elif [ $LOCAL = $BASE ]; then
                log "building $2 bundle"
                GIT_CHECK=1
        elif [ $REMOTE = $BASE ]; then
                log "Local $2 repo manually updated."
        else
                log "Local $2 repo diverged"
        fi
	exit $GIT_CHECK

}
#takes md5sum of files and then md5sums those files.
# $1 is the path to the directory to be checked. 
# $2 is the path to the md5sum directory
# $3 name of the repository
check_md5sum ()
{
	local CHECK_PATH=$1
	local MD5_PATH=$2
	local REPO_NAME=$3
	local MD5_CHECK=0
	local NEW_MD5=$(find  $CHECK_PATH -type f -exec md5sum {} + | awk '{print $1}' | sort | md5sum| awk '{ print $1 }')
	if [ -f $MD5_PATH/$REPO_NAME ]; then
		#load  md5sum file and compare
		CURRENT_MD5=$(<$MD5_PATH/$REPO_NAME)
		if [ "$CURRENT_MD5" = "$NEW_MD5" ]; then
			log "$REPO_NAME is current."
		else
			log "building $REPO_NAME bundle"
			MD5_CHECK=1
			if [ $CHECK_ONLY = 0 ]; then
				echo $NEW_MD5 > $MD5_PATH/$REPO_NAME
			fi
		fi
	else
		MD5_CHECK=1
		echo $NEW_MD5 > $MD5_PATH/$REPO_NAME
		log "Creating new repo for $REPO_NAME"
	fi
	exit $MD5_CHECK
}

#main functions. 
#checks piuio git 
if [ $BUILD_PIUIO = 0 ]; then
	STATUS=$(check_git $PIUIO_PATH 'piuio')
	BUILD_PIUIO=$?
	log $STATUS
else
	log 'Forcing update of piuio package'
fi
#check stepmania's git
if [ $BUILD_SM = 0 ]; then
	STATUS=$(check_git $SM_PATH 'stepmania')
	BUILD_SM=$?
	log $STATUS
else
	log 'Forcing update of stepmania package'
fi

#checks each directory in the themes directory, determines if it's a git repo or needs an md5sum check and then makes the check
#if it needs to be updated it will add the theme name to $BUILD_THEME_NAMES and set BUILD_THEME to 1. 
if [ $BUILD_THEME = 0 ]; then
	
        cd "$THEME_PATH"
        for THEME in **
        do
		log "Checking Theme: $THEME"
		if [ -d $THEME_PATH/$THEME/.git ]; then
			STATUS=$(check_git $THEME_PATH/$THEME $THEME) 
		else
			STATUS=$(check_md5sum $THEME_PATH/$THEME $THEME_PATH/.md5sum $THEME)

		fi
                BUILD_CHECK=$?
                if [ $BUILD_CHECK = 1 ]; then
	                BUILD_THEME=1
			BUILD_THEME_NAMES+=($THEME)
                fi
		log $STATUS
	done
else
        cd "$THEME_PATH"
        for THEME in **
        do
		BUILD_THEME_NAMES+=($THEME)
	done
	log 'Forcing update of theme packages'
fi
#if we're running a check only we skip this.  
#Otherwise we run the specific build routines that were either forced by a flag or 
#Were determined to be out of date. 
if [ $CHECK_ONLY = 0 ]; then
	if [ $BUILD_SM = 1 ]; then
		build_sm
	fi
	if [ $BUILD_THEME = 1 ]; then
		bundle_piu_theme
	fi
        if [ $BUILD_PIUIO = 1 ]; then
                bundle_piuio
        fi
fi

exit 0