# Load AwesomePrint if available
begin
  require 'awesome_print'
rescue LoadError
end

module SemanticLogger
  module Formatters
    class Color
      # Parameters:
      #  Any valid AwesomePrint option for rendering data.
      #  These options can also be changed be creating a `~/.aprc` file.
      #  See: https://github.com/michaeldv/awesome_print
      #
      #  Note: The option :multiline is set to false if not supplied.
      #  Note: Has no effect if Awesome Print is not installed.
      def initialize(options={})
        @ai_options             = options.dup
        @ai_options[:multiline] = false unless @ai_options.has_key?(:multiline)
      end

      # Adds color to the default log formatter
      # Example:
      #   SemanticLogger.add_appender(io: $stdout, formatter: :color)
      def call(log, logger)
        colors      = SemanticLogger::AnsiColors
        level_color = colors::LEVEL_MAP[log.level]

        # Header with date, time, log level and process info
        message     = "#{log.formatted_time} #{level_color}#{log.level_to_s}#{colors::CLEAR} [#{log.process_info}]"

        # Tags
        message << ' ' << log.tags.collect { |tag| "[#{level_color}#{tag}#{colors::CLEAR}]" }.join(' ') if log.tags && (log.tags.size > 0)

        # Duration
        message << " (#{colors::BOLD}#{log.duration_human}#{colors::CLEAR})" if log.duration

        # Class / app name
        message << " #{level_color}#{log.name}#{colors::CLEAR}"

        # Log message
        message << " -- #{log.message}" if log.message

        # Payload: Colorize the payload if the AwesomePrint gem is loaded
        if log.has_payload?
          payload = log.payload
          message << ' -- ' <<
            if !defined?(AwesomePrint) || !payload.respond_to?(:ai)
              payload.inspect
            else
              payload.ai(@ai_options) rescue payload.inspect
            end
        end

        # Exceptions
        if log.exception
          message << " -- Exception: #{colors::BOLD}#{log.exception.class}: #{log.exception.message}#{colors::CLEAR}\n"
          message << log.backtrace_to_s
        end
        message
      end

    end
  end
end