rabbitmq-cluster

#!/bin/bash

set -u
set -e

CLUSTER=${CLUSTER:-''}
MESSAGE_TTL=${MESSAGE_TTL:-''} # milliseconds
FQDN=$(hostname -s).wot.io
ENV_PRODPASSWORD=${PRODPASSWORD:-''}
ENV_WOTPASSWORD=${WOTPASSWORD:-''}
ENV_ADMINPASSWORD=${ADMINPASSWORD:-''}

export RABBITMQNODE_NAME=rabbit@$(hostname -s)

set_message_policy() {
	local vhost=${1:?'Usage: set_message_policy <vhost>'}

	# Set a message TTL on queues and exchanges if requested.
	if [ "${MESSAGE_TTL}" != "" ]; then
		rabbitmqctl set_policy -p ${vhost} --apply-to all msg-ttl-${MESSAGE_TTL} '.*' '{"message-ttl":'''${MESSAGE_TTL}'''}'
	fi
}

echo "Starting RabbitMQ on ${FQDN}"

echo "Configuring RabbitMQ for LDAP auth"
sed  -i 's/%LDAP_SERVER%/'${LDAP_SERVER:-"ldap.wot.io"}'/;
		s/%RABBITMQ_PASSWORD%/'${RABBITMQ_PASSWORD}'/' /etc/rabbitmq/rabbitmq.config

echo "Setting RabbitMQ Erlang cookie"
install -o rabbitmq -g rabbitmq -m 0400 /.erlang.cookie /var/lib/rabbitmq/.erlang.cookie

echo "Starting RabbitMQ"
runuser -l rabbitmq '/usr/lib/rabbitmq/bin/rabbitmq-server' &

sleep 10

CONFIGURED_VHOSTS="$(rabbitmqctl list_vhosts | wc -l)"
if [ "$CONFIGURED_VHOSTS" -lt "5" ]; then # 3 actual vhosts (/, wot, management) plus two lines from rabbitmqctl output
	if [[ -z "$CLUSTER" ]]; then
		echo "Loading passwords"
		[ -f /.passwords ] && source /.passwords

		# Passwords from the environment override the embedded passwords.
		PRODPASSWORD=${ENV_PRODPASSWORD:-${PRODPASSWORD}}
		WOTPASSWORD=${ENV_WOTPASSWORD:-${WOTPASSWORD}}
		ADMINPASSWORD=${ENV_ADMINPASSWORD:-${ADMINPASSWORD}}

		if [ -z "${PRODPASSWORD}" ] || [ -z "${WOTPASSWORD}" ] || [ -z "${ADMINPASSWORD}" ]; then
			echo "Local RabbitMQ passwords cannot be set."
			exit 1
		fi

		echo "Running as master"
		sleep 10
		echo "Configuring master"
		rabbitmqctl add_vhost management
		rabbitmqctl add_user prod $PRODPASSWORD
		rabbitmqctl add_user wot $WOTPASSWORD
		rabbitmqctl add_user admin $ADMINPASSWORD
		rabbitmqctl set_user_tags prod wot-observer
		rabbitmqctl set_user_tags wot management WOT wot-observer
		rabbitmqctl set_user_tags admin administrator WOT wot-administrator
		rabbitmqctl set_permissions admin '.*' '.*' '.*'
		rabbitmqctl set_permissions -p management admin '.*' '.*' '.*'
		rabbitmqctl set_permissions -p management wot '.*' '.*' '.*'

		set_message_policy /
		set_message_policy management

		env | grep "^VHOSTS.*" | awk -F "=" '{print $2}' | while read -r line ; do
			rabbitmqctl add_vhost $line
			rabbitmqctl set_permissions -p $line admin '.*' '.*' '.*'
			rabbitmqctl set_permissions -p $line prod '.*' '.*' '.*'
			rabbitmqctl set_permissions -p $line wot '.*' '.*' '.*'

			set_message_policy $line
		done

		echo "RabbitMQ master configured... running..."
	else
		echo "Joining cluster at $CLUSTER"
		rabbitmqctl stop_app
		rabbitmqctl reset
		rabbitmqctl join_cluster $CLUSTER
		rabbitmqctl start_app
	fi
else
	echo "RabbitMQ already configured. Running."
fi

/bin/bash -l