From ed493f9c3a63d94ef10b149829fe3a752c3532e3 Mon Sep 17 00:00:00 2001
From: KagurazakaNyaa <dev@kagurazakanyaa.com>
Date: Thu, 18 Jan 2024 23:28:03 +0800
Subject: [PATCH] Allow user skip unbound healthcheck

---
 data/Dockerfiles/unbound/healthcheck.sh | 10 ++++++++++
 generate_config.sh                      |  4 ++++
 2 files changed, 14 insertions(+)

diff --git a/data/Dockerfiles/unbound/healthcheck.sh b/data/Dockerfiles/unbound/healthcheck.sh
index ea94f63b..760aa02b 100644
--- a/data/Dockerfiles/unbound/healthcheck.sh
+++ b/data/Dockerfiles/unbound/healthcheck.sh
@@ -1,5 +1,10 @@
 #!/bin/bash
 
+# Skipping DNS check
+if [[ "${SKIP_DNS_CHECK}" =~ ^([yY][eE][sS]|[yY])+$ ]]; then
+    SKIP_DNS_CHECK=y
+fi
+
 # Declare log function for logfile inside container
 function log_to_file() {
     echo "$(date +"%Y-%m-%d %H:%M:%S"): $1" > /var/log/healthcheck.log
@@ -66,6 +71,11 @@ function check_netcat() {
 
 }
 
+if [[ ${SKIP_DNS_CHECK} == "y" ]]; then
+    log_to_file "Healthcheck: ALL CHECKS WERE SKIPPED! Unbound is healthy!"
+    exit 0
+fi
+
 # run checks, if check is not returning 0 (return value if check is ok), healthcheck will exit with 1 (marked in docker as unhealthy)
 check_ping
 
diff --git a/generate_config.sh b/generate_config.sh
index 2986f168..0c8a9bcf 100755
--- a/generate_config.sh
+++ b/generate_config.sh
@@ -363,6 +363,10 @@ SKIP_IP_CHECK=n
 
 SKIP_HTTP_VERIFICATION=n
 
+# Skip DNS check in Unbound container - y/n
+
+SKIP_DNS_CHECK=n
+
 # Skip ClamAV (clamd-mailcow) anti-virus (Rspamd will auto-detect a missing ClamAV container) - y/n
 
 SKIP_CLAMD=${SKIP_CLAMD}