# Azure CI Dispatcher
# ------------------
# This workflow allows Azure CI to be centralized in a single PR check.
# It detects which subtrees (from a super-repo structure) were changed in a
# pull request, and automatically requests Azure CI runs for the corresponding
# subtrees.
#
# For any given subtree, if an upstream subtree is also included in the PR,
# it will not run CI for the downstream subtree.
# Eg. A PR that touches rocprim and rocthrust will only trigger rocprim CI.
#
# Requires an Azure Personal Access Token with permissions to manage builds.
# The token should be stored in the repository secrets as `AZ_PAT`.

name: Trigger Azure CI

on:
  pull_request_target:
    types:
      - opened
      - synchronize
      - reopened
      - ready_for_review
    branches:
      - develop
      - staging
      - main
      - release-staging/rocm-rel-7.*
    paths-ignore:  # Do not trigger full tests on documentation-only changes
      - '**/*.md'
      - '**/*.rtf'
      - '**/*.rst'
      - '**/.markdownlint-ci2.yaml'
      - '**/.readthedocs.yaml'
      - '**/.spellcheck.local.yaml'
      - '**/.wordlist.txt'
      - 'docs/**'
      - 'projects/*/docs/**'

concurrency:
  group: azure-ci-dispatcher-${{ github.event.pull_request.number || github.ref }}
  cancel-in-progress: true

jobs:
  dispatch-azure-ci:
    name: Trigger Azure CI
    if: github.repository == 'ROCm/rocm-systems' && github.event.pull_request.draft == false
    runs-on: ubuntu-latest
    steps:
    - name: Generate a token
      id: generate-token
      uses: actions/create-github-app-token@df432ceedc7162793a195dd1713ff69aefc7379e # v2.0.6
      with:
        app-id: ${{ secrets.APP_ID }}
        private-key: ${{ secrets.APP_PRIVATE_KEY }}
        owner: ${{ github.repository_owner }}
        repositories: |
          rocm-systems

    - name: Wait until refs/pull/${{ github.event.pull_request.number }}/merge exists
      run: |
        merge_ref="refs/pull/${{ github.event.pull_request.number }}/merge"
        check_merge_ref() {
          git ls-remote "https://github.com/ROCm/rocm-systems" "$merge_ref" | grep -q "$merge_ref"
        }

        max_attempts=10
        attempt=0

        while [ $attempt -lt $max_attempts ]; do
          if check_merge_ref; then
            echo "$merge_ref found."
            break
          else
            retry_delay=60
            echo "$merge_ref not found. Retrying in $retry_delay seconds..."
            sleep $retry_delay
          fi
          attempt=$((attempt + 1))
        done

        if [ $attempt -ge $max_attempts ]; then
          echo "$merge_ref not found. Maximum attempts reached."
          exit 1
        fi

    - name: Checkout code
      uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
      with:
        ref: refs/pull/${{ github.event.pull_request.number }}/merge
        sparse-checkout: .github
        sparse-checkout-cone-mode: true
        token: ${{ steps.generate-token.outputs.token }}

    - name: Install dependencies
      run: |
        python -m pip install --upgrade pip
        pip install -r .github/requirements.txt

    - name: Detect changed subtrees
      id: detect
      env:
        GH_TOKEN: ${{ steps.generate-token.outputs.token }}
      run: |
        python .github/scripts/pr_detect_changed_subtrees.py \
          --repo "${{ github.repository }}" \
          --pr "${{ github.event.pull_request.number }}" \
          --config ".github/repos-config.json" \
          --require-monorepo-source

    - name: Cancel in-progress/not-started runs for current PR
      id: cancel-in-progress
      if: steps.detect.outputs.subtrees
      run: |
        pr_number=${{ github.event.pull_request.number }}
        pr_filter_query="branchName=refs/pull/$pr_number/merge&repositoryType=GitHub&repositoryId=ROCm/rocm-systems&api-version=7.1"

        res=$(curl -sSX GET "https://dev.azure.com/ROCm-CI/ROCm-CI/_apis/build/builds?$pr_filter_query" \
          -H "Content-Type: application/json")

        runs=$(echo "$res" | jq -r ".value[] | select((.status == \"inProgress\" or .status == \"notStarted\") and .definition.name != \"rocm-ci-caller\") | .id")

        if [ -z "$runs" ]; then
          echo "No in-progress/not-started runs found for ROCm/rocm-systems PR #$pr_number"
          echo "status=false" >> $GITHUB_OUTPUT
          exit 0
        fi

        echo "Found in-progress/not-started runs for ROCm/rocm-systems PR #$pr_number: $runs"
        echo "status=true" >> $GITHUB_OUTPUT

        for run_id in $runs; do
          echo "Cancelling run ID: $run_id"
          echo "Run URL: https://dev.azure.com/ROCm-CI/ROCm-CI/_build/results?buildId=$run_id"
          response=$(curl -sSX PATCH "https://dev.azure.com/ROCm-CI/ROCm-CI/_apis/build/builds/$run_id?api-version=7.1" \
            -u ":${{ secrets.AZ_PAT }}" \
            -H "Content-Type: application/json" \
            -d '{"status": "cancelling"}')

          if [ $? -ne 0 ]; then
            echo "Failed to cancel run ID: $run_id"
          else
            echo "Cancelled run ID: $run_id"
          fi
        done

    - name: Rerun previous failed/cancelled runs for current PR merge commit
      id: rerun-failed
      if: steps.detect.outputs.subtrees && steps.cancel-in-progress.outputs.status == 'false'
      run: |
        pr_number=${{ github.event.pull_request.number }}
        pr_merge_sha=$(curl -sSX GET "https://api.github.com/repos/ROCm/rocm-systems/git/ref/pull/${pr_number}/merge" | jq -r '.object.sha')
        pr_filter_query="branchName=refs/pull/$pr_number/merge&repositoryType=GitHub&repositoryId=ROCm/rocm-systems&api-version=7.1"

        res=$(curl -sSX GET "https://dev.azure.com/ROCm-CI/ROCm-CI/_apis/build/builds?$pr_filter_query" \
          -H "Content-Type: application/json")

        failed_runs_info=$(echo "$res" | jq -r ".value[] |
          select((.result == \"failed\" or .result == \"canceled\")
          and (.sourceVersion | contains(\"$pr_merge_sha\")))
          | {id: .id, name: .definition.name}")
        success_runs_info=$(echo "$res" | jq -r ".value[] |
          select((.result == \"succeeded\")
          and (.sourceVersion | contains(\"$pr_merge_sha\")))
          | {id: .id, name: .definition.name}")

        failed_run_ids=$(echo "$failed_runs_info" | jq -r '.id')
        failed_project_names=$(echo "$failed_runs_info" | jq -r '.name')
        success_run_ids=$(echo "$success_runs_info" | jq -r '.id')
        success_project_names=$(echo "$success_runs_info" | jq -r '.name')

        if [ -z "$failed_run_ids" ]; then
          echo "No failed/cancelled runs found for ROCm/rocm-systems PR #$pr_number at merge commit $pr_merge_sha"
          echo "status=false" >> $GITHUB_OUTPUT
          exit 0
        fi

        echo "Found failed/cancelled runs for ROCm/rocm-systems PR #$pr_number at merge commit $pr_merge_sha: ${failed_run_ids[*]}"
        echo "Found successful runs for ROCm/rocm-systems PR #$pr_number at merge commit $pr_merge_sha: ${success_run_ids[*]}"
        echo "status=true" >> $GITHUB_OUTPUT

        new_run_ids=()

        for run_id in $failed_run_ids; do
          echo "Rerunning failed run ID: $run_id"
          echo "Run URL: https://dev.azure.com/ROCm-CI/ROCm-CI/_build/results?buildId=$run_id"
          response=$(curl -sSX PATCH "https://dev.azure.com/ROCm-CI/ROCm-CI/_apis/build/builds/$run_id?retry=true&api-version=7.1" \
            -u ":${{ secrets.AZ_PAT }}" \
            -H "Content-Type: application/json")

          if [ $? -ne 0 ]; then
            echo "Failed to rerun run ID: $run_id"
          else
            echo "Rerun requested for run ID: $run_id"
            new_run_ids+=("$run_id")
          fi
        done

        echo "run_ids=${new_run_ids[*]}" >> $GITHUB_OUTPUT
        echo "project_names=${failed_project_names[*]}" >> $GITHUB_OUTPUT
        echo "success_run_ids=${success_run_ids[*]}" >> $GITHUB_OUTPUT
        echo "success_project_names=${success_project_names[*]}" >> $GITHUB_OUTPUT

    - name: Start new Azure CI runs
      id: dispatch
      if: steps.detect.outputs.subtrees && (steps.cancel-in-progress.outputs.status == 'true' || steps.rerun-failed.outputs.status == 'false')
      env:
        GH_TOKEN: ${{ steps.generate-token.outputs.token }}
      run: |
        echo "${{ steps.detect.outputs.subtrees }}" > changed_subtrees.txt

        python .github/scripts/azure_resolve_subtree_deps.py \
          --subtree-file changed_subtrees.txt \
          > resolved_subtrees.txt

        run_ids=()
        project_names=()

        while IFS= read -r line; do
          IFS='=' read -r project_name definition_id <<< "$line"
          echo "Requesting run for $project_name with definition ID $definition_id"

          max_attempts=3
          retry_delay=5
          attempt=1
          success=false

          while [ $attempt -le $max_attempts ]; do
            response=$(curl -sSX POST https://dev.azure.com/ROCm-CI/ROCm-CI/_apis/pipelines/$definition_id/runs?api-version=7.1 \
              -u ":${{ secrets.AZ_PAT }}" \
              -H "Content-Type: application/json" \
              -d '{
                "resources": {
                  "repositories": {
                    "self": {
                      "refName": "refs/pull/${{ github.event.pull_request.number }}/merge"
                    }
                  }
                }
              }')

            if [ $? -eq 0 ]; then
              success=true
              break
            fi

            echo "Attempt $attempt failed. Retrying in $retry_delay seconds..."
            sleep $retry_delay
            attempt=$((attempt + 1))
          done

          if [ "$success" = true ]; then
            run_id=$(echo "$response" | jq -r '.id' || echo "null")
            if [ "$run_id" != "null" ]; then
              echo "Run ID for $project_name: $run_id"
              echo "Run URL: https://dev.azure.com/ROCm-CI/ROCm-CI/_build/results?buildId=$run_id"
              run_ids+=("$run_id")
              project_names+=("$project_name")
            else
              echo "Failed to request run for $project_name"
            fi
          else
            echo "Failed to request run for $project_name after $max_attempts attempts"
          fi

          echo ""
        done < resolved_subtrees.txt

        echo "run_ids=${run_ids[*]}" >> $GITHUB_OUTPUT
        echo "project_names=${project_names[*]}" >> $GITHUB_OUTPUT

    - name: Create summary check
      env:
        GH_TOKEN: ${{ steps.generate-token.outputs.token }}
        PR_TITLE: ${{ toJSON(github.event.pull_request.title) }}
      run: |
        if [[ -n "${{ steps.dispatch.outputs.run_ids }}" && -n "${{ steps.dispatch.outputs.project_names }}" ]]; then             # If new runs were started
          run_ids=(${{ steps.dispatch.outputs.run_ids }})
          project_names=(${{ steps.dispatch.outputs.project_names }})
        elif [[ -n "${{ steps.rerun-failed.outputs.run_ids }}" && -n "${{ steps.rerun-failed.outputs.project_names }}" ]]; then   # If reruns were requested
          run_ids=(${{ steps.rerun-failed.outputs.run_ids }})
          project_names=(${{ steps.rerun-failed.outputs.project_names }})
          success_run_ids=(${{ steps.rerun-failed.outputs.success_run_ids }})
          success_project_names=(${{ steps.rerun-failed.outputs.success_project_names }})
        else
          echo "No run IDs or project names found, creating empty summary check."
        fi

        pr_number=${{ github.event.pull_request.number }}
        pr_head_sha=${{ github.event.pull_request.head.sha }}
        pr_merge_sha=$(curl -sSX GET "https://api.github.com/repos/ROCm/rocm-systems/git/ref/pull/${pr_number}/merge" | jq -r '.object.sha')

        newline=$'\n'

        summary="PR: [$PR_TITLE #$pr_number](${{ github.event.pull_request.html_url }})${newline}${newline}"
        summary+="HEAD: [$pr_head_sha](https://github.com/ROCm/rocm-systems/commit/$pr_head_sha)${newline}${newline}"
        summary+="MERGE: [$pr_merge_sha](https://github.com/ROCm/rocm-systems/commit/$pr_merge_sha)${newline}${newline}"

        if [[ -z "${run_ids[*]}" && -z "${success_run_ids[*]}" ]]; then
          summary+="### No Azure CI runs were started for this PR.${newline}${newline}"
          gh_output=$(gh api repos/ROCm/rocm-systems/check-runs \
            -f "name=Azure CI Summary" \
            -f "head_sha=$pr_head_sha" \
            -f "status=completed" \
            -f "conclusion=neutral" \
            -f "output[title]=Azure CI Summary" \
            -f "output[summary]=$summary")
          echo "Created empty summary check with ID: $(echo "$gh_output" | jq -r '.id')"
          echo "Summary check URL: https://github.com/ROCm/rocm-systems/pull/$pr_number/checks?check_run_id=$(echo "$gh_output" | jq -r '.id')"
          exit 0
        fi

        summary+="### Pipelines triggered for this PR${newline}${newline}"
        summary+="| Project | Run ID | Status |${newline}"
        summary+="|--------------|--------|--------|${newline}"

        if [[ -n "${success_run_ids[*]}" ]]; then
          for i in "${!success_project_names[@]}"; do
            summary+="| ${success_project_names[i]} | [${success_run_ids[i]}](https://dev.azure.com/ROCm-CI/ROCm-CI/_build/results?buildId=${success_run_ids[i]}) | success |${newline}"
          done
        fi

        for i in "${!project_names[@]}"; do
          summary+="| ${project_names[i]} | [${run_ids[i]}](https://dev.azure.com/ROCm-CI/ROCm-CI/_build/results?buildId=${run_ids[i]}) | pending |${newline}"
        done

        summary+="${newline}${newline}"
        summary+="### Rerun instructions${newline}${newline}"
        summary+="To request Azure to rerun jobs, click the \`Re-run all jobs\` button on the [corresponding \`Trigger Azure CI\` run](https://github.com/ROCm/rocm-systems/actions/runs/${{ github.run_id }}).${newline}${newline}"
        summary+="If there are any pending runs for this PR, they will be cancelled, and new runs will be started.${newline}${newline}"
        summary+="If there are no pending runs, but there are existing failed or cancelled runs for this PR and merge SHA, the existing runs will be rerun.${newline}${newline}"
        summary+="Otherwise, new runs will be started.${newline}${newline}"

        text=""

        if [[ -n "${success_run_ids[*]}" ]]; then
          for i in "${!success_run_ids[@]}"; do
            text+="${success_run_ids[i]}=success;"
          done
        fi

        for i in "${!project_names[@]}"; do
          text+="${run_ids[i]}=pending;"
        done

        gh_output=$(gh api repos/ROCm/rocm-systems/check-runs \
          -f "name=Azure CI Summary" \
          -f "head_sha=$pr_head_sha" \
          -f "status=in_progress" \
          -f "output[title]=Azure CI Summary" \
          -f "output[summary]=$summary" \
          -f "output[text]=$text")

        echo "Created summary check with ID: $(echo "$gh_output" | jq -r '.id')"
        echo "Summary check URL: https://github.com/ROCm/rocm-systems/pull/$pr_number/checks?check_run_id=$(echo "$gh_output" | jq -r '.id')"