rack-attack 0.1.0 → 0.2.0

data/README.md

@@ -82,7 +82,11 @@ Note that `req` is a [Rack::Request](http://rack.rubyforge.org/doc/classes/Rack/
 Customize the response of throttled requests using an object that adheres to the [Rack app interface](http://rack.rubyforge.org/doc/SPEC.html).
 
     Rack:Attack.throttled_response = lambda do |env|
-      env['rack.attack.throttled'] # name and other data about the matched throttle
+      # name and other data about the matched throttle
+      env['rack.attack.matched']
+      env['rack.attack.match_type']
+      env['rack.attack.match_data']
+
       [ 503, {}, ['Throttled']]
     end
 
@@ -97,9 +101,9 @@ Similarly for blacklisted responses:
 
 Rack::Attack uses the [ActiveSupport::Notifications](http://api.rubyonrails.org/classes/ActiveSupport/Notifications.html) API if available.
 
-You can subscribe to 'rack.attack.{blacklist,throttle,whitelist}' events and log it, graph it, etc:
+You can subscribe to 'rack.attack' events and log it, graph it, etc:
 
-    ActiveSupport::Notifications.subscribe('rack.attack.blacklist') do |name, start, finish, request_id, req|
+    ActiveSupport::Notifications.subscribe('rack.attack') do |name, start, finish, request_id, req|
       puts req.inspect
     end
 
@@ -120,3 +124,9 @@ Rack::Attack complements `iptables` and nginx's [limit_zone module](http://wiki.
 Thanks to [Kickstarter](https://github.com/kickstarter) for sponsoring Rack::Attack development
 
 [![Travis CI](https://secure.travis-ci.org/ktheory/rack-attack.png)](http://travis-ci.org/ktheory/rack-attack)
+
+## License
+
+Copyright (c) 2012 Kickstarter, Inc
+
+Released under an (MIT License](http://opensource.org/licenses/MIT)

data/lib/rack/attack.rb

@@ -34,7 +34,7 @@ module Rack::Attack
       @notifier ||= ActiveSupport::Notifications if defined?(ActiveSupport::Notifications)
       @blacklisted_response ||= lambda {|env| [503, {}, ['Blocked']] }
       @throttled_response   ||= lambda {|env|
-        retry_after = env['rack.attack.matched'][:period] rescue nil
+        retry_after = env['rack.attack.match_data'][:period] rescue nil
         [503, {'Retry-After' => retry_after}, ['Retry later']]
       }
 
@@ -75,8 +75,8 @@ module Rack::Attack
       end
     end
 
-    def instrument(type, payload)
-      notifier.instrument("rack.attack.#{type}", payload) if notifier
+    def instrument(req)
+      notifier.instrument('rack.attack', req) if notifier
     end
 
    def clear!

data/lib/rack/attack/check.rb

@@ -10,8 +10,9 @@ module Rack
       def [](req)
         block[req].tap {|match|
           if match
-            req.env["rack.attack.matched"] = {type => name}
-            Rack::Attack.instrument(type, req)
+            req.env["rack.attack.matched"] = name
+            req.env["rack.attack.match_type"] = type
+            Rack::Attack.instrument(req)
           end
         }
       end

data/lib/rack/attack/throttle.rb

@@ -23,8 +23,10 @@ module Rack
         count = cache.count(key, period)
         (count > limit).tap do |throttled|
           if throttled
-            req.env['rack.attack.matched'] = {:throttle => name, :count => count, :period => period, :limit => limit}
-            Rack::Attack.instrument(:throttle, req)
+            req.env['rack.attack.matched']    = name
+            req.env['rack.attack.match_type'] = :throttle
+            req.env['rack.attack.match_data'] = {:count => count, :period => period, :limit => limit}
+            Rack::Attack.instrument(req)
           end
         end
       end

data/lib/rack/attack/version.rb

@@ -1,5 +1,5 @@
 module Rack
   module Attack
-    VERSION = '0.1.0'
+    VERSION = '0.2.0'
   end
 end

data/spec/rack_attack_spec.rb

@@ -37,7 +37,8 @@ describe 'Rack::Attack' do
         last_response.status.must_equal 503
       end
       it "should tag the env" do
-        last_request.env['rack.attack.matched'].must_equal({:blacklist => "ip #{@bad_ip}"})
+        last_request.env['rack.attack.matched'].must_equal "ip #{@bad_ip}"
+        last_request.env['rack.attack.match_type'].must_equal :blacklist
       end
 
       allow_ok_requests
@@ -57,7 +58,8 @@ describe 'Rack::Attack' do
           last_response.status.must_equal 200
         end
         it "should tag the env" do
-          last_request.env['rack.attack.matched'].must_equal({:whitelist => 'good ua'})
+          last_request.env['rack.attack.matched'].must_equal 'good ua'
+          last_request.env['rack.attack.match_type'].must_equal :whitelist
         end
       end
     end
@@ -86,7 +88,9 @@ describe 'Rack::Attack' do
         last_response.status.must_equal 503
       end
       it 'should tag the env' do
-        last_request.env['rack.attack.matched'].must_equal({:throttle => 'ip/sec', :count => 2, :limit => 1, :period => 1})
+        last_request.env['rack.attack.matched'].must_equal 'ip/sec'
+        last_request.env['rack.attack.match_type'].must_equal :throttle
+        last_request.env['rack.attack.match_data'].must_equal({:count => 2, :limit => 1, :period => 1})
       end
       it 'should set a Retry-After header' do
         last_response.headers['Retry-After'].must_equal 1

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rack-attack
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -121,7 +121,6 @@ files:
 - lib/rack/attack/whitelist.rb
 - lib/rack/attack.rb
 - Rakefile
-- LICENSE
 - README.md
 - spec/rack_attack_spec.rb
 - spec/spec_helper.rb

data/LICENSE

@@ -1,20 +0,0 @@
-Copyright (c) 2012 Aaron Suggs
-
-Permission is hereby granted, free of charge, to any person obtaining
-a copy of this software and associated documentation files (the
-"Software"), to deal in the Software without restriction, including
-without limitation the rights to use, copy, modify, merge, publish,
-distribute, sublicense, and/or sell copies of the Software, and to
-permit persons to whom the Software is furnished to do so, subject to
-the following conditions:
-
-The above copyright notice and this permission notice shall be
-included in all copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
-EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
-MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
-NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
-LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
-OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
-WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.