From 88f852927fa4fd1dcd72a02b8e71c92e04ab9726 Mon Sep 17 00:00:00 2001
From: Darshit Chanpura <dchanp@amazon.com>
Date: Thu, 7 Sep 2023 17:02:08 -0400
Subject: [PATCH] Adds automation script to add issue label

Signed-off-by: Darshit Chanpura <dchanp@amazon.com>
---
 .github-automation/copy-issue-labels-to-pr.py | 67 +++++++++++++++++++
 .github/workflows/pr-automation.yml           | 14 ++++
 2 files changed, 81 insertions(+)
 create mode 100644 .github-automation/copy-issue-labels-to-pr.py
 create mode 100644 .github/workflows/pr-automation.yml

diff --git a/.github-automation/copy-issue-labels-to-pr.py b/.github-automation/copy-issue-labels-to-pr.py
new file mode 100644
index 0000000000000..2dd23bb278b53
--- /dev/null
+++ b/.github-automation/copy-issue-labels-to-pr.py
@@ -0,0 +1,67 @@
+#!/usr/bin/python3
+# SPDX-License-Identifier: Apache-2.0
+
+import itertools
+import json
+import re
+import sys
+from github import Github
+
+# We don't want to copy all labels on linked issues; only those in this subset.
+COPYABLE_LABELS = {
+  "enhancement",
+  "documentation"
+}
+
+# Returns a pull request extracted from Github's event JSON.
+def get_pr(event):
+    # --- Extract PR from event JSON ---
+    # `check_run`/`check_suite` does not include any direct reference to the PR
+    # that triggered it in its event JSON. We have to extrapolate it using the head
+    # SHA that *is* there.
+
+    # Get head SHA from event JSON
+    pr_head_sha = event["check_suite"]["head_sha"]
+
+    # Find the repo PR that matches the head SHA we found
+    return {pr.head.sha: pr for pr in repo.get_pulls()}[pr_head_sha]
+
+# Get all issue numbers related to a PR.
+def get_related_issues(pr):
+    # Regex to pick out closing keywords.
+    regex = re.compile("(close[sd]?|fix|fixe[sd]?|resolve[sd]?)\s*:?\s+#(\d+)", re.I)
+
+    # Extract all associated issues from closing keyword in PR
+    for verb, num in regex.findall(pr.body):
+        yield int(num)
+
+    # Extract all associated issues from PR commit messages
+    for c in pr.get_commits():
+        for verb, num in regex.findall(c.commit.message):
+            yield int(num)
+
+# Get inputs from shell
+(token, repository, path) = sys.argv[1:4]
+
+# Initialize repo
+repo = Github(token).get_repo(repository)
+
+# Open Github event JSON
+with open(path) as f:
+    event = json.load(f)
+
+# Get the PR we're working on.
+pr = get_pr(event)
+pr_labels = {label.name for label in pr.labels}
+
+# Get every label on every linked issue.
+issues = get_related_issues(pr)
+issues_labels = [repo.get_issue(n).labels for n in issues]
+issues_labels = {l.name for l in itertools.chain(*issues_labels)}
+
+# Find the set of all labels we want to copy that aren't already set on the PR.
+unset_labels = COPYABLE_LABELS & issues_labels - pr_labels
+
+# If there are any labels we need to add, add them.
+if len(unset_labels) > 0:
+    pr.set_labels(*list(unset_labels))
diff --git a/.github/workflows/pr-automation.yml b/.github/workflows/pr-automation.yml
new file mode 100644
index 0000000000000..dbaab469e9f12
--- /dev/null
+++ b/.github/workflows/pr-automation.yml
@@ -0,0 +1,14 @@
+# Automation that runs upon pull requests being submitted.
+name: automation-pr
+
+on:
+  pull_request_target:
+    types: [opened, reopened, synchronize, converted_to_draft, edited]
+
+jobs:
+  mirror-labels-to-pr:
+#    if: github.repository == 'opensearch-project/OpenSearch'
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+        run: .github-automation/copy-issue-labels-to-pr.py ${{ secrets.GITHUB_TOKEN }} $GITHUB_REPOSITORY $GITHUB_EVENT_PATH