rack-attack/lib/rack/attack/check.rb

# frozen_string_literal: true

module Rack
  class Attack
    class Check
      attr_reader :name, :block, :type
      def initialize(name, options = {}, &block)
        @name = name
        @block = block
        @type = options.fetch(:type, nil)
      end

      def matched_by?(request)
        block.call(request).tap do |match|
          if match
            request.env["rack.attack.matched"] = name
            request.env["rack.attack.match_type"] = type
            Rack::Attack.instrument(request)
          end
        end
      end
    end
  end
end