aderyabin/sniffer

# frozen_string_literal: true

module Sniffer
  module Adapters
    # Curl adapter
    module CurlAdapter
      def http_with_sniffer(verb)
        sniffer_request(verb)

        http_without_sniffer(verb)

        bm = Benchmark.realtime do
          @res = http_without_sniffer(verb)
        end

        sniffer_response(bm)

        @res
      end

      def http_post_with_sniffer(*args)
        sniffer_request(:POST, *args)

        bm = Benchmark.realtime do
          @res = http_post_without_sniffer(*args)
        end

        sniffer_response(bm)

        @res
      end

      # Only used when prepending, see all_prepend.rb
      module Prepend
        include CurlAdapter

        def http(verb)
          sniffer_request(verb)

          super(verb)

          bm = Benchmark.realtime do
            @res = super(verb)
          end

          sniffer_response(bm)

          @res
        end

        def http_post(*args)
          sniffer_request(:POST, *args)

          bm = Benchmark.realtime do
            @res = super(*args)
          end

          sniffer_response(bm)

          @res
        end
      end

      private

      def data_item
        @data_item ||= Sniffer::DataItem.new if Sniffer.enabled?
      end

      # rubocop:disable Metrics/AbcSize, Metrics/MethodLength
      def sniffer_request(verb, *args)
        return unless data_item

        uri = URI(url)
        query = uri.path
        query += "?#{uri.query}" if uri.query

        data_item.request = Sniffer::DataItem::Request.new(host: uri.host,
                                                           method: verb,
                                                           query: query,
                                                           headers: headers.collect.to_h,
                                                           body: args.join("&"),
                                                           port: uri.port)

        Sniffer.store(data_item)
      end
      # rubocop:enable Metrics/AbcSize, Metrics/MethodLength

      def sniffer_response(timing)
        return unless data_item

        _, *http_headers = header_str.split(/[\r\n]+/).map(&:strip)
        http_headers = Hash[http_headers.flat_map { |s| s.scan(/^(\S+): (.+)/) }]

        data_item.response = Sniffer::DataItem::Response.new(status: status.to_i,
                                                             headers: http_headers,
                                                             body: body_str,
                                                             timing: timing)

        Sniffer.notify_response(data_item)
      end
    end
  end
end

if defined?(::Curl::Easy)
  if defined?(Sniffer::Adapters::CurlAdapter::PREPEND)
    Curl::Easy.prepend Sniffer::Adapters::CurlAdapter::Prepend
  else
    Curl::Easy.class_eval do
      include Sniffer::Adapters::CurlAdapter
      alias_method :http_without_sniffer, :http
      alias_method :http, :http_with_sniffer

      alias_method :http_post_without_sniffer, :http_post
      alias_method :http_post, :http_post_with_sniffer
    end
  end
end