#!/bin/bash
set -e

echo "-------------------------------------------------------------------------"
echo "Hosting Account Setup"
echo "-------------------------------------------------------------------------"

# doing this at ~
cwd=$(pwd)
HOSTINGACCOUNT_ARG=${1}
HOSTINGACCOUNT_ARG_LENGTH=${#1}

# Get directory of this script
# https://stackoverflow.com/questions/59895/how-can-i-get-the-source-directory-of-a-bash-script-from-within-the-script-itsel
pushd . > /dev/null
SCRIPT_PATH="${BASH_SOURCE[0]}"
if ([ -h "${SCRIPT_PATH}" ]); then
  while([ -h "${SCRIPT_PATH}" ]); do cd `dirname "$SCRIPT_PATH"`;
  SCRIPT_PATH=`readlink "${SCRIPT_PATH}"`; done
fi
cd `dirname ${SCRIPT_PATH}` > /dev/null
SCRIPT_PATH=`pwd`;
popd  > /dev/null

# Ensure STAGE_NAME environment variable set
if [ -z "$STAGE_NAME" ]; then
  echo "------ERROR: STAGE_NAME environment variable has to be set-------------"
  exit 1
else
  if [[ "$HOSTINGACCOUNT_ARG_LENGTH" != "12" ]]; then
    echo "------ERROR: Argument ACCOUNT_ID should have 12 digits-------------\n"
    exit 1
  fi
fi

# Locate installation directory of Service Workbench
# TODO Allow user to specify SWB install dir as an envt var
for f in ~ ~/environment; do
  if [ -d ${f}/service-workbench-on-aws ]; then
    echo "SWB found at $f"
    SWB_PATH="${f}/service-workbench-on-aws"
  fi
done
if [ -z $SWB_PATH ]; then
  echo "Cannot find Service Workbench installed at ~ or ~/environment"
  exit 1
fi

# This will get the serverless stack variables and 
# export to CFN_PARAM_* variables
# a `hosting-account-env-vars` file is created and 
# then sourced
echo ">> 1/4 - Copying Loading Env Vars"
cd ${SWB_PATH}/main/solution/backend
BACKEND_STACK=$(pnpx sls info --verbose --stage $STAGE_NAME | grep 'stack: ' | sed 's/stack\: //g')
BACKEND_STACK_INFO=$(aws cloudformation describe-stacks --stack-name $BACKEND_STACK)
ENV_VARS=$(echo $BACKEND_STACK_INFO | jq '.Stacks[].Outputs[] | {ParameterKey: .OutputKey, ParameterValue: .OutputValue} | select(.OutputKey != "ServiceEndpoint")  | flatten | "export CFN_PARAM_\(.[0])=\(.[1])"' | sed "s/\"//g; s/\=/\=\"/g; s/$/\";/")
echo $ENV_VARS > ${SCRIPT_PATH}/hosting-account-env-vars;
source ${SCRIPT_PATH}/hosting-account-env-vars

echo ">> 2/4 - Preparing CFn Args"
CFN_FILE_PARAM=${SCRIPT_PATH}/hosting-account-cfn-args-$STAGE_NAME.json
cp ${SCRIPT_PATH}/example-hosting-account-params.json $CFN_FILE_PARAM

# Namespace
sed -i "s/CFN_PARAM_StageName/$STAGE_NAME/g" $CFN_FILE_PARAM

# CentralAccountId == Same account for the Quick Start
#SAME_ACCOUNT_ID=$(echo $CFN_PARAM_WorkflowLoopRunnerRoleArn | sed "s/arn:aws:iam:://g; s/:role\/$STAGE_NAME-va-sw-WorkflowLoopRunner//g")
sed -i "s/CFN_PARAM_CentralAccountId/$HOSTINGACCOUNT_ARG/g" $CFN_FILE_PARAM

# ApiHandlerRoleArn (using | as separator as Arn contains slash)
sed -i "s|CFN_PARAM_ApiHandlerRoleArn|$CFN_PARAM_ApiHandlerRoleArn|g" $CFN_FILE_PARAM
sed -i "s|ApiHandlerRoleArn|ApiHandlerArn|g" $CFN_FILE_PARAM

# WorkflowLoopRunnerRoleArn
sed -i "s|CFN_PARAM_WorkflowLoopRunnerRoleArn|$CFN_PARAM_WorkflowLoopRunnerRoleArn|g" $CFN_FILE_PARAM
sed -i "s|WorkflowLoopRunnerRoleArn|WorkflowRoleArn|g" $CFN_FILE_PARAM


STACK_NAME=swb-hosting-compute-$HOSTINGACCOUNT_ARG-$STAGE_NAME-stack
echo ">> 3/4 - Creating stack: ${STACK_NAME}"
STACK_ID=$(aws cloudformation create-stack \
    --stack-name $STACK_NAME \
    --template-body file://${SWB_PATH}/addons/addon-base-raas/packages/base-raas-cfn-templates/src/templates/onboard-account.cfn.yml \
    --parameters file://$CFN_FILE_PARAM \
    --capabilities CAPABILITY_NAMED_IAM | jq '. | flatten | (.[0])' | sed "s/\"//g;")
    
echo ">> 4/4 - Waiting for creation to complete ${STACK_ID} ..."
aws cloudformation wait stack-create-complete --stack-name $STACK_NAME    
# if [[ $? -eq 0 ]]; then
#     # Wait for create-stack to finish
#     echo  "Waiting for create-stack command to complete"
#     CREATE_STACK_STATUS=$(aws cloudformation describe-stacks --stack-name ${STACK_NAME} --query 'Stacks[0].StackStatus' --output text)
#     while [[ $CREATE_STACK_STATUS == "REVIEW_IN_PROGRESS" ]] || [[ $CREATE_STACK_STATUS == "CREATE_IN_PROGRESS" ]]
#     do
#         # Wait 30 seconds and then check stack status again
#         sleep 30
#         CREATE_STACK_STATUS=$(aws --region cloudformation describe-stacks --stack-name ${STACK_NAME} --query 'Stacks[0].StackStatus' --output text)
#     done
# fi

echo ">> Stack Output:"
aws cloudformation describe-stacks --stack-name $STACK_NAME --query "Stacks[0].Outputs[].{Key: OutputKey, Value: OutputValue}" --output table 

# getting back to the working dir
cd $cwd