#!/bin/sh

# Copyright (C) 2014-2015  Codethink Limited
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; version 2 of the License.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program.  If not, see <http://www.gnu.org/licenses/>.

set -e

ROOT="$1"

enable(){
    ln -sf "/usr/lib/systemd/system/openstack-neutron-$1.service" \
            "$ROOT/etc/systemd/system/multi-user.target.wants/openstack-neutron-$1.service"
}

unrecognised_value(){
    eval echo Unrecognised value \$$1 for $1 >&2
    exit 1
}


##########################################################################
# Check variables
##########################################################################


if [ -z "$NEUTRON_SERVICE_USER" -a \
     -z "$NEUTRON_SERVICE_PASSWORD" -a \
     -z "$NEUTRON_DB_USER" -a \
     -z "$NEUTRON_DB_PASSWORD" -a \
     -z "$METADATA_PROXY_SHARED_SECRET" -a \
     -z "$NOVA_SERVICE_USER" -a \
     -z "$NOVA_SERVICE_PASSWORD" -a \
     -z "$RABBITMQ_HOST" -a \
     -z "$RABBITMQ_USER" -a \
     -z "$RABBITMQ_PASSWORD" -a \
     -z "$RABBITMQ_PORT" -a \
     -z "$CONTROLLER_HOST_ADDRESS" -a \
     -z "$MANAGEMENT_INTERFACE_IP_ADDRESS" -a \
     -z "$KEYSTONE_TEMPORARY_ADMIN_TOKEN" ]; then
     # No Neutron options defined, do nothing.
     exit 0
fi

if [ -z "$NEUTRON_SERVICE_USER" -o \
     -z "$NEUTRON_SERVICE_PASSWORD" -o \
     -z "$NEUTRON_DB_USER" -o \
     -z "$NEUTRON_DB_PASSWORD" -o \
     -z "$METADATA_PROXY_SHARED_SECRET" -o \
     -z "$NOVA_SERVICE_USER" -o \
     -z "$NOVA_SERVICE_PASSWORD" -o \
     -z "$RABBITMQ_HOST" -o \
     -z "$RABBITMQ_USER" -o \
     -z "$RABBITMQ_PASSWORD" -o \
     -z "$RABBITMQ_PORT" -o \
     -z "$CONTROLLER_HOST_ADDRESS" -o \
     -z "$MANAGEMENT_INTERFACE_IP_ADDRESS" -o \
     -z "$KEYSTONE_TEMPORARY_ADMIN_TOKEN" ]; then
    echo Some options required for Neutron were defined, but not all.
    exit 1
fi

#############################################
# Ensure /var/run is an appropriate symlink #
#############################################

if ! link="$(readlink "$ROOT/var/run")" || [ "$link" != ../run ]; then
    rm -rf "$ROOT/var/run"
    ln -s ../run "$ROOT/var/run"
fi

###################
# Enable services #
###################

if [ x"${NEUTRON_ENABLE_CONTROLLER=True}" = xTrue -o \
     x"${NEUTRON_ENABLE_MANAGER=True}" = xTrue -o \
     x"${NEUTRON_ENABLE_AGENT=True}" = xTrue ]; then
    enable config-setup
fi

case "${NEUTRON_ENABLE_CONTROLLER}" in
True|yes|y)
    enable config-setup
    enable db-setup
    enable server
    ;;
False|no|n|'')
    ;;
*)
    unrecognised_value NEUTRON_ENABLE_CONTROLLER
    ;;
esac

case "${NEUTRON_ENABLE_MANAGER}" in
True|yes|y)
    enable config-setup
    enable ovs-cleanup
    enable dhcp-agent
    enable l3-agent
    enable plugin-openvswitch-agent
    enable metadata-agent
    ;;
False|no|n|'')
    ;;
*)
    unrecognised_value NEUTRON_ENABLE_MANAGER
    ;;
esac

case "${NEUTRON_ENABLE_AGENT}" in
True|yes|y)
    enable config-setup
    enable plugin-openvswitch-agent
    ;;
False|no|n|'')
    ;;
*)
    unrecognised_value NEUTRON_ENABLE_AGENT
    ;;
esac

##########################################################################
# Generate config variable shell snippet
##########################################################################

OPENSTACK_DATA="$ROOT/etc/openstack"
mkdir -p "$OPENSTACK_DATA"

python <<'EOF' >"$OPENSTACK_DATA/neutron.conf"
import os, sys, yaml

nova_configuration={
    'NEUTRON_SERVICE_USER': os.environ['NEUTRON_SERVICE_USER'],
    'NEUTRON_SERVICE_PASSWORD': os.environ['NEUTRON_SERVICE_PASSWORD'],
    'NEUTRON_DB_USER': os.environ['NEUTRON_DB_USER'],
    'NEUTRON_DB_PASSWORD': os.environ['NEUTRON_DB_PASSWORD'],
    'METADATA_PROXY_SHARED_SECRET': os.environ['METADATA_PROXY_SHARED_SECRET'],
    'NOVA_SERVICE_USER': os.environ['NOVA_SERVICE_USER'],
    'NOVA_SERVICE_PASSWORD': os.environ['NOVA_SERVICE_PASSWORD'],
    'RABBITMQ_HOST': os.environ['RABBITMQ_HOST'],
    'RABBITMQ_USER': os.environ['RABBITMQ_USER'],
    'RABBITMQ_PASSWORD': os.environ['RABBITMQ_PASSWORD'],
    'RABBITMQ_PORT': os.environ['RABBITMQ_PORT'],
    'CONTROLLER_HOST_ADDRESS': os.environ['CONTROLLER_HOST_ADDRESS'],
    'MANAGEMENT_INTERFACE_IP_ADDRESS': os.environ['MANAGEMENT_INTERFACE_IP_ADDRESS'],
    'KEYSTONE_TEMPORARY_ADMIN_TOKEN': os.environ['KEYSTONE_TEMPORARY_ADMIN_TOKEN'],
}

yaml.dump(nova_configuration, sys.stdout, default_flow_style=False)
EOF