🐛 Source Amazon Seller Partner: fix oauth consent url (#9849)

airbytehq · Nov 17, 2023 · 182c42c · 182c42c
1 parent 0642e70
commit 182c42c
Showing 1 changed file with 63 additions and 2 deletions.
diff --git a/airbyte-oauth/src/main/java/io/airbyte/oauth/flows/AmazonSellerPartnerOAuthFlow.java b/airbyte-oauth/src/main/java/io/airbyte/oauth/flows/AmazonSellerPartnerOAuthFlow.java
@@ -25,8 +25,59 @@
  */
 public class AmazonSellerPartnerOAuthFlow extends BaseOAuth2Flow {
 
-  private static final String AUTHORIZE_URL = "https://sellercentral.amazon.com/apps/authorize/consent";
   private static final String ACCESS_TOKEN_URL = "https://api.amazon.com/auth/o2/token";
+  private static final String sellerEuropeUrl = "https://sellercentral-europe.amazon.com";
+  private static Map<String, String> vendorCentralUrls = ImmutableMap.<String, String>builder()
+      .put("CA", "https://vendorcentral.amazon.ca")
+      .put("US", "https://vendorcentral.amazon.com")
+      .put("DE", "https://vendorcentral.amazon.de")
+      .put("MX", "https://vendorcentral.amazon.com.mx")
+      .put("JP", "https://vendorcentral.amazon.co.jp")
+      .put("IT", "https://vendorcentral.amazon.it")
+      .put("AU", "https://vendorcentral.amazon.com.au")
+      .put("BR", "https://vendorcentral.amazon.com.br")
+      .put("BE", "https://vendorcentral.amazon.com.be")
+      .put("ES", "https://vendorcentral.amazon.es")
+      .put("UK", "https://vendorcentral.amazon.co.uk")
+      .put("GB", "https://vendorcentral.amazon.co.uk")
+      .put("NL", "https://vendorcentral.amazon.nl")
+      .put("PL", "https://vendorcentral.amazon.pl")
+      .put("FR", "https://vendorcentral.amazon.fr")
+      .put("IN", "https://www.vendorcentral.in")
+      .put("SE", "https://vendorcentral.amazon.se")
+      .put("SG", "https://vendorcentral.amazon.com.sg")
+      .put("AE", "https://vendorcentral.amazon.me")
+      .put("TR", "https://vendorcentral.amazon.com.tr")
+      .put("SA", "https://vendorcentral.amazon.me")
+      .put("EG", "https://vendorcentral.amazon.me")
+      .put("ZA", "https://vendorcentral.amazon.co.za")
+      .build();
+
+  private static Map<String, String> sellerCentralUrls = ImmutableMap.<String, String>builder()
+      .put("CA", "https://sellercentral.amazon.ca")
+      .put("US", "https://sellercentral.amazon.com")
+      .put("MX", "https://sellercentral.amazon.com.mx")
+      .put("BR", "https://sellercentral.amazon.com.br")
+      .put("ES", sellerEuropeUrl)
+      .put("UK", sellerEuropeUrl)
+      .put("GB", sellerEuropeUrl)
+      .put("FR", sellerEuropeUrl)
+      .put("NL", "https://sellercentral.amazon.nl")
+      .put("DE", sellerEuropeUrl)
+      .put("IT", sellerEuropeUrl)
+      .put("SE", "https://sellercentral.amazon.se")
+      .put("ZA", "https://sellercentral.amazon.co.za")
+      .put("PL", "https://sellercentral.amazon.pl")
+      .put("EG", "https://sellercentral.amazon.eg")
+      .put("TR", "https://sellercentral.amazon.com.tr")
+      .put("SA", "https://sellercentral.amazon.sa")
+      .put("AE", "https://sellercentral.amazon.ae")
+      .put("IN", "https://sellercentral.amazon.in")
+      .put("BE", "https://sellercentral.amazon.com.be")
+      .put("SG", "https://sellercentral.amazon.sg")
+      .put("AU", "https://sellercentral.amazon.com.au")
+      .put("JP", "https://sellercentral.amazon.co.jp")
+      .build();
 
   @Override
   protected String getClientIdUnsafe(final JsonNode oauthConfig) {
@@ -65,7 +116,17 @@ public String getSourceConsentUrl(final UUID workspaceId,
     }
 
     try {
-      return new URIBuilder(AUTHORIZE_URL)
+      final String accountType = getConfigValueUnsafe(inputOAuthConfiguration, "account_type");
+      final String region = getConfigValueUnsafe(inputOAuthConfiguration, "region");
+      String consentUrl;
+
+      if ("Vendor".equals(accountType)) {
+        consentUrl = vendorCentralUrls.get(region);
+      } else {
+        consentUrl = sellerCentralUrls.get(region);
+      }
+
+      return new URIBuilder(consentUrl + "/apps/authorize/consent")
           /*
            * Airbyte Amazon Seller Partner `application_id`, to provide native OAuth integration for
            * 3rd-parties.