class Sentry::Rails::LogSubscribers::ActiveRecordSubscriber

def add_db_config_attributes(attributes, db_config)
  return unless db_config
  attributes[:db_system] = db_config[:adapter] if db_config[:adapter]
  if db_config[:database]
    db_name = db_config[:database]
    if db_config[:adapter] == "sqlite3" && db_name.include?("/")
      db_name = File.basename(db_name)
    end
    attributes[:db_name] = db_name
  end
  attributes[:server_address] = db_config[:host] if db_config[:host]
  attributes[:server_port] = db_config[:port] if db_config[:port]
  attributes[:server_socket_address] = db_config[:socket] if db_config[:socket]
end

def build_log_message(statement_name)
  if statement_name && statement_name != "SQL"
    "Database query: #{statement_name}"
  else
    "Database query"
  end
end

def extract_db_config(payload)
  connection = payload[:connection]
  return unless connection
  extract_db_config_from_connection(connection)
end

def extract_db_config_fallback(connection)
  connection.config if connection.respond_to?(:config)
end

def extract_db_config_from_connection(connection)
  if connection.pool.respond_to?(:db_config)
    db_config = connection.pool.db_config
    if db_config.respond_to?(:configuration_hash)
      return db_config.configuration_hash
    elsif db_config.respond_to?(:config)
      return db_config.config
    end
  end
  extract_db_config_fallback(connection)
end

def extract_db_config_from_connection(connection)
  if connection.pool.respond_to?(:spec)
    spec = connection.pool.spec
    if spec.respond_to?(:config)
      return spec.config
    end
  end
  extract_db_config_fallback(connection)
end

def sql(event)
  return unless Sentry.initialized?
  return if EXCLUDED_NAMES.include?(event.payload[:name])
  sql = event.payload[:sql]
  statement_name = event.payload[:name]
  # Rails 5.0.0 doesn't include :cached in the payload, it was added in Rails 5.1
  cached = event.payload.fetch(:cached, false)
  connection_id = event.payload[:connection_id]
  db_config = extract_db_config(event.payload)
  attributes = {
    sql: sql,
    duration_ms: duration_ms(event),
    cached: cached
  }
  binds = event.payload[:binds]
  if Sentry.configuration.send_default_pii && !binds&.empty?
    type_casted_binds = type_casted_binds(event)
    binds.each_with_index do |bind, index|
      name = bind.is_a?(Symbol) ? bind : bind.name
      attributes["db.query.parameter.#{name}"] = type_casted_binds[index].to_s
    end
  end
  attributes[:statement_name] = statement_name if statement_name && statement_name != "SQL"
  attributes[:connection_id] = connection_id if connection_id
  add_db_config_attributes(attributes, db_config)
  message = build_log_message(statement_name)
  log_structured_event(
    message: message,
    level: :info,
    attributes: attributes
  )
end

def type_casted_binds(event)
  event.payload[:type_casted_binds].call
end

def type_casted_binds(event)
  event.payload[:type_casted_binds]
end