#!/bin/bash -ex
#
# run.alestic.com/apt/rightscale
#
# Convert /etc/apt/sources.list to use RightScale archive mirror with multiverse
#
# Pass in YYYY/MM/DD to use snapshot of the mirror as of a fixed date
#

date=$1
if [ "$date" = "" ]; then
  path="ubuntu"
else
  path="ubuntu_daily/$date"
fi

availability_zone=$(wget -qO- http://169.254.169.254/2009-04-04/meta-data/placement/availability-zone/)
if [ "$availability_zone" = "us-west-1" ]; then
  suffixes='1 2'
elif [ "$availability_zone" = "eu-west-1" ]; then
  suffixes=''
else
  suffixes='1 2 3'
fi

codename=$(lsb_release -cs)
sudo mv /etc/apt/sources.list /etc/apt/sources.list.$(date +%Y%m%d-%H%M)
for suffix in '' $suffixes; do
  cat <<EOF | sudo tee -a /etc/apt/sources.list
deb http://ec2-us-east-mirror$suffix.rightscale.com/$path $codename main restricted universe multiverse
deb http://ec2-us-east-mirror$suffix.rightscale.com/$path $codename-updates main restricted universe multiverse
EOF
done
  cat <<EOF | sudo tee -a /etc/apt/sources.list
deb http://security.ubuntu.com/ubuntu $codename-security main restricted universe multiverse
EOF
sudo apt-get update