_exit_handler_reason=""
_exit_handler(){
  exit_code="$?"

  if [ "${exit_code}" -eq 0 ]; then
    echo "Stacco cloud-init handler finished"
    cfn-signal -s true "${AWS_INSTANCE_WAIT_HANDLE}"
  else
    sleep 1000
    echo "Stacco cloud-init handler aborted (${exit_code}: ${_exit_handler_reason})"
    cfn-signal -e "${exit_code}" -r "${_exit_handler_reason}" "${AWS_INSTANCE_WAIT_HANDLE}"
  fi

  exit "${exit_code}"
}
trap _exit_handler EXIT
set -e

export HOME=/root
cd "$HOME"

echo "configuring hostname from aws metadata service"
host_title="${AWS_STACK_NAME}-$(echo "${AWS_INSTANCE_LOGICAL_NAME}" | sed 's/LaunchConfiguration$//')"
private_hostname=$(curl -sL http://169.254.169.254/latest/meta-data/local-hostname)
short_private_hostname=$(echo "${private_hostname}" | cut -d'.' -f 1)
echo "${host_title}" > /etc/hostname
hostname -b -F /etc/hostname
sed -i '/127\.0\./d' /etc/hosts
cat >>/etc/hosts <<EOF
127.0.0.1 localhost
127.0.1.1 ${private_hostname} ${host_title} ${short_private_hostname}
EOF


if [ -n "$LOGENTRIES_ACCOUNT_KEY" ]; then
  le clean
  rm -f /etc/default/logentries_not_to_be_run

  echo "starting remote logging"
  host_key=$(docker-utils get-host-uuid "${AWS_INSTANCE_LOGICAL_NAME}.${AWS_STACK_NAME}")

  le init --account-key="${LOGENTRIES_ACCOUNT_KEY}" --agent-key="${host_key}"
  le register --force --name="${host_title}" --hostname="${private_hostname}"
  service logentries start

  logs_to_follow=(syslog messages dmesg auth.log boot.log daemon.log dpkg.log kern.log cron secure faillog cloud-init-output.log)
  for log_name in "${logs_to_follow[@]}"; do
    log_path="/var/log/${log_name}"
    if [ -f "${log_path}" ]; then
      log_name=$(basename "${log_path}" '.log')
      le follow --name="${log_name}" "${log_path}"
    fi
  done

  service logentries restart
fi