# frozen_string_literal: true

# Copyright (C) 2015-2020 MongoDB Inc.
#
# Licensed under the Apache License, Version 2.0 (the 'License');
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an 'AS IS' BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

module Mongo
  class Monitoring
    # Subscribes to command events and logs them.
    #
    # @since 2.1.0
    class CommandLogSubscriber
      include Loggable

      # @return [ Hash ] options The options.
      attr_reader :options

      # Constant for the max number of characters to print when inspecting
      # a query field.
      #
      # @since 2.1.0
      LOG_STRING_LIMIT = 250

      # Create the new log subscriber.
      #
      # @example Create the log subscriber.
      #   CommandLogSubscriber.new
      #
      # @param [ Hash ] options The options.
      #
      # @option options [ Logger ] :logger An optional custom logger.
      #
      # @since 2.1.0
      def initialize(options = {})
        @options = options
      end

      # Handle the command started event.
      #
      # @example Handle the event.
      #   subscriber.started(event)
      #
      # @param [ CommandStartedEvent ] event The event.
      #
      # @since 2.1.0
      def started(event)
        return unless logger.debug?

        _prefix = prefix(event,
                         connection_generation: event.connection_generation,
                         connection_id: event.connection_id,
                         server_connection_id: event.server_connection_id)
        log_debug("#{_prefix} | STARTED | #{format_command(event.command)}")
      end

      # Handle the command succeeded event.
      #
      # @example Handle the event.
      #   subscriber.succeeded(event)
      #
      # @param [ CommandSucceededEvent ] event The event.
      #
      # @since 2.1.0
      def succeeded(event)
        return unless logger.debug?

        log_debug("#{prefix(event)} | SUCCEEDED | #{'%.3f' % event.duration}s")
      end

      # Handle the command failed event.
      #
      # @example Handle the event.
      #   subscriber.failed(event)
      #
      # @param [ CommandFailedEvent ] event The event.
      #
      # @since 2.1.0
      def failed(event)
        return unless logger.debug?

        log_debug("#{prefix(event)} | FAILED | #{event.message} | #{event.duration}s")
      end

      private

      def format_command(args)
        truncating? ? truncate(args) : args.inspect
      rescue Exception
        '<Unable to inspect arguments>'
      end

      def prefix(event, connection_generation: nil, connection_id: nil,
                 server_connection_id: nil)
        extra = [ connection_generation, connection_id ].compact.join(':')
        extra = if extra == ''
                  nil
                else
                  "conn:#{extra}"
                end
        extra += " sconn:#{server_connection_id}" if server_connection_id
        "#{event.address} req:#{event.request_id}#{extra && " #{extra}"} | " +
          "#{event.database_name}.#{event.command_name}"
      end

      def truncate(command)
        ((s = command.inspect).length > LOG_STRING_LIMIT) ? "#{s[0..LOG_STRING_LIMIT]}..." : s
      end

      def truncating?
        @truncating ||= (options[:truncate_logs] != false)
      end
    end
  end
end
