Refactor user agent

starkbank · Apr 28, 2020 · 1de0a34 · 1de0a34
1 parent 1ba62f5
commit 1de0a34
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/main/java/com/starkbank/User.java b/src/main/java/com/starkbank/User.java
@@ -7,6 +7,7 @@
 
 public abstract class User extends Resource{
     public static Project defaultUser = null;
+    public static String userAgentOverride = null;
 
     public final String pem;
     public final String environment;

diff --git a/src/main/java/com/starkbank/utils/Response.java b/src/main/java/com/starkbank/utils/Response.java
@@ -18,6 +18,8 @@
 import java.time.Instant;
 import java.util.Map;
 
+import static com.starkbank.User.userAgentOverride;
+
 
 public final class Response {
 
@@ -96,7 +98,7 @@ private static HttpResponse prepareFetch(String path, String method, JsonObject
                 .setHeader("Access-Id", user.accessId())
                 .setHeader("Access-Time", accessTime)
                 .setHeader("Access-Signature", signature.toBase64())
-                .setHeader("User-Agent", "Java-" + System.getProperty("java.version") + "-SDK-0.1.2")
+                .setHeader("User-Agent", getUserAgent())
                 .setHeader("Content-Type", "application/json");
 
         if (method.equals("POST") || method.equals("PATCH")) {
@@ -121,4 +123,8 @@ private static String host(Project user, String version) {
         }
     }
 
+    private static String getUserAgent() {
+        String javaUserAgent = "Java-" + System.getProperty("java.version") + "-SDK-0.1.2";
+        return (userAgentOverride == null) ? javaUserAgent : userAgentOverride;
+    }
 }