#!/bin/sh

# upgrade_unifi_controller.sh
# Easy UniFi Controller Upgrade Script for Unix/Linux Systems
# by Steve Jenkins (stevejenkins.com)
# Version 2.3
# Last Updated January 8, 2017

# REQUIREMENTS
# 1) Assumes you already have any version of UniFi Controller installed 
#    and running on your system.
# 2) Assumes a user named "ubnt" owns the /opt/UniFi directory.
# 3) Requires a service start/stop script to properly shut down and 
#    restart the UniFi controller before and after upgrade. I've written
#    compatible startup scrips for SysV and systemd systems at 
#    http://wp.me/p1iGgP-2wl
# 4) Requires wget command to fetch the software from UBNT's download site.

# USAGE
# Modify the "UNIFI_DOWNLOAD_URL" variable below using the full URL of
# the UniFi Controller zip file on UBNT's download site. Optionally modify
# any of the additional variables below (defaults should work fine),
# then run the script!

# CONFIGURATION OPTIONS
VERSION=$(/sbin/e-smith/db configuration getprop unifi VersionToInstall || echo "5.6.30")
UNIFI_DOWNLOAD_URL=http://dl.ubnt.com/unifi/$VERSION/UniFi.unix.zip
UNIFI_ARCHIVE_FILENAME=UniFi.unix.zip
UNIFI_OWNER=ubnt
UNIFI_SERVICE=unifi
UNIFI_PARENT_DIR=/opt
UNIFI_DIR=/opt/UniFi
UNIFI_BACKUP_DIR=/opt/UniFi_bak
TEMP_DIR=/tmp

#### SHOULDN'T HAVE TO MODIFY PAST THIS POINT ####

#today date
TODAY=$(date +%Y-%m-%d-%H%M%S)

# Create progress dots function
show_dots() {
	while ps $1 >/dev/null ; do
	printf "."
	sleep 1
	done
	printf "\n"
}

# Let's DO this!
printf "Upgrading UniFi Controller...\n"

# Retrieve the updated zip archive from UBNT (overwriting any previous version)
printf "\nDownloading %s from UBNT..." "$UNIFI_DOWNLOAD_URL"
cd $TEMP_DIR || exit
wget -qq $UNIFI_DOWNLOAD_URL -O $UNIFI_ARCHIVE_FILENAME &
show_dots $!

# Check to make sure we have a downloaded file to work with

if [ -f "$UNIFI_ARCHIVE_FILENAME" ]; then

	# Archive file exists, extract and install it

	# Stop the local UniFi Controller service
	printf "\n"
	/usr/bin/systemctl stop $UNIFI_SERVICE
	
	# Remove previous backup directory (if it exists)
#	if [ -d "$UNIFI_BACKUP_DIR" ]; then
#		printf "\nRemoving previous backup directory...\n"
#		rm -rf $UNIFI_BACKUP_DIR
#	fi
	# remove previous backup and leave the 3 last
	ls -tp $UNIFI_BACKUP_DIR/ | grep -v '/$' | tail -n +3 | xargs -d '\n' -r rm --
	
	# Move existing UniFi directory to backup location
	printf "\nMoving existing UniFi Controller directory to backup location...\n"
	mv $UNIFI_DIR "$UNIFI_BACKUP_DIR/UniFi_$TODAY"

	# Extract new version
	printf "\nExtracting downloaded software..."
	unzip -qq $TEMP_DIR/$UNIFI_ARCHIVE_FILENAME -d $UNIFI_PARENT_DIR &
	show_dots $!
	
	# Jump into the backup directory
	cd "$UNIFI_BACKUP_DIR/UniFi_$TODAY" || exit
	
	# Create an archive of the existing data directory
	printf "\nBacking up existing UniFi Controller data..."
	tar zcf $TEMP_DIR/unifi_data_bak.tar.gz data/ &
	show_dots $!
	
	# Extract the data into the new directory
	printf "\nExtracting UniFi Controller backup data to new directory..."
	tar zxf $TEMP_DIR/unifi_data_bak.tar.gz -C $UNIFI_DIR &
	show_dots $!
	
	# Enforce proper ownership of UniFi directory
	chown -R $UNIFI_OWNER:$UNIFI_OWNER $UNIFI_DIR
	
	# Restart the local UniFi Controller service
	printf "\n"
	/usr/bin/systemctl start $UNIFI_SERVICE 
	
	# All done!
	printf "\nUpgrade of UniFi Controller complete!\n"

	exit 0

else

	# Archive file doesn't exist, warn and exit
	printf "\nUniFi Controller software not found! Please check download link.\n"

	exit 1
fi