# encoding: utf-8
# This file is distributed under New Relic's license terms.
# See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
# frozen_string_literal: true

require_relative 'instrumentation'

module NewRelic::Agent::Instrumentation
  module Curb
    module Chain
      def self.instrument!
        Curl::Easy.class_eval do
          include NewRelic::Agent::Instrumentation::Curb::Easy

          def http_head_with_newrelic(*args, &blk)
            http_head_with_tracing { http_head_without_newrelic(*args, &blk) }
          end

          alias_method(:http_head_without_newrelic, :http_head)
          alias_method(:http_head, :http_head_with_newrelic)

          def http_post_with_newrelic(*args, &blk)
            http_post_with_tracing { http_post_without_newrelic(*args, &blk) }
          end

          alias_method(:http_post_without_newrelic, :http_post)
          alias_method(:http_post, :http_post_with_newrelic)

          def http_put_with_newrelic(*args, &blk)
            http_put_with_tracing { http_put_without_newrelic(*args, &blk) }
          end

          alias_method(:http_put_without_newrelic, :http_put)
          alias_method(:http_put, :http_put_with_newrelic)

          # Hook the #http method to set the verb.
          def http_with_newrelic(verb)
            http_with_tracing(verb) { http_without_newrelic(verb) }
          end

          alias_method(:http_without_newrelic, :http)
          alias_method(:http, :http_with_newrelic)

          # Hook the #perform method to mark the request as non-parallel.
          def perform_with_newrelic
            perform_with_tracing { perform_without_newrelic }
          end

          alias_method(:perform_without_newrelic, :perform)
          alias_method(:perform, :perform_with_newrelic)

          # Record the HTTP verb for future #perform calls
          def method_with_newrelic(verb)
            method_with_tracing(verb) { method_without_newrelic(verb) }
          end

          alias_method(:method_without_newrelic, :method)
          alias_method(:method, :method_with_newrelic)

          # We override this method in order to ensure access to header_str even
          # though we use an on_header callback
          def header_str_with_newrelic
            header_str_with_tracing { header_str_without_newrelic }
          end

          alias_method(:header_str_without_newrelic, :header_str)
          alias_method(:header_str, :header_str_with_newrelic)
        end

        Curl::Multi.class_eval do
          include NewRelic::Agent::Instrumentation::Curb::Multi

          # Add CAT with callbacks if the request is serial
          def add_with_newrelic(curl)
            add_with_tracing(curl) { add_without_newrelic(curl) }
          end

          alias_method(:add_without_newrelic, :add)
          alias_method(:add, :add_with_newrelic)

          # Trace as an External/Multiple call if the first request isn't serial.
          def perform_with_newrelic(&blk)
            perform_with_tracing { perform_without_newrelic(&blk) }
          end

          alias_method(:perform_without_newrelic, :perform)
          alias_method(:perform, :perform_with_newrelic)
        end
      end
    end
  end
end