#!/bin/bash
basedir=".."
outputdir="output/adguardhome"
path="${basedir}/cache_domains.json"

export IFS=" "

if ! command -v jq >/dev/null; then
	cat <<-EOF
		This script requires jq to be installed.
		Your package manager should be able to find it
	EOF
	exit 1
fi

cachenamedefault="disabled"
combinedoutput=$(jq -r ".combined_output" config.json)

while read -r line; do
	ip=$(jq ".ips[\"${line}\"]" config.json)
	declare "cacheip${line}"="${ip}"
done <<<"$(jq -r ".ips | to_entries[] | .key" config.json)"

while read -r line; do
	name=$(jq -r ".cache_domains[\"${line}\"]" config.json)
	declare "cachename${line}"="${name}"
done <<<"$(jq -r ".cache_domains | to_entries[] | .key" config.json)"

rm -rf ${outputdir}
mkdir -p ${outputdir}
while read -r entry; do
	unset cacheip
	unset cachename
	key=$(jq -r ".cache_domains[$entry].name" ${path})
	cachename="cachename${key}"
	if [ -z "${!cachename}" ]; then
		cachename=${cachenamedefault}
	fi
	if [[ ${cachename} == "disabled" ]]; then
		continue
	fi
	cacheipname="cacheip${!cachename}"
	cacheip=$(jq -r "if type == \"array\" then .[] else . end" <<<"${!cacheipname}" | xargs)
	while read -r fileid; do
		while read -r filename; do
			destfilename=${filename//txt/conf}
			outputfile=${outputdir}/${destfilename}
			touch "${outputfile}"
			while read -r fileentry; do
				# Ignore comments and newlines
				if [[ ${fileentry} == \#* ]] || [[ -z ${fileentry} ]]; then
					continue
				fi
				domainprefix="|"
				if [[ $fileentry =~ ^\*\. ]]; then
					domainprefix="||"
				fi
				parsed="${fileentry#\*\.}"
				if grep -q "${domainprefix}${parsed}^\$dnsrewrite" "${outputfile}"; then
					continue
				fi
				for i in ${cacheip}; do
					echo "${domainprefix}${parsed}^\$dnsrewrite=${i}" >>"${outputfile}"
					echo "${domainprefix}${parsed}^\$dnstype=AAAA" >>"${outputfile}"
				done
			done <<<"$(cat ${basedir}/"${filename}" | sort)"
		done <<<"$(jq -r ".cache_domains[${entry}].domain_files[${fileid}]" ${path})"
	done <<<"$(jq -r ".cache_domains[${entry}].domain_files | to_entries[] | .key" ${path})"
done <<<"$(jq -r ".cache_domains | to_entries[] | .key" ${path})"

if [[ ${combinedoutput} == "true" ]]; then
	for file in "${outputdir}"/*; do f=${file//${outputdir}\//} && f=${f//.conf/} && echo "# ${f^}" >>${outputdir}/lancache.conf && cat "${file}" >>${outputdir}/lancache.conf && rm "${file}"; done
fi

cat <<EOF
Configuration generation completed.

Please copy the following files:
- ./${outputdir}/*.conf to /opt/adguardhome/work/userfilters/
- Navigate to Adguard Home -> Filters -> DNS blocklists -> Add blocklist -> Add a custom list
- Add list for each service or utilise the combined output for a single list
EOF