#!/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/config.yaml
download_blob "${remote_config}" "${local_config}"

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}")
aks_oidc_issuer=$(yq '.k8s.oidcIssuer' "${local_config}")
image_registry=$(yq '.imageRegistry' "${local_config}")
resource_group="$cluster-infra"
subscription_id=$(yq '.azure.controlplaneSubscriptionId' "${local_config}")
job_id=$(yq '.jobId' "${local_config}")

hosted_zone_name="observability-$env.pingcap.cloud"
if [ "$env" == "prod" ]; then
    hosted_zone_name="observability.tidbcloud.com"
fi
echo "hosted_zone_name: $hosted_zone_name"

remote_iam_config=$(yq '.jobOutput.basic-iam' "${local_config}")
remote_blob_config=$(yq '.jobOutput.basic-blobstorage' "${local_config}")
remote_addon_config=$(yq '.jobOutput.basic-addon' "${local_config}")
local_iam_config=/tmp/iam-config.yaml
local_blob_config=/tmp/blob-config.yaml
local_addon_config=/tmp/addon-config.yaml
download_blob ${remote_iam_config} ${local_iam_config}
download_blob ${remote_blob_config} ${local_blob_config}
download_blob ${remote_addon_config} ${local_addon_config}

regional_server_role_arn=$(yq '.iam.regionalServer.clientId' "${local_iam_config}")
diag_bucket_name=$(yq '.blobstorage.container.diag' "${local_blob_config}")

oidc_value=$(yq '.k8s.oidcIssuer' "${local_config}" )
zone=$(yq '.jobOptions.o11y.zone' "$local_config")

endpoint_service_id=$(yq '.addon.privateLinkServiceName' "${local_addon_config}")
waf_id=$(yq '.addon.rsWAFID' "${local_addon_config}")
nginx_read_ingress_class="nginx-read"
nginx_write_ingress_class="nginx-write"
storage_account_name=$(yq '.blobstorage.accountName' "${local_blob_config}")
storage_resource_group=$(yq '.blobstorage.resourceGroup' "${local_blob_config}")

args=$(cat <<EOF
{
  "regionalServerRoleARN": "$regional_server_role_arn",
  "diagBucketName": "$diag_bucket_name",
  "oidcValue": "$oidc_value",
  "zone": "$zone",
  "endpointServiceID": "$endpoint_service_id",
  "wafACLARN": "$waf_id",
  "nginxReadIngressClass": "$nginx_read_ingress_class",
  "nginxWriteIngressClass": "$nginx_write_ingress_class",
  "imageRegistry": "$image_registry",
  "hostedZoneName": "$hosted_zone_name",
  "storageAccountName": "$storage_account_name",
  "subscriptionID": "$subscription_id",
  "resourceGroup": "$resource_group",
  "storageResourceGroup": "$storage_resource_group"
}
EOF
)

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

gh workflow run regional-server-cd.yaml --repo ${o11y_repo} --ref ${git_ref} -f env=${env} -f region=${region} -f vendor=${vendor} -f git-ref=${git_ref}

# Wait services ready
declare -A ss
ss["deploy/regionalserver -n observability"]="false"
ss["deploy/data-server -n observability"]="false"

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