#!/bin/bash
#
# Copyright (c) Authors: https://www.armbian.com/authors
#
# This file is licensed under the terms of the GNU General Public
# License version 2. This program is licensed "as is" without any
# warranty of any kind, whether express or implied.

. /etc/armbian-release
. /etc/os-release
. /lib/init/vars.sh
. /lib/lsb/init-functions
. /usr/lib/armbian/armbian-common

case "$1" in
	start)

	# Run a q&d benchmark to be able to identify cards way too slow easily
	if command -v "iozone" > /dev/null 2>&1; then
	echo -e "\n### quick iozone test:$(cd /root; iozone -e -I -a -s 1M -r 4k -i 0 -i 1 -i 2 | grep '^            1024' | sed 's/            1024      //')" >> $Log
	fi

	# Bluetooth tweaks
	case ${BOARD} in
		cubietruck)
			systemctl enable brcm40183-patch && service brcm40183-patch start
			;;
		bananapim2zero|bananapim2plus|nanopiair|nanopim1plus|orangepizeroplus2*|orangepiwin|nanopineoplus2|nanopik2)
			update-rc.d ap6212-bluetooth defaults
			;;
	esac

	# Debian Buster desktop bugfix https://bugs.debian.org/cgi-bin/bugreport.cgi?bug=923289
	[[ -d /etc/lightdm ]] && dpkg-reconfigure libgdk-pixbuf2.0-0 && update-mime-database /usr/share/mime

	# update console info
	setupcon --save

	# add unique installation id to the image release file
	sed '1s/^/IMAGE_UUID=/' /proc/sys/kernel/random/uuid >> /etc/armbian-image-release

	# disable passing locale environment via ssh
	sed -e '/AcceptEnv LANG/ s/^#*/#/' -i /etc/ssh/sshd_config

	# Enable Armbian Live patch if installed
	[[ -f /etc/systemd/system/armbian-live-patch.service ]] && systemctl --no-reload enable armbian-live-patch.service

	# SSH Keys creation
	if [[ "${OPENSSHD_REGENERATE_HOST_KEYS}" = true ]]; then
		rm -f /etc/ssh/ssh_host*
		read entropy_before </proc/sys/kernel/random/entropy_avail
		dpkg-reconfigure openssh-server >/dev/null 2>&1
		service ssh restart
		read entropy_after </proc/sys/kernel/random/entropy_avail
		echo -e "\n### [firstrun] Recreated SSH keys (entropy: ${entropy_before} ${entropy_after})" >>${Log}
	else
		echo -e "\n### [firstrun] SSH host keys unchanged" >>${Log}
	fi

	# get rid of purple background color in newt apps whiptail, partimage, debconf ... Reverting to Debian look.
	[[ -f /etc/newt/palette ]] && sed -e 's/magenta/blue/g' -i /etc/newt/palette

	# randomize mac in armbianEnv.txt
	if [[ -f /boot/armbianEnv.txt ]]; then
		generate_random_mac
		sed  -i "s/^ethaddr=.*/ethaddr=$MACADDR/" /boot/armbianEnv.txt
		generate_random_mac
		sed  -i "s/^eth1addr=.*/eth1addr=$MACADDR/" /boot/armbianEnv.txt
	fi

	# hardware workarounds per family
	case $LINUXFAMILY in
		sun7i|sun8i)
			# set some mac address for BT
			[[ -n $(lsmod | grep -E "dhd|brcmfmac" ) ]] && \
				(MACADDR=$(printf '43:29:B1:%02X:%02X:%02X\n' $[RANDOM%256] $[RANDOM%256] $[RANDOM%256]) ; \
				sed -i "s/^MAC_ADDR=.*/MAC_ADDR=${MACADDR}/" /etc/default/brcm40183 ;\
				sed -i "s/^MAC_ADDR=.*/MAC_ADDR=${MACADDR}/" /etc/default/ap6212 \
				echo -e "\n### [firstrun] Use MAC address ${MACADDR} for Bluetooth from now" >>${Log})

			case $BOARD_NAME in
				"NanoPi Air"|"NanoPi M1 Plus"|"Nanopi Neo Plus 2")
					# change serial port for BT on FriendlyELEC H3 boards
					sed -i "s/^PORT=.*/PORT=ttyS3/" /etc/default/ap6212
					;;
				"Orange Pi Zero"|"NanoPi Duo"|"Sunvell R69")
					generate_random_mac
					echo "options xradio_wlan macaddr=${MACADDR}" >/etc/modprobe.d/xradio_wlan.conf
					echo -e "\n### [firstrun] Use MAC address ${MACADDR} for Wi-Fi from now" >>${Log}
					cd /etc/network/ && ln -sf interfaces.network-manager interfaces
					(modprobe -r xradio_wlan && sleep 1 && modprobe xradio_wlan) &
					;;
			esac

			# trigger red or blue LED as user feedback
			echo heartbeat >/sys/class/leds/*red*/trigger 2>/dev/null || echo heartbeat >/sys/class/leds/*blue*/trigger 2>/dev/null
			;;

		x86|arm64)
			[[ -n "$(dmesg | grep "No NVIDIA GPU found")" ]] && sudo apt-get -y -qq purge nvidia-dkms-510 nvidia-driver-510 nvidia-settings nvidia-common >> /dev/null
		;;
		*)

	esac

	# varios temporary hardware workarounds
	[[ "${LINUXFAMILY}" == mvebu ]] && set_fixed_mac
	[[ "${LINUXFAMILY}" == meson ]] && set_fixed_mac
	[[ "${LINUXFAMILY}" == meson64 ]] && set_fixed_mac
	[[ "${BOARD}" == nanopi-r6* ]] && set_fixed_mac

	# adjust hostname for rpi's
	if [[ "${BOARD}" == rpi4b && "$(hostname)" == rpi4b ]]; then
		BOARD_NAME=$(cat /proc/device-tree/model | tr '\0' '\n' | sed -E 's/ Rev [0-9.]+$//')
		[[ "$BOARD_NAME" == *"Zero 2 W"* ]] && NEW_HOSTNAME="rpizero2w"
		[[ "$BOARD_NAME" == *"Pi 5 Model B"* ]] && NEW_HOSTNAME="rpi5b"
		[[ "$BOARD_NAME" == *"Pi 3 Model B"* ]] && NEW_HOSTNAME="rpi3b"
		[[ "$BOARD_NAME" == *"Pi 400"* ]] && NEW_HOSTNAME="rpi400"
		# Update /etc/hosts to reflect the new hostname
		sed -i "s/$(hostname)/$NEW_HOSTNAME/g" /etc/hosts
		# Change the hostname
		hostnamectl set-hostname "$NEW_HOSTNAME"
	fi

	# Remove KDE Neon base files upgrade pin
	[[ -f /etc/apt/preferences.d/99-neon-base-files ]] && rm -f /etc/apt/preferences.d/99-neon-base-files

	systemctl disable armbian-firstrun
	exit 0
	;;

	*)
	echo "Usage: $0 start"
	exit 0
	;;
esac