#!/bin/sh
# Copyright (c) 2025 Roy Marples
# All rights reserved

# systemd-resolved subscriber for resolvconf

# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
#     * Redistributions of source code must retain the above copyright
#       notice, this list of conditions and the following disclaimer.
#     * Redistributions in binary form must reproduce the above
#       copyright notice, this list of conditions and the following
#       disclaimer in the documentation and/or other materials provided
#       with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

[ -f "/usr/pkg/etc"/resolvconf.conf ] || exit 0
. "/usr/pkg/etc/resolvconf.conf" || exit 1

case "${systemd_resolved:-NO}" in
[Yy][Ee][Ss]|[Tt][Rr][Uu][Ee]|[Oo][Nn]|1) ;;
*) exit 0;;
esac

[ -n "$RESOLVCONF" ] || eval "$(/usr/pkg/sbin/resolvconf -v)"
NL="
"

: ${systemd_resolved_conf:=/run/systemd/resolved.conf.d/60-resolvconf.conf}
: ${systemd_delegate_dir:=/run/systemd/dns-delegate.d}

# Try to ensure that config dirs exist
if command -v config_mkdirs >/dev/null 2>&1; then
	config_mkdirs "$systemd_resolved_conf" "$systemd_delegate_dir/x"
else
	/usr/pkg/sbin/resolvconf -D "$systemd_resolved_conf" "$systemd_delegate_dir/x"
fi

header="# Generated by resolvconf$NL"
header="${header}$NL"
header="${header}[Resolve]$NL"

conf="$header"
# We emit blank values to force them to reset on SIGHUP
conf="${conf}DNS=$NAMESERVERS$NL"
# Indicate these nameservers are for all domain lookups by using ~.
conf="${conf}Domains=$SEARCH${NAMESERVERS:+ ~.}$NL"

printf %s "$conf" >"$systemd_resolved_conf"

# DNS Delegates requires https://github.com/systemd/systemd/pull/34368
rm -f "$systemd_delegate_dir/resolvconf-"*".dns-delegate"
header="# Generated by resolvconf$NL"
header="${header}$NL"
header="${header}[Delegate]$NL"
for d in $DOMAINS; do
	dn="${d%%:*}"
	ns="${d#*:}"
	dconf="${header}Domains="
	search=false
	for sd in $SEARCH; do
		if [ "$sd" = "$dn" ]; then
			search=true
			break
		fi
	done
	if ! $search; then
		dconf="${dconf}~"
	fi
	dconf="${dconf}$dn$NL"
	dconf="${dconf}DNS="
	while [ -n "$ns" ]; do
		dconf="$dconf${ns%%,*} "
		[ "$ns" = "${ns#*,}" ] && break
		ns="${ns#*,}"
	done
	dconf="$dconf$NL"
	printf %s "$dconf" >"$systemd_delegate_dir/resolvconf-$dn.dns-delegate"
done

pid=$(pidof systemd-resolved)
if [ -n "$pid" ]; then
	kill -HUP $pid
fi
