From dc4c9d0a956c6b4f9f0b93327990f4219378a0fb Mon Sep 17 00:00:00 2001
From: Mihai Moiseanu <mihai.moiseanu@gmail.com>
Date: Tue, 5 Mar 2024 12:09:15 +0200
Subject: [PATCH] add header for uk traffic

---
 .../core/APIs/implementation/OAuthApiImpl.java     |  3 +++
 src/main/java/com/mangopay/core/Configuration.java | 14 +++++++++++++-
 src/main/java/com/mangopay/core/RestTool.java      |  4 +++-
 3 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/mangopay/core/APIs/implementation/OAuthApiImpl.java b/src/main/java/com/mangopay/core/APIs/implementation/OAuthApiImpl.java
index 0b6fa19f..ba4a3f31 100644
--- a/src/main/java/com/mangopay/core/APIs/implementation/OAuthApiImpl.java
+++ b/src/main/java/com/mangopay/core/APIs/implementation/OAuthApiImpl.java
@@ -52,6 +52,9 @@ public OAuthToken createToken() throws Exception {
         }
         rest.addRequestHttpHeader("Content-Type", "application/x-www-form-urlencoded");
         rest.addRequestHttpHeader("User-Agent",  String.format("MangoPay V2 SDK Java %s", getRoot().getConfig().getVersion()));
+        if(root.getConfig().isUkHeaderFlag()){
+            rest.addRequestHttpHeader("x-tenant-id", "uk");
+        }
         OAuthToken response = rest.request(OAuthToken.class, null, urlMethod, requestType, requestData);
 
         return response;
diff --git a/src/main/java/com/mangopay/core/Configuration.java b/src/main/java/com/mangopay/core/Configuration.java
index 5b1622dc..7f104316 100644
--- a/src/main/java/com/mangopay/core/Configuration.java
+++ b/src/main/java/com/mangopay/core/Configuration.java
@@ -45,6 +45,11 @@ public class Configuration {
      * Mangopay SDK Version
      */
     private String version;
+
+    /**
+     * Set to true for uk traffic
+     */
+    private boolean ukHeaderFlag = false;
     
 
     public String getClientId() {
@@ -115,7 +120,14 @@ public void setReadTimeout(int readTimeout) {
         this.readTimeout = readTimeout;
     }
 
-    
+    public boolean isUkHeaderFlag() {
+        return ukHeaderFlag;
+    }
+
+    public void setUkHeaderFlag(boolean ukHeaderFlag) {
+        this.ukHeaderFlag = ukHeaderFlag;
+    }
+
     /**
      * Get Mangopay SDK Version
      * @return String Mangopay Version
diff --git a/src/main/java/com/mangopay/core/RestTool.java b/src/main/java/com/mangopay/core/RestTool.java
index bfbe7ce8..1f5d85f0 100644
--- a/src/main/java/com/mangopay/core/RestTool.java
+++ b/src/main/java/com/mangopay/core/RestTool.java
@@ -679,7 +679,9 @@ private Map<String, String> getHttpHeaders(String restUrl) throws Exception {
         }
 
         httpHeaders.put("User-Agent",  String.format("MangoPay V2 SDK Java %s", root.getConfig().getVersion()));
-
+        if(this.root.getConfig().isUkHeaderFlag()) {
+            httpHeaders.put("x-tenant-id", "uk");
+        }
         return httpHeaders;
     }