From 5db7509b52ca35dac1a075ed9daaf32c30756e04 Mon Sep 17 00:00:00 2001
From: manuasir <manuel.jimenez@wazuh.com>
Date: Fri, 19 Jul 2019 18:06:27 +0200
Subject: [PATCH] Fixed cluster configuration

---
 elasticsearch/config/config_cluster.sh | 55 ++++++++++++--------------
 1 file changed, 26 insertions(+), 29 deletions(-)

diff --git a/elasticsearch/config/config_cluster.sh b/elasticsearch/config/config_cluster.sh
index 2222099e..406cdf30 100644
--- a/elasticsearch/config/config_cluster.sh
+++ b/elasticsearch/config/config_cluster.sh
@@ -3,39 +3,36 @@
 
 elastic_config_file="/usr/share/elasticsearch/config/elasticsearch.yml"
 
+remove_single_node_conf(){
+  if grep -Fq "discovery.type" $1; then
+    sed -i '/discovery.type\: /d' $1 
+  fi
+}
 
-# If Elasticsearch cluster is enable
-if [[ $ELASTIC_CLUSTER == "true" ]]
-then
-  
-  # Set the cluster.name and discovery.zen.minimun_master_nodes variables
-  sed -i 's:cluster.name\: "docker-cluster":cluster.name\: "'$CLUSTER_NAME'":g' $elastic_config_file
-  #sed -i 's:discovery.zen.minimum_master_nodes\: 1:discovery.zen.minimum_master_nodes\: '$CLUSTER_NUMBER_OF_MASTERS':g' $elastic_config_file
+# If Elasticsearch cluster is enable, then set up the elasticsearch.yml
+if [[ $ELASTIC_CLUSTER == "true" && $CLUSTER_NODE_MASTER != "" && $CLUSTER_NODE_DATA != "" && $CLUSTER_NODE_INGEST != "" ]];then
 
-  # Add the cluster configuration
-  echo "
-#cluster node
-node:
-  master: ${CLUSTER_NODE_MASTER}
-  data: ${CLUSTER_NODE_DATA}
-  ingest: ${CLUSTER_NODE_INGEST}
-  name: ${CLUSTER_NODE_NAME}
-  max_local_storage_nodes: ${CLUSTER_MAX_NODES}
+  remove_single_node_conf $elastic_config_file
 
-bootstrap:
-  memory_lock: ${CLUSTER_MEMORY_LOCK} 
+  # Remove the old configuration
+  sed -i '/# cluster node/,/# end cluster config/d' $elastic_config_file
 
-cluster.initial_master_nodes:
-  - '${CLUSTER_INITIAL_MASTER_NODES}'
+  # Add the current cluster configuration
+cat > $elastic_config_file << EOF
+# cluster node
+network.host: 0.0.0.0
+node.name: $CLUSTER_NODE_NAME
+node.master: $CLUSTER_NODE_MASTER
 
-" >> $elastic_config_file
-else
-
-cat >> $elastic_config_file <<'EOF'
-cluster.initial_master_nodes:
-  - 'elasticsearch'
+cluster.initial_master_nodes: 
+  - $CLUSTER_INITIAL_MASTER_NODES
+# end cluster config" 
 EOF
 
-# echo 'discovery.type: single-node'
-
-fi
+# If the cluster is disabled, then set a single-node configuration
+else
+  sed -i '/# cluster node/,/# end cluster config/d' $elastic_config_file
+  # If it's not already configured
+  remove_single_node_conf $elastic_config_file
+  echo "discovery.type: single-node" >> $elastic_config_file
+fi
\ No newline at end of file