#1863 Add parameter to configure index location. Add warning is permi…

…ssion problem is detected
TheHive-Project · Mar 23, 2021 · 61057cc · 61057cc
1 parent f261d33
commit 61057cc
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/package/docker/entrypoint b/package/docker/entrypoint
@@ -1,6 +1,7 @@
 #!/bin/bash
 CQL_HOSTNAMES=${TH_CQL_HOSTNAMES:-cassandra}
 BDB_DIRECTORY=${TH_BDB_DIRECTORY:-/data/db}
+INDEX_DIRECTORY=${TH_INDEX_DIRECTORY:-/data/index}
 HDFS_URL=${TH_HDFS_URL}
 STORAGE_DIRECTORY=${TH_STORAGE_DIRECTORY:-/data/files}
 test "${TH_NO_CONFIG_SECRET}" == 1
@@ -60,6 +61,7 @@ do
     "--cql-username")      shift; CQL_USERNAME=$1 ;;
     "--cql-password")      shift; CQL_PASSWORD=$1 ;;
     "--bdb-directory")     shift; BDB_DIRECTORY=$1 ;;
+    "--index-directory")   shift; INDEX_DIRECTORY=$1 ;;
     "--no-config-storage") CONFIG_STORAGE=0 ;;
     "--hdfs-url")          shift; HDFS_URL=$1 ;;
     "--storage-directory") shift; STORAGE_DIRECTORY=$1 ;;
@@ -103,6 +105,12 @@ then
       echo "storage.backend = berkeleyje"             >> ${CONFIG_FILE}
       echo "storage.directory = \"${BDB_DIRECTORY}\"" >> ${CONFIG_FILE}
       echo "berkeleyje.freeDisk = 1"                  >> ${CONFIG_FILE}
+      if test -e "${BDB_DIRECTORY}"
+      then
+        test -w "${BDB_DIRECTORY}" || echo "WARNING the directory used to store database ($BDB_DIRECTORY) is not writable"
+      else
+        mkdir -p "${BDB_DIRECTORY}" || echo "WARNING the directory used to store database ($BDB_DIRECTORY) is not writable"
+      fi
     else
       echo "Using cassandra address = ${CQL[@]}"
       echo "storage.backend = cql"            >> ${CONFIG_FILE}
@@ -120,7 +128,16 @@ then
       echo "Waiting until Cassandra DB is up"
       sleep 30   # Sleep until cassandra Db is up
     fi
+    echo "index.search.backend = lucene"                   >> ${CONFIG_FILE}
+    echo "index.search.directory = \"${INDEX_DIRECTORY}\"" >> ${CONFIG_FILE}
+    if test -e "${INDEX_DIRECTORY}"
+    then
+      test -w "${INDEX_DIRECTORY}" || echo "WARNING the directory used to store index ($INDEX_DIRECTORY) is not writable"
+    else
+      mkdir -p "${INDEX_DIRECTORY}" || echo "WARNING the directory used to store index ($INDEX_DIRECTORY) is not writable"
+    fi
     echo "}" >> ${CONFIG_FILE}
+    if
   fi
 
   if test "${CONFIG_STORAGE}" = 1
@@ -140,6 +157,12 @@ then
       mkdir -p "${STORAGE_DIRECTORY}"
       echo "provider: localfs"                           >> ${CONFIG_FILE}
       echo "localfs.directory: \"${STORAGE_DIRECTORY}\"" >> ${CONFIG_FILE}
+      if test -e "${STORAGE_DIRECTORY}"
+      then
+        test -w "${STORAGE_DIRECTORY}" || echo "WARNING the directory used to store files ($STORAGE_DIRECTORY) is not writable"
+      else
+        mkdir -p "${STORAGE_DIRECTORY}" || echo "WARNING the directory used to store files ($STORAGE_DIRECTORY) is not writable"
+      fi
     fi
     echo "}" >> ${CONFIG_FILE}
   fi