#!/bin/bash

# Save trace setting
XTRACE=$(set +o | grep xtrace)
set +o xtrace

DIST_PACKAGES="/usr/local/lib/python2.7/dist-packages"
ENTRY_POINT="$DIST_PACKAGES/glance_store-*.dist-info/entry_points.txt"
NEUTRON_ENTRY_POINT="/opt/stack/neutron/neutron.egg-info/entry_points.txt"
ML2_CONF="/etc/neutron/plugins/ml2/ml2_conf.ini"
NOVA_CPU_CONF="/etc/nova/nova-cpu.conf"

# Functions
# ------------

function set_aws_region_and_az {
    local conf_file=$1
    iniset $conf_file AWS region_name ${AWS_REGION_NAME}
    iniset $conf_file AWS az ${AWS_AVAILABILITY_ZONE}
}

function configure_glance {
    set_aws_region_and_az $GLANCE_API_CONF
    iniset $GLANCE_API_CONF DEFAULT show_multiple_locations True
    iniset $GLANCE_API_CONF glance_store default_store aws
    iniset $GLANCE_API_CONF glance_store stores aws

    sudo crudini --set $ENTRY_POINT glance_store.drivers glance.store.aws.Store glance_store._drivers.aws:Store
    sudo crudini --set $ENTRY_POINT glance_store.drivers aws glance_store._drivers.aws:Store
    value=$(sed -n "/, 'aws'/p" $DIST_PACKAGES/glance_store/backend.py)
    if [[ $value == "" ]]; then
        sudo sed -i -e "s/, 'vsphere'/, 'vsphere', 'aws'/g" $DIST_PACKAGES/glance_store/backend.py
    fi
}

function configure_cinder {
    set_aws_region_and_az $CINDER_CONF
    iniset $CINDER_CONF AWS volume_driver ${AWS_CINDER_DRIVER}
    iniset $CINDER_CONF DEFAULT enable_force_upload ${ENABLE_FORCE_UPLOAD}
    iniset $CINDER_CONF DEFAULT enabled_backends AWS
    iniset $CINDER_CONF DEFAULT volumes_dir ${VOLUME_DIR}
    iniset $CINDER_CONF DEFAULT os_privileged_user_name ${CINDER_USER}
    iniset $CINDER_CONF DEFAULT default_volume_type AWS
    iniset $CINDER_CONF DEFAULT os_region_name ${OS_REGION_NAME}
}

function configure_nova {
    set_aws_region_and_az $NOVA_CONF
    iniset $NOVA_CONF DEFAULT compute_driver ${AWS_COMPUTE_DRIVER}
    iniset $NOVA_CONF DEFAULT glance_api_insecure True
    iniset $NOVA_CONF DEFAULT cinder_api_insecure True
    iniset $NOVA_CONF DEFAULT vnc_enabled True
    iniset $NOVA_CONF DEFAULT osapi_compute_extension ${COMPUTE_EXTENSION}
    iniset $NOVA_CONF DEFAULT force_dhcp_release True

    iniset $NOVA_CONF keystone_authtoken identity_uri "http://${SERVICE_HOST}/identity"
    iniset $NOVA_CONF keystone_authtoken username nova
    iniset $NOVA_CONF keystone_authtoken admin_user nova
    iniset $NOVA_CONF keystone_authtoken password $SERVICE_PASSWORD
    iniset $NOVA_CONF keystone_authtoken project_name $SERVICE_PROJECT_NAME
    iniset $NOVA_CONF keystone_authtoken admin_password $SERVICE_PASSWORD
    iniset $NOVA_CONF keystone_authtoken admin_tenant_name $SERVICE_PROJECT_NAME
    iniset $NOVA_CONF keystone_authtoken region_name ${OS_REGION_NAME}

    set_aws_region_and_az $NOVA_CPU_CONF
    iniset $NOVA_CPU_CONF keystone_authtoken identity_uri "http://${SERVICE_HOST}/identity"
    iniset $NOVA_CPU_CONF keystone_authtoken username nova
    iniset $NOVA_CPU_CONF keystone_authtoken admin_user nova
    iniset $NOVA_CPU_CONF keystone_authtoken password $SERVICE_PASSWORD
    iniset $NOVA_CPU_CONF keystone_authtoken project_name $SERVICE_PROJECT_NAME
    iniset $NOVA_CPU_CONF keystone_authtoken admin_password $SERVICE_PASSWORD
    iniset $NOVA_CPU_CONF keystone_authtoken admin_tenant_name $SERVICE_PROJECT_NAME
    iniset $NOVA_CPU_CONF keystone_authtoken region_name ${OS_REGION_NAME}

}

function configure_neutron {
    set_aws_region_and_az $NEUTRON_CONF
    iniset $NEUTRON_CONF DEFAULT core_plugin ${CORE_PLUGIN}
    iniset $NEUTRON_CONF DEFAULT service_plugins ${AWS_SERVICE_PLUGIN}
    iniset $NEUTRON_CONF nova region_name ${OS_REGION_NAME}
    iniset $NEUTRON_CONF keystone_authtoken auth_uri "http://${SERVICE_HOST}/identity"

    iniset $ML2_CONF ml2 type_drivers ${ML2_DRIVERS}
    iniset $ML2_CONF ml2 tenant_network_types ${ML2_TENANT}
    iniset $ML2_CONF ml2_type_flat flat_networks "*"
    iniset $ML2_CONF ml2_type_vlan network_vlan_ranges ${ML2_VLAN_TYPE}
    iniset $ML2_CONF ml2 mechanism_drivers aws
    iniset $ML2_CONF ml2 extension_drivers subnet_az

    sudo crudini --set $NEUTRON_ENTRY_POINT neutron.ml2.mechanism_drivers aws neutron.plugins.ml2.drivers.aws.mechanism_aws:AwsMechanismDriver
    sudo crudini --set $NEUTRON_ENTRY_POINT neutron.service_plugins aws_router neutron.services.l3_router.aws_router_plugin:AwsRouterPlugin
}

# Restore xtrace
$XTRACE

## Local variables:
## mode: shell-script
## End: