#!/bin/bash

set -e

source common.sh

if [ -z "$1" ]; then
  echo "ERROR: Configuration file not passed"
  echo "Please use ./deploy-infra.sh PATH/TO/CONFIG to run this script"
  exit 1
else
  echo "Using $1 file for config"
  source $1
fi

if [ $ACCOUNT_ID == None ]; then
    echo "Couldn't find Account ID, please check if AWS Profile $CLI_PROFILE is set"
    exit 1
fi

if [ ! -f "$KEY_PATH" ]; then
    echo "Key file not found at $KEY_PATH"
    exit 1
fi

# Deploy the CloudFormation template
echo -e "\n\n=========== Deploying main.yml ==========="
aws cloudformation deploy \
  --region $REGION \
  --profile $CLI_PROFILE \
  --stack-name $NEW_STACK_NAME \
  --template-file cloudformation/infrastructure.yml \
  --no-fail-on-empty-changeset \
  --capabilities CAPABILITY_NAMED_IAM \
  --parameter-overrides \
    EC2InstanceType=$DEFAULT_EC2_INSTANCE_TYPE \
    ValidatorEC2InstanceType=$VALIDATOR_EC2_INSTANCE_TYPE \
    RPCEC2InstanceType=$RPC_EC2_INSTANCE_TYPE \
    BuildEC2InstanceType=$BUILD_EC2_INSTANCE_TYPE \
    KeyName=$AWS_KEY_PAIR_NAME \
    EC2AMI=$EC2_AMI_ID \
    NumberOfValidators=$NUMBER_OF_VALIDATORS \
    VolumeSize=$VOLUME_SIZE \
    RPCVolumeSize=$RPC_VOLUME_SIZE

# If the deploy succeeded, get the IP, create inventory and configure the created instances
if [ $? -eq 0 ]; then
  # Install additional Ansible roles from requirements
  ansible-galaxy install -r requirements.yml

  ASG=$(get_aws_export $NEW_STACK_NAME "AutoScalingGroup")

  VALIDATORS=""

  INSTANCES=$(aws autoscaling describe-auto-scaling-instances --profile $CLI_PROFILE \
    --query "AutoScalingInstances[?AutoScalingGroupName=='${ASG}'].InstanceId" --output text);

  for ID in $INSTANCES
  do
    IP=$(aws ec2 describe-instances --instance-ids $ID --query "Reservations[].Instances[].PublicIpAddress" --profile $CLI_PROFILE --output text)
    VALIDATORS+="$IP\n"
  done

  RPC_NODES=$(get_aws_export $NEW_STACK_NAME "RPCPublicIp")

  BUILD_SERVER=$(get_aws_export $NEW_STACK_NAME "BuildPublicIp")

  BUILD_INSTANCE_ID=$(get_aws_export $NEW_STACK_NAME "BuildInstanceId")

  mkdir -p $DATA_PATH

  echo -e "[build]\n$BUILD_SERVER\n\n[validators]\n$VALIDATORS\n[rpc]\n$RPC_NODES" > $INVENTORY_PATH

  # Build binaries if AMI not specified or a custom proposals parameter is passed
  if [ -z "$EC2_AMI_ID" ]
  then
    echo -e "\n\n=========== Compile joystream-node on build server ==========="
    ansible-playbook -i $INVENTORY_PATH --private-key $KEY_PATH build-code.yml \
      --extra-vars "branch_name=$BRANCH_NAME git_repo=$GIT_REPO build_local_code=$BUILD_LOCAL_CODE
                    data_path=$DATA_PATH runtime_profile=$RUNTIME_PROFILE"
  fi

  if [ -z "$EC2_AMI_ID" ]
  then
    echo -e "\n\n=========== Install additional utils on build server ==========="
    ansible-playbook -i $INVENTORY_PATH --private-key $KEY_PATH setup-build-server.yml
  fi

  echo -e "\n\n=========== Configure and start new validators and rpc node ==========="
  ansible-playbook -i $INVENTORY_PATH --private-key $KEY_PATH configure-network.yml \
    --extra-vars "local_dir=$LOCAL_CODE_PATH network_suffix=$NETWORK_SUFFIX
                  data_path=$DATA_PATH number_of_validators=$NUMBER_OF_VALIDATORS
                  deployment_type=$DEPLOYMENT_TYPE
                  initial_balances_file=$INITIAL_BALANCES_PATH
                  initial_members_file=$INITIAL_MEMBERS_PATH
                  skip_chain_setup=$SKIP_CHAIN_SETUP"

  echo -e "\n\n=========== Delete Build instance ==========="
  DELETE_RESULT=$(aws ec2 terminate-instances --instance-ids $BUILD_INSTANCE_ID --profile $CLI_PROFILE)
  echo $DELETE_RESULT

fi