From 60bdff75ec8b3ec7a7b5724e5f9ea4c8ccad4741 Mon Sep 17 00:00:00 2001
From: roll <quek.ruoling@gmail.com>
Date: Mon, 29 Dec 2025 20:09:36 +0800
Subject: [PATCH] [MISC][RL] Update pipeline script to exit on 401

---
 .github/workflows/trigger-gitlab-pipeline.yml | 38 ++++++++++++++-----
 1 file changed, 28 insertions(+), 10 deletions(-)

diff --git a/.github/workflows/trigger-gitlab-pipeline.yml b/.github/workflows/trigger-gitlab-pipeline.yml
index a030a14..569f55a 100644
--- a/.github/workflows/trigger-gitlab-pipeline.yml
+++ b/.github/workflows/trigger-gitlab-pipeline.yml
@@ -65,6 +65,7 @@ jobs:
           GITLAB_PROJECT_ID: ${{ vars.GITLAB_PROJECT_ID }}
           SLEEP_DURATION_MINUTES: 15
           CHECK_INTERVAL_MINUTES: 5
+          MAX_RETRY_ATTEMPTS: 6
           DOWNSTREAM_PIPELINE_NAME: library
           DOWNSTREAM_JOB_NAME: build-dist-job
         run: |
@@ -74,27 +75,44 @@ jobs:
 
           num_attempts=1
           job_status=""
-          until [[ $num_attempts -gt 5 || $job_status == "success" || $job_status == "failed" || $job_status == "canceled" || $job_status == "skipped" ]]
+          until [[ $num_attempts -gt $MAX_RETRY_ATTEMPTS || $job_status =~ ^(success|failed|canceled|skipped)$ ]]
           do
             echo "Attempt: $num_attempts"
 
-            downstream_pipeline_id=$(curl -sS --request GET --header "PRIVATE-TOKEN: $GITLAB_PAT" \
-              --url "$GITLAB_ENDPOINT/$GITLAB_PROJECT_ID/pipelines/$PIPELINE_ID/bridges" \
-              | jq -r '[ .[] | select( .name | contains("'"$DOWNSTREAM_PIPELINE_NAME"'")) ]' | jq -r '.[].downstream_pipeline.id')
+            pipeline_result=$(curl -sS --request GET --header "PRIVATE-TOKEN: $GITLAB_PAT" \
+              --write-out "\n%{http_code}" --url "$GITLAB_ENDPOINT/$GITLAB_PROJECT_ID/pipelines/$PIPELINE_ID/bridges")
+            pipeline_result_body=$(echo "$pipeline_result" | sed '$d')
+            pipeline_result_code=$(echo "$pipeline_result" | tail -n 1)
+
+            if [[ pipeline_result_code -eq 401 ]]; then
+              echo "Please check if the GITLAB_PAT has sufficient permissions, or regenerate it if it has expired"
+              exit 1
+            fi
+
+            downstream_pipeline_id=$(echo "$pipeline_result_body" | jq -r '[ .[] | select( .name | contains("'"$DOWNSTREAM_PIPELINE_NAME"'")) ]' | jq -r '.[].downstream_pipeline.id')
             echo "($DOWNSTREAM_PIPELINE_NAME) Downstream pipeline id: $downstream_pipeline_id"
 
-            job_status=$(curl -sS --header "PRIVATE-TOKEN: $GITLAB_PAT" \
-              --url "$GITLAB_ENDPOINT/$GITLAB_PROJECT_ID/pipelines/$downstream_pipeline_id/jobs" \
-              | jq -r '[ .[] | select( .name | contains("'"$DOWNSTREAM_JOB_NAME"'")) ]' | jq -r '.[].status')
+            job_status_result=$(curl -sS --header "PRIVATE-TOKEN: $GITLAB_PAT" \
+              --write-out "\n%{http_code}" --url "$GITLAB_ENDPOINT/$GITLAB_PROJECT_ID/pipelines/$downstream_pipeline_id/jobs")
+            job_status_result_body=$(echo "$job_status_result" | sed '$d')
+            job_status_result_code=$(echo "$job_status_result" | tail -n 1)
+
+            if [[ job_status_result_code -eq 401 ]]; then
+              echo "Please check if the GITLAB_PAT has sufficient permissions, or regenerate it if it has expired"
+              exit 1
+            fi
+
+            job_status=$(echo "$job_status_result_body" | jq -r '[ .[] | select( .name | contains("'"$DOWNSTREAM_JOB_NAME"'")) ]' | jq -r '.[].status')
             echo "($DOWNSTREAM_JOB_NAME) Job status: $job_status"
 
-            if [[ $job_status == "running" ]]; then
-              num_attempts=$((num_attempts+1))
+            num_attempts=$((num_attempts+1))
+
+            if [[ $num_attempts -lt $MAX_RETRY_ATTEMPTS && ! $job_status =~ ^(success|failed|canceled|skipped)$ ]]; then
               sleep $(($CHECK_INTERVAL_MINUTES*60))
             fi
           done
 
-          [[ $num_attempts -gt 5 ]] && echo "Number of retries exceeded, please head to the GitLab job to check what happened" && exit 1
+          [[ $num_attempts -gt $MAX_RETRY_ATTEMPTS ]] && echo "Number of retries exceeded, please head to the GitLab job to check what happened" && exit 1
 
           [[ $job_status == "canceled" ]] && echo "GitLab job was cancelled"