Skip to content

Label PRs by Author Org Membership #8128

Label PRs by Author Org Membership

Label PRs by Author Org Membership #8128

Workflow file for this run

# Label PRs by Author Org Membership
# -------------
# This workflow is designed to automatically label pull requests based on the author's membership in the ROCm organization.
name: "Label PRs by Author Org Membership"
on:
workflow_dispatch:
schedule:
- cron: "*/30 * * * *" # every 30 minutes
jobs:
label-prs:
if: github.repository == 'ROCm/rocm-libraries'
runs-on: ubuntu-24.04
permissions:
contents: read
pull-requests: write
env:
ORG_TO_CHECK: ROCm
ORG_LABEL: "organization: ROCm"
STREAMHPC_TEAM: streamhpc
STREAMHPC_LABEL: "organization: streamhpc"
EXTERNAL_LABEL: "external contribution"
steps:
- name: Generate GitHub App token
id: generate-token
uses: actions/create-github-app-token@bcd2ba49218906704ab6c1aa796996da409d3eb1 # v3.2.0
with:
app-id: ${{ secrets.APP_ID }}
private-key: ${{ secrets.APP_PRIVATE_KEY }}
owner: ${{ github.repository_owner }}
- name: Checkout code
uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
with:
sparse-checkout: |
.github
${{ github.event.inputs.subrepo-prefix }}
sparse-checkout-cone-mode: true
token: ${{ steps.generate-token.outputs.token }}
fetch-depth: 0 #for subtree operations
- name: Get open PR numbers
env:
GH_TOKEN: ${{ steps.generate-token.outputs.token }}
run: |
gh pr list --state open --json number,labels \
--jq '[.[] | select(
([.labels[].name] | index(env.ORG_LABEL) | not) and
([.labels[].name] | index(env.EXTERNAL_LABEL) | not)
) | .number] | .[]' \
> pr_numbers.txt
- name: Label PRs based on author membership
env:
GH_TOKEN: ${{ steps.generate-token.outputs.token }}
run: |
while read pr; do
echo "Checking PR #$pr"
labels=$(gh pr view "$pr" --json labels --jq '.labels[].name')
if echo "$labels" | grep -qFx "${{ env.ORG_LABEL }}" \
|| echo "$labels" | grep -qFx "${{ env.STREAMHPC_LABEL }}" \
|| echo "$labels" | grep -qFx "${{ env.EXTERNAL_LABEL }}"; then
echo "PR #$pr already labeled, skipping."
continue
fi
author=$(gh pr view "$pr" --json author --jq '.author.login')
if gh api orgs/${{ env.ORG_TO_CHECK }}/members/$author --silent; then
if gh api orgs/${{ env.ORG_TO_CHECK }}/teams/${{ env.STREAMHPC_TEAM }}/members/$author --silent; then
gh pr edit "$pr" --add-label "${{ env.STREAMHPC_LABEL }}"
else
gh pr edit "$pr" --add-label "${{ env.ORG_LABEL }}"
fi
else
gh pr edit "$pr" --add-label "${{ env.EXTERNAL_LABEL }}"
fi
done < pr_numbers.txt