# encoding: utf-8

module OneApm
  module Agent
    module Instrumentation
      module QueueTime
        unless defined?(OA_REQUEST_START_HEADER)
          OA_REQUEST_START_HEADER    = 'HTTP_X_REQUEST_START'.freeze
          OA_QUEUE_START_HEADER      = 'HTTP_X_QUEUE_START'.freeze
          OA_MIDDLEWARE_START_HEADER = 'HTTP_X_MIDDLEWARE_START'.freeze
          OA_ALL_QUEUE_METRIC        = 'WebFrontend/QueueTime'.freeze
          # any timestamps before this are thrown out and the parser
          # will try again with a larger unit (2000/1/1 UTC)
          OA_EARLIEST_ACCEPTABLE_TIME = Time.at(946684800)

          OA_CANDIDATE_HEADERS = [
            OA_REQUEST_START_HEADER,
            OA_QUEUE_START_HEADER,
            OA_MIDDLEWARE_START_HEADER
          ].freeze

          OA_DIVISORS = [1_000_000, 1_000, 1]
        end

        module_function

        def parse_frontend_timestamp(headers, now=Time.now)
          earliest = nil

          OA_CANDIDATE_HEADERS.each do |header|
            if headers[header]
              parsed = parse_timestamp(timestamp_string_from_header_value(headers[header]))
              if parsed && (!earliest || parsed < earliest)
                earliest = parsed
              end
            end
          end

          if earliest && earliest > now
            OneApm::Manager.logger.debug("Negative queue time detected, treating as zero: start=#{earliest.to_f} > now=#{now.to_f}")
            earliest = now
          end

          earliest
        end

        def timestamp_string_from_header_value(value)
          case value
          when /^\s*([\d+\.]+)\s*$/ then $1
          # following regexp intentionally unanchored to handle
          # (ie ignore) leading server names
          when /t=([\d+\.]+)/       then $1
          end
        end

        def parse_timestamp(string)
          OA_DIVISORS.each do |divisor|
            begin
              t = Time.at(string.to_f / divisor)
              return t if t > OA_EARLIEST_ACCEPTABLE_TIME
            rescue RangeError
              # On Ruby versions built with a 32-bit time_t, attempting to
              # instantiate a Time object in the far future raises a RangeError,
              # in which case we know we've chosen the wrong divisor.
            end
          end

          nil
        end
      end
    end
  end
end