#!/usr/bin/env bash
#
# Copyright (c) Cognitect, Inc.
# All rights reserved.

set -e

. /opt/datomic.app.deploy/bin/export-environment

. $(dirname $0)/common-aws-functions
start_sec=$(/bin/date +%s.%N)
msg "Started $(basename $0) at $(/bin/date "+%F %T")"

# get this instance's ID
INSTANCE_ID=$(get_instance_id)
if [ $? != 0 -o -z "${INSTANCE_ID}" ]; then
    error_exit "Unable to get this instance's ID; cannot continue."
fi

msg "Running AWS CLI with region: ${INSTANCE_ID}"

# Usage: register <EC2 instance ID> <Target Group ARN>
#
#   Registers <EC2 instance ID> with <Target Group ARN>
register() {
    local instance_id=$1
    local target_group_arn=$2

    if [[ ! -z "${target_group_arn}" ]]; then
        msg "Registering $instance_id with $target_group_arn"
        register_instance $instance_id $target_group_arn
        if [ $? != 0 ]; then
            error_exit "Failed to register instance"
        else
            msg "Instance is registering"
        fi
    fi
}

register $INSTANCE_ID $DATOMIC_TARGET_GROUP_ARN
register $INSTANCE_ID $DATOMIC_INVOKE_TARGET_GROUP_ARN
register $INSTANCE_ID $DATOMIC_HTTP_DIRECT_TARGET_GROUP_ARN

finish_msg
exit 0