1 files changed, 66 insertions, 0 deletions
diff --git a/id/server/idserverlib/src/main/java/at/gv/egovernment/moa/id/protocols/pvp2x/utils/AttributeExtractor.java b/id/server/idserverlib/src/main/java/at/gv/egovernment/moa/id/protocols/pvp2x/utils/AttributeExtractor.java
new file mode 100644
index 000000000..a59fc17c5
--- /dev/null
+++ b/id/server/idserverlib/src/main/java/at/gv/egovernment/moa/id/protocols/pvp2x/utils/AttributeExtractor.java
@@ -0,0 +1,66 @@
+package at.gv.egovernment.moa.id.protocols.pvp2x.utils;
+
+import java.util.Iterator;
+import java.util.List;
+
+import at.gv.egovernment.moa.id.auth.data.AuthenticationSession;
+import at.gv.egovernment.moa.id.auth.data.ExtendedSAMLAttribute;
+import at.gv.egovernment.moa.id.auth.validator.parep.ParepValidator;
+
+public class AttributeExtractor {
+	
+	public static String extractSAMLAttributeOA(String name, 
+			AuthenticationSession authSession) {
+		List extAttributes = authSession.getExtendedSAMLAttributesOA();
+		if(extAttributes == null) {
+			return null;
+		}
+		Iterator extAttributesIt = extAttributes.iterator();
+		String value = null;
+		while(extAttributesIt.hasNext()) {
+			Object attr = extAttributesIt.next();
+			if(attr instanceof ExtendedSAMLAttribute) {
+				ExtendedSAMLAttribute extAttribute = (ExtendedSAMLAttribute) attr;
+				if(extAttribute.getName().equals(name)) {
+					if(extAttribute.getValue() instanceof String) {
+						return extAttribute.getValue().toString();
+					}
+					break;
+				}
+			}
+		}
+		return null;
+	}
+	
+	public static String extractSAMLAttributeAUTH(String name, 
+			AuthenticationSession authSession) {
+		List extAttributes = authSession.getExtendedSAMLAttributesAUTH();
+		if(extAttributes == null) {
+			return null;
+		}
+		Iterator extAttributesIt = extAttributes.iterator();
+		String value = null;
+		while(extAttributesIt.hasNext()) {
+			Object attr = extAttributesIt.next();
+			if(attr instanceof ExtendedSAMLAttribute) {
+				ExtendedSAMLAttribute extAttribute = (ExtendedSAMLAttribute) attr;
+				if(extAttribute.getName().equals(name)) {
+					if(extAttribute.getValue() instanceof String) {
+						return extAttribute.getValue().toString();
+					}
+					break;
+				}
+			}
+		}
+		return null;
+	}
+	
+	public static String extractSAMLAttributeBOTH(String name, 
+			AuthenticationSession authSession) {
+		String value = extractSAMLAttributeOA(name, authSession);
+		if(value == null) {
+			value = extractSAMLAttributeAUTH(name, authSession);
+		}
+		return value;
+	}
+}