install.sh

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

# Try re-running with sudo
[ "$UID" -eq 0 ] || exec sudo bash "$0" "$@"

# Required to do when unit files are changed or removed
function reloadSystemd() {
	echo "Reloading systemd manager configuration..." 
	systemctl daemon-reload
}

SPATH=$(dirname "$0") # Path to the script
REMOTE_URL=https://raw.githubusercontent.com/jacklul/pihole-updatelists # Remote URL that serves raw files from the repository
GIT_BRANCH=master # Git branch to use, user can specify custom branch as first argument
SYSTEMD=$(pidof systemd >/dev/null && echo "1" || echo "0") # Is systemd available?
SYSTEMD_INSTALLED=$([ -f "/etc/systemd/system/pihole-updatelists.timer" ] && echo "1" || echo "0") # Is systemd timer installed already?
DOCKER=$([ "$(grep "docker" < /proc/1/cgroup)" == "" ] && echo "0" || echo "1") # Is this a Docker installation?

if [ "$1" == "uninstall" ]; then	# Simply remove the files and reload systemd (if available)
	rm -v /usr/local/sbin/pihole-updatelists
	rm -v /etc/bash_completion.d/pihole-updatelists
	rm -vf /etc/cron.d/pihole-updatelists
	rm -vf /etc/systemd/system/pihole-updatelists.service
	rm -vf /etc/systemd/system/pihole-updatelists.timer
	
	if [ -f "/var/tmp/pihole-updatelists.old" ]; then
		rm -v /var/tmp/pihole-updatelists.old
	fi

	if [ "$SYSTEMD" == 1 ]; then
		reloadSystemd
	fi

	exit 0
elif [ "$1" == "docker" ]; then	# Force Docker install
	DOCKER=1
elif [ "$1" == "systemd" ]; then # Force systemd unit files installation
	SYSTEMD=1
elif [ "$1" == "crontab" ]; then # Force crontab installation
	SYSTEMD=0
elif [ "$1" != "" ]; then	# Install using different branch
	GIT_BRANCH=$1

	if ! wget -q --spider "$REMOTE_URL/$GIT_BRANCH/install.sh" ; then
		echo "Invalid branch: ${GIT_BRANCH}"
		exit 1
	fi
fi

# Do not install systemd unit files inside Docker container
if [ "$DOCKER" == 1 ]; then
	SYSTEMD=0
fi

# We require some stuff before continuing
command -v php >/dev/null 2>&1 || { echo "This script requires PHP-CLI to run, install it with 'sudo apt install php-cli'."; exit 1; }
[[ $(php -v | head -n 1 | cut -d " " -f 2 | cut -f1 -d".") -lt 7 ]] && { echo "Detected PHP version lower than 7.0, make sure php-cli package is up to date!"; exit 1; }

# Use local files when possible, otherwise install from remote repository
if \
	[ "$GIT_BRANCH" == "master" ] && \
	[ -f "$SPATH/pihole-updatelists.php" ] && \
	[ -f "$SPATH/pihole-updatelists.conf" ] && \
	[ -f "$SPATH/pihole-updatelists.service" ] && \
	[ -f "$SPATH/pihole-updatelists.timer" ] && \
	[ -f "$SPATH/pihole-updatelists.bash" ] \
; then
	if [ ! -d "/usr/local/sbin" ]; then
		mkdir -vp /usr/local/sbin && chmod -v 0755 /usr/local/sbin
	fi

	if [ -f "/usr/local/sbin/pihole-updatelists" ]; then
		if ! cmp -s "$SPATH/pihole-updatelists.php" "/usr/local/sbin/pihole-updatelists"; then
			echo "Backing up previous version..."
			cp -v /usr/local/sbin/pihole-updatelists /var/tmp/pihole-updatelists.old && \
			chmod -v -x /var/tmp/pihole-updatelists.old
		fi
	fi

	cp -v "$SPATH/pihole-updatelists.php" /usr/local/sbin/pihole-updatelists && \
	chmod -v +x /usr/local/sbin/pihole-updatelists
	
	if [ ! -f "/etc/pihole-updatelists.conf" ]; then
		cp -v "$SPATH/pihole-updatelists.conf" /etc/pihole-updatelists.conf
	fi

	if [ "$SYSTEMD" == 1 ]; then
		cp -v "$SPATH/pihole-updatelists.service" /etc/systemd/system
		cp -v "$SPATH/pihole-updatelists.timer" /etc/systemd/system
	fi
	
	if [ ! -d "/etc/bash_completion.d" ]; then
		mkdir -vp /etc/bash_completion.d
	fi

	cp -v "$SPATH/pihole-updatelists.bash" /etc/bash_completion.d/pihole-updatelists

	# Convert line endings when dos2unix command is available
	command -v dos2unix >/dev/null 2>&1 && dos2unix /usr/local/sbin/pihole-updatelists /etc/bash_completion.d/pihole-updatelists
elif [ "$REMOTE_URL" != "" ] && [ "$GIT_BRANCH" != "" ]; then
	if [ ! -d "/usr/local/sbin" ]; then
		mkdir -vp /usr/local/sbin && chmod -v 0755 /usr/local/sbin
	fi
	
	if [ -f "/usr/local/sbin/pihole-updatelists" ]; then
		wget -nv -O /tmp/pihole-updatelists.php "$REMOTE_URL/$GIT_BRANCH/pihole-updatelists.php"

		if ! cmp -s "/tmp/pihole-updatelists.php" "/usr/local/sbin/pihole-updatelists"; then
			echo "Backing up previous version..."
			cp -v /usr/local/sbin/pihole-updatelists /var/tmp/pihole-updatelists.old && \
			chmod -v -x /var/tmp/pihole-updatelists.old
		fi

		mv -v /tmp/pihole-updatelists.php /usr/local/sbin/pihole-updatelists && \
		chmod -v +x /usr/local/sbin/pihole-updatelists
	else
		wget -nv -O /usr/local/sbin/pihole-updatelists "$REMOTE_URL/$GIT_BRANCH/pihole-updatelists.php" && \
		chmod -v +x /usr/local/sbin/pihole-updatelists
	fi

	if [ ! -f "/etc/pihole-updatelists.conf" ]; then
		wget -nv -O /etc/pihole-updatelists.conf "$REMOTE_URL/$GIT_BRANCH/pihole-updatelists.conf"
	fi

	if [ "$SYSTEMD" == 1 ]; then
		wget -nv -O /etc/systemd/system/pihole-updatelists.service "$REMOTE_URL/$GIT_BRANCH/pihole-updatelists.service"
		wget -nv -O /etc/systemd/system/pihole-updatelists.timer "$REMOTE_URL/$GIT_BRANCH/pihole-updatelists.timer"
	fi

	if [ ! -d "/etc/bash_completion.d" ]; then
		mkdir -vp /etc/bash_completion.d
	fi

	wget -nv -O /etc/bash_completion.d/pihole-updatelists "$REMOTE_URL/$GIT_BRANCH/pihole-updatelists.bash"
else
	echo "Missing required files for installation!"
	exit 1
fi

if [ -f "/usr/local/sbin/pihole-updatelists.old" ]; then
	rm -v /usr/local/sbin/pihole-updatelists.old
fi

# Install schedule related files
if [ "$SYSTEMD" == 1 ]; then
	if [ -f "/etc/cron.d/pihole-updatelists" ]; then
		# Comment out the existing cron job
		sed "s/^#*/#/" -i /etc/cron.d/pihole-updatelists
	fi

	if [ "$SYSTEMD_INSTALLED" == 0 ]; then
		echo "Enabling and starting pihole-updatelists.timer..."
		systemctl enable pihole-updatelists.timer && systemctl start pihole-updatelists.timer
	else
		reloadSystemd
	fi
else
	if [ ! -f "/etc/cron.d/pihole-updatelists" ]; then
		echo "# Pi-hole's Lists Updater by Jack'lul
# https://github.com/jacklul/pihole-updatelists

#30 3 * * 6   root   /usr/local/sbin/pihole-updatelists
" > /etc/cron.d/pihole-updatelists
		sed "s/#30 /$((1 + RANDOM % 58)) /" -i /etc/cron.d/pihole-updatelists

		echo "Created crontab (/etc/cron.d/pihole-updatelists)"
	fi
fi

# Docker-related tasks
if [ "$DOCKER" == 1 ]; then
	[ ! -d "/etc/s6-overlay/s6-rc.d/_postFTL" ] && { echo "Missing /etc/s6-overlay/s6-rc.d/_postFTL directory!"; exit 1; }
	[ ! -f "/usr/local/bin/_postFTL.sh" ] && { echo "Missing /usr/local/bin/_postFTL.sh file!"; exit 1; }
	
	mkdir -v /etc/pihole-updatelists
	
	if [ -f "$SPATH/docker.sh" ]; then
		cp -v "$SPATH/docker.sh" /usr/local/bin/_updatelists.sh
	elif [ "$REMOTE_URL" != "" ]; then
		wget -nv -O /usr/local/bin/_updatelists.sh "$REMOTE_URL/$GIT_BRANCH/docker.sh"
	else
		echo "Missing required file (docker.sh) for installation!"
	fi

	chmod -v +x /usr/local/bin/_updatelists.sh

	echo "#!/command/execlineb" > /etc/s6-overlay/s6-rc.d/_postFTL/up
	echo "background { bash -ec \"/usr/local/bin/_updatelists.sh && /usr/local/bin/_postFTL.sh\" }" >> /etc/s6-overlay/s6-rc.d/_postFTL/up
	echo "Modified /etc/s6-overlay/s6-rc.d/_postFTL/up to launch pihole-updatelists first!"

	echo "alias pihole-updatelists='/usr/local/sbin/pihole-updatelists --config=/etc/pihole-updatelists/pihole-updatelists.conf --env'" >> /root/.bashrc
	echo "Created alias for pihole-updatelists command in /root/.bashrc"
fi