Add an option to GithubCreatePRCommand to add labels

A list of labels can be added when the PR is created
box · Sep 23, 2024 · 629debe · 629debe
1 parent 3ab619a
commit 629debe
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 3 deletions.
diff --git a/cli/src/main/java/com/box/l10n/mojito/cli/command/GithubCreatePRCommand.java b/cli/src/main/java/com/box/l10n/mojito/cli/command/GithubCreatePRCommand.java
@@ -83,6 +83,13 @@ public class GithubCreatePRCommand extends Command {
       description = "Enable auto-merge with the specified method")
   EnableAutoMergeType enableAutoMerge = EnableAutoMergeType.NONE;
 
+  @Parameter(
+      names = {"--labels"},
+      required = false,
+      variableArity = true,
+      description = "The PR labels")
+  List<String> labels;
+
   enum EnableAutoMergeType {
     SQUASH,
     MERGE,
@@ -98,12 +105,17 @@ public boolean shouldShowInCommandList() {
   protected void execute() throws CommandException {
     try {
 
-      GHPullRequest pr =
-          githubClients.getClient(owner).createPR(repository, title, head, base, body, reviewers);
+      GithubClient githubClient = githubClients.getClient(owner);
+
+      GHPullRequest pr = githubClient.createPR(repository, title, head, base, body, reviewers);
+
       consoleWriter.a("PR created: ").fg(Ansi.Color.CYAN).a(pr.getHtmlUrl().toString()).println();
       if (!EnableAutoMergeType.NONE.equals(enableAutoMerge)) {
-        githubClients.getClient(owner).enableAutoMerge(pr, GithubClient.AutoMergeType.SQUASH);
+        githubClient.enableAutoMerge(pr, GithubClient.AutoMergeType.SQUASH);
       }
+
+      githubClient.addLabelsToPR(pr, labels);
+
     } catch (GithubException e) {
       throw new CommandException(e);
     }

diff --git a/common/src/main/java/com/box/l10n/mojito/github/GithubClient.java b/common/src/main/java/com/box/l10n/mojito/github/GithubClient.java
@@ -433,6 +433,16 @@ public void enableAutoMerge(GHPullRequest pullRequest, AutoMergeType autoMergeTy
     }
   }
 
+  public void addLabelsToPR(GHPullRequest pullRequest, List<String> labels) {
+    if (labels != null) {
+      try {
+        pullRequest.addLabels(labels.toArray(new String[0]));
+      } catch (IOException e) {
+        throw new GithubException("Can't add labels to PR", e);
+      }
+    }
+  }
+
   public String getOwner() {
     return owner;
   }