#!/bin/bash

# Copyright (c) Contributors to the Open 3D Engine Project.
# For complete copyright and license terms please see the LICENSE at the root of this distribution.
#
# SPDX-License-Identifier: Apache-2.0 OR MIT
#

# Deploy the CDK applications for AWS gems (Linux only)

SOURCE_DIRECTORY=$(dirname "$0")
PATH=$SOURCE_DIRECTORY/python:$PATH
GEM_DIRECTORY=$SOURCE_DIRECTORY/Gems

DeployCDKApplication()
{
    # Deploy the CDK application for a specific AWS gem
    GEM_NAME=$1
    ADDITIONAL_ARGUMENTS=$2
    echo [cdk_deployment] Deploy the CDK application for the $GEM_NAME gem
    pushd $GEM_DIRECTORY/$GEM_NAME/cdk

    # Revert the CDK application code to a stable state using the provided commit ID
    if ! git checkout $COMMIT_ID -- .;
    then
        echo [git_checkout] Failed to checkout the CDK application for the $GEM_NAME gem using commit ID $COMMIT_ID
        popd
        exit 1
    fi

    # Install required packages for the CDK application
    if ! python -m pip install -r requirements.txt;
    then
        echo [cdk_deployment] Failed to install required packages for the $GEM_NAME gem
        popd
        exit 1
    fi

    # Deploy the CDK application
    if ! cdk deploy $ADDITIONAL_ARGUMENTS --require-approval never;
    then
        echo [cdk_deployment] Failed to deploy the CDK application for the $GEM_NAME gem
        popd
        exit 1
    fi
    popd
}

# Create and activate a virtualenv for the CDK deployment
if ! python/python.sh -m venv .env;
then
    echo [cdk_bootstrap] Failed to create a virtualenv for the CDK deployment
    exit 1
fi
if ! source .env/bin/activate;
then
    echo [cdk_bootstrap] Failed to activate the virtualenv for the CDK deployment
    exit 1
fi

echo [cdk_installation] Install nvm $NVM_VERSION
curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/$NVM_VERSION/install.sh | bash
export NVM_DIR="$HOME/.nvm"
[ -s "$NVM_DIR/nvm.sh" ] && \. "$NVM_DIR/nvm.sh"  # This loads nvm
[ -s "$NVM_DIR/bash_completion" ] && \. "$NVM_DIR/bash_completion"  # This loads nvm bash_completion
echo [cdk_installation] Install the current version of nodejs
nvm install node

echo [cdk_installation] Install the latest version of CDK
if ! npm uninstall -g aws-cdk;
then
    echo [cdk_bootstrap] Failed to uninstall the current version of CDK
    exit 1
fi
if ! npm install -g aws-cdk@latest;
then
    echo [cdk_bootstrap] Failed to install the latest version of CDK
    exit 1
fi

# Set temporary AWS credentials from the assume role
credentials=$(aws sts assume-role --query Credentials.[SecretAccessKey,SessionToken,AccessKeyId] --output text --role-arn $ASSUME_ROLE_ARN --role-session-name o3de-Automation-session)
AWS_SECRET_ACCESS_KEY=$(echo "$credentials" | cut -d' ' -f1)
AWS_SESSION_TOKEN=$(echo "$credentials" | cut -d' ' -f2)
AWS_ACCESS_KEY_ID=$(echo "$credentials" | cut -d' ' -f3)

O3DE_AWS_DEPLOY_ACCOUNT=$(echo "$ASSUME_ROLE_ARN" | cut -d':' -f5)

# Bootstrap and deploy the CDK applications
echo [cdk_bootstrap] Bootstrap CDK
if ! cdk bootstrap aws://$O3DE_AWS_DEPLOY_ACCOUNT/$O3DE_AWS_DEPLOY_REGION;
then
    echo [cdk_bootstrap] Failed to bootstrap CDK
    exit 1
fi

if ! DeployCDKApplication AWSCore "-c disable_access_log=true -c remove_all_storage_on_destroy=true --all";
then
    exit 1
fi
if ! DeployCDKApplication AWSClientAuth;
then
    exit 1
fi
if ! DeployCDKApplication AWSMetrics "-c batch_processing=true";
then
    exit 1
fi

exit 0