class InboundRequestsLoggerMiddleware
  attr_accessor :only_state_change, :path_regexp, :skip_request_body_regexp, :skip_response_body_regexp

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

  def call(env)
    request = ActionDispatch::Request.new(env)
    logging = log?(env, request)
    if logging
      env["INBOUND_REQUEST_LOG"] = InboundRequestLog.from_request(request, skip_request_body: skip_request_body?(env))
      request.body.rewind
    end
    status, headers, body = @app.call(env)
    if logging
      updates = {response_code: status, ended_at: Time.current}
      updates[:response_body] = if skip_response_body?(env)
        "[Skipped]"
      else
        parsed_body(body)
      end
      # this usually works. let's be optimistic.
      begin
        env["INBOUND_REQUEST_LOG"].update_columns(updates)
      rescue JSON::GeneratorError => _e # this can be raised by activerecord if the string is not UTF-8.
        env["INBOUND_REQUEST_LOG"].update_columns(updates.except(:response_body))
      end
    end
    [status, headers, body]
  end

  private

  def skip_request_body?(env)
    skip_request_body_regexp && env["PATH_INFO"] =~ skip_request_body_regexp
  end

  def skip_response_body?(env)
    skip_response_body_regexp && env["PATH_INFO"] =~ skip_response_body_regexp
  end

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

  def parsed_body(body)
    return unless body.present?

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

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