class ElasticAPM::Middleware

def call(env)
  begin
    if running? && !path_ignored?(env)
      transaction = start_transaction(env)
    end
    resp = @app.call env
  rescue InternalError
    raise # Don't report ElasticAPM errors
  rescue ::Exception => e
    context = ElasticAPM.build_context(rack_env: env, for_type: :error)
    ElasticAPM.report(e, context: context, handled: false)
    raise
  ensure
    if transaction
      if resp
        status, headers, _body = resp
        transaction.add_response(status, headers: headers.dup)
        transaction&.outcome = Transaction::Outcome.from_http_status(status)
      else
        transaction&.outcome = Transaction::Outcome::FAILURE
      end
    end
    ElasticAPM.end_transaction http_result(status)
  end
  resp
end

def config
  @config ||= ElasticAPM.agent.config
end

def http_result(status)
  status && "HTTP #{status.to_s[0]}xx"
end

def initialize(app)
  @app = app
end

def path_ignored?(env)
  return true if config.ignore_url_patterns.any? do |r|
    r.match(env['PATH_INFO'])
  end
  return true if config.transaction_ignore_urls.any? do |r|
    r.match(env['PATH_INFO'])
  end
  false
end

def running?
  ElasticAPM.running?
end

def start_transaction(env)
  context = ElasticAPM.build_context(rack_env: env, for_type: :transaction)
  ElasticAPM.start_transaction 'Rack', 'request',
    context: context,
    trace_context: trace_context(env)
end

def trace_context(env)
  TraceContext.parse(env: env)
rescue TraceContext::InvalidTraceparentHeader => e
  warn e.message
  nil
end