From 2ad7d912b3d0b782862622600c1fb82374a5892b Mon Sep 17 00:00:00 2001
From: Terrtia <or1994@hotmail.fr>
Date: Wed, 24 Jun 2020 14:27:10 +0200
Subject: [PATCH] fix: [reset_AIL] add helper + fix soft reset

---
 reset_AIL.sh | 71 ++++++++++++++++++++++++++++++++++------------------
 1 file changed, 47 insertions(+), 24 deletions(-)

diff --git a/reset_AIL.sh b/reset_AIL.sh
index aac94352..50832ac5 100755
--- a/reset_AIL.sh
+++ b/reset_AIL.sh
@@ -6,6 +6,29 @@ GREEN="\\033[1;32m"
 
 [ -z "$AIL_HOME" ] && echo "Needs the env var AIL_HOME. Run the script from the virtual environment." && exit 1;
 
+function helptext {
+    echo -e $GREEN"
+
+              .o.            ooooo      ooooo
+             .888.           \`888'      \`888'
+            .8\"888.           888        888
+           .8' \`888.          888        888
+          .88ooo8888.         888        888
+         .8'     \`888.        888        888       o
+        o88o     o8888o   o  o888o   o  o888ooooood8
+
+         Analysis Information Leak framework
+    "$DEFAULT"
+    Use this script to reset AIL (DB + stored items):
+
+    Usage:
+    -----
+    reset_AIL.sh
+      [--softReset]               Keep All users accounts
+      [-h  | --help]              Help
+    "
+}
+
 function reset_dir {
   # Access dirs and delete
   cd $AIL_HOME
@@ -23,16 +46,6 @@ function reset_dir {
     popd
   fi
 
-  if [ $userInput -eq $num ]
-  then
-    if [ -d DATA_ARDB/ ]; then
-      pushd DATA_ARDB/
-      rm -r *
-      echo 'cleaned DATA_ARDB'
-      popd
-    fi
-  fi
-
   if [ -d logs/ ]; then
     pushd logs/
     rm *
@@ -97,29 +110,36 @@ function flush_DB_keep_user {
   bash ${AIL_BIN}LAUNCH.sh -k
 }
 
+function validate_reset {
+  echo -e $RED"WARNING: DELETE AIL DATA"$DEFAULT
+
+  # Make sure the reseting is intentional
+  num=$(( ( RANDOM % 100 )  + 1 ))
+
+  echo -e $RED"To reset the platform, enter the following number: "$DEFAULT $num
+  read userInput
+
+  if [ $userInput -eq $num ]
+  then
+      echo "Reseting AIL..."
+  else
+      echo "Wrong number"
+      exit 1;
+  fi
+}
+
 function soft_reset {
+  validate_reset;
   reset_dir;
   flush_DB_keep_user;
 }
 
 #If no params,
 [[ $@ ]] || {
-    # Make sure the reseting is intentional
-    num=$(( ( RANDOM % 100 )  + 1 ))
-
-    echo -e $RED"To reset the platform, enter the following number: "$DEFAULT $num
-    read userInput
-
-    if [ $userInput -eq $num ]
-    then
-        echo "Reseting AIL..."
-    else
-        echo "Wrong number"
-        exit 1;
-    fi
+    validate_reset;
 
     num=$(( ( RANDOM % 100 )  + 1 ))
-    echo -e $RED"If yes you want to delete the DB , enter the following number: "$DEFAULT $num
+    echo -e $RED"If you want to delete the DB , enter the following number: "$DEFAULT $num
     read userInput
 
     reset_dir;
@@ -143,6 +163,9 @@ while [ "$1" != "" ]; do
     case $1 in
         --softReset )           soft_reset;
                                 ;;
+        -h | --help )           helptext;
+                                exit
+                                ;;
         * )                     exit 1
     esac
     shift