# File lib/rack/attack.rb, line 20 def blacklist(name, &block) self.blacklists[name] = Blacklist.new(name, block) end
# File lib/rack/attack.rb, line 72 def blacklisted?(req) blacklists.any? do |name, blacklist| blacklist[req] end end
# File lib/rack/attack.rb, line 33 def blacklists; @blacklists ||= {}; end
# File lib/rack/attack.rb, line 94 def cache @cache ||= Cache.new end
# File lib/rack/attack.rb, line 51 def call(env) req = Rack::Request.new(env) if whitelisted?(req) @app.call(env) elsif blacklisted?(req) blacklisted_response[env] elsif throttled?(req) throttled_response[env] else tracked?(req) @app.call(env) end end
# File lib/rack/attack.rb, line 98 def clear! @whitelists, @blacklists, @throttles = {}, {}, {} end
# File lib/rack/attack.rb, line 90 def instrument(req) notifier.instrument('rack.attack', req) if notifier end
# File lib/rack/attack.rb, line 37 def new(app) @app = app # Set defaults @notifier ||= ActiveSupport::Notifications if defined?(ActiveSupport::Notifications) @blacklisted_response ||= lambda {|env| [401, {}, ["Unauthorized\n"]] } @throttled_response ||= lambda {|env| retry_after = env['rack.attack.match_data'][:period] rescue nil [429, {'Retry-After' => retry_after.to_s}, ["Retry later\n"]] } self end
# File lib/rack/attack.rb, line 24 def throttle(name, options, &block) self.throttles[name] = Throttle.new(name, options, block) end
# File lib/rack/attack.rb, line 78 def throttled?(req) throttles.any? do |name, throttle| throttle[req] end end
# File lib/rack/attack.rb, line 34 def throttles; @throttles ||= {}; end
# File lib/rack/attack.rb, line 28 def track(name, &block) self.tracks[name] = Track.new(name, block) end
# File lib/rack/attack.rb, line 84 def tracked?(req) tracks.each_value do |tracker| tracker[req] end end
# File lib/rack/attack.rb, line 16 def whitelist(name, &block) self.whitelists[name] = Whitelist.new(name, block) end
Generated with the Darkfish Rdoc Generator 2.