From 574b64b33b2d52a4c2ed209f9258a685d5747474 Mon Sep 17 00:00:00 2001
From: Terrtia <or1994@hotmail.fr>
Date: Tue, 30 Jul 2019 15:21:27 +0200
Subject: [PATCH] fix: [reset_ail] fix directory name

---
 reset_AIL.sh | 148 ++++++++++++++++++++++++++++++++++++++-------------
 1 file changed, 111 insertions(+), 37 deletions(-)

diff --git a/reset_AIL.sh b/reset_AIL.sh
index e87dfb67..44582e1a 100755
--- a/reset_AIL.sh
+++ b/reset_AIL.sh
@@ -6,52 +6,126 @@ GREEN="\\033[1;32m"
 
 [ -z "$AIL_HOME" ] && echo "Needs the env var AIL_HOME. Run the script from the virtual environment." && exit 1;
 
-# Make sure the reseting is intentional
-num=$(( ( RANDOM % 100 )  + 1 ))
+function soft_reset {
+  # Access dirs and delete
+  cd $AIL_HOME
 
-echo -e $RED"To reset the platform, enter the following number: "$DEFAULT $num 
-read userInput
+  # Kill all screens
+  screen -ls | grep Detached | cut -d. -f1 | awk '{print $1}' | xargs kill
 
-if [ $userInput -eq $num ]
-then
-    echo "Reseting AIL..."
-else
-    echo "Wrong number"
-    exit 1;
-fi
+  set -e
 
+  # Access dirs and delete
+  cd $AIL_HOME
 
-# Kill all screens
-screen -ls | grep Detached | cut -d. -f1 | awk '{print $1}' | xargs kill
+  if [ -d indexdir/ ]; then
+    pushd indexdir/
+    rm -r *
+    echo 'cleaned indexdir'
+    popd
+  fi
 
-set -e
+  if [ $userInput -eq $num ]
+  then
+    if [ -d DATA_ARDB/ ]; then
+      pushd DATA_ARDB/
+      rm -r *
+      echo 'cleaned DATA_ARDB'
+      popd
+    fi
+  fi
 
-# Access dirs and delete
-cd $AIL_HOME
+  if [ -d logs/ ]; then
+    pushd logs/
+    rm *
+    echo 'cleaned logs'
+    popd
+  fi
 
-pushd dumps/
-rm *
-echo 'cleaned dumps'
-popd
+  if [ -d PASTES/ ]; then
+    pushd PASTES/
+    rm -r *
+    echo 'cleaned PASTES'
+    popd
+  fi
 
-pushd indexdir/
-rm -r *
-echo 'cleaned indexdir'
-popd
+  if [ -d HASHS/ ]; then
+    pushd HASHS/
+    rm -r *
+    echo 'cleaned HASHS'
+    popd
+  fi
 
-pushd LEVEL_DB_DATA/
-rm -r *
-echo 'cleaned LEVEL_DB_DATA'
-popd
+  if [ -d CRAWLED_SCREESHOT/ ]; then
+    pushd CRAWLED_SCREESHOT/
+    rm -r *
+    echo 'cleaned CRAWLED_SCREESHOT'
+    popd
+  fi
 
-pushd logs/
-rm *
-echo 'cleaned logs'
-popd
+  if [ -d temp/ ]; then
+    pushd temp/
+    rm -r *
+    echo 'cleaned temp'
+    popd
+  fi
 
-pushd PASTES/
-rm -r *
-echo 'cleaned PASTES'
-popd
+  if [ -d var/www/submitted/ ]; then
+    pushd var/www/submitted
+    rm -r *
+    echo 'cleaned submitted'
+    popd
+  fi
 
-echo -e $GREEN"* AIL has been reset *"$DEFAULT
+  echo -e $GREEN"* AIL has been reset *"$DEFAULT
+
+  exit
+}
+
+#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
+
+    num=$(( ( RANDOM % 100 )  + 1 ))
+    echo -e $RED"If yes you want to delete the DB , enter the following number: "$DEFAULT $num
+    read userInput
+
+    set -e
+
+    soft_reset;
+
+    if [ $userInput -eq $num ]
+    then
+      if [ -d DATA_ARDB/ ]; then
+        pushd DATA_ARDB/
+        rm -r *
+        echo 'cleaned DATA_ARDB'
+        popd
+      fi
+    fi
+
+    echo -e $GREEN"* AIL has been reset *"$DEFAULT
+
+    exit
+}
+
+while [ "$1" != "" ]; do
+    case $1 in
+        --softReset )           soft_reset;
+                                      ;;
+        * )                     exit 1
+    esac
+    shift
+done