#!/usr/bin/env bash

set -o errexit -o nounset

TOP_DIR=$(cd "$(dirname "$0")/.." && pwd)

source "$TOP_DIR/config/paths"
source "$CONFIG_DIR/openstack"
source "$CONFIG_DIR/localrc"
source "$LIB_DIR/functions.guest.sh"

indicate_current_auto

exec_logfile

function set_apt_proxy {
    local PRX_KEY="Acquire::http::Proxy"
    local APT_FILE=/etc/apt/apt.conf

    if [ -f $APT_FILE ] && grep -q $PRX_KEY $APT_FILE; then
        # apt proxy has already been set (by preseed/kickstart)
        if [ -n "${VM_PROXY-}" ]; then
            # Replace with requested proxy
            sudo sed -i "s#^\($PRX_KEY\).*#\1 \"$VM_PROXY\";#" $APT_FILE
        else
            # No proxy requested -- remove
            sudo sed -i "s#^$PRX_KEY.*##" $APT_FILE
        fi
    elif [ -n "${VM_PROXY-}" ]; then
        # Proxy requested, but none configured: add line
        echo "$PRX_KEY \"$VM_PROXY\";" | sudo tee -a $APT_FILE
    fi
}

set_apt_proxy

# Get apt index files
sudo apt-get update

function ubuntu_cloud_archive {
    # cloud-keyring to verify packages from ubuntu-cloud repo
    sudo apt-get install ubuntu-cloud-keyring

    #--------------------------------------------------------------------------
    # Enable the OpenStack repository
    # http://docs.openstack.org/mitaka/install-guide-ubuntu/environment-packages.html
    #--------------------------------------------------------------------------

    # Install packages needed for add-apt-repository
    sudo apt-get -y install software-properties-common \
                            python-software-properties
    sudo add-apt-repository -y "cloud-archive:$OPENSTACK_RELEASE"
}

function ubuntu_cloud_staging {
    #--------------------------------------------------------------------------
    # Enable the OpenStack repository
    # https://launchpad.net/~ubuntu-cloud-archive/+archive/ubuntu/mitaka-staging
    #
    # NOTE: Using pre-release staging ppa is not documented in install-guide
    #--------------------------------------------------------------------------

    sudo apt-key adv --keyserver keyserver.ubuntu.com --recv-keys 9F68104E

    cat << DEB |sudo tee /etc/apt/sources.list.d/cloudarchive-$OPENSTACK_RELEASE.list
deb http://ppa.launchpad.net/ubuntu-cloud-archive/$OPENSTACK_RELEASE/ubuntu trusty main
deb-src http://ppa.launchpad.net/ubuntu-cloud-archive/$OPENSTACK_RELEASE/ubuntu trusty main
DEB
}

# precise needs the cloud archive, and so does trusty for non-Icehouse releases
if grep -qs DISTRIB_CODENAME=precise /etc/lsb-release ||
        [ "$OPENSTACK_RELEASE" != "icehouse" ]; then
    if [[ "$OPENSTACK_RELEASE" =~ staging ]]; then
        echo "Enabling the Ubuntu cloud staging ppa."
        ubuntu_cloud_staging
    else
        echo "Enabling the Ubuntu cloud archive."
        ubuntu_cloud_archive
    fi

    # Get index files only for ubuntu-cloud repo but keep standard lists
    src_list=cloudarchive-$OPENSTACK_RELEASE.list
    if [ -f "/etc/apt/sources.list.d/$src_list" ]; then
        sudo apt-get update \
            -o Dir::Etc::sourcelist="sources.list.d/$src_list" \
            -o Dir::Etc::sourceparts="-" -o APT::Get::List-Cleanup="0"
    else
        echo "ERROR: apt source not found: /etc/apt/sources.list.d/$src_list"
        exit 1
    fi
fi