From 2e3b4dea824af204c49ee3310896faf6d107a8c8 Mon Sep 17 00:00:00 2001 From: Matthias Piepkorn <mpiepk@gmail.com> Date: Fri, 27 Jan 2017 22:53:54 +0000 Subject: [PATCH] Fix .gitignore, add .gitattributes --- src/main/java/org/keycloak/protocol/cas/mappers/GroupMembershipMapper.java | 25 ++++++++++++++++++++----- 1 files changed, 20 insertions(+), 5 deletions(-) diff --git a/src/main/java/org/keycloak/protocol/cas/mappers/GroupMembershipMapper.java b/src/main/java/org/keycloak/protocol/cas/mappers/GroupMembershipMapper.java index 665a6e8..6d9c8ac 100644 --- a/src/main/java/org/keycloak/protocol/cas/mappers/GroupMembershipMapper.java +++ b/src/main/java/org/keycloak/protocol/cas/mappers/GroupMembershipMapper.java @@ -1,15 +1,14 @@ package org.keycloak.protocol.cas.mappers; +import org.keycloak.models.GroupModel; import org.keycloak.models.ProtocolMapperModel; +import org.keycloak.models.UserSessionModel; +import org.keycloak.models.utils.ModelToRepresentation; import org.keycloak.protocol.cas.CASLoginProtocol; -import org.keycloak.protocol.oidc.OIDCLoginProtocol; import org.keycloak.protocol.oidc.mappers.OIDCAttributeMapperHelper; import org.keycloak.provider.ProviderConfigProperty; -import java.util.ArrayList; -import java.util.HashMap; -import java.util.List; -import java.util.Map; +import java.util.*; public class GroupMembershipMapper extends AbstractCASProtocolMapper { private static final List<ProviderConfigProperty> configProperties = new ArrayList<ProviderConfigProperty>(); @@ -48,6 +47,22 @@ return "Map user group membership"; } + @Override + public void setAttribute(Map<String, Object> attributes, ProtocolMapperModel mappingModel, UserSessionModel userSession) { + List<String> membership = new LinkedList<>(); + boolean fullPath = useFullPath(mappingModel); + for (GroupModel group : userSession.getUser().getGroups()) { + if (fullPath) { + membership.add(ModelToRepresentation.buildGroupPath(group)); + } else { + membership.add(group.getName()); + } + } + String protocolClaim = mappingModel.getConfig().get(OIDCAttributeMapperHelper.TOKEN_CLAIM_NAME); + + attributes.put(protocolClaim, membership); + } + public static boolean useFullPath(ProtocolMapperModel mappingModel) { return "true".equals(mappingModel.getConfig().get("full.path")); } -- Gitblit v1.9.1