fix to return a single value from pac4j for certain attributes

TIER · Sep 9, 2021 · 1609b6c · 1609b6c
1 parent 11664ec
commit 1609b6c
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/pac4j-module/src/main/java/net/unicon/shibui/pac4j/BetterSAML2Profile.java b/pac4j-module/src/main/java/net/unicon/shibui/pac4j/BetterSAML2Profile.java
@@ -13,19 +13,26 @@ public BetterSAML2Profile(final SimpleProfileMapping simpleProfileMapping) {
         this.profileMapping = simpleProfileMapping;
     }
 
+    private String getSingleAttributeValue(Object attribute) {
+        if (attribute instanceof Collection) {
+            return (String) ((Collection)attribute).toArray()[0];
+        }
+        return (String) attribute;
+    }
+
     @Override
     public String getEmail() {
-        return (String) getAttribute(profileMapping.getEmail());
+        return getSingleAttributeValue(getAttribute(profileMapping.getEmail()));
     }
 
     @Override
     public String getFamilyName() {
-        return (String) getAttribute(profileMapping.getLastName());
+        return getSingleAttributeValue(getAttribute(profileMapping.getLastName()));
     }
 
     @Override
     public String getFirstName() {
-        return (String) getAttribute(profileMapping.getFirstName());
+        return getSingleAttributeValue(getAttribute(profileMapping.getFirstName()));
     }
 
     public List<String> getGroups() {
@@ -42,4 +49,4 @@ public String getUsername() {
         }
     }
 
-}
+}