#!/usr/bin/env bash
set -eou pipefail

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

timeout=${JOB_TIMEOUT:=300} # 5min
interval=${JOB_CHECK_INTERVAL:=10}

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

argocd_repo=tidbcloud/app-delivery
region=$(yq '.region' "${local_config}")
export AWS_REGION=$region
logicalRegionName=$(yq '.name' "${local_config}")
env=$(yq '.env' "${local_config}")
cloud=$(yq '.cloudProvider' "${local_config}")
cluster=$(yq '.k8s.name' "${local_config}")
jobId=$(yq '.jobId' "${local_config}")
remote_output=$(yq '.outputPath' "${local_config}")
image_registry=$(yq '.imageRegistry' "${local_config}")
controlplane_account=$(yq '.aws.controlplaneAccount' "${local_config}")
tidb_image_registry=$(yq ".jobOptions.dataplaneManager.tidbImageRegistry // \"${image_registry}\"" "${local_config}")
is_byoc=$(yq '.jobOptions.isByoc // "false"' "${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
dataplaneManagerRoleArn=$(yq '.dataplane_manager.roleArn' "${local_iam_config}")

additionalArgs=$(cat <<EOF
{
  "isByoc": "$is_byoc",
  "roleARN": "$dataplaneManagerRoleArn",
  "imageRegistry": "$image_registry",
  "tidbImageRegistry": "$tidb_image_registry",
  "fullManagerServiceRole": "arn:aws:iam::${controlplane_account}:role/full-manager-service-role"
}
EOF
)
echo "additionalArgs: $additionalArgs"

# Deploy application services by invoking deploy-new-region action in $argocd_repo
common_params="-f cloud=${cloud} -f region=${region} -f env=${env} -f cluster=${cluster} -f jobId=${jobId} -f logicalRegionName=${logicalRegionName}"

export GH_REPO="${argocd_repo}"
gh workflow run deploy-new-region-next-gen-o11y-agent-control-server.yaml ${common_params} -f additionalArgs="${additionalArgs}"

# Wait services ready
declare -A services
services["deploy/o11y-agent-control-server"]="false"
namespace="observability"
wait_for_services_ready services namespace
if [[ $? -eq 1 ]]; then
    echo "Services are not ready"
    exit 1
fi