#!/bin/bash

if [[ ${CONTROLLERFEATURES} == *federation* ]]; then

cat > ${WORKSPACE}/install_federation_rabbit.sh <<EOF

    echo "Install rabbit server ..."
    sudo yum install -y erlang
    sudo yum install -y rabbitmq-server # Technically we should only install on the first ODL

    echo "Starting rabbit server ..."
    sudo chkconfig rabbitmq-server
    sudo service rabbitmq-server start

    echo "Add federation user to rabbit server ..."
    sudo /usr/sbin/rabbitmqctl add_user federation federation
    sudo rabbitmqctl set_permissions -p / federation ".*" ".*" ".*"

EOF

cat > ${WORKSPACE}/configure_federation_rabbit.sh <<EOF

    echo "Configuring rabbit broker for ODL ..."
    SERVICECONF=\$(find "/tmp/${BUNDLEFOLDER}/" -name "federation-service-impl-*config.xml")
    sed -ie "s/<mqBrokerIp>.*</<mqBrokerIp>\$1</g" \${SERVICECONF}
    sed -ie "s/<site-ip>.*</<site-ip>\$2</g" \${SERVICECONF}
    sed -ie "s/CONTROL_QUEUE_.*</CONTROL_QUEUE_\$2</g" \${SERVICECONF}
    sed -ie "s/guest/federation/g" \${SERVICECONF}
    cat \${SERVICECONF}

EOF

    NUM_ODLS_PER_SITE=$((NUM_ODL_SYSTEM / NUM_OPENSTACK_SITES))
    echo "Copying config files to ODL Controller folder"
    for i in `seq 1 ${NUM_OPENSTACK_SITES}`
    do
        FIRST_ODL_IN_SITE=ODL_SYSTEM_$(((i - 1) * NUM_ODLS_PER_SITE + 1))_IP
        RABBIT_SERVER_IP=${!FIRST_ODL_IN_SITE} # We install Rabbit on the first ODL in each site
        CONTROLLERIP=ODL_SYSTEM_$(((i - 1) * NUM_ODLS_PER_SITE + j))_IP

        scp ${WORKSPACE}/install_federation_rabbit.sh ${RABBIT_SERVER_IP}:/tmp/
        ssh ${RABBIT_SERVER_IP} 'bash /tmp/install_federation_rabbit.sh'

        for j in `seq 1 ${NUM_ODLS_PER_SITE}`
        do
            ODL_IP=ODL_SYSTEM_$(((i - 1) * NUM_ODLS_PER_SITE + j))_IP
            if [ ${NUM_ODLS_PER_SITE} -gt 1 ]; then
                HA_PROXY_IP=OPENSTACK_HAPROXY_${i}_IP
                SITE_IP=${!HA_PROXY_IP}
            else
                SITE_IP=${RABBIT_SERVER_IP}
            fi
            echo "Setting rabbit client site IP to ${SITE_IP} and rabbit servier IP to ${RABBIT_SERVER_IP} on ${!ODL_IP}"

            scp ${WORKSPACE}/configure_federation_rabbit.sh ${!ODL_IP}:/tmp/
            ssh ${!ODL_IP} 'bash /tmp/configure_federation_rabbit.sh' ${RABBIT_SERVER_IP} ${SITE_IP}
        done
    done
fi