From f01a7118369ee4c51d0ea66185ce3d973308d4aa Mon Sep 17 00:00:00 2001
From: Ming Di Leom <2809763-curben@users.noreply.gitlab.com>
Date: Wed, 16 Oct 2024 09:49:48 +0000
Subject: [PATCH] fix: handle empty upstream data

---
 src/script.sh | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/script.sh b/src/script.sh
index ee0fd60..17c86e7 100644
--- a/src/script.sh
+++ b/src/script.sh
@@ -108,7 +108,6 @@ sed -e "s/^/||/g" -e "s/$/\$document/g" | \
 sed "1i $COMMENT_UBO" | \
 sed "1s/Blocklist/Blocklist (Vivaldi)/" > "../public/botnet-filter-vivaldi.txt"
 
-
 ## Hash comment
 # awk + head is a workaround for sed prepend
 COMMENT=$(printf "$COMMENT_UBO" | sed "s/^!/#/g" | awk '{printf "%s\\n", $0}' | head -c -2)
@@ -145,6 +144,17 @@ done < "feodo-ip.txt"
 set -x
 
 
+# upstream may provide empty data
+if [ ! -s "feodo-ip.txt" ]; then
+  printf "$COMMENT_UBO\n! END 0 entries\n" > "../public/botnet-filter.txt"
+  printf "$COMMENT_UBO\n! END 0 entries\n" > "../public/botnet-filter-agh.txt"
+  printf "$COMMENT_UBO\n! END 0 entries\n" > "../public/botnet-filter-ag.txt"
+  printf "$COMMENT_UBO\n! END 0 entries\n" > "../public/botnet-filter-vivaldi.txt"
+  printf "$COMMENT\n# END 0 entries\n" > "../public/botnet-filter-dnscrypt-blocked-ips.txt"
+  echo "# END 0 entries" > "../public/botnet-filter-suricata.rules"
+  echo "# END 0 entries" > "../public/botnet-filter-splunk.csv"
+fi
+
 sed -i "1i $COMMENT" "../public/botnet-filter-suricata.rules"
 sed -i "1s/Blocklist/Suricata Ruleset/" "../public/botnet-filter-suricata.rules"