class Middleware::MetricsMiddleware

def call(env)
  start_time = Time.now
  @status, @headers, @response = @app.call(env)
  end_time = Time.now
  # If the url contains any CSS or JavaScript files then do not collect metrics for them
  block_words = ["css", "assets", "jpg", "png", "jpeg", "ico"]
  if block_words.any? { |word| env['PATH_INFO'].include?(word) }
    return [@status, @headers, @response]
  end
  response_time = end_time - start_time
  #Prometheus Stuff
  if env['PATH_INFO'] == '/connect/internal/metrics'
    #Do something before each scrape
    if defined? Resque.redis
      app_name = ENV['DEIS_APP'].present? ? "#{ENV['DEIS_APP']}" : "#{Rails.application.class.parent_name}"
      begin
        Resque.redis.ping
        Prometheus::REDIS_CONNECTION.set({connection:'redis',name:app_name},1)
        Prometheus::FINISHED_JOBS.set({type:'resque',name:app_name},Resque.info[:processed])
        Prometheus::PENDING_JOBS.set({type:'resque',name:app_name},Resque.info[:pending])
        Prometheus::ACTIVE_WORKERS.set({type:'resque',name:app_name},Resque.info[:working])
        Prometheus::WORKERS.set({type:'resque',name:app_name},Resque.info[:workers])
        Prometheus::FAILED_JOBS.set({type:'resque',name:app_name},Resque.info[:failed])
      rescue Redis::CannotConnectError
          Prometheus::REDIS_CONNECTION.set({connection:'redis',name:app_name},0)
      end
      if ZuoraConnect.configuration.custom_prometheus_update_block != nil
        ZuoraConnect.configuration.custom_prometheus_update_block.call()
      end
    end
  end
  # Writing to telegraf: Handle 404 and 500 requests
  if @status == 404 || @status == 304
    # Getting the endpoint and content_type
    request_path = @status == 404 ? "ActionController#RoutingError" : env["action_controller.instance"].present? ? "#{env["action_controller.instance"].class}##{env["action_controller.instance"].action_name}" : ""
    # Uncomment following block of code for handling engine requests/requests without controller
    # else
    #   # Handling requests which do not have controllers (engines)
    #   if env["SCRIPT_NAME"].present?
    #     controller_path = "#{env['SCRIPT_NAME'][1..-1]}"
    #     controller_path = controller_path.sub("/", "::")
    #     request_path = "#{controller_path}#UnknownAction"
    #   end
    content_type = @headers['Content-Type'].split(';')[0] if @headers['Content-Type']
    ZuoraConnect::AppInstanceBase.write_to_telegraf("endpoint_name": request_path, "method_name": env['REQUEST_METHOD'], "status_code": @status, "response_time": response_time,  "content_type": content_type, "direction": "inbound") if request_path.present?
  end
  [@status, @headers, @response]
end

def initialize(app)
  @app = app
end