Contents

class InboundRequestLoggerMiddleware
  attr_accessor :only_state_change, :path_regexp

  def initialize(app, only_state_change: true, path_regexp: /.*/)
    @app = app
    self.only_state_change = only_state_change
    self.path_regexp = path_regexp
  end

  def call(env)
    request = ActionDispatch::Request.new(env)
    logging = log?(env, request)
    if logging
      @inbound_request_log = InboundRequestLog.from_request(request)
      env["inbound_request_log"] = @inbound_request_log
      request.body.rewind
    end
    status, headers, body = @app.call(env)
    if logging
      @inbound_request_log.update_columns(response_body: parsed_body(body), response_code: status, ended_at: Time.current)
    end
    [status, headers, body]
  end

  private

  def log?(env, request)
    env["REQUEST_URI"] =~ path_regexp && (!only_state_change || request_with_state_change?(request))
  end

  def parsed_body(body)
    return unless body.present?

    if body.respond_to?(:body)
      JSON.parse(body.body)
    elsif body.respond_to?(:[])
      JSON.parse(body[0])
    else
      body
    end
  rescue JSON::ParserError
    body
  end

  def request_with_state_change?(request)
    request.post? || request.put? || request.patch? || request.delete?
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
rails_api_logger-0.4.1	lib/rails_api_logger/inbound_requests_logger_middleware.rb
rails_api_logger-0.4.0	lib/rails_api_logger/inbound_requests_logger_middleware.rb