#!/usr/bin/env bash

set -euo pipefail

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

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

o11y_repo=tidbcloud/observability
git_ref="main"
region=$(yq '.region' "${local_config}")
logical_region_name=$(yq '.name' "${local_config}")
env=$(yq '.env' "${local_config}")
vendor=$(yq '.cloudProvider' "${local_config}")
cluster=$(yq '.k8s.name' "${local_config}")
jobId=$(yq '.jobId' "${local_config}")
output=$(yq '.outputPath' "${local_config}")
account=$(yq '.aws.controlplaneAccount' "${local_config}")
imageRegistry=$(yq '.imageRegistry' "${local_config}")

o11y_role_arn="arn:aws:iam::${account}:role/regional-server-${cluster}"

args=$(cat <<EOF
{
  "roleARN": "${o11y_role_arn}",
  "imageRegistry": "${imageRegistry}",
  "account": "${account}"
}
EOF
)
echo "args: $args"

# Create pull request by invoking new-region workflow.
common_params="-f vendor=${vendor} -f region=${region} -f logicalRegionName=${logical_region_name} -f env=${env} -f cluster=${cluster} -f jobId=${jobId}"
gh workflow run new-region-drhouse.yaml --repo ${o11y_repo} --ref ${git_ref} ${common_params} -f args="${args}"
wait_workflow_run_to_merge new-region-drhouse.yaml ${o11y_repo}
if [[ $? -eq 1 ]]; then
    echo "Workflow failed"
    exit 1
fi

# Deploy application services by invoking drhouse-cd workflow
gh workflow run drhouse-cd.yaml --repo ${o11y_repo} --ref ${git_ref} -f env=${env} -f region=${logical_region_name} -f vendor=${vendor} -f git-ref=${git_ref}

# Wait services ready
declare -A ss
ss["sts/diagnosis-query -n observability"]="false"
ss["sts/o11ydiagnosis-deltalake -n observability"]="false"

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