class ElasticAPM::Injectors::SinatraInjector

def compile_template(engine, data, opts, *args, &block)
  opts[:__elastic_apm_template_name] =
    case data
    when Symbol then data.to_s
    else format('Inline %s', engine)
    end
  compile_template_without_apm(engine, data, opts, *args, &block)
end

def dispatch!(*args, &block)
  dispatch_without_apm!(*args, &block).tap do
    next unless (transaction = ElasticAPM.current_transaction)
    next unless (route = env['sinatra.route'])
    transaction.name = route
  end
end

def install
  ::Sinatra::Base.class_eval do
    alias dispatch_without_apm! dispatch!
    alias compile_template_without_apm compile_template
    def dispatch!(*args, &block)
      dispatch_without_apm!(*args, &block).tap do
        next unless (transaction = ElasticAPM.current_transaction)
        next unless (route = env['sinatra.route'])
        transaction.name = route
      end
    end
    def compile_template(engine, data, opts, *args, &block)
      opts[:__elastic_apm_template_name] =
        case data
        when Symbol then data.to_s
        else format('Inline %s', engine)
        end
      compile_template_without_apm(engine, data, opts, *args, &block)
    end
  end
end