change binary download distribution

Author: amaanbs

src/main/java/com/browserstack/local/Local.java

@@ -55,9 +55,9 @@ public void start(Map<String, String> options) throws Exception {
         startOptions = options;
         LocalBinary lb;
         if (options.get("binarypath") != null) {
-            lb = new LocalBinary(options.get("binarypath"));
+            lb = new LocalBinary(options.get("binarypath"), options.get("key"));
         } else {
-            lb = new LocalBinary("");
+            lb = new LocalBinary("", options.get("key"));
         }
         binaryPath = lb.getBinaryPath();
 

src/main/java/com/browserstack/local/LocalBinary.java

@@ -3,8 +3,11 @@
 import org.apache.commons.io.FileUtils;
 import org.apache.commons.io.IOUtils;
 
+import org.json.JSONObject;
+
 import java.io.IOException;
 import java.io.InputStream;
+import java.io.OutputStream;
 import java.io.BufferedReader;
 import java.io.InputStreamReader;
 import java.io.File;
@@ -15,14 +18,24 @@
 import java.util.zip.GZIPInputStream;
 import java.util.zip.ZipException;
 
+import java.lang.StringBuilder;
+
 class LocalBinary {
 
     private static final String BIN_URL = "https://www.browserstack.com/local-testing/downloads/binaries/";
 
-    private String httpPath;
+    private String binaryFileName;
+
+    private String sourceUrl;
 
     private String binaryPath;
 
+    private Boolean fallbackEnabled = false;
+
+    private Throwable downloadFailureThrowable = null;
+
+    private String key;
+
     private boolean isOSWindows;
 
     private final String orderedPaths[] = {
@@ -31,7 +44,8 @@ class LocalBinary {
             System.getProperty("java.io.tmpdir")
     };
 
-    LocalBinary(String path) throws LocalException {
+    LocalBinary(String path, String key) throws LocalException {
+        this.key = key;
         initialize();
         if (path != "") {
             getBinaryOnPath(path);
@@ -65,8 +79,7 @@ private void initialize() throws LocalException {
             throw new LocalException("Failed to detect OS type");
         }
 
-        String sourceURL = BIN_URL;
-        httpPath = sourceURL + binFileName;
+        this.binaryFileName = binFileName;
     }
 
     private boolean isAlpine() {
@@ -167,8 +180,40 @@ private boolean makePath(String path) {
         }
     }
 
+    private void fetchSourceUrl() {
+        URL url = new URL("https://local.browserstack.com/binary/api/v1/endpoint");
+        URLConnection connection = url.openConnection();
+        
+        connection.setDoOutput(true);
+        connection.setRequestProperty("Content-Type", "application/json");
+        connection.setRequestProperty("User-Agent", "browserstack-local-java/" + Local.getPackageVersion());
+        connection.setRequestProperty("Accept", "application/json");
+        if (fallbackEnabled) connection.setRequestProperty("X-Local-Fallback-Cloudflare", "true");
+
+        String jsonInput = "{\"auth_token\": " + key + (fallbackEnabled ? (", \"error_message\": " + downloadFailureThrowable.getMessage()) : "") + "}";
+
+        try (OutputStream os = connection.getOutputStream()) {
+            byte[] input = jsonInput.getBytes("utf-8");
+            os.write(input, 0, input.length);
+        }
+
+        try (InputStream is = connection.getInputStream();
+             BufferedReader reader = new BufferedReader(new InputStreamReader(is, "utf-8"))) {
+            StringBuilder response = new StringBuilder();
+            String line;
+            while ((line = reader.readLine()) != null) {
+                response.append(line.trim());
+            }
+            String responseBody = response.toString();
+            JSONObject json = new JSONObject(responseBody);
+            this.sourceUrl = json.getJSONObject("data").getString("endpoint");
+        }
+    }
+
     private void downloadBinary(String destParentDir, Boolean custom) throws LocalException {
         try {
+            fetchSourceUrl();
+
             String source = destParentDir;
             if (!custom) {
                 if (!new File(destParentDir).exists())
@@ -179,13 +224,20 @@ private void downloadBinary(String destParentDir, Boolean custom) throws LocalEx
                     source += ".exe";
                 }
             }
-            URL url = new URL(httpPath);
+            URL url = new URL(sourceUrl + '/' + binaryFileName);
 
             File f = new File(source);
-            newCopyToFile(url, f);
-
+            try {
+                newCopyToFile(url, f);
+            } catch (IOException e) {
+                if (fallbackEnabled) throw e;
+                /* Binary download failed due to a server error */
+                fallbackEnabled = true;
+                downloadFailureThrowable = e;
+                downloadBinary(destParentDir, custom);
+            }
             changePermissions(binaryPath);
-        } catch (Exception e) {
+        } catch (Throwable e) {
             throw new LocalException("Error trying to download BrowserStackLocal binary: " + e.getMessage());
         }
     }