From 5e29f8c850583d387032bc98fcfd672aebdd63ed Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?F=C3=A9lix=20Aime?= <felix.aime@kaspersky.com>
Date: Thu, 25 Mar 2021 15:42:51 +0100
Subject: [PATCH] Update install.sh with new checks

---
 install.sh | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/install.sh b/install.sh
index c74c324..eb59e6d 100644
--- a/install.sh
+++ b/install.sh
@@ -282,11 +282,13 @@ install_package() {
        apt-key add - < Release.key
        rm Release.key && sudo apt-get update
        apt-get install zeek -y
-    elif [[ $1 == "nodejs" ]]; then
+    elif [[ $1 == "node" ]]; then
        curl -sL https://deb.nodesource.com/setup_14.x | bash
        apt-get install -y nodejs
     elif [[ $1 == "dig" ]]; then
        apt-get install -y dnsutils
+    elif [[ $1 == "pip" ]]; then
+       apt-get install -y python3-pip
    fi
 }
 
@@ -300,7 +302,9 @@ check_dependencies() {
          "/usr/bin/dig"
          "/usr/bin/suricata"
          "/usr/bin/unclutter"
-         "/usr/bin/sqlite3")
+         "/usr/bin/sqlite3"
+         "/usr/bin/pip"
+         "/usr/bin/node")
 
    echo -e "\e[39m[+] Checking dependencies...\e[39m"
    for bin in "${bins[@]}"
@@ -312,8 +316,6 @@ check_dependencies() {
            install_package ${bin##*/}
       fi
    done
-   echo -e "\e[39m[+] Install NodeJS...\e[39m"
-   install_package nodejs
    echo -e "\e[39m[+] Install Python packages...\e[39m"
    python3 -m pip install -r "$SCRIPT_PATH/assets/requirements.txt"
 }