#!/bin/bash
set -euo pipefail

brewPrefix="$(brew --prefix)"
dotnetDir="${brewPrefix}/opt/dotnet@8"
dotnetPackage="dotnet@8"
dnsDir="/opt/technitium/dns"

dnsTar="${dnsDir}/DnsServerPortable.tar.gz"
dnsUrl="https://download.technitium.com/dns/DnsServerPortable.tar.gz"

sudo mkdir -p "${dnsDir}"
sudo chown -R "$(id -u):$(id -g)" "${dnsDir}"
installLog="${dnsDir}/install.log"
echo "" >${installLog}

echo ""
echo "==============================="
echo "Technitium DNS Server Installer"
echo "==============================="

echo ""
echo "Installing ASP.NET Core Runtime..."
brew bundle install --upgrade --file <(echo -e "brew \"${dotnetPackage}\"\nbrew \"libmsquic\"") >>${installLog} 2>&1

echo ""
echo "Downloading Technitium DNS Server..."

if curl -o ${dnsTar} --fail ${dnsUrl} >>${installLog} 2>&1; then
	if [ -d ${dnsDir} ]; then
		echo "Updating Technitium DNS Server..."
	else
		echo "Installing Technitium DNS Server..."
	fi

	tar -zxf ${dnsTar} -C ${dnsDir} >>${installLog} 2>&1

	sudo mkdir -p /etc/dns
	if launchctl list com.technitium.dns >/dev/null 2>&1; then
		echo "Restarting launchd service..."
		launchctl kickstart -kp system/com.technitium.dns >>${installLog} 2>&1
	else
		echo "Configuring launchd service..."
		sudo tee /Library/LaunchDaemons/com.technitium.dns.plist >/dev/null <<EOF
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
<plist version="1.0">
	<dict>
		<key>Label</key>
		<string>com.technitium.dns</string>
		<key>ProgramArguments</key>
		<array>
			<string>${dotnetDir}/bin/dotnet</string>
			<string>/opt/technitium/dns/DnsServerApp.dll</string>
			<string>/etc/dns</string>
		</array>
		<key>RunAtLoad</key>
		<true/>
		<key>StandardOutPath</key>
		<string>/Library/Logs/com.technitium.dns.out.log</string>
		<key>StandardErrorPath</key>
		<string>/Library/Logs/com.technitium.dns.err.log</string>
		<key>EnvironmentVariables</key>
		<dict>
			<key>DYLD_FALLBACK_LIBRARY_PATH</key>
			<string>${brewPrefix}/lib</string>
		</dict>
		<key>KeepAlive</key>
		<dict>
			<key>SuccessfulExit</key>
			<false/>
		</dict>
		<key>ThrottleInterval</key>
		<integer>5</integer>
	</dict>
</plist>
EOF
		sudo launchctl bootstrap system /Library/LaunchDaemons/com.technitium.dns.plist 2>&1 | tee -a ${installLog}
		sudo launchctl enable system/com.technitium.dns 2>&1 | tee -a ${installLog}
	fi

	echo ""
	echo "Technitium DNS Server was installed successfully!"
	echo "Open http://$(hostname):5380/ to access the web console."
	echo ""
	echo "Donate! Make a contribution by becoming a Patron: https://www.patreon.com/technitium"
	echo ""
else
	echo ""
	echo "Failed to download Technitium DNS Server from: ${dnsUrl}"
	exit 1
fi