#!/bin/bash

######################################################################
# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. #
# SPDX-License-Identifier: MIT-0                                     #
######################################################################

source ./fun.sh

echo ""
echo "Configuring EC2 Instance Template ..."

EC2_TEMPLATE_CONFIG=./bootstrap/ec2-template-config.json
EC2_TEMPLATE=./bootstrap/launch-template-${EC2_TEMPLATE_NAME}.json
EC2_TEMPLATE_TMP=./bootstrap/launch-template-${EC2_TEMPLATE_NAME}.json.tmp
cp -f $EC2_TEMPLATE_CONFIG $EC2_TEMPLATE

echo ""
echo "Updating template ${EC2_TEMPLATE} based on ec2.conf ..."

# LaunchTemplateName
echo ""
echo "LaunchTemplateName ..."
sed -i.tmp "s/\"LaunchTemplateName\": \"\"/\"LaunchTemplateName\": \"$EC2_TEMPLATE_NAME\"/g" $EC2_TEMPLATE

# SubnetId 
echo ""
echo "SubnetId ..."
#EC2_SUBNET_ID=$(subnet)
if [ "$EC2_SUBNET_ID" == "" ]; then
	sed -i.tmp "/\"SubnetId\": \"\"/d" $EC2_TEMPLATE
else
	sed -i.tmp "s/\"SubnetId\": \"\"/\"SubnetId\": \"$EC2_SUBNET_ID\"/g" $EC2_TEMPLATE
fi

# ImageId
echo ""
echo "ImageId ..."
sed -i.tmp "s/\"ImageId\": \"\"/\"ImageId\": \"$EC2_IMAGE_ID\"/g" $EC2_TEMPLATE

# VolumeSize
echo ""
echo "VolumeSize ..."
sed -i.tmp "s/\"VolumeSize\": 0/\"VolumeSize\": $EC2_VOLUME_SIZE_GB/g" $EC2_TEMPLATE

# DeviceName
echo ""
echo "DeviceName ..."
sed -i.tmp "s#\"DeviceName\": \"/dev/sda1\"#\"DeviceName\": \"${EC2_DEVICE_NAME}\"#g" $EC2_TEMPLATE

# InstanceType
echo ""
echo "InstanceType ..."
cat $EC2_TEMPLATE | jq --arg EC2_INSTANCE_TYPE "$EC2_INSTANCE_TYPE" '.LaunchTemplateData.InstanceType = $EC2_INSTANCE_TYPE' > $EC2_TEMPLATE_TMP
cp -f $EC2_TEMPLATE_TMP $EC2_TEMPLATE

# IamInstanceProfile.Name
echo ""
echo "IamInstanceProfile.Name"
instance_profile
cat $EC2_TEMPLATE | jq --arg EC2_INSTANCE_PROFILE_NAME "$EC2_INSTANCE_PROFILE_NAME" '.LaunchTemplateData.IamInstanceProfile.Name = $EC2_INSTANCE_PROFILE_NAME' > $EC2_TEMPLATE_TMP
cp -f $EC2_TEMPLATE_TMP $EC2_TEMPLATE

#sed  -i.tmp "s/\"Key\": \"\"/\"Key\": \"Name\"/g" $EC2_TEMPLATE
#sed  -i.tmp "s/\"Value\": \"\"/\"Value\": \"$EC2_INSTANCE_NAME\"/g" $EC2_TEMPLATE

# KeyName
echo ""
echo "KeyName ..."
keypair
sed  -i.tmp "s/\"KeyName\": \"\"/\"KeyName\": \"$EC2_KEY_NAME\"/g" $EC2_TEMPLATE

# AssociatePublicIpAddress
echo ""
echo "AssociatePublicIpAddress ..."
sed  -i.tmp "s/\"AssociatePublicIpAddress\": true/\"AssociatePublicIpAddress\": $EC2_ASSIGN_PUBLIC_IP/g" $EC2_TEMPLATE

# SecurityGroup
echo ""
echo "Security group ..."
EC2_SG_ID=$(security_group quiet)
cat $EC2_TEMPLATE | jq --arg EC2_SG_ID "$EC2_SG_ID" '.LaunchTemplateData.NetworkInterfaces[].Groups[0] = $EC2_SG_ID' > $EC2_TEMPLATE_TMP
cp -f $EC2_TEMPLATE_TMP $EC2_TEMPLATE

rm -f $EC2_TEMPLATE_TMP

# Display config
echo ""
echo "Generated instance template:"
echo ""
cat $EC2_TEMPLATE | jq

echo ""
echo "Please review the template above, and if needed modify it in file $EC2_TEMPLATE"
echo "Once satisfied, you can run ./template-create.sh"
echo "and ./template-launch.sh [template_name] to launch an instance from this template."
echo ""