# Copyright (c) 2016 SolarWinds, LLC.
# All rights reserved.

module AppOpticsAPM
  module Inst
    module ConnectionAdapters
      module Utils

        def extract_trace_details(sql, name = nil, binds = [])
          opts = {}
          if AppOpticsAPM::Config[:sanitize_sql]
            # Sanitize SQL and don't report binds
            opts[:Query] = AppOpticsAPM::Util.sanitize_sql(sql)
          else
            # Report raw SQL and any binds if they exist
            opts[:Query] = sql.to_s
            opts[:QueryArgs] = binds.map { |col, val| [col.name, val.to_s] } unless binds.empty?
          end

          opts[:Name] = name.to_s if name
          opts[:Backtrace] = AppOpticsAPM::API.backtrace if AppOpticsAPM::Config[:active_record][:collect_backtraces]

          config = ActiveRecord::Base.connection_config
          if config
            opts[:Database]   = config[:database] if config.key?(:database)
            opts[:RemoteHost] = config[:host]     if config.key?(:host)
            adapter_name = config[:adapter]

            case adapter_name
            when /mysql/i
              opts[:Flavor] = 'mysql'
            when /^postgres|^postgis/i
              opts[:Flavor] = 'postgresql'
            end
          end
        rescue StandardError => e
          AppOpticsAPM.logger.debug "[appoptics_apm/rails] Exception raised capturing ActiveRecord KVs: #{e.inspect}"
          AppOpticsAPM.logger.debug e.backtrace.join('\n')
        ensure
          return opts
        end

        # We don't want to trace framework caches.  Only instrument SQL that
        # directly hits the database.
        def ignore_payload?(name)
          %w(SCHEMA EXPLAIN CACHE).include?(name.to_s) ||
            (name && name.to_sym == :skip_logging) ||
            name == 'ActiveRecord::SchemaMigration Load'
        end

        def execute_with_appoptics(sql, name = nil)
          if AppOpticsAPM.tracing? && !ignore_payload?(name)

            opts = extract_trace_details(sql, name)
            AppOpticsAPM::API.trace('activerecord', opts, :ar_started) do
              execute_without_appoptics(sql, name)
            end
          else
            execute_without_appoptics(sql, name)
          end
        end

        def exec_query_with_appoptics(sql, name = nil, binds = [])
          if AppOpticsAPM.tracing? && !ignore_payload?(name)

            opts = extract_trace_details(sql, name, binds)
            AppOpticsAPM::API.trace('activerecord', opts, :ar_started) do
              exec_query_without_appoptics(sql, name, binds)
            end
          else
            exec_query_without_appoptics(sql, name, binds)
          end
        end

        def exec_delete_with_appoptics(sql, name = nil, binds = [])
          if AppOpticsAPM.tracing? && !ignore_payload?(name)

            opts = extract_trace_details(sql, name, binds)
            AppOpticsAPM::API.trace('activerecord', opts, :ar_started) do
              exec_delete_without_appoptics(sql, name, binds)
            end
          else
            exec_delete_without_appoptics(sql, name, binds)
          end
        end

        def exec_update_with_appoptics(sql, name = nil, binds = [])
          if AppOpticsAPM.tracing? && !ignore_payload?(name)

            opts = extract_trace_details(sql, name, binds)
            AppOpticsAPM::API.trace('activerecord', opts, :ar_started) do
              exec_update_without_appoptics(sql, name, binds)
            end
          else
            exec_update_without_appoptics(sql, name, binds)
          end
        end

        def exec_insert_with_appoptics(sql, name = nil, binds = [], *args)
          if AppOpticsAPM.tracing? && !ignore_payload?(name)

            opts = extract_trace_details(sql, name, binds)
            AppOpticsAPM::API.trace('activerecord', opts, :ar_started) do
              exec_insert_without_appoptics(sql, name, binds, *args)
            end
          else
            exec_insert_without_appoptics(sql, name, binds, *args)
          end
        end

        def begin_db_transaction_with_appoptics
          AppOpticsAPM::API.trace('activerecord', { :Query => 'BEGIN', :Flavor => :mysql }, :ar_started) do
            begin_db_transaction_without_appoptics
          end
        end
      end # Utils
    end
  end
end