Add trap to catch Ctrl+C

yubiuser · Jul 21, 2020 · 04d337c · 04d337c
1 parent 88816ac
commit 04d337c
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/pihole_adlist_tool b/pihole_adlist_tool
@@ -91,9 +91,18 @@ remove_temp_database () {
 rm -f $TEMP_DB
 }
 
+# cleanup on trap
+cleanup_on_trap () {
+remove_temp_database
+echo -e "\\n\\n${bold}User-abort detected!${normal} Removing temporary database.\n" 
+exit 1
+}
+
 ### warm-up ######
 
 
+trap cleanup_on_trap INT
+
 # getopts flags and assing arguments to variables
 
 while getopts 'd:t:s:uh' flag; do