generate-configs.sh

#!/bin/bash

is_easticsearch_cluster_defined() {
  [[ -n "$ES_CLUSTER_NAME" ]]
}

is_cors_enabled() {
  [[ -n "$CORS_ENABLED" ]]
}

is_using_elasticsearch_config_file() {
  [[ -f /etc/graylog2-elasticsearch.yml ]]
}

is_dir_not_empty() {
  local dir=$1

  [[ "$(ls -A $dir)" ]]
}

install_plugins() {
  local string=$1
  local plugins=(${string//,/ })
  for i in "${!plugins[@]}"
  do
    /opt/elasticsearch/bin/plugin --install ${plugins[i]} \
      && mv /opt/elasticsearch/plugins/* /opt/graylog2-server/plugins
  done
}

enable_es_config_file() {
  sed -i -e "s/#elasticsearch_config_file = \/etc\/graylog2-elasticsearch.yml/elasticsearch_config_file = \/etc\/graylog2-elasticsearch.yml/" /etc/graylog2.conf
}

enable_cluster_in_graylog() {
  sed -i -e "s/#elasticsearch_cluster_name = graylog2/elasticsearch_cluster_name = ${ES_CLUSTER_NAME}/" /etc/graylog2.conf
  sed -i -e "s/#elasticsearch_node_master = false/elasticsearch_node_master = false/" /etc/graylog2.conf
  sed -i -e "s/#elasticsearch_node_data = false/elasticsearch_node_data = false/" /etc/graylog2.conf
  sed -i -e "s/elasticsearch_discovery_zen_ping_unicast_hosts = 127.0.0.1:9300/elasticsearch_discovery_zen_ping_unicast_hosts = ${ES_CLUSTER_HOSTS}/" /etc/graylog2.conf
}

dont_start_elasticsearch() {
  sed -i -e "s/autorestart=true ;es/autorestart=false ;es\nautostart=false ;es/" /etc/supervisor/conf.d/supervisord-graylog.conf
}

enable_cors() {
  sed -i -e "s/#rest_enable_cors = true/rest_enable_cors = true/" /etc/graylog2.conf
}

is_defined() {
  local var=$1
  [[ -n "$var" ]]
}

define_the_damn_admin_password() {
  sed -i -e "s/root_password_sha2 = $(echo -n admin | sha256sum | awk '{print $1}')/root_password_sha2 = $GRAYLOG2_ADMIN_PASSWORD/" /etc/graylog2.conf
}

replace_admin_user() {
  local user=$1

  sed -i -e "s/# root_username = admin/root_username = $user/" /etc/graylog2.conf
}

main() {
  is_defined "$GRAYLOG2_ES_PLUGINS" \
    && is_dir_not_empty /opt/graylog2-server/plugins \
    || install_plugins $GRAYLOG2_ES_PLUGINS

  is_defined "$GRAYLOG2_ADMIN_USER" \
    && replace_admin_user $GRAYLOG2_ADMIN_USER

  is_defined "$GRAYLOG2_ADMIN_PASSWORD" \
    && define_the_damn_admin_password

  # eventually the bash retardedness will go to the python file. :-/
  generate-graylog2-es

  is_using_elasticsearch_config_file \
    && enable_es_config_file

  is_easticsearch_cluster_defined \
    && enable_cluster_in_graylog \
    && dont_start_elasticsearch

  is_cors_enabled \
    && enable_cors

  return 0
}

main