#!/bin/bash

################################################ SETUP ################################################
deps=(libasound2-dev libevdev-dev liblua5.3-dev libjack-jackd2-dev pkg-config libssl-dev gcc make wget git)
user=$(whoami)                  # for bypassing user check replace "$(whoami)" with "root".

tmp_path=$(mktemp -d)           # Repo download path
updater_dir=/etc/midimonster-updater       # Updater download + config path
updater_file=$updater_dir/updater.conf

latest_version=$(curl --silent "https://api.github.com/repos/cbdevnet/midimonster/releases/latest" | grep '"tag_name":' | sed -E 's/.*"([^"]+)".*/\1/')

makeargs=all                    # Build args

normal=$(tput sgr0)
dim=$(tput dim)
bold=$(tput bold)
uline=$(tput smul)
c_red=$(tput setaf 1)
c_green=$(tput setaf 2)
c_mag=$(tput setaf 5)

VAR_DESTDIR=""                  # Unused
VAR_PREFIX="/usr"
VAR_PLUGINS="$VAR_PREFIX/lib/midimonster"		# Reassigned in func. ARGS to update "$VAR_PREFIX" when an argument is set.
VAR_DEFAULT_CFG="/etc/midimonster/midimonster.cfg"
VAR_EXAMPLE_CFGS="$VAR_PREFIX/share/midimonster"	# Reassigned in func. ARGS to update "$VAR_PREFIX" when an argument is set.


################################################ SETUP ################################################

############################################## FUNCTIONS ##############################################

ARGS () {
	for i in "$@"
	do
		case $i in
			--prefix=*)
				VAR_PREFIX="${i#*=}"
				VAR_PREFIX_I="1"
				VAR_PLUGINS="$VAR_PREFIX/lib/midimonster"
				VAR_EXAMPLE_CFGS="$VAR_PREFIX/share/midimonster"
				VAR_EXAMPLE_CFGS="$VAR_PREFIX/share/midimonster"
			;;
			--plugins=*)
				VAR_PLUGINS="${i#*=}"
				VAR_PLUGINS_I="1"
			;;
			--defcfg=*)
				VAR_DEFAULT_CFG="${i#*=}"
				VAR_DEFAULT_CFG_I="1"
			;;
			--examples=*)
				VAR_EXAMPLE_CFGS="${i#*=}"
				VAR_EXAMPLE_CFGS_I="1"
			;;
			--dev)
				NIGHTLY=1
				NIGHTLY_I="1"
			;;
			-d|--default)
				VAR_PREFIX_I="1"
				VAR_PLUGINS_I="1"
				VAR_DEFAULT_CFG_I="1"
				VAR_EXAMPLE_CFGS_I="1"
				NIGHTLY_I="1"
				NIGHTLY=1
			;;
			-fu|--forceupdate)
				UPDATER_FORCE="1"
			;;    
			-h|--help|*)		# Help messages
				printf "${bold}Usage:${normal} ${0} ${c_green}[OPTIONS]${normal}"
				printf "\n	${c_green}--prefix${normal} ${c_red}<path>${normal}		Sets the prefix		${c_mag}Default:${normal} ${dim}%s  ${normal}" "$VAR_PREFIX"
				printf "\n	${c_green}--plugins${normal} ${c_red}<path>${normal}	Sets the plugin path	${c_mag}Default:${normal} ${dim}%s  ${normal}" "$VAR_PLUGINS"
				printf "\n	${c_green}--defcfg${normal} ${c_red}<path>${normal}		Sets the config path	${c_mag}Default:${normal} ${dim}%s  ${normal}" "$VAR_DEFAULT_CFG"
				printf "\n	${c_green}--examples${normal} ${c_red}<path>${normal}	Sets the example configs path	${c_mag}Default:${normal} ${dim}%s  ${normal}\n" "$VAR_EXAMPLE_CFGS"
				printf "\n	${c_green}--dev${normal}		Install nightly version"
				printf "\n   ${c_green}-d,  --default${normal}	Use default values to install"
				printf "\n   ${c_green}-fu, --forceupdate${normal}	Force the  updater to update without a version check"
				printf "\n   ${c_green}-h,  --help${normal}		Show this message"
				printf "\n	${uline}${bold}${c_mag}Each argument can be overwritten by another, the last one is used!${normal}\n"
				exit 1
			;;
		esac
		shift
	done
}

INSTALL-DEPS () {           # Install deps from array "$deps"
	for t in ${deps[@]}; do
	    if [ $(dpkg-query -W -f='${Status}' $t 2>/dev/null | grep -c "ok installed") -eq 0 ]; then
	        printf "Installing %s\n" "$t"
	        apt-get install $t;
	        printf "Done.\n";
	    else
	        printf "%s already installed!\n" "$t"
	    fi
	done
	printf "\n"
}

NIGHTLY_CHECK () {			# Asks for nightly version
	if [ -z "$NIGHTLY_I" ]; then
		read -p "Do you want to install the latest development version? (y/n)? " magic
		case "$magic" in
			y|Y)
				printf "OK! You´re a risky person ;D\n"
				NIGHTLY=1
				;;
			n|N)
				printf "That´s OK - installing the latest stable version for you ;-)\n"
				NIGHTLY=0
				;;
			*)
				printf "${bold}Invalid input -- INSTALLING LATEST STABLE VERSION!${normal}\n"
				NIGHTLY=0
				;;
		esac
	fi

	# Roll back to last tag if we're not on a nightly build
	if [ "$NIGHTLY" != 1 ]; then
		cd "$tmp_path"
		printf "Finding latest stable version...\n"
		Iversion=$(git describe --abbrev=0)
		printf "Starting Git checkout to %s...\n" "$Iversion"
		git checkout -f -q $Iversion
	fi
}

INSTALL-PREP () {
	(
		printf "Starting download...\n"
		git clone https://github.com/cbdevnet/midimonster.git "$tmp_path" # Gets Midimonster
		printf "\nInitializing repository...\n"
		cd $tmp_path
		git init $tmp_path
		printf "\n"
	)
	NIGHTLY_CHECK
	printf "Preparation done.\n\n"
	printf "${bold}If you don't know what you're doing, just hit enter a few times.${normal}\n\n"
	if [ -z "$VAR_PREFIX_I" ]; then
		read -e -i "$VAR_PREFIX" -p "PREFIX (Install root directory): " input # Reads VAR_PREFIX then update containing vars
		VAR_PREFIX="${input:-$VAR_PREFIX}"
		VAR_PLUGINS="$VAR_PREFIX/lib/midimonster"	# Update prefix-containing variables.
		VAR_EXAMPLE_CFGS="$VAR_PREFIX/share/midimonster"	# Update prefix-containing variables.
	fi

	if [ -z "$VAR_PLUGINS_I" ]; then
		read -e -i "$VAR_PLUGINS" -p "PLUGINS (Plugin directory): " input # Reads VAR_PLUGINS
		VAR_PLUGINS="${input:-$VAR_PLUGINS}"
	fi

	if [ -z "$VAR_DEFAULT_CFG_I" ]; then
		read -e -i "$VAR_DEFAULT_CFG" -p "Default config path: " input # Reads VAR_DEFAULT_CFG
		VAR_DEFAULT_CFG="${input:-$VAR_DEFAULT_CFG}"
	fi

	if [ -z "$VAR_EXAMPLE_CFGS_I" ]; then
		read -e -i "$VAR_EXAMPLE_CFGS" -p "Example config directory: " input # Reads VAR_EXAMPLE_CFGS
		VAR_EXAMPLE_CFGS="${input:-$VAR_EXAMPLE_CFGS}"
	fi

	UPDATER_SAVE

	export PREFIX=$VAR_PREFIX
	export PLUGINS=$VAR_PLUGINS
	export DEFAULT_CFG=$VAR_DEFAULT_CFG
	export DESTDIR=$VAR_DESTDIR
	export EXAMPLES=$VAR_EXAMPLE_CFGS
}

UPDATER-PREP () {
	(
		printf "Starting download...\n"
		git clone https://github.com/cbdevnet/midimonster.git "$tmp_path" # Gets Midimonster
		printf "\nInitializing repository...\n"
		cd $tmp_path
		git init $tmp_path
	)
	NIGHTLY_CHECK
	printf "Preparation done.\n\n"
	rm -rf "$VAR_PLUGINS/"

	UPDATER_SAVE

	export PREFIX=$VAR_PREFIX
	export PLUGINS=$VAR_PLUGINS
	export DEFAULT_CFG=$VAR_DEFAULT_CFG
	export DESTDIR=$VAR_DESTDIR
	export EXAMPLES=$VAR_EXAMPLE_CFGS
}

UPDATER () {
	installed_version="$(midimonster --version)"
	#installed_version="MIDIMonster v0.3-40-gafed325" # FOR TESTING ONLY!
	if [[ "$installed_version" =~ "$latest_version" ]]; then
		printf "Newest Version is already installed! ${bold}($installed_version)${normal}\n\n"
		ERROR
	else
		printf "The installed Version ${bold}´$installed_version´${normal} equals not the newest stable version ${bold}´$latest_version´${normal} (Maybe you are running a development(NIGHTLY) version?)\n\n"
	fi

	UPDATER-PREP
	INSTALL-RUN
	DONE
}

INSTALL-RUN () {                                    # Build
	cd "$tmp_path"
	make clean
	make $makeargs
	make install
}

UPDATER_SAVE () {                                   # Saves file for the auto updater in this script
	rm -rf $updater_dir
	printf "Saving updater to %s/updater.sh\n" "$update_dir"
	mkdir -p "$updater_dir"
	wget https://raw.githubusercontent.com/cbdevnet/midimonster/master/installer.sh -O $updater_dir/updater.sh
	printf "Creating symlink to updater in /usr/bin/midimonster-updater\n"
	ln -s "$updater_dir/updater.sh" "/usr/bin/midimonster-updater"
	chmod +x "$updater_dir/updater.sh"
	printf "Exporting updater config to %s\n" "$updater_file"
	printf "VAR_PREFIX=%s\nVAR_PLUGINS=%s\nVAR_DEFAULT_CFG=%s\nVAR_DESTDIR=%s\nVAR_EXAMPLE_CFGS=%s\n" "$VAR_PREFIX" "$VAR_PLUGINS" "$VAR_DEFAULT_CFG" "$VAR_DESTDIR" "$VAR_EXAMPLE_CFGS" > $updater_file
}

ERROR () {
	printf "\nAborting...\n"
	CLEAN
	printf "Exiting...\n"
	exit 1
}

DONE () {
	printf "\nDone.\n"
	CLEAN
	exit 0
}

CLEAN () {
	printf "\nCleaning...\n"
	rm -rf $tmp_path
}

############################################## FUNCTIONS ##############################################


################################################ Main #################################################
trap ERROR SIGINT SIGTERM SIGKILL
ARGS "$@"	# Parse arguments
clear

# Check if $user = root!
if [ "$user" != "root" ]; then
	printf "Installer must be run as root\n"
	ERROR
fi

if [ $(wget -q --spider http://github.com) $? -eq 1 ]; then
	printf "You need a connection to the internet\n"
	ERROR
fi

# Forceupdate		# Now only with default config because source imports all and overwrites the args.. [WIP!]
if [ "$UPDATER_FORCE" = "1" ]; then
	printf "Forcing the updater to start...\n\n"
	if [ -f $updater_file ]; then
		. $updater_file
		ARGS "$@"	# Parse arguments again to compensate overwrite from source /\
		printf "Successfully imported settings from %s\n" "$updater_file"
	fi

	UPDATER-PREP
	INSTALL-RUN
	DONE
fi

# Check if updater config file exist and import it (overwrites default values!)
if [ -f $updater_file ]; then
	printf "Starting updater...\n\n"
	. $updater_file
	ARGS "$@"	# Parse arguments again to compensate overwrite from source /\
	printf "Successfully imported settings from %s\n" "$updater_file"

		# Check if binary $VAR_PREFIX/bin/midimonster exist. If yes start updater else skip.
	if [ -x "$VAR_PREFIX/bin/midimonster" ]; then
		UPDATER
	else
		printf "MIDIMonster binary not found, skipping updater.\nYou can force an update with --forceupdate\n"
	fi
fi

#Normal install
INSTALL-DEPS
INSTALL-PREP
INSTALL-RUN
DONE