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

require 'new_relic/agent/transaction/segment'
require 'new_relic/agent/datastores/metric_helper'
require 'new_relic/agent/database'

module NewRelic
  module Agent
    class Transaction
      class DatastoreSegment < Segment
        attr_reader :product, :operation, :collection, :sql_statement, :nosql_statement, :port_path_or_id
        attr_accessor :host, :database_name

        def initialize product, operation, collection = nil, host = nil, port_path_or_id = nil, database_name=nil
          @product = product
          @operation = operation
          @collection = collection
          @sql_statement = nil
          @nosql_statement = nil
          @host = host
          self.port_path_or_id = port_path_or_id
          @database_name = database_name ? database_name.to_s : nil
          super Datastores::MetricHelper.scoped_metric_for(product, operation, collection)
        end

        def port_path_or_id= ppi
          @port_path_or_id = ppi ? ppi.to_s : nil
        end

        def notice_sql sql
          _notice_sql sql
        end

        # @api private
        def _notice_sql sql, config=nil, explainer=nil, binds=nil, name=nil
          return unless record_sql?
          @sql_statement = Database::Statement.new sql, config, explainer, binds, name, host, port_path_or_id, database_name
        end

        def notice_nosql_statement nosql_statement
          return unless record_sql?
          @nosql_statement = nosql_statement
        end

        private

        def record_metrics
          @unscoped_metrics = Datastores::MetricHelper.unscoped_metrics_for(product, operation, collection, host, port_path_or_id)
          super
        end

        def segment_complete
          add_segment_parameters
          notice_sql_statement if sql_statement
          notice_statement if nosql_statement
        end

        def add_segment_parameters
          instance_reporting_enabled = NewRelic::Agent.config[:'datastore_tracer.instance_reporting.enabled']
          db_name_reporting_enabled = NewRelic::Agent.config[:'datastore_tracer.database_name_reporting.enabled']
          return unless instance_reporting_enabled || db_name_reporting_enabled

          params = {}
          add_instance_parameters params if instance_reporting_enabled
          add_database_name_parameter params if db_name_reporting_enabled

          NewRelic::Agent.instance.transaction_sampler.add_node_parameters params
        end

        def add_instance_parameters params
          params[:host] = host if host
          params[:port_path_or_id] = port_path_or_id if port_path_or_id
        end

        def add_database_name_parameter(params)
          params[:database_name] = database_name if database_name
        end

        def notice_sql_statement
          NewRelic::Agent.instance.transaction_sampler.notice_sql_statement(sql_statement, duration)
          NewRelic::Agent.instance.sql_sampler.notice_sql_statement(sql_statement.dup, name, duration)
          nil
        end

        def notice_statement
          NewRelic::Agent.instance.transaction_sampler.notice_nosql_statement(nosql_statement, duration)
          nil
        end

        def record_sql?
          transaction_state.is_sql_recorded?
        end
      end
    end
  end
end