mirror of https://github.com/jacekkow/keycloak-protocol-cas
| commit | author | age | ||
| 513246 | 1 | package org.keycloak.protocol.cas.utils; |
| MP | 2 | |
| 3 | import org.keycloak.protocol.cas.representations.CasServiceResponse; | |
| 4 | ||
| 5 | import javax.xml.bind.JAXBElement; | |
| 6 | import javax.xml.bind.annotation.XmlAccessType; | |
| 7 | import javax.xml.bind.annotation.XmlAccessorType; | |
| 8 | import javax.xml.bind.annotation.XmlAnyElement; | |
| 9 | import javax.xml.bind.annotation.XmlSchema; | |
| 10 | import javax.xml.bind.annotation.adapters.XmlAdapter; | |
| 11 | import javax.xml.namespace.QName; | |
| 12 | import java.util.ArrayList; | |
| 0ad1a9 | 13 | import java.util.Collection; |
| 513246 | 14 | import java.util.List; |
| MP | 15 | import java.util.Map; |
| 16 | ||
| 17 | /** | |
| 18 | * Transforms the attribute map of the AuthenticationSuccess object (which can contain either simple values or | |
| 19 | * lists) to a flat list of XML nodes, where the key is the node name.<br> | |
| 20 | * Lists output multiple XML nodes with the same name. | |
| 21 | */ | |
| 22 | public final class AttributesMapAdapter extends XmlAdapter<AttributesMapAdapter.AttributeWrapperType, Map<String, Object>> { | |
| 23 | @Override | |
| 24 | public AttributeWrapperType marshal(Map<String, Object> v) throws Exception { | |
| 25 | return new AttributeWrapperType(v); | |
| 26 | } | |
| 27 | ||
| 28 | @Override | |
| 29 | public Map<String, Object> unmarshal(AttributeWrapperType v) throws Exception { | |
| 30 | throw new IllegalStateException("not implemented"); | |
| 31 | } | |
| 32 | ||
| 33 | @XmlAccessorType(XmlAccessType.FIELD) | |
| 34 | static class AttributeWrapperType { | |
| 35 | @XmlAnyElement | |
| 36 | private final List<JAXBElement<String>> elements; | |
| 37 | ||
| 38 | AttributeWrapperType(Map<String, Object> attributes) { | |
| 39 | this.elements = new ArrayList<>(); | |
| 40 | for (Map.Entry<String, Object> entry : attributes.entrySet()) { | |
| 0ad1a9 | 41 | if (entry.getValue() instanceof Collection) { |
| MP | 42 | for (Object item : ((Collection) entry.getValue())) { |
| 513246 | 43 | addElement(entry.getKey(), item); |
| MP | 44 | } |
| 45 | } else { | |
| 46 | addElement(entry.getKey(), entry.getValue()); | |
| 47 | } | |
| 48 | } | |
| 49 | } | |
| 50 | ||
| 51 | private void addElement(String name, Object value) { | |
| 52 | if (value != null) { | |
| 53 | String namespace = CasServiceResponse.class.getPackage().getAnnotation(XmlSchema.class).namespace(); | |
| 54 | elements.add(new JAXBElement<>(new QName(namespace, name), String.class, value.toString())); | |
| 55 | } | |
| 56 | } | |
| 57 | } | |
| 58 | } | |
