Configurable context root for keycloak API

virtualmarc · virtualmarc · commit 3b15592fc34c · 2022-07-20T13:46:05.000+02:00
diff --git a/scribejava-apis/src/main/java/com/github/scribejava/apis/KeycloakApi.java b/scribejava-apis/src/main/java/com/github/scribejava/apis/KeycloakApi.java
@@ -1,29 +1,39 @@
 package com.github.scribejava.apis;
 
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ConcurrentMap;
+
 import com.github.scribejava.apis.openid.OpenIdJsonTokenExtractor;
 import com.github.scribejava.core.builder.api.DefaultApi20;
 import com.github.scribejava.core.extractors.TokenExtractor;
 import com.github.scribejava.core.model.OAuth2AccessToken;
 
-import java.util.concurrent.ConcurrentHashMap;
-import java.util.concurrent.ConcurrentMap;
-
 public class KeycloakApi extends DefaultApi20 {
 
     private static final ConcurrentMap<String, KeycloakApi> INSTANCES = new ConcurrentHashMap<>();
 
     private final String baseUrlWithRealm;
+    private final String contextRoot;
 
     protected KeycloakApi(String baseUrlWithRealm) {
+        this(baseUrlWithRealm, "auth");
+    }
+
+    protected KeycloakApi(String baseUrlWithRealm, String contextRoot) {
         this.baseUrlWithRealm = baseUrlWithRealm;
+        this.contextRoot = contextRoot;
     }
 
     public static KeycloakApi instance() {
         return instance("http://localhost:8080/", "master");
     }
 
     public static KeycloakApi instance(String baseUrl, String realm) {
-        final String defaultBaseUrlWithRealm = composeBaseUrlWithRealm(baseUrl, realm);
+        return instance(baseUrl, realm, "auth");
+    }
+
+    public static KeycloakApi instance(String baseUrl, String realm, String contextRoot) {
+        final String defaultBaseUrlWithRealm = composeBaseUrlWithRealm(baseUrl, realm, contextRoot);
 
         //java8: switch to ConcurrentMap::computeIfAbsent
         KeycloakApi api = INSTANCES.get(defaultBaseUrlWithRealm);
@@ -37,8 +47,10 @@ public static KeycloakApi instance(String baseUrl, String realm) {
         return api;
     }
 
-    protected static String composeBaseUrlWithRealm(String baseUrl, String realm) {
-        return baseUrl + (baseUrl.endsWith("/") ? "" : "/") + "auth/realms/" + realm;
+    protected static String composeBaseUrlWithRealm(String baseUrl, String realm, String contextRoot) {
+        return baseUrl + (baseUrl.endsWith("/") ? "" : "/") +
+            (contextRoot != null && !contextRoot.isEmpty() ? contextRoot + (contextRoot.endsWith("/") ? "" : "/") : "") +
+            "realms/" + realm;
     }
 
     @Override
