Initial Commit

buildnpush2iris.sh

@@ -0,0 +1,77 @@
+#!/bin/bash
+# Courtesy of SOCFortress
+
+# Help
+Help()
+{
+   # Display Help
+   echo "This script builds the DFIR-IRIS module of the current directory and installs it to DFIR-IRIS. If you run it for the first time or change something in the module configuration template make sure to run the -a switch."
+   echo
+   echo "Syntax: ./buildnpush2iris [-a|h]"
+   echo "options:"
+   echo "a     Also install the module to the iris-web_app_1 container. Only required on initial install or when changes to config template were made."
+   echo "h     Print this Help."
+   echo
+}
+
+Run()
+{
+    echo "[BUILDnPUSH2IRIS] Starting the build and push process.."
+    SEARCH_DIR='./dist'
+    get_recent_file () {
+        FILE=$(ls -Art1 ${SEARCH_DIR} | tail -n 1)
+        if [ ! -f ${FILE} ]; then
+            SEARCH_DIR="${SEARCH_DIR}/${FILE}"
+            get_recent_file
+        fi
+        echo $FILE
+        exit
+    }
+
+    python3.9 setup.py bdist_wheel
+
+    latest=$(get_recent_file)
+    module=${latest#"./dist/"}
+
+    echo "[BUILDnPUSH2IRIS] Found latest module file: $latest"
+    echo "[BUILDnPUSH2IRIS] Copy module file to worker container.."
+    docker cp $latest iris-web-worker-1:/iriswebapp/dependencies/$module
+    echo "[BUILDnPUSH2IRIS] Installing module in worker container.."
+    docker exec -it iris-web-worker-1 /bin/sh -c "pip3 install dependencies/$module --force-reinstall"
+    echo "[BUILDnPUSH2IRIS] Restarting worker container.."
+    docker restart iris-web-worker-1
+
+    if [ "$a_Flag" = true ] ; then
+        echo "[BUILDnPUSH2IRIS] Copy module file to app container.."
+        docker cp $latest iris-web-app-1:/iriswebapp/dependencies/$module
+        echo "[BUILDnPUSH2IRIS] Installing module in app container.."
+        docker exec -it iris-web-app-1 /bin/sh -c "pip3 install dependencies/$module --force-reinstall"
+        echo "[BUILDnPUSH2IRIS] Restarting app container.."
+        docker restart iris-web-app-1
+    fi
+
+    echo "[BUILDnPUSH2IRIS] Completed!"
+}
+
+a_Flag=false
+
+while getopts ":ha" option; do
+   case $option in
+      h) # display Help
+         Help
+         exit;;
+      a) # Enter a name
+         echo "[BUILDnPUSH2IRIS] Pushing to Worker and App container!"
+         a_Flag=true
+         Run
+         exit;;
+     \?) # Invalid option
+         echo "ERROR: Invalid option"
+         exit;;
+
+   esac
+done
+
+echo "[BUILDnPUSH2IRIS] Pushing to Worker container only!"
+Run
+exit