Give the option of specifying customer ids to upload to

Author: chewy-zlai

distribution/build_and_upload_artifacts.sh

@@ -6,6 +6,8 @@ function print_usage() {
     echo "  --all       Build and upload all artifacts (GCP and AWS)"
     echo "  --gcp       Build and upload only GCP artifacts"
     echo "  --aws       Build and upload only AWS artifacts"
+    echo "  --aws_customer_ids <customer_id>  Specify AWS customer IDs to upload artifacts to."
+    echo "  --gcp_customer_ids <customer_id>  Specify GCP customer IDs to upload artifacts to."
     echo "  -h, --help  Show this help message"
 }
 
@@ -34,6 +36,24 @@ while [[ $# -gt 0 ]]; do
             print_usage
             exit 0
             ;;
+        --aws_customer_ids)
+            if [[ -z $2 ]]; then
+                echo "Error: --customer_ids requires a value"
+                print_usage
+                exit 1
+            fi
+            INPUT_AWS_CUSTOMER_IDS=("$2")
+            shift 2
+            ;;
+        --gcp_customer_ids)
+            if [[ -z $2 ]]; then
+                echo "Error: --customer_ids requires a value"
+                print_usage
+                exit 1
+            fi
+            INPUT_GCP_CUSTOMER_IDS=("$2")
+            shift 2
+            ;;
         *)
             echo "Unknown option: $1"
             print_usage
@@ -213,22 +233,24 @@ function upload_to_aws() {
   done
 }
 
-# check if $1 (single customer id mode) has been set
-if [ -z "$1" ]; then
-  if [ "$BUILD_AWS" = true ]; then
-    upload_to_aws "${AWS_CUSTOMER_IDS[@]}"
-  fi
-  if [ "$BUILD_GCP" = true ]; then
-    upload_to_gcp "${GCP_CUSTOMER_IDS[@]}"
-  fi
-else
-  if [ "$BUILD_AWS" = true ]; then
-    upload_to_aws "$1"
-  fi
-  if [ "$BUILD_GCP" = true ]; then
-    upload_to_gcp "$1"
+
+if [ "$BUILD_AWS" = true ]; then
+  if [ -z "$INPUT_AWS_CUSTOMER_IDS" ]; then
+      echo "No customer ids provided for AWS. Using default: ${AWS_CUSTOMER_IDS[*]}"
+    else
+      AWS_CUSTOMER_IDS=("${INPUT_AWS_CUSTOMER_IDS[@]}")
+    fi
+  upload_to_aws "${AWS_CUSTOMER_IDS[@]}"
+fi
+if [ "$BUILD_GCP" = true ]; then
+  if [ -z "$INPUT_GCP_CUSTOMER_IDS" ]; then
+    echo "No customer ids provided for GCP. Using default: ${GCP_CUSTOMER_IDS[*]}"
+  else
+    GCP_CUSTOMER_IDS=("${INPUT_GCP_CUSTOMER_IDS[@]}")
   fi
+  upload_to_gcp "${GCP_CUSTOMER_IDS[@]}"
 fi
 
+
 # Cleanup wheel stuff
 rm ./*.whl