Add a mainBranch configuration for the plugin

Summary: See #2 Differential Revision: https://phabricator.tdrhq.com/D10062
screenshotbot · Sep 20, 2024 · 15596a7 · 15596a7
1 parent 26c5596
commit 15596a7
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 1 deletion.
diff --git a/.idea/misc.xml b/.idea/misc.xml
diff --git a/plugin/src/main/java/io/screenshotbot/gradle/plugin/AbstractIntegrationBuilder.java b/plugin/src/main/java/io/screenshotbot/gradle/plugin/AbstractIntegrationBuilder.java
@@ -110,6 +110,7 @@ protected void prepareTask(Task task, Project project, String mode) {
                     it.mode = mode;
                     it.hostname = extension.getHostname();
                     it.batch = extension.getBatch();
+                    it.mainBranch = extension.getMainBranch();
 
                     it.mustRunAfter(inputTaskName);
                     it.doFirst((innerTask) -> {

diff --git a/plugin/src/main/java/io/screenshotbot/gradle/plugin/ScreenshotbotPlugin.java b/plugin/src/main/java/io/screenshotbot/gradle/plugin/ScreenshotbotPlugin.java
@@ -53,6 +53,17 @@ public String getBatch() {
         public void setBatch(String batch) {
             this.batch = batch;
         }
+
+        public String getMainBranch() {
+            return mainBranch;
+        }
+
+        public void setMainBranch(String mainBranch) {
+            this.mainBranch = mainBranch;
+        }
+
+        private String mainBranch;
+
     }
     @Override
     public void apply(Project target) {

diff --git a/plugin/src/main/java/io/screenshotbot/gradle/plugin/UploadScreenshotsTask.java b/plugin/src/main/java/io/screenshotbot/gradle/plugin/UploadScreenshotsTask.java
@@ -15,6 +15,8 @@ public class UploadScreenshotsTask extends BaseRecorderTask {
     public String mode = "record";
     public String batch = null;
 
+    public String mainBranch = null;
+
     @Inject
     public UploadScreenshotsTask(ExecOperations execOperations) {
         super(execOperations);
@@ -48,6 +50,10 @@ public void uploadChannel() {
                 args.add(this.batch);
             }
 
+            if (this.mainBranch != null && this.mainBranch.length() > 0) {
+                args.add("--main-branch");
+                args.add(this.mainBranch);
+            }
             it.setArgs(args);
         });
     }