io.phasetwo.keycloak.config.RealmAttributesConfigLoader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of keycloak-events Show documentation
The newest version!
package io.phasetwo.keycloak.config;

import com.fasterxml.jackson.core.type.TypeReference;
import jakarta.persistence.EntityManager;
import jakarta.persistence.TypedQuery;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Collectors;
import lombok.extern.jbosslog.JBossLog;
import org.keycloak.connections.jpa.JpaConnectionProvider;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.RealmModel;
import org.keycloak.models.jpa.entities.RealmAttributeEntity;
import org.keycloak.util.JsonSerialization;

@JBossLog
public class RealmAttributesConfigLoader {

  public static final String REALM_ATTRIBUTE_CONFIG_PREFIX = "_providerConfig";

  public static  Optional loadConfiguration(
      KeycloakSession session, String realm, String providerId, Class clazz) {
    return loadConfiguration(session, realm, providerId).map(s -> safeConvert(s, clazz));
  }

  public static Optional loadConfiguration(
      KeycloakSession session, String realm, String providerId) {
    RealmModel r = session.realms().getRealmByName(realm);
    if (r == null) return Optional.empty();
    return Optional.of(r.getAttribute(getKey(providerId)));
  }

  public static  List loadConfigurations(
      KeycloakSession session, String realm, String providerId, Class clazz) {
    return loadConfigurations(session, realm, providerId).stream()
        .map(s -> safeConvert(s, clazz))
        .collect(Collectors.toList());
  }

  public static List loadConfigurations(
      KeycloakSession session, String realm, String providerId) {
    log.debugf(
        "loading configurations for realm=%s, provider=%s. using query %s",
        realm, providerId, getKey(providerId));
    EntityManager em = session.getProvider(JpaConnectionProvider.class).getEntityManager();
    TypedQuery query =
        em.createQuery(
            "SELECT ra FROM RealmAttributeEntity ra WHERE ra.realm.name = :realm AND ra.name LIKE :name ORDER BY ra.name",
            RealmAttributeEntity.class);
    query.setParameter("name", "" + getKey(providerId) + "%");
    query.setParameter("realm", realm);
    return query
        .getResultStream()
        .filter(e -> realm.equals(e.getRealm().getName()))
        .map(RealmAttributeEntity::getValue)
        .collect(Collectors.toList());
  }

  private static String getKey(String providerId) {
    return String.format("%s.%s", REALM_ATTRIBUTE_CONFIG_PREFIX, providerId);
  }

  private static  T safeConvert(String s, Class clazz) {
    try {
      return JsonSerialization.readValue(s, clazz);
    } catch (Exception e) {
      log.warn(e);
      return null;
    }
  }

  public static Map safeConvertToMap(String s) {
    try {
      return JsonSerialization.readValue(s, new TypeReference>() {});
    } catch (Exception e) {
      log.warn(e);
      return null;
    }
  }
}