Change Binary Download Distribution

Author: amaanbs

BrowserStackLocal/BrowserStackLocal/BrowserStackTunnel.cs

@@ -8,6 +8,12 @@
 using System.Security.Principal;
 using Newtonsoft.Json.Linq;
 
+
+using System.Collections.Generic;
+using System.Net.Http;
+using System.Threading.Tasks;
+using Newtonsoft.Json;
+
 namespace BrowserStack
 {
   public enum LocalState { Idle, Connecting, Connected, Error, Disconnected };
@@ -16,7 +22,11 @@ public class BrowserStackTunnel : IDisposable
   {
     static readonly string uname = Util.GetUName();
     static readonly string binaryName = GetBinaryName();
-    static readonly string downloadURL = "https://www.browserstack.com/local-testing/downloads/binaries/" + binaryName;
+
+    static readonly string userAgent = "";
+    static readonly string sourceUrl = null;
+    private bool isFallbackEnabled = false;
+    private Exception downloadFailureException = null;
 
     static readonly string homepath = !IsWindows() ?
                                         Environment.GetFolderPath(Environment.SpecialFolder.Personal) :
@@ -41,7 +51,7 @@ static bool IsDarwin(string osName)
     {
       return osName.Contains("darwin");
     }
-    
+
 
     static bool IsWindows()
     {
@@ -102,8 +112,9 @@ public virtual void addBinaryArguments(string binaryArguments)
       this.binaryArguments = binaryArguments;
     }
 
-    public BrowserStackTunnel()
+    public BrowserStackTunnel(string userAgent)
     {
+      userAgent = userAgent;
       localState = LocalState.Idle;
       output = new StringBuilder();
     }
@@ -121,7 +132,7 @@ public virtual void fallbackPaths()
     public void modifyBinaryPermission()
     {
       if (!IsWindows())
-       {
+      {
         try
         {
           using (Process proc = Process.Start("/bin/bash", $"-c \"chmod 0755 {this.binaryAbsolute}\""))
@@ -143,16 +154,59 @@ public void modifyBinaryPermission()
       }
     }
 
-    public void downloadBinary()
+    private string fetchSourceUrl(string accessKey)
+    {
+      var url = "https://local.browserstack.com/binary/api/v1/endpoint";
+
+      using (var client = new HttpClient())
+      {
+        var data = new Dictionary<string, object>
+        {
+            { "auth_token", accessKey }
+        };
+        client.DefaultRequestHeaders.Add("Content-Type", "application/json");
+        client.DefaultRequestHeaders.Add("user-agent", userAgent);
+
+        if (isFallbackEnabled)
+        {
+          data["error_message"] = downloadFailureException.Message;
+          client.DefaultRequestHeaders.Add("X-Local-Fallback-Cloudflare", "true");
+        }
+
+        string jsonData = JsonConvert.SerializeObject(data);
+        var content = new StringContent(jsonData, Encoding.UTF8, "application/json");
+
+        HttpResponseMessage response = await client.PostAsync(url, content);
+
+        response.EnsureSuccessStatusCode();
+
+        string responseString = await response.Content.ReadAsStringAsync();
+
+        dynamic jsonResponse = JsonConvert.DeserializeObject(responseString);
+
+        Console.WriteLine("Response JSON:");
+        Console.WriteLine(jsonResponse);
+
+        if (jsonResponse.error != null)
+        {
+          throw new Exception(jsonResponse.error);
+        }
+        return jsonResponse.data.endpoint;
+      }
+    }
+
+    public void downloadBinary(string accessKey)
     {
       string binaryDirectory = Path.Combine(this.binaryAbsolute, "..");
       //string binaryAbsolute = Path.Combine(binaryDirectory, binaryName);
 
+      string sourceDownloadUrl = fetchSourceUrl(accessKey);
+
       Directory.CreateDirectory(binaryDirectory);
 
       using (var client = new WebClient())
       {
-        client.DownloadFile(downloadURL, this.binaryAbsolute);
+        client.DownloadFile(sourceDownloadUrl + "/" + binaryName, this.binaryAbsolute);
       }
 
       if (!File.Exists(binaryAbsolute))
@@ -163,7 +217,7 @@ public void downloadBinary()
       modifyBinaryPermission();
     }
 
-    public virtual void Run(string accessKey, string folder, string logFilePath, string processType)
+    public virtual void Run(string accessKey, string folder, string logFilePath, string processType, bool fallbackEnabled, Exception failureException)
     {
       string arguments = "-d " + processType + " ";
       if (folder != null && folder.Trim().Length != 0)
@@ -176,7 +230,9 @@ public virtual void Run(string accessKey, string folder, string logFilePath, str
       }
       if (!File.Exists(binaryAbsolute))
       {
-        downloadBinary();
+        isFallbackEnabled = fallbackEnabled;
+        downloadFailureException = failureException;
+        downloadBinary(accessKey);
       }
 
       if (process != null)

BrowserStackLocal/BrowserStackLocal/Local.cs

@@ -14,6 +14,9 @@ public class Local
     private string argumentString = "";
     private string customBinaryPath = "";
     private string bindingVersion = "";
+    private string userAgent = "browserstack-local-csharp";
+    private bool isFallbackEnabled = false;
+    private Exception downloadFailureException = null;
 
     protected BrowserStackTunnel tunnel = null;
     private static KeyValuePair<string, string> emptyStringPair = new KeyValuePair<string, string>();
@@ -144,11 +147,11 @@ public static string GetVersionStringFromAssemblyEmbedded(string pAssemblyDispla
     public Local()
     {
        bindingVersion = GetVersionStringFromAssemblyEmbedded("BrowserStackLocal");
-       tunnel = new BrowserStackTunnel();
+       userAgent = userAgent + "/" + bindingVersion;
+       tunnel = new BrowserStackTunnel(userAgent);
     }
     public void start(List<KeyValuePair<string, string>> options)
     {
-      tunnel.basePathsIndex = -1;
       foreach (KeyValuePair<string, string> pair in options)
       {
         string key = pair.Key;
@@ -174,38 +177,61 @@ public void start(List<KeyValuePair<string, string>> options)
 
       argumentString += "-logFile \"" + customLogPath + "\" ";
       argumentString += "--source \"c-sharp:" + bindingVersion + "\" ";
-      tunnel.addBinaryPath(customBinaryPath);
       tunnel.addBinaryArguments(argumentString);
-      while (true)
+
+      DownloadVerifyAndRunBinary();
+    }
+
+    private void DownloadVerifyAndRunBinary()
+    {
+      tunnel.basePathsIndex = -1;
+      tunnel.addBinaryPath(customBinaryPath);
+      try
       {
-        bool except = false;
-        try
-        {
-          tunnel.Run(accessKey, folder, customLogPath, "start");
-        }
-        catch (System.ComponentModel.Win32Exception)
-        {
-          except = true;
-        }
-        catch (Exception e)
+        while (true)
         {
-          except = true;
-          Console.WriteLine(e.ToString());
+          bool except = false;
+          try
+          {
+            tunnel.Run(accessKey, folder, customLogPath, "start", isFallbackEnabled, downloadFailureException);
+          }
+          catch (System.ComponentModel.Win32Exception)
+          {
+            except = true;
+          }
+          catch (Exception e)
+          {
+            except = true;
+            Console.WriteLine(e.ToString());
+          }
+          if (except)
+          {
+            tunnel.fallbackPaths();
+          }
+          else
+          {
+            break;
+          }
         }
-        if (except)
+      }
+      catch (Exception err)
+      {
+        if (!isFallbackEnabled)
         {
-          tunnel.fallbackPaths();
+          isFallbackEnabled = true;
+          downloadFailureException = err;
+          downloadVerifyAndRunBinary();
         }
         else
         {
-          break;
+          throw err;
         }
       }
     }
 
     public void stop()
     {
-      tunnel.Run(accessKey, folder, customLogPath, "stop");
+      tunnel.Run(accessKey, folder, customLogPath, "stop", isFallbackEnabled, downloadFailureException);
       tunnel.Kill();
     }
   }