Merge pull request #7 from amboss-mededu/pull-multiple-attributes

Pulling all atrributes from SAML
amboss-mededu · Mar 14, 2022 · fb30d1c · fb30d1c
2 parents 8bfe056 + 81ad829
commit fb30d1c
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/samlsp/session.go b/samlsp/session.go
@@ -87,3 +87,15 @@ func AttributeFromContext(ctx context.Context, name string) string {
 	}
 	return sa.GetAttributes().Get(name)
 }
+
+func GetAllAttributesFromContext(ctx context.Context, name string) []string {
+	s := SessionFromContext(ctx)
+	if s == nil {
+		return []string{""}
+	}
+	sa, ok := s.(SessionWithAttributes)
+	if !ok {
+		return []string{""}
+	}
+	return sa.GetAttributes().GetAll(name)
+}
diff --git a/samlsp/session_jwt.go b/samlsp/session_jwt.go
@@ -145,3 +145,11 @@ func (a Attributes) Get(key string) string {
 	}
 	return v[0]
 }
+
+func (a Attributes) GetAll(key string) []string {
+	if a == nil {
+		return []string{""}
+	}
+	v := a[key]
+	return v
+}