[ci] Trigger rerun on failure of workflow Dev/Beta Browser Images

Signed-off-by: Viet Nguyen Duc <[email protected]>

.github/workflows/update-dev-beta-browser-images.yml

@@ -2,10 +2,25 @@ name: Update Dev/Beta Browser Images
 
 on:
   workflow_dispatch:
+    inputs:
+      rerunFailedOnly:
+        description: 'Rerun only failed jobs'
+        required: false
+        type: boolean
+        default: true
   schedule:
     # Trigger build every 2 days
     - cron: '0 2 */2 * *'
 
+permissions: write-all
+
+env:
+  GH_CLI_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+  GH_CLI_TOKEN_PR: ${{ secrets.SELENIUM_CI_TOKEN || secrets.GITHUB_TOKEN }}
+  RUN_ID: ${{ github.run_id }}
+  RERUN_FAILED_ONLY: ${{ github.event.inputs.rerunFailedOnly || true }}
+  RUN_ATTEMPT: ${{ github.run_attempt }}
+
 jobs:
 
   deploy:
@@ -94,3 +109,33 @@ jobs:
         docker run --rm $NAME/standalone-$BROWSER:$CHANNEL bash -c "$BROWSER_VERSION_COMMAND"
         echo "$BROWSER_CAPS $CHANNEL_CAPS WebDriver version:"
         docker run --rm $NAME/standalone-$BROWSER:$CHANNEL bash -c "$DRIVER_VERSION_COMMAND"
+
+
+  rerun-workflow-when-failure:
+    name: Rerun workflow when failure
+    needs:
+      - deploy
+    if: failure() && ( github.run_attempt < 3 )
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@main
+      - name: Install GitHub CLI
+        run: |
+          sudo apt update
+          sudo apt install gh
+      - name: Authenticate GitHub CLI for PR
+        if: github.event_name == 'pull_request'
+        run: |
+          echo "$GH_CLI_TOKEN_PR" | gh auth login --with-token
+      - name: Authenticate GitHub CLI
+        if: github.event_name != 'pull_request'
+        run: |
+          echo "$GH_CLI_TOKEN" | gh auth login --with-token
+      - name: Rerun workflow when failure
+        run: |
+          echo "Rerun workflow ID $RUN_ID in attempt #$(($RUN_ATTEMPT + 1))"
+          gh workflow run rerun-failed.yml \
+            --repo $GITHUB_REPOSITORY \
+            --raw-field runId=$RUN_ID \
+            --raw-field rerunFailedOnly=$RERUN_FAILED_ONLY