#!/bin/bash

# set exit on error to true
set -e
# load .env
set -o allexport
source ./scripts/.env set

REGION=us-east-1
CURR_BRANCH=$(git branch --show-current)

function authenticate {
    account_number=$1
    role_name=$2
    profile_name=$3
    echo Loading account credentials for Account $account_number with Role: $role_name...
    ada cred update --profile="${profile_name}" --account="${account_number}" --role=${role_name} --provider=isengard --once
    aws configure set region $REGION --profile $profile_name
}

function triggerProjectBatch {
    account_number=$1
    role_name=$2
    profile_name=$3
    project_name=$4
    target_branch=$5
    authenticate $account_number $role_name $profile_name
    echo AWS Account: $account_number
    echo Project: $project_name 
    echo Target Branch: $target_branch
    RESULT=$(aws codebuild start-build-batch --profile="${profile_name}" --project-name $project_name --source-version=$target_branch \
     --environment-variables-override name=BRANCH_NAME,value=$target_branch,type=PLAINTEXT \
     --query 'buildBatch.id' --output text)
    echo "https://$REGION.console.aws.amazon.com/codesuite/codebuild/$account_number/projects/$project_name/batch/$RESULT?region=$REGION"
}

function triggerProject {
    account_number=$1
    role_name=$2
    profile_name=$3
    project_name=$4
    target_branch=$5
    authenticate $account_number $role_name $profile_name
    echo AWS Account: $account_number
    echo Project: $project_name 
    echo Target Branch: $target_branch
    RESULT=$(aws codebuild start-build --profile="${profile_name}" --project-name $project_name --source-version=$target_branch \
     --environment-variables-override name=BRANCH_NAME,value=$target_branch,type=PLAINTEXT \
     --query 'build.id' --output text)
    echo "https://$REGION.console.aws.amazon.com/codesuite/codebuild/$account_number/projects/$project_name/build/$RESULT?region=$REGION"
}

function cloudE2EBeta {
    echo Running Beta E2E Test Suite
    E2E_ROLE_NAME=CodebuildDeveloper
    E2E_PROFILE_NAME=AmplifyAPIE2EBeta
    E2E_PROJECT_NAME=amplify-codegen-e2e-workflow
    TARGET_BRANCH=$CURR_BRANCH
    triggerProjectBatch $E2E_ACCOUNT_BETA $E2E_ROLE_NAME $E2E_PROFILE_NAME $E2E_PROJECT_NAME $TARGET_BRANCH
}

function cloudE2E {
    echo Running Prod E2E Test Suite
    E2E_ROLE_NAME=CodebuildDeveloper
    E2E_PROFILE_NAME=AmplifyAPIE2EProd
    E2E_PROJECT_NAME=amplify-codegen-e2e-workflow
    TARGET_BRANCH=$CURR_BRANCH
    triggerProjectBatch $E2E_ACCOUNT_PROD $E2E_ROLE_NAME $E2E_PROFILE_NAME $E2E_PROJECT_NAME $TARGET_BRANCH
}

function cleanupStaleResourcesBeta {
  echo Running Beta E2E resource stale resource cleanup
  CLEANUP_ROLE_NAME=CodebuildDeveloper
  CLEANUP_PROFILE_NAME=AmplifyAPIE2EBeta
  CLEANUP_PROJECT_NAME=amplify-codegen-cleanup-workflow
  TARGET_BRANCH=$CURR_BRANCH
  triggerProject $E2E_ACCOUNT_BETA $CLEANUP_ROLE_NAME $CLEANUP_PROFILE_NAME $CLEANUP_PROJECT_NAME $TARGET_BRANCH
}

function cleanupStaleResources {
  echo Running Prod E2E resource stale resource cleanup
  CLEANUP_ROLE_NAME=CodebuildDeveloper
  CLEANUP_PROFILE_NAME=AmplifyAPIE2EProd
  CLEANUP_PROJECT_NAME=amplify-codegen-cleanup-workflow
  TARGET_BRANCH=$CURR_BRANCH
  triggerProject $E2E_ACCOUNT_PROD $CLEANUP_ROLE_NAME $CLEANUP_PROFILE_NAME $CLEANUP_PROJECT_NAME $TARGET_BRANCH
}

function triggerProjectBatchWithDebugSession {
  account_number=$1
  role_name=$2
  profile_name=$3
  project_name=$4
  target_branch=$5
  authenticate $account_number $role_name $profile_name
  echo AWS Account: $account_number
  echo Project: $project_name 
  echo Target Branch: $target_branch
  debug_spec=$(cat .codebuild/debug_workflow.yml)
  RESULT=$(aws codebuild start-build-batch --region=$REGION --profile="${profile_name}" --project-name $project_name --source-version=$target_branch \
    --debug-session-enabled \
    --buildspec-override "$debug_spec" \
    --environment-variables-override name=BRANCH_NAME,value=$target_branch,type=PLAINTEXT \
    --query 'buildBatch.id' --output text)
  echo "https://$REGION.console.aws.amazon.com/codesuite/codebuild/$account_number/projects/$project_name/batch/$RESULT?region=$REGION"
}

function cloudE2EDebug {
  echo Generating the debug E2E buildspec
  if [ $# -eq 0 ]; then
    echo "Please provide the batch build id of codebuild"
    exit 1
  fi
  generatedDebugSpecForFailedTests $1
  echo Running Prod E2E Test Suite
  E2E_ROLE_NAME=CodebuildDeveloper
  E2E_PROFILE_NAME=AmplifyAPIE2EProd
  E2E_PROJECT_NAME=amplify-codegen-e2e-workflow
  TARGET_BRANCH=$CURR_BRANCH
  triggerProjectBatchWithDebugSession $E2E_ACCOUNT_PROD $E2E_ROLE_NAME $E2E_PROFILE_NAME $E2E_PROJECT_NAME $TARGET_BRANCH
}

function generatedDebugSpecForFailedTests {
  # Get temporary access for the account
  E2E_ROLE_NAME=CodebuildDeveloper
  E2E_PROFILE_NAME=AmplifyAPIE2EProd
  authenticate $E2E_ACCOUNT_PROD $E2E_ROLE_NAME "$E2E_PROFILE_NAME"
  local batch_build_id=$1
  echo "Getting failed test suites"
  failed_tests=$(aws codebuild batch-get-build-batches --profile="$E2E_PROFILE_NAME" --ids "$batch_build_id" --region us-east-1 --query 'buildBatches[0].buildGroups' | jq -c '.[] | select(.currentBuildSummary.buildStatus == "FAILED").identifier')
  if [ -z "$failed_tests" ]; then
    echo "No failed tests found in batch $1"
    exit 0
  fi
  echo $failed_tests | xargs yarn ts-node ./scripts/split-e2e-tests.ts --debug
}

function authenticateWithE2EProfile {
    E2E_ROLE_NAME=CodebuildDeveloper
    E2E_PROFILE_NAME=AmplifyAPIE2EProd
    authenticate $E2E_ACCOUNT_PROD $E2E_ROLE_NAME $E2E_PROFILE_NAME
}