#!/bin/bash

# Not to fail on old key we will not update package list from Z-Wave repo - this will be done later after key installation
rm -f /etc/apt/sources.list.d/z-wave-me.list

# Install additional packages that help to add Z-Wave.Me repository
apt-get update -y --allow-releaseinfo-change
apt-get install dirmngr apt-transport-https gnupg wget -y

# Check for distro
apt-get install lsb_release >/dev/null 2>&1 # try the built-in tool first (if present)

distro=$(lsb_release -a 2>/dev/null | grep Codename | awk '{print $2}')
if [ "${distro}" == "" -o "${distro}" == "n/a" ]
then
	distro=$(cat /etc/os-release | grep VERSION_CODENAME | awk -F"=" '{print $2}')
fi

distro_id=$(lsb_release -a 2>/dev/null | grep "Distributor ID" | awk '{print $3}' | tr '[:upper:]' '[:lower:]')
if [ "${distro_id}" == "" ]
then
	distro_id=$(cat /etc/os-release | grep -E "^ID=" | awk -F"=" '{print $2}')
fi

# Cancel installation if distribution is not supported
if [ "${distro}" != "buster" -a "${distro}" != "bullseye" -a "${distro}" != "bookworm" -a "${distro}" != "focal" -a "${distro}" != "jammy" ]
then
	echo
	echo "Your Linux distribution is currently not supported by Z-Way"
	echo "Please contact Z-Wave.Me support for more information"
	echo "Supported Raspbian or Debian distributions are Buster, Bullseye and Bookworm"
	echo "Supported Ubuntu distribution are 20.04 and upper"
	echo
	exit 1
fi

# Check architecture	
architecture=$(uname -m)

case "${architecture}" in
	armhf | armv6l | armv7l)
		arch_tag=""
		arch_apt=""
		additional_packets="z-way-full${arch_apt} webif${arch_apt}"
		install_webif="ok"
		# fix distro_id for Raspberry OS
		if [ "${distro_id}" == "debian" -o "${distro_id}" == "ubuntu" ]
		then
			distro_id="raspbian"
		fi
		if [ "$distro" == "bookworm" ] 
		then
		    distro="bullseye"
		    # Add old debian repository
		    echo "deb https://deb.debian.org/debian bullseye main" | sudo tee /etc/apt/sources.list.d/bullseye.list
		
		    # Create /etc/apt/preferences.d/bullseye-pin with low priority for all packages except libwebsockets16
		    echo "Package: *" | sudo tee /etc/apt/preferences.d/bullseye-pin
		    echo "Pin: release a=bullseye" | sudo tee -a /etc/apt/preferences.d/bullseye-pin
		    echo "Pin-Priority: -1" | sudo tee -a /etc/apt/preferences.d/bullseye-pin
		
		    echo "Package: libwebsockets16" | sudo tee -a /etc/apt/preferences.d/bullseye-pin
		    echo "Pin: release a=bullseye" | sudo tee -a /etc/apt/preferences.d/bullseye-pin
		    echo "Pin-Priority: 990" | sudo tee -a /etc/apt/preferences.d/bullseye-pin
		
		    sudo apt update

		    f=/boot/config.txt; u="enable_uart=1"; grep -q "^$u$" $f || (sudo sed -i "/^enable_uart/d" $f; echo $u | sudo tee -a $f; echo -e "\e[31mUART is enabled, reboot your RPI...\e[0m")
		fi
		;;

	aarch64)
		arch_tag="[arch=armhf]"
		arch_apt=":armhf"
		install_webif="ok"
		additional_packets="z-way-full${arch_apt} webif${arch_apt}"
		# fix distro_id for Raspberry OS
		if [ "${distro_id}" == "debian" -o "${distro_id}" == "ubuntu" ]
		then
			distro_id="raspbian"
		fi
		#temporary support for the debian bookworm distribution
		if [ "$distro" == "bookworm" ] 
		then
		    distro="bullseye"
		    # Add old debian repository
		    echo "deb https://deb.debian.org/debian bullseye main" | sudo tee /etc/apt/sources.list.d/bullseye.list
		
		    cat <<EOF | sudo tee /etc/apt/preferences.d/bullseye-pin
Package: *
Pin: release a=bullseye
Pin-Priority: 100
EOF
		
		    sudo apt update
		fi
		;;

	x86_64)
		arch_tag="[arch=amd64]"
		arch_apt=""
		install_webif=""
		additional_packets=""
		
		if [ "${distro}" == "bullseye" -o "${distro}" == "bookworm" ]
		then
			distro="buster"
		fi

		#temporary support for the ubuntu jammy distribution
		if [ "$distro" == "jammy" ] 
		then
		    distro="focal"
		    # Add old debian repository
		    echo "deb http://archive.ubuntu.com/ubuntu/ focal main restricted" | sudo tee /etc/apt/sources.list.d/focal.list
		
		    cat <<EOF | sudo tee /etc/apt/preferences.d/focal-pin
Package: *
Pin: release a=focal
Pin-Priority: -1

Package: libssl1.1
Pin: release a=focal
Pin-Priority: 990
EOF
		    sudo apt update
		fi
		;;
		*)
		echo
		echo "Your operating system architecture is not supported or unknown"
		echo "Please contact Z-Wave.Me support for more information"
		echo "Supported architectures: armhf, aarch64, x86_64"
		echo "Your architecture: ${architecture} and your distro is ${distro_id} ${distro}"
		echo
		exit 1
		;;
esac
minimal_packets="z-way-server${arch_apt} zbw${arch_apt}"

# Add Z-Wave.Me repository
apt-key adv --keyserver hkp://keyserver.ubuntu.com:80 --recv-keys 5b2f88a91611e683
echo "deb ${arch_tag} https://repo.z-wave.me/z-way/${distro_id} ${distro} main" >/etc/apt/sources.list.d/z-wave-me.list

# Update packages list
apt-get update -y

# Install Z-Way
if [ -z "$ZWAY_UPIF" ]
then # don't update webif if running from webif
	apt-get install --reinstall -y ${minimal_packets} ${additional_packets}
else
	apt-get install --reinstall -y ${minimal_packets}
fi

# Override Z-Way/zbw/webif configs from the package (keep private files)
CONF_FILES=$(dpkg-query -W -f='${Conffiles}\n' z-way-server webif zbw | awk 'OFS="  "{print $2,$1}' | LANG=C md5sum --quiet -c 2>/dev/null | cut -f 1 -d :)
if [ -n "$CONF_FILES" ]
then
	echo "Package configuration files were changes:" $CONF_FILES
	
	RESTORE_CONFIG_FILE="yes"
	
	if [ -z "$ZWAY_UPIF" ]
	then
		read -p "Restore config files to the package default state? (type 'yes' to restore): " RESTORE_CONFIG_FILE < /dev/tty
	fi
	
	if [ "$RESTORE_CONFIG_FILE" == "yes" ]
	then
		echo "Replacing them by package configuration files"
		TMP_DIR=$(mktemp -d)
		chown _apt:root $TMP_DIR
		pushd $TMP_DIR
		apt-get download z-way-server${arch_apt}
		
		dpkg-deb --fsys-tarfile ./z-way-server*.deb | sudo tar -C / -x ./opt/z-way-server/config/Defaults.xml ./opt/z-way-server/config.xml ./etc/z-way/box_type ./etc/logrotate.d/z-way-server ./etc/init.d/z-way-server
		if [ "${install_webif}" != "" ]
		then
			apt-get download webif${arch_apt}
			dpkg-deb --fsys-tarfile ./webif*.deb | sudo tar -C / -x ./etc/webif.conf ./etc/mongoose/mongoose.conf
		fi
		
		#  no config files in zbw package
		# apt-get download zbw
		# dpkg-deb --fsys-tarfile ./zbw*.deb | sudo tar -C / -x ...
		
		popd
		rm -R $TMP_DIR
	else
		echo "Keeping your changed configuration files"
	fi
fi