Making the config API more flexible

roman · Sep 7, 2009 · e61cfd5 · e61cfd5
1 parent 9ec3067
commit e61cfd5
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/lib/warden_oauth/manager.rb b/lib/warden_oauth/manager.rb
@@ -9,11 +9,16 @@ def self.included(base)
 
       def oauth(service, &block)
         config = Warden::OAuth::Config.new
-        yield config
+        if block_given?
+          if block.arity == 1 
+            yield config 
+          else
+            config.instance_eval(&block)
+          end
+        end
         config.check_requirements
         config.provider_name = service
         Warden::OAuth::Strategy.build(service, config)
-        #@config[:oauth_services][service] = config
       end
 
       module ClassMethods