Download binary at custom path if not present already

Author: AdityaHirapara

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

@@ -53,12 +53,13 @@ public Local() {
      */
     public void start(Map<String, String> options) throws Exception {
         startOptions = options;
+        LocalBinary lb;
         if (options.get("binarypath") != null) {
-            binaryPath = options.get("binarypath");
+            lb = new LocalBinary(options.get("binarypath"));
         } else {
-            LocalBinary lb = new LocalBinary();
-            binaryPath = lb.getBinaryPath();
+            lb = new LocalBinary("");
         }
+        binaryPath = lb.getBinaryPath();
 
         makeCommand(options, "start");
 
@@ -106,12 +107,13 @@ public void stop() throws Exception {
     * @param options Options supplied for the Local instance
     **/
     public void stop(Map<String, String> options) throws Exception {
+        LocalBinary lb;
         if (options.get("binarypath") != null) {
-            binaryPath = options.get("binarypath");
+            lb = new LocalBinary(options.get("binarypath"));
         } else {
-            LocalBinary lb = new LocalBinary();
-            binaryPath = lb.getBinaryPath();
+            lb = new LocalBinary("");
         }
+        binaryPath = lb.getBinaryPath();
         makeCommand(options, "stop");
         proc = runCommand(command);
         proc.waitFor();

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

@@ -24,9 +24,13 @@ class LocalBinary {
             System.getProperty("java.io.tmpdir")
     };
 
-    LocalBinary() throws LocalException {
+    LocalBinary(String path) throws LocalException {
         initialize();
-        getBinary();
+        if (path != "") {
+            getBinaryOnPath(path);
+        } else {
+            getBinary();
+        }
         checkBinary();
     }
 
@@ -111,6 +115,14 @@ private boolean validateBinary() throws LocalException{
         }
     }
 
+    private void getBinaryOnPath(String path) throws LocalException {
+        binaryPath = path;
+
+        if (!new File(binaryPath).exists()) {
+            downloadBinary(binaryPath, true);
+        }
+    }
+
     private void getBinary() throws LocalException {
         String destParentDir = getAvailableDirectory();
         binaryPath = destParentDir + "/BrowserStackLocal";
@@ -120,7 +132,7 @@ private void getBinary() throws LocalException {
         }
 
         if (!new File(binaryPath).exists()) {
-            downloadBinary(destParentDir);
+            downloadBinary(destParentDir, false);
         }
     }
 
@@ -147,23 +159,26 @@ private boolean makePath(String path) {
         }
     }
 
-    private void downloadBinary(String destParentDir) throws LocalException {
+    private void downloadBinary(String destParentDir, Boolean custom) throws LocalException {
         try {
-            if (!new File(destParentDir).exists())
-                new File(destParentDir).mkdirs();
-
-            URL url = new URL(httpPath);
-            String source = destParentDir + "/BrowserStackLocal";
-            if (isOSWindows) {
-                source += ".exe";
+            String source = destParentDir;
+            if (!custom) {
+                if (!new File(destParentDir).exists())
+                    new File(destParentDir).mkdirs();
+
+                source = destParentDir + "/BrowserStackLocal";
+                if (isOSWindows) {
+                    source += ".exe";
+                }
             }
+            URL url = new URL(httpPath);
 
             File f = new File(source);
             FileUtils.copyURLToFile(url, f);
 
             changePermissions(binaryPath);
         } catch (Exception e) {
-            throw new LocalException("Error trying to download BrowserStackLocal binary");
+            throw new LocalException("Error trying to download BrowserStackLocal binary: " + e.getMessage());
         }
     }