Allow specifying vaccine ref to use in GitLab

DataDog · Feb 12, 2025 · c7ff282 · c7ff282
1 parent 6b55226
commit c7ff282
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
@@ -180,4 +180,4 @@ vaccine:
   stage: vaccine
   needs: [create-multiarch-lib-injection-image]
   script: |
-    .gitlab/scripts/vaccine.sh
+    .gitlab/scripts/vaccine.sh master
diff --git a/.gitlab/scripts/vaccine.sh b/.gitlab/scripts/vaccine.sh
@@ -6,14 +6,16 @@ GH_VACCINE_PAT=$(vault kv get -field=vaccine-token kv/k8s/gitlab-runner/dd-trace
 REPO="Datadog/vaccine"
 POLL_INTERVAL=60  # seconds
 
+REF="${1:-master}"
+
 # Trigger workflow
 echo "Triggering workflow..."
 TRIGGER_RESPONSE=$(curl -X POST \
   -H "Accept: application/vnd.github.v3+json" \
   -H "Authorization: token $GH_VACCINE_PAT" \
   -w "\n%{http_code}" \
   "https://api.github.com/repos/$REPO/actions/workflows/vaccine.yml/dispatches" \
-  -d '{"ref":"master", "inputs": {"commit_sha": "'$CI_COMMIT_SHA'"}}' 2>&1)
+  -d '{"ref":"'${REF}'", "inputs": {"commit_sha": "'$CI_COMMIT_SHA'"}}' 2>&1)
 
 HTTP_STATUS=$(echo "$TRIGGER_RESPONSE" | tail -n1)
 if [ "$HTTP_STATUS" -ne 204 ]; then