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

set -e

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

cd $(dirname $(realpath $0))/..

. $(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: deregister <EC2 instance ID> <Target Group ARN>
#
#   Deregisters <EC2 instance ID> from <Target Group ARN>
deregister() {
    local instance_id=$1
    local target_group_arn=$2

    if [[ ! -z "${target_group_arn}" ]]; then
        msg "Deregistering $instance_id from $target_group_arn"
        deregister_instance $instance_id $target_group_arn
        if [ $? != 0 ]; then
            error_exit "Failed to deregister instance"
        else
            msg "Instance is deregistering"
        fi
    fi
}

deregister $INSTANCE_ID $DATOMIC_TARGET_GROUP_ARN
deregister $INSTANCE_ID $DATOMIC_INVOKE_TARGET_GROUP_ARN
deregister $INSTANCE_ID $DATOMIC_HTTP_DIRECT_TARGET_GROUP_ARN

finish_msg
exit 0