#!/bin/bash

declare -a regions=("us-west-2" "us-east-1")
declare -a repos=(
  "third-party/busybox"
  "third-party/fatedier/frps"
  "third-party/fatedier/frpc"
  "third-party/prom/node-exporter"
  "third-party/prom/prometheus"
  "third-party/victoriametrics/vmagent"
  "third-party/timberio/vector"
  "third-party/jetstack/cert-manager-controller"
  "third-party/jetstack/cert-manager-webhook"
  "third-party/jetstack/cert-manager-cainjector"
  "third-party/elastic/filebeat-oss-tidb-module"
  "third-party/sig-storage/csi-attacher"
  "third-party/sig-storage/csi-snapshotter"
  "third-party/kube-state-metrics/kube-state-metrics"
  "third-party/sig-storage/livenessprobe"
  "third-party/sig-storage/csi-node-driver-registrar"
  "third-party/sig-storage/csi-provisioner"
  "third-party/sig-storage/csi-resizer"
  "third-party/ebs-csi-driver/aws-ebs-csi-driver"
  "third-party/amazon/aws-load-balancer-controller"
  "third-party/grafana/grafana"
  "third-party/autoscaling/cluster-autoscaler"
  "tidbcloud/gotty"
  "tidbcloud/mysqlclient"
  "tidbcloud/aws-pca-issuer"
  "tidbcloud/audit"
  "tidbcloud/advanced-statefulset"
  "tidbcloud/sli-reporter"
  "tidbcloud/cluster-proxy"
  "tidbcloud/dataflow-agent"
  "tidbcloud/pd"
  "tidbcloud/tikv"
  "tidbcloud/tidb"
  "tidbcloud/tiflash"
  "tidbcloud/tidb-lightning"
  "tidbcloud/ticdc"
  "tidbcloud/br"
  "tidbcloud/tidb-monitor-initializer"
  "tidbcloud/tidb-operator"
  "tidbcloud/tidb-backup-manager"
  "tidbcloud/tidb-monitor-reloader"
  "tidbcloud/serverless/tikv"
  "tidbcloud/serverless/tiflash"
  "tidbcloud/serverless/tidb"
  "tidbcloud/serverless/pd"
  "tidbcloud/vector"
)

export AWS_PROFILE=dbaas-dev

for region in "${regions[@]}"; do
  for repo in "${repos[@]}"; do
    aws ecr set-repository-policy --repository-name "$repo" --policy-text file://dev-ecr-repo-policy.json --region "$region" --force --no-cli-pager || echo "Failed to set policy for $repo in $region"
    sleep 0.2
  done
done