[java] dynamically obtain current git branch for visual by default

Author: titusfortner

java/main/src/main/java/com/saucelabs/saucebindings/GitManager.java

@@ -0,0 +1,24 @@
+package com.saucelabs.saucebindings;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStreamReader;
+
+public class GitManager {
+    private static String currentBranch = "_default_";
+
+    public static String getCurrentBranch() {
+        try {
+            Process process = Runtime.getRuntime().exec("git rev-parse --abbrev-ref HEAD");
+            process.waitFor();
+            BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
+            String line = reader.readLine();
+            if (!line.contains("fatal:")) {
+                currentBranch = line;
+            }
+        } catch (IOException | InterruptedException e) {
+            // Ignore exception and use default
+        }
+        return currentBranch;
+    }
+}

java/main/src/main/java/com/saucelabs/saucebindings/options/VisualOptions.java

@@ -1,6 +1,7 @@
 package com.saucelabs.saucebindings.options;
 
 import com.saucelabs.saucebindings.CITools;
+import com.saucelabs.saucebindings.GitManager;
 import com.saucelabs.saucebindings.SauceSession;
 import com.saucelabs.saucebindings.SystemManager;
 import lombok.AccessLevel;
@@ -73,7 +74,7 @@ public MutableCapabilities toCapabilities() {
         }
 
         if (branch == null) {
-            branch = "_default_";
+            branch = GitManager.getCurrentBranch();
         }
 
         capabilityManager.addCapabilities();

java/main/src/test/java/com/saucelabs/saucebindings/options/VisualOptionsTest.java

@@ -2,6 +2,7 @@
 
 import com.saucelabs.saucebindings.Browser;
 import com.saucelabs.saucebindings.CITools;
+import com.saucelabs.saucebindings.GitManager;
 import com.saucelabs.saucebindings.SaucePlatform;
 import com.saucelabs.saucebindings.SystemManager;
 import org.junit.Test;
@@ -21,7 +22,7 @@ public void createsDefaultVisualOptions() {
 
         MutableCapabilities visualCapabilities = new MutableCapabilities();
         visualCapabilities.setCapability("apiKey", System.getenv("SCREENER_API_KEY"));
-        visualCapabilities.setCapability("branch", "_default_");
+        visualCapabilities.setCapability("branch", GitManager.getCurrentBranch());
         visualCapabilities.setCapability("projectName", CITools.getBuildName());
 
         assertEquals(visualCapabilities, visualOptions.toCapabilities().getCapability("sauce:visual"));