From e87842bcb554623398ef2b84820b42406841d39e Mon Sep 17 00:00:00 2001
From: David Li <li.davidm96@gmail.com>
Date: Fri, 7 Mar 2025 13:21:08 +0900
Subject: [PATCH 1/2] ci: add action to assign milestone to PR/issue

Fixes #456.
---
 .github/workflows/dev_pr.yml          |  7 +++
 .github/workflows/dev_pr_milestone.sh | 70 +++++++++++++++++++++++++++
 2 files changed, 77 insertions(+)
 create mode 100644 .github/workflows/dev_pr_milestone.sh

diff --git a/.github/workflows/dev_pr.yml b/.github/workflows/dev_pr.yml
index 8da3ab5fd5..353e17dc7e 100644
--- a/.github/workflows/dev_pr.yml
+++ b/.github/workflows/dev_pr.yml
@@ -75,3 +75,10 @@ jobs:
           script: |
             const scripts = require(`${process.env.GITHUB_WORKSPACE}/.github/workflows/dev_pr.js`);
             await scripts.check_linked_issue({core, github, context, issue: ${{ steps.title-format.outputs.result }}});
+
+      - name: Assign milestone
+        if: '! github.event.pull_request.draft'
+        env:
+          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        run: |
+          ./.github/workflows/dev_pr_milestone.sh "${GITHUB_REPOSITORY}" ${{github.event.number}}
diff --git a/.github/workflows/dev_pr_milestone.sh b/.github/workflows/dev_pr_milestone.sh
new file mode 100644
index 0000000000..b6876b4b08
--- /dev/null
+++ b/.github/workflows/dev_pr_milestone.sh
@@ -0,0 +1,70 @@
+#!/usr/bin/env bash
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# Assign a milestone to the given PR based on the open milestones and known
+# releases.
+
+set -euo pipefail
+
+main() {
+  local -r repo="${1}"
+  local -r pr_number="${2}"
+  echo "On ${repo} pull ${pr_number}"
+
+  local -r existing_milestone=$(gh pr view "${pr_number}" \
+    --json milestone \
+    -t '{{if .milestone}}{{.milestone.title}}{{end}}')
+
+  if [[ -n "${existing_milestone}" ]]; then
+    echo "PR has milestone: ${existing_milestone}"
+    local -r milestone="${existing_milestone}"
+  else
+    local -r milestone=$(
+      gh api "/repos/${repo}/milestones" |
+        jq --raw-output '.[] | .title' |
+        grep -E '^[0-9]+\.[0-9]+\.[0-9]+$'
+      head -n1
+    )
+
+    echo "Assigning milestone: ${milestone}"
+    gh pr edit "${pr_number}" -m "${milestone}"
+  fi
+
+  local -r repo_owner=$(echo "${repo}" | cut -d'/' -f1)
+  local -r repo_name=$(echo "${repo}" | cut -d'/' -f2)
+  local -r graphql_query="{
+      repository(owner: \"${repo_owner}\", name: \"${repo_name}\") {
+        pullRequest(number: ${pr_number}) {
+          closingIssuesReferences(first: 5) {
+            edges {
+              node {
+                number
+              }
+            }
+          }
+        }
+      }
+    }"
+  local -r linked_issues=$(gh api graphql -f query="${graphql_query}" | jq -r '.data.repository.pullRequest.closingIssuesReferences.edges | .[].node.number')
+  for issue in ${linked_issues}; do
+    echo "Linked issue: ${issue}"
+    gh issue edit "${issue}" --milestone "${milestone}"
+  done
+}
+
+main "$@"

From 9c6f10312bcd100d8bbefd0e99a944a2536de2c8 Mon Sep 17 00:00:00 2001
From: David Li <li.davidm96@gmail.com>
Date: Fri, 7 Mar 2025 01:26:10 -0500
Subject: [PATCH 2/2] Update .github/workflows/dev_pr.yml

Co-authored-by: Sutou Kouhei <kou@cozmixng.org>
---
 .github/workflows/dev_pr.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/dev_pr.yml b/.github/workflows/dev_pr.yml
index 353e17dc7e..34b3363c50 100644
--- a/.github/workflows/dev_pr.yml
+++ b/.github/workflows/dev_pr.yml
@@ -81,4 +81,4 @@ jobs:
         env:
           GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         run: |
-          ./.github/workflows/dev_pr_milestone.sh "${GITHUB_REPOSITORY}" ${{github.event.number}}
+          ./.github/workflows/dev_pr_milestone.sh "${GITHUB_REPOSITORY}" ${{ github.event.number }}