#!/bin/bash

die() { printf '%s\n' "$*" >&2; exit 1; }

sql_string()
{
	printf '%s' "'${1//\'/\'\'}'"
}

powerdns_sqlite_add_replace_record()
{
	local fqdn="$1.$3"
	fqdn=${fqdn: -64 : 64}
	local sql_new_domain="$(sql_string "$fqdn")"
	local sql_ip_address="$(sql_string "$2")"
	local sql_domain="$(sql_string "$3")"
	local record_type="$(sql_string "$4")"

	DBDIR=/etc/powerdns
	DBNAME=powerdns.sqlite3
	DB=$DBDIR/$DBNAME

	test -r $DB    && test -w $DB    || die "Wrong permissions on $DB"
	test -r $DBDIR && test -w $DBDIR || die "Wrong permissions on $DBDIR"

	sqlite3 $DB <<END
${SQL_ECHO:+.echo on}
BEGIN;
	DELETE FROM records WHERE type=$record_type AND name=$sql_new_domain;

	INSERT INTO records

		(domain_id,
		name,
		type,
		content,
		ttl,
		prio)

		SELECT
			id,
			$sql_new_domain,
			$record_type,
			$sql_ip_address,
			3600,
			0
		FROM domains
		WHERE name=$sql_domain;
COMMIT;
END
	r=$?
	[ $r = 0 ] || return $r

	printf '%s %s\n' "$fqdn" "$2"
}

add()
{
	local record_type
	case "$2" in
		*.*.*.*) record_type=A ;;
		*:*) record_type=AAAA ;;
		*) exit 1 ;;
	esac

	powerdns_sqlite_add_replace_record "$1" "$2" "$domain" "$record_type"
}

main()
{
	add "$subdomain" "$ip_address"

	set -- $SSH_ORIGINAL_COMMAND
	while [ $# -ge 1 ]; do
		d=$1
		shift

		case "$d" in
			*.*) continue;;
		esac

		add "$d.$subdomain" "$ip_address"
	done
}

PEM_DEST=$HOME/public_rsync

eval "$(samizdat-ssh-uid --copy-pem "$PEM_DEST")"

domain=${SSH_CLIENT_DOMAIN}
subdomain=${SSH_CLIENT_FINGERPRINT}
ip_address=${SSH_CLIENT%% *}

main "$@"