Add ability to use multiple instances of middleware

Author: fatkodima

lib/rack/attack.rb

@@ -96,9 +96,16 @@ def reset!
 
     attr_reader :configuration
 
-    def initialize(app)
+    def initialize(app, &block)
       @app = app
-      @configuration = self.class.configuration
+      @configuration =
+        if block_given?
+          configuration = Configuration.new
+          configuration.instance_exec(&block)
+          configuration
+        else
+          self.class.configuration
+        end
     end
 
     def call(env)

spec/acceptance/rails_middleware_spec.rb

@@ -16,5 +16,17 @@
       @app.initialize!
       assert @app.middleware.include?(Rack::Attack)
     end
+
+    it "can be configured via a block" do
+      @app.middleware.delete(Rack::Attack)
+      @app.middleware.use(Rack::Attack) do
+        blocklist_ip("1.2.3.4")
+      end
+      get "/", {}, "REMOTE_ADDR" => "1.2.3.4"
+      assert_equal 403, last_response.status
+
+      get "/", {}, "REMOTE_ADDR" => "4.3.2.1"
+      assert_equal 200, last_response.status
+    end
   end
 end