From 4f1aa2d8ff067f0430cae80d5159e5de3f53bd84 Mon Sep 17 00:00:00 2001
From: Manfred Moser <manfred@simpligility.ca>
Date: Thu, 25 Jan 2024 13:57:38 -0800
Subject: [PATCH] Add github milestone labeling

---
 .github/bin/retry               | 21 +++++++++
 .github/workflows/milestone.yml | 83 +++++++++++++++++++++++++++++++++
 2 files changed, 104 insertions(+)
 create mode 100755 .github/bin/retry
 create mode 100644 .github/workflows/milestone.yml

diff --git a/.github/bin/retry b/.github/bin/retry
new file mode 100755
index 000000000..9e9bd9da4
--- /dev/null
+++ b/.github/bin/retry
@@ -0,0 +1,21 @@
+#!/usr/bin/env bash
+
+set -euo pipefail
+
+x() {
+    echo "+ $*" >&2
+    "$@"
+}
+
+max_retry_time_seconds=$(( 30 * 60 ))
+retry_delay_seconds=10
+
+END=$(( $(date +%s) + ${max_retry_time_seconds} ))
+
+while (( $(date +%s) < $END )); do
+    x "$@" && exit 0
+    sleep "${retry_delay_seconds}"
+done
+
+echo "$0: retrying [$*] timed out" >&2
+exit 1
diff --git a/.github/workflows/milestone.yml b/.github/workflows/milestone.yml
new file mode 100644
index 000000000..6b04264e7
--- /dev/null
+++ b/.github/workflows/milestone.yml
@@ -0,0 +1,83 @@
+name: milestone
+
+on:
+  push:
+    branches: [main]
+
+defaults:
+  run:
+    shell: bash --noprofile --norc -euo pipefail {0}
+
+jobs:
+  set-milestone:
+    if: github.repository_owner == 'trinodb'
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+      - name: Get milestone from pom.xml
+        run: |
+          .github/bin/retry ./mvnw -v
+          MILESTONE_NUMBER="$(./mvnw help:evaluate -Dexpression=project.version -q -DforceStdout | cut -d- -f1)"
+          echo "Setting PR milestone to ${MILESTONE_NUMBER}"
+          echo "MILESTONE_NUMBER=${MILESTONE_NUMBER}" >> $GITHUB_ENV
+      - name: Set milestone to PR
+        uses: actions/github-script@v7
+        with:
+          github-token: ${{secrets.GITHUB_TOKEN}}
+          script: |
+            // Get pull request number from commit sha
+            // 'listPullRequestsAssociatedWithCommit()' lists the merged pull request
+            // https://docs.github.com/en/rest/reference/repos#list-pull-requests-associated-with-a-commit
+            // and https://octokit.github.io/rest.js/v19#repos-list-pull-requests-associated-with-commit
+            const pr_response = await github.rest.repos.listPullRequestsAssociatedWithCommit({
+                owner: context.repo.owner,
+                repo: context.repo.repo,
+                commit_sha: context.sha
+            })
+            if (pr_response.data.length === 0) {
+                console.log('Pull request not found for commit', context.sha)
+                return
+            }
+            if (pr_response.data.length > 1) {
+                console.log(pr_response.data)
+                throw 'Expected 1 pull request but found: ' + pr_response.data.length
+            }
+            const pr_number = pr_response.data[0].number
+            console.log('Found pull request', pr_number, 'for commit', context.sha)
+
+            // Get milestone
+            const {
+                MILESTONE_NUMBER
+            } = process.env
+            console.log('Milestone number to set', MILESTONE_NUMBER)
+
+            // Find milestone with matching title
+            // See https://octokit.github.io/rest.js/v19#pagination
+            const milestones = await github.paginate(github.rest.issues.listMilestones, {
+                owner: context.repo.owner,
+                repo: context.repo.repo,
+                state: 'all'
+            })
+            let milestone = milestones.find(milestone => milestone.title === MILESTONE_NUMBER)
+            console.log('Found milestone with title', MILESTONE_NUMBER, milestone)
+
+            // Create new milestone if it doesn't exist
+            if (!milestone) {
+                const create_response = await github.rest.issues.createMilestone({
+                    owner: context.repo.owner,
+                    repo: context.repo.repo,
+                    title: MILESTONE_NUMBER
+                })
+                milestone = create_response.data
+                console.log('Created new milestone with title', MILESTONE_NUMBER, milestone)
+            }
+
+            // Set milestone to PR
+            await github.rest.issues.update({
+                owner: context.repo.owner,
+                repo: context.repo.repo,
+                milestone: milestone.number,
+                issue_number: pr_number
+            })
+            console.log('Added PR', pr_number, 'to milestone', MILESTONE_NUMBER)