#!/usr/bin/env bash

remote_config=${CONFIG_PATH}
local_config=/tmp/base-config.yaml

bucket_name=${remote_config#s3://}
bucket_name=${bucket_name%%/*}
output_bucket_region=$(aws s3api get-bucket-location --bucket=$bucket_name --query="LocationConstraint" \
  --region=us-west-2 --output=text --no-cli-pager)
aws s3 cp "${remote_config}" "${local_config}" --region=$output_bucket_region

disabled=$(yq '.jobOptions.tcidm.disabled' "${local_config}")
if [[ $disabled == "true" ]]; then
  echo "job is disabled, exit"
  exit 0
fi

argocd_repo=tidbcloud/app-delivery
region=$(yq '.region' "${local_config}")
logicalRegionName=$(yq '.name' "${local_config}")
zones=$(yq -ocsv '.jobOptions.regionalServer.availabilityZones' "${local_config}") # us-west-2a,us-west-2b,us-west-2c
env=$(yq '.env' "${local_config}")
cloud=$(yq '.cloudProvider' "${local_config}")
cluster=$(yq '.k8s.name' "${local_config}")
jobId=$(yq '.jobId' "${local_config}")
imageRegistry=$(yq '.imageRegistry' "${local_config}")
awsAssumeRoleExternalId=$(yq -ojson '.jobOptions.tcidm.awsAssumeRoleExternalId // 123' "${local_config}")

remote_iam_config=$(yq '.jobOutput.iam' "${local_config}")
local_iam_config=/tmp/iam-config.yaml
aws s3 cp ${remote_iam_config} ${local_iam_config} --region=$output_bucket_region
dmRoleArn=$(yq '.tcidm.dmRoleArn' "${local_iam_config}")
importRoleArn=$(yq '.tcidm.importRoleArn' "${local_iam_config}")
dmAssumeRoleName=$(yq '.tcidm.dmAssumeRoleName' "${local_iam_config}")
importAssumeRoleName=$(yq '.tcidm.importAssumeRoleName' "${local_iam_config}")

# Deploy tcidm by invoking deploy-new-region action in $argocd_repo
common_params="-f provider=${cloud} -f region=${region} -f zones=${zones} -f env=${env} -f cluster=${cluster} -f jobId=${jobId} -f logicalRegionName=${logicalRegionName}"
export GH_REPO="${argocd_repo}"

additional_args=$(cat <<EOF
{
  "imageRegistry": "${imageRegistry}"
}
EOF
)
echo "additional args for tcidm-regional-service: ${additional_args}"
gh workflow run deploy-new-region-tcidm-regional-service.yaml ${common_params} -f additionalArgs="${additional_args}"

additional_args=$(cat <<EOF
{
  "imageRegistry": "${imageRegistry}",
  "awsRoleArn": "${dmRoleArn}",
  "awsAssumeRoleName": "${dmAssumeRoleName}",
  "awsAssumeRoleExternalId": "${awsAssumeRoleExternalId}",
  "cloud": "${cloud}"
}
EOF
)
echo "additional args for dm-controller: ${additional_args}"
gh workflow run deploy-new-region-dm-controller.yaml ${common_params} -f additionalArgs="${additional_args}"

additional_args=$(cat <<EOF
{
  "imageRegistry": "${imageRegistry}",
  "awsRoleArn": "${importRoleArn}",
  "awsAssumeRoleName": "${importAssumeRoleName}",
  "awsAssumeRoleExternalId": "${awsAssumeRoleExternalId}",
  "cloud": "${cloud}"
}
EOF
)
echo "additional args for import-controller: ${additional_args}"
gh workflow run deploy-new-region-import-controller.yaml ${common_params} -f additionalArgs="${additional_args}"

ROOT=$(cd $(dirname "${BASH_SOURCE[0]}") && pwd)
source "$ROOT/lib.sh"

declare -A services
services["deploy/regional-tcidm-service"]="false"
services["deploy/dm-controller"]="false"
services["deploy/import-controller"]="false"
namespace="tcidm"

wait_for_services_ready services namespace
if [[ $? -eq 1 ]]; then
    echo "Services are not ready"
    exit 1
fi