Contents

# Copyright (c) 2020 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
# frozen_string_literal: true

module Contrast
  module Agent
    # This class is instantiated when we receive a request and the agent is enabled to process
    # that request. It holds the ruleset that we perform filtering operations on (currently
    # prefilter and postfilter).
    class RequestHandler
      include Contrast::Components::Interface
      access_component :agent, :logging, :scope

      attr_reader :ruleset, :context

      def initialize context
        @context = context
        @ruleset = AGENT.ruleset
      end

      def send_activity_messages
        Contrast::Utils::GemfileReader.instance.generate_library_usage(context.activity)
        [context.server_activity, context.activity, context.observed_route].each do |message|
          Contrast::Utils::ServiceSenderUtil.push_to_ready_queue(message)
        end
      end

      # If the response is streaming, we should only perform filtering on our stream safe rules
      def stream_safe_postfilter
        stream_safe_ruleset = ruleset.select(&:stream_safe?)
        postfilter_ruleset = Contrast::Agent::RuleSet.new(stream_safe_ruleset)
        postfilter_ruleset.postfilter
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
contrast-agent-3.12.2	lib/contrast/agent/request_handler.rb
contrast-agent-3.12.1	lib/contrast/agent/request_handler.rb
contrast-agent-3.12.0	lib/contrast/agent/request_handler.rb