# frozen_string_literal: true

# Copyright The OpenTelemetry Authors
#
# SPDX-License-Identifier: Apache-2.0

module OpenTelemetry
  module Instrumentation
    module ActiveRecord
      module Patches
        # Module to prepend to ActiveRecord::Base for instrumentation
        module PersistenceClassMethods
          def self.prepended(base)
            class << base
              prepend ClassMethods
            end
          end

          # Contains ActiveRecord::Persistence::ClassMethods to be patched
          module ClassMethods
            def create(attributes = nil, &block)
              tracer.in_span("#{self}.create") do
                super
              end
            end

            def create!(attributes = nil, &block)
              tracer.in_span("#{self}.create!") do
                super
              end
            end

            def instantiate(attributes, column_types = {}, &block)
              tracer.in_span("#{self}.instantiate") do
                super
              end
            end

            def update(id = :all, attributes) # rubocop:disable Style/OptionalArguments
              tracer.in_span("#{self}.update") do
                super
              end
            end

            def destroy(id)
              tracer.in_span("#{self}.destroy") do
                super
              end
            end

            def delete(id_or_array)
              tracer.in_span("#{self}.delete") do
                super
              end
            end

            private

            def tracer
              ActiveRecord::Instrumentation.instance.tracer
            end
          end
        end
      end
    end
  end
end
