read attributes from config

1 files changed, 8 insertions, 4 deletions

diff --git a/src/saml-idp/main.rb b/src/saml-idp/main.rb
index 03538c5..86aa6bb 100644
--- a/src/saml-idp/main.rb
+++ b/src/saml-idp/main.rb
@@ -25,14 +25,18 @@ class Configuration
 end
 
 class User
+  def initialize(attributes)
+    @attributes = attributes
+  end
+
   def name_id_for(name_id_format)
-    $config[:email]
+    @attributes[:email]
   end
 
   def assertion_attributes_for(request)
     {
-      Username: $config[:email],
-      MemberOf: $config[:member_of]
+      Username: @attributes[:email],
+      MemberOf: @attributes[:member_of]
     }
   end
 end
@@ -108,7 +112,7 @@ class IdentityProvider
     saml_request = binding_for(request).deserialize(params)
     @builder = nil
     url, saml_params = saml_request.response_for(
-      User.new,
+      User.new($config),
       binding: :http_post,
       relay_state: params[:RelayState]
     ) do |builder|