mirror of https://github.com/jacekkow/keycloak-protocol-cas

Doccrazy
2017-01-27 1482f2bf59b0e26b74f1e34631d5366e1abaff96
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;
13 import java.util.List;
14 import java.util.Map;
15
16 /**
17  * Transforms the attribute map of the AuthenticationSuccess object (which can contain either simple values or
18  * lists) to a flat list of XML nodes, where the key is the node name.<br>
19  * Lists output multiple XML nodes with the same name.
20  */
21 public final class AttributesMapAdapter extends XmlAdapter<AttributesMapAdapter.AttributeWrapperType, Map<String, Object>> {
22     @Override
23     public AttributeWrapperType marshal(Map<String, Object> v) throws Exception {
24         return new AttributeWrapperType(v);
25     }
26
27     @Override
28     public Map<String, Object> unmarshal(AttributeWrapperType v) throws Exception {
29         throw new IllegalStateException("not implemented");
30     }
31
32     @XmlAccessorType(XmlAccessType.FIELD)
33     static class AttributeWrapperType {
34         @XmlAnyElement
35         private final List<JAXBElement<String>> elements;
36
37         AttributeWrapperType(Map<String, Object> attributes) {
38             this.elements = new ArrayList<>();
39             for (Map.Entry<String, Object> entry : attributes.entrySet()) {
40                 if (entry.getValue() instanceof List) {
41                     for (Object item : ((List) entry.getValue())) {
42                         addElement(entry.getKey(), item);
43                     }
44                 } else {
45                     addElement(entry.getKey(), entry.getValue());
46                 }
47             }
48         }
49
50         private void addElement(String name, Object value) {
51             if (value != null) {
52                 String namespace = CasServiceResponse.class.getPackage().getAnnotation(XmlSchema.class).namespace();
53                 elements.add(new JAXBElement<>(new QName(namespace, name), String.class, value.toString()));
54             }
55         }
56     }
57 }