From ab4bce4787093428305647ad547ed61c8d4809fd Mon Sep 17 00:00:00 2001
From: DL6ER <dl6er@dl6er.de>
Date: Sat, 11 Sep 2021 21:56:44 +0200
Subject: [PATCH] Allow users to force recovery even when checks are okay using
 "pihole -g -r recover force"

Signed-off-by: DL6ER <dl6er@dl6er.de>
---
 gravity.sh | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/gravity.sh b/gravity.sh
index 2d9b5dee..c2033fcf 100755
--- a/gravity.sh
+++ b/gravity.sh
@@ -858,7 +858,9 @@ database_recovery() {
     echo -ne "  ${INFO} ${str}..."
     if result="$(pihole-FTL sqlite3 "${gravityDBfile}" "PRAGMA foreign_key_check" 2>&1)"; then
       echo -e "${OVER}  ${TICK} ${str} - no errors found"
-      return
+      if [[ $1 != "force" ]]; then
+        return
+      fi
     else
       echo -e "${OVER}  ${CROSS} ${str} - errors found:"
       while IFS= read -r line ; do echo "  - $line"; done <<< "$result"
@@ -942,7 +944,7 @@ if [[ "${recreate_database:-}" == true ]]; then
 fi
 
 if [[ "${recover_database:-}" == true ]]; then
-  database_recovery
+  database_recovery "$4"
 fi
 
 # Move possibly existing legacy files to the gravity database