88#
99# Two modes:
1010# --tag <tag> Find the successful CI run triggered by a tag push.
11- # --branch <branch> Find the latest successful CI run on a branch.
11+ # --branch <branch> Find the latest qualifying successful CI run on a branch.
1212#
1313# Outputs the run ID on stdout. All diagnostic messages go to stderr.
1414# When --head-sha is passed, a second line with the run's head SHA is printed.
@@ -24,11 +24,14 @@ Usage:
2424Options:
2525 --tag <tag> Find run by git tag (requires local git repo with the tag)
2626 --branch <branch> Find latest successful run on the given branch
27+ --artifact <glob> Require an unexpired artifact matching this shell glob
28+ (repeatable; branch mode only)
2729 --head-sha Also print the run's head commit SHA (second line)
2830
2931Examples:
3032 $0 --tag v13.0.1 NVIDIA/cuda-python
3133 $0 --branch main NVIDIA/cuda-python
34+ $0 --branch 12.9.x --artifact 'cuda-bindings-python313-*' NVIDIA/cuda-python
3235 $0 --branch main --head-sha NVIDIA/cuda-python "CI"
3336EOF
3437 exit 1
3841MODE=" "
3942REF=" "
4043HEAD_SHA_FLAG=0
44+ ARTIFACT_PATTERNS=()
4145
4246while [[ $# -gt 0 ]]; do
4347 case " ${1} " in
4448 --tag)
49+ [[ $# -ge 2 ]] || usage
4550 [[ -n " ${MODE} " && " ${MODE} " != " tag" && { echo " Error: --tag and --branch are mutually exclusive" >&2 ; exit 1; }
4651 MODE=" tag" ; REF=" ${2} " ; shift 2 ;;
4752 --branch)
53+ [[ $# -ge 2 ]] || usage
4854 [[ -n " ${MODE} " && " ${MODE} " != " branch" && { echo " Error: --tag and --branch are mutually exclusive" >&2 ; exit 1; }
4955 MODE=" branch" ; REF=" ${2} " ; shift 2 ;;
56+ --artifact)
57+ [[ $# -ge 2 ]] || usage
58+ ARTIFACT_PATTERNS+=(" ${2} " ; shift 2 ;;
5059 --head-sha)
5160 HEAD_SHA_FLAG=1; shift ;;
5261 -h|--help)
@@ -69,6 +78,11 @@ WORKFLOW_NAME="${1:-CI}"
6978
7079if [[ -z " ${REPOSITORY} " ; then usage; fi
7180
81+ if [[ " ${MODE} " != " branch" && ${# ARTIFACT_PATTERNS[@]} -gt 0 ]]; then
82+ echo " Error: --artifact is only supported with --branch" >&2
83+ exit 1
84+ fi
85+
7286# ── Prerequisite checks ──
7387if [[ -z " ${GH_TOKEN:- } " ; then
7488 echo " Error: GH_TOKEN environment variable is required" >&2
@@ -86,28 +100,90 @@ done
86100if [[ " ${MODE} " == " branch" ; then
87101 echo " Looking up latest successful '${WORKFLOW_NAME} ' run on branch: ${REF} " >&2
88102
89- RUN_ID=$( gh run list \
90- -b " ${REF} "
91- -L 1 \
92- -w " ${WORKFLOW_NAME} "
93- -s success \
94- -R " ${REPOSITORY} "
95- --json databaseId \
96- | jq -r ' .[0].databaseId // empty' )
103+ RUN_DATA=$( gh run list \
104+ --repo " ${REPOSITORY} "
105+ --branch " ${REF} "
106+ --workflow " ${WORKFLOW_NAME} "
107+ --status success \
108+ --json databaseId,workflowName,status,conclusion,headSha,headBranch,createdAt,url \
109+ --limit 100)
110+
111+ CANDIDATE_RUNS=$( echo " ${RUN_DATA} " | jq -r \
112+ --arg branch " ${REF} " '
113+ map(select(
114+ .headBranch == $branch
115+ and .conclusion == "success"
116+ ))
117+ | sort_by(.createdAt, .databaseId)
118+ | reverse
119+ | .[]
120+ | [.databaseId, .headSha, .createdAt]
121+ | @tsv
122+ ' )
123+
124+ if [[ -z " ${CANDIDATE_RUNS} " ; then
125+ echo " Error: No successful '${WORKFLOW_NAME} ' run found on branch '${REF} '" >&2
126+ exit 1
127+ fi
128+
129+ if [[ ${# ARTIFACT_PATTERNS[@]} -gt 0 ]]; then
130+ echo " Requiring unexpired artifacts matching:" >&2
131+ printf ' - %s\n' " ${ARTIFACT_PATTERNS[@]} " >&2
132+ fi
133+
134+ RUN_ID=" "
135+ HEAD_SHA=" "
136+ while IFS= read -r candidate; do
137+ IFS=$' \t ' read -r candidate_id candidate_sha candidate_created_at <<< " ${candidate}"
138+
139+ missing_patterns=()
140+ if [[ ${# ARTIFACT_PATTERNS[@]} -gt 0 ]]; then
141+ if ! ARTIFACT_NAMES=$( gh api --paginate \
142+ " repos/${REPOSITORY} /actions/runs/${candidate_id} /artifacts?per_page=100"
143+ --jq ' .artifacts[] | select(.expired == false) | .name' ) ; then
144+ echo " Error: Failed to list artifacts for run ${candidate_id} " >&2
145+ exit 1
146+ fi
147+
148+ for pattern in " ${ARTIFACT_PATTERNS[@]} " ; do
149+ pattern_matched=0
150+ while IFS= read -r artifact_name; do
151+ # The caller supplies a shell glob, so the RHS must remain unquoted.
152+ # shellcheck disable=SC2053
153+ if [[ -n " ${artifact_name} " && " ${artifact_name} " == ${pattern} ]]; then
154+ pattern_matched=1
155+ break
156+ fi
157+ done <<< " ${ARTIFACT_NAMES}"
158+ if [[ ${pattern_matched} == 0 ]]; then
159+ missing_patterns+=(" ${pattern} "
160+ fi
161+ done
162+ fi
163+
164+ if [[ ${# missing_patterns[@]} -gt 0 ]]; then
165+ echo " Skipping run ${candidate_id} (${candidate_created_at} ); missing unexpired artifact(s): ${missing_patterns[*]} " >&2
166+ continue
167+ fi
168+
169+ RUN_ID=" ${candidate_id} "
170+ HEAD_SHA=" ${candidate_sha} "
171+ break
172+ done <<< " ${CANDIDATE_RUNS}"
97173
98174 if [[ -z " ${RUN_ID} " ; then
99- echo " Error: No successful '${WORKFLOW_NAME} ' run found on branch '${REF} '" >&2
175+ echo " Error: No successful '${WORKFLOW_NAME} ' run on branch '${REF} ' has all required artifacts " >&2
100176 exit 1
101177 fi
102178
103179 echo " Found run ID: ${RUN_ID} " >&2
104180 echo " ${RUN_ID} "
105181
106182 if [[ " ${HEAD_SHA_FLAG} " == 1 ]]; then
107- HEAD_SHA= $( gh run view " ${RUN_ID } " \
108- -R " ${REPOSITORY} " \
109- --json headSha \
110- | jq -r ' .headSha ' )
183+ if [[ -z " ${HEAD_SHA } " ]] ; then
184+ echo " Error: Run ${RUN_ID} has no head SHA " >&2
185+ exit 1
186+ fi
111187 echo " Head SHA: ${HEAD_SHA} " >&2
112188 echo " ${HEAD_SHA} "
113189 fi
0 commit comments