Sync team labels from issues to PRs when PRs are opened

Signed-off-by: stianst <stianst@gmail.com>

Author: stianst

src/main/java/org/keycloak/gh/bot/SyncPRTeamLabels.java

@@ -0,0 +1,47 @@
+package org.keycloak.gh.bot;
+
+import io.quarkiverse.githubapp.event.PullRequest;
+import org.jboss.logging.Logger;
+import org.keycloak.gh.bot.utils.CommitUtils;
+import org.kohsuke.github.*;
+import org.slf4j.LoggerFactory;
+
+import java.io.IOException;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+import java.util.stream.Collectors;
+
+public class SyncPRTeamLabels {
+
+    private static final Logger logger = Logger.getLogger(SyncPRTeamLabels.class);
+    private static final org.slf4j.Logger log = LoggerFactory.getLogger(SyncPRTeamLabels.class);
+
+    void onOpen(@PullRequest.Opened GHEventPayload.PullRequest payload) throws IOException {
+        GHPullRequest pullRequest = payload.getPullRequest();
+        List<String> commitMessages = pullRequest.listCommits().toList().stream().map(i -> i.getCommit().getMessage()).toList();
+
+        Set<String> teamLabelsToAdd = new HashSet<>();
+
+        for (String commitMessage : commitMessages) {
+            Integer issuerNumber = CommitUtils.getIssuerNumber(commitMessage);
+            if (issuerNumber != null) {
+                GHRepository repository = payload.getRepository();
+                GHIssue issue = repository.getIssue(issuerNumber);
+                logger.infov("PR {0} {1} linked to issue {2} {3}", pullRequest.getNumber(), pullRequest.getTitle(), issue.getNumber(), issue.getTitle());
+
+                issue.getLabels().stream().map(GHLabel::getName).filter(l -> l.startsWith("team/")).forEach(teamLabelsToAdd::add);
+            }
+        }
+
+        if (!teamLabelsToAdd.isEmpty()) {
+            logger.infov("Adding labels {0} to PR {1}", String.join(",", teamLabelsToAdd), pullRequest.getNumber());
+            pullRequest.addLabels(teamLabelsToAdd.toArray(new String[0]));
+        } else {
+            logger.infov("No team labels found for PR {0}", pullRequest.getNumber());
+        }
+    }
+
+
+
+}

src/main/java/org/keycloak/gh/bot/utils/CommitUtils.java

@@ -0,0 +1,21 @@
+package org.keycloak.gh.bot.utils;
+
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+public class CommitUtils {
+
+    private static final Pattern LINKED_ISSUE_PATTERN = Pattern.compile("(([c|C]loses)|([f|F]ixes)|([r|R]esolves)) #([0-9]*)");
+
+    public static Integer getIssuerNumber(String commitMessage) {
+        Matcher matcher = LINKED_ISSUE_PATTERN.matcher(commitMessage);
+        if (matcher.find() && matcher.groupCount() == 5) {
+            try {
+                return Integer.valueOf(matcher.group(5));
+            } catch (NumberFormatException e) {
+            }
+        }
+        return null;
+    }
+
+}

src/test/java/org/keycloak/gh/bot/utils/CommitUtilsTest.java

@@ -0,0 +1,20 @@
+package org.keycloak.gh.bot.utils;
+
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+
+public class CommitUtilsTest {
+
+    @Test
+    public void testCloses() {
+        Assertions.assertEquals(1232456, CommitUtils.getIssuerNumber("This is some commit message\nCloses #1232456 and something else"));
+        Assertions.assertEquals(1232456, CommitUtils.getIssuerNumber("This is some commit message\ncloses #1232456 and something else"));
+        Assertions.assertEquals(1232456, CommitUtils.getIssuerNumber("This is some commit message\nFixes #1232456 and something else\nSomething else"));
+        Assertions.assertEquals(1232456, CommitUtils.getIssuerNumber("This is some commit message\nfixes #1232456 and something else\nSomething else"));
+        Assertions.assertEquals(1232456, CommitUtils.getIssuerNumber("Resolves #1232456 and something else\nSomething else"));
+        Assertions.assertEquals(1232456, CommitUtils.getIssuerNumber("resolves #1232456 and something else\nSomething else"));
+        Assertions.assertNull(CommitUtils.getIssuerNumber("Some message\nResolves 1232456"));
+        Assertions.assertNull(CommitUtils.getIssuerNumber("Some message\nResolves: #1232456"));
+    }
+
+}