die(){
  echo "error: $1" >&2
  exit 1
}

set -e

export HOME=/root
cd "$HOME"

stop rsyslog

echo "configuring hostname from cloudformation and aws metadata service"
configured_hostname="${CFN_INSTANCE_HOSTNAME}"
configured_hostname_short=$(echo "${configured_hostname}" | cut -d'.' -f 1)
private_hostname=$(curl -sL http://169.254.169.254/latest/meta-data/local-hostname)
private_hostname_short=$(echo "${private_hostname}" | cut -d'.' -f 1)
private_ipv4=$(curl -fsL http://169.254.169.254/latest/meta-data/local-ipv4)
public_hostname=$(curl -fsL http://169.254.169.254/latest/meta-data/public-hostname || :)
public_hostname_short=$(echo "${public_hostname}" | cut -d'.' -f 1)
public_ipv4=$(curl -fsL http://169.254.169.254/latest/meta-data/public-ipv4 || :)

echo "${configured_hostname_short}" > /etc/hostname
hostname -b -F /etc/hostname
sed -i '/127\.0\./d' /etc/hosts
cat >>/etc/hosts <<EOF
127.0.0.1 localhost
${private_ipv4} ${configured_hostname} ${configured_hostname_short}
127.0.1.1 ${private_hostname} ${private_hostname_short}
EOF

if [ -n "${public_hostname}" ]; then
cat >>/etc/hosts <<EOF
127.0.2.1 ${public_hostname} ${public_hostname_short}
EOF
fi

rm -f /etc/rsyslog.d/21-cloudinit.conf

cat >/etc/rsyslog.d/91-receive-container-logs.conf <<EOF
module(load="imfile")
module(load="imtcp" MaxSessions="500")
module(load="imudp")

input(type="imtcp" port="514")
input(type="imudp" port="514")
EOF

if [ -n "$PAPERTRAIL_HOST" ]; then
curl -sL "https://papertrailapp.com/tools/papertrail-bundle.pem" > /etc/ssl/certs/papertrail.pem
cat >/etc/rsyslog.d/99-papertrail.conf <<EOF
\$PreserveFQDN on
\$DefaultNetstreamDriverCAFile /etc/ssl/certs/papertrail.pem
\$ActionSendStreamDriver gtls
\$ActionSendStreamDriverMode 1
\$ActionSendStreamDriverAuthMode x509/name
\$ActionSendStreamDriverPermittedPeer *.papertrailapp.com
*.*          @@${PAPERTRAIL_HOST}
EOF
fi

chown -R root:syslog /var/log/upstart

start rsyslog


provision_volume() {
  device="$1"; shift
  mntpt="$1"; shift

  mkfs.btrfs "${device}" 2>/dev/null || :

  mount "${device}" /mnt 2>/dev/null || :
  btrfs filesystem resize max /mnt 2>/dev/null || :
  umount /mnt 2>/dev/null || :

  mkdir -p "${mntpt}"
  mount -t btrfs -o "rw,noatime,space_cache" "${device}" "${mntpt}"
}

if [ -b "/dev/xvdc" ]; then
  provision_volume "/dev/xvdc" "/var/lib/docker"
fi

if [ -b "/dev/xvdd" ]; then
  provision_volume "/dev/xvdd" "/var/lib/docker/vfs/dir"
fi

if ! hash docker; then
  echo "installing docker..."
  echo 'deb http://get.docker.io/ubuntu docker main' > /etc/apt/sources.list.d/docker.list
  apt-key adv --keyserver 'hkp://pgp.mit.edu:80' --recv-keys '36A1D7869245C8950F966E92D8576A8BA88D21E9'
  apt-get update
  apt-get install -qy lxc-docker
  gpasswd -a ubuntu docker
fi

echo "logging into ${DOCKER_REGISTRY_URL}..."
cat >"$HOME/.dockercfg" <<EOF
{"${DOCKER_REGISTRY_URL}": {
  "auth": "${DOCKER_REGISTRY_AUTH}",
  "email": "${DOCKER_REGISTRY_EMAIL}"
}}
EOF