#!/bin/bash

workflow="deploy-new-region-cluster-autoscaler.yaml"
argocd_repo=tidbcloud/app-delivery

# get input
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

region=$(yq '.region' "${local_config}")
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}")
output=$(yq '.outputPath' "${local_config}")

# get role
reoute_iam_output=$(yq -e '.jobOutput.iam' $local_config)
local_iam_output=/tmp/iam-output.yaml
aws s3 cp $reoute_iam_output $local_iam_output --region=$output_bucket_region
cluster_autoscaler_role_arn=$(yq -e '.clusterAutoscaler.roleArn' $local_iam_output)

common_params="-f env=${env} -f region=${region} -f cluster=${cluster} -f jobId=${jobId} -f logicalRegionName=${logicalRegionName}"
gh workflow -R $argocd_repo run $workflow -f role-arn=${cluster_autoscaler_role_arn} $common_params

# wait services come ready
declare -A services
services["cluster-autoscaler"]="false"
ns=kube-system
timeout=${JOB_TIMEOUT:=300} # 5min
interval=${JOB_CHECK_INTERVAL:=10}
start_time=$(date +%s)
echo "Waiting all services ready..."
while [[ $(($(date +%s) - $start_time)) -lt $timeout ]]; do
  all_ready=true
  for svc in "${!services[@]}"; do
    if [[ ${services[$svc]} == "true" ]]; then
      continue
    else
      all_ready=false
    fi
    ready=$(kubectl -n "${ns}" get deploy "${svc}" -ojsonpath='{.status.readyReplicas}' 2>/dev/null)
    if [[ $? -eq 0 ]]; then
      if [[ $ready -ge 1 ]]; then
        services[$svc]="true"
        echo "Service $svc is ready"
      fi
    fi
  done

  if $all_ready; then
    echo "All services ready!"
    exit 0
  else
    sleep "${interval}"
  fi
done
# Report unready services
echo "Following services are unready:"
for svc in "${!services[@]}"; do
  if [[ ${services[$svc]} == "false" ]]; then
    echo $svc
  fi
done
exit 1