Merge branch 'development_preview'

1 files changed, 18 insertions, 5 deletions

diff --git a/id/oa/src/main/java/at/gv/egovernment/moa/id/demoOA/servlet/pvp2/DemoApplication.java b/id/oa/src/main/java/at/gv/egovernment/moa/id/demoOA/servlet/pvp2/DemoApplication.java
index aeb4d8eac..e36a880ba 100644
--- a/id/oa/src/main/java/at/gv/egovernment/moa/id/demoOA/servlet/pvp2/DemoApplication.java
+++ b/id/oa/src/main/java/at/gv/egovernment/moa/id/demoOA/servlet/pvp2/DemoApplication.java
@@ -198,11 +198,7 @@ public class DemoApplication extends HttpServlet {
 				
 			}
 			
-			//set assertion
-			org.w3c.dom.Document doc = SAML2Utils.asDOMDocument(samlResponse);
-			String assertion = DOMUtils.serializeNode(doc);				
-			bean.setAssertion(assertion);
-			
+	
 			if (samlResponse.getStatus().getStatusCode().getValue().equals(StatusCode.SUCCESS_URI)) {
 		
 				List<org.opensaml.saml2.core.Assertion> saml2assertions = new ArrayList<org.opensaml.saml2.core.Assertion>();
@@ -245,12 +241,28 @@ public class DemoApplication extends HttpServlet {
 			
 				}
 				
+				samlResponse.getAssertions().clear();
+				samlResponse.getAssertions().addAll(saml2assertions);
+				
+				//set assertion
+				org.w3c.dom.Document doc = SAML2Utils.asDOMDocument(samlResponse);
+				String assertion = DOMUtils.serializeNode(doc);				
+				bean.setAssertion(assertion);	
+				
+				String principleId = null;
 				String givenName = null;
 				String familyName = null;
 				String birthday = null;
 				
 				for (org.opensaml.saml2.core.Assertion saml2assertion : saml2assertions) {
 					
+					try {
+						principleId = saml2assertion.getSubject().getNameID().getValue();
+						
+					} catch (Exception e) {
+						log.warn("Can not read SubjectNameId", e);
+					}
+					
 					//loop through the nodes to get what we want
 					List<AttributeStatement> attributeStatements = saml2assertion.getAttributeStatements();
 					for (int i = 0; i < attributeStatements.size(); i++)
@@ -277,6 +289,7 @@ public class DemoApplication extends HttpServlet {
 					
 				}
 									
+				bean.setPrincipleId(principleId);
 				bean.setDateOfBirth(birthday);
 				bean.setFamilyName(familyName);
 				bean.setGivenName(givenName);