From 1c28ad6f0070ff81efef44aaa694488d4bff60c4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Christian=20K=C3=B6nig?= <ckoenig@posteo.de>
Date: Mon, 3 Jan 2022 21:34:37 +0100
Subject: [PATCH] Skipp building index if no enabled adlists are present
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Christian König <ckoenig@posteo.de>
---
 gravity.sh | 28 ++++++++++++++++++----------
 1 file changed, 18 insertions(+), 10 deletions(-)

diff --git a/gravity.sh b/gravity.sh
index a6ab3c86..e886530d 100755
--- a/gravity.sh
+++ b/gravity.sh
@@ -85,19 +85,27 @@ generate_gravity_database() {
 
 # Copy data from old to new database file and swap them
 gravity_swap_databases() {
-  local str copyGravity
-  str="Building tree"
-  echo -ne "  ${INFO} ${str}..."
+  local str copyGravity enabled_adlists
 
-  # The index is intentionally not UNIQUE as poor quality adlists may contain domains more than once
-  output=$( { sqlite3 "${gravityTEMPfile}" "CREATE INDEX idx_gravity ON gravity (domain, adlist_id);"; } 2>&1 )
-  status="$?"
+  # Only build an index if there are enabled adlists
+  enabled_adlists="$(sqlite3 "${gravityTEMPfile}" "Select Count(id) from vw_adlist;")"
+  if [[ enabled_adlists -ge 1 ]]; then
+    str="Building tree"
+    echo -ne "  ${INFO} ${str}..."
 
-  if [[ "${status}" -ne 0 ]]; then
-    echo -e "\\n  ${CROSS} Unable to build gravity tree in ${gravityTEMPfile}\\n  ${output}"
-    return 1
+    # The index is intentionally not UNIQUE as poor quality adlists may contain domains more than once
+    output=$( { sqlite3 "${gravityTEMPfile}" "CREATE INDEX idx_gravity ON gravity (domain, adlist_id);"; } 2>&1 )
+    status="$?"
+
+    if [[ "${status}" -ne 0 ]]; then
+        echo -e "\\n  ${CROSS} Unable to build gravity tree in ${gravityTEMPfile}\\n  ${output}"
+        return 1
+    fi
+    echo -e "${OVER}  ${TICK} ${str}"
+  else
+      str="No enabled adlists. Skipping building tree"
+      echo -ne "  ${INFO} ${str}..."
   fi
-  echo -e "${OVER}  ${TICK} ${str}"
 
   str="Swapping databases"
   echo -ne "  ${INFO} ${str}..."