easy-going-rails/lib/formatters/ansi_formatter.rb

require_relative 'ansi_wrapper'
require_relative 'ansi_colors'

require 'io/console'
require 'amazing_print'
require 'json'

# wraps meanwhile takes care of ansi colors
class AnsiFormatter < SemanticLogger::Formatters::Color
  include AnsiColors

  ANSI_DEBUG = CLEAR + TEXT_GRAY_400
  ANSI_INFO = CLEAR + TEXT_GRAY_100
  ANSI_WARN = CLEAR + BG_YELLOW + TEXT_BLACK
  ANSI_ERROR = CLEAR + BG_RED + TEXT_WHITE
  ANSI_FATAL = CLEAR + BG_MAGENTA + BOLD + TEXT_WHITE
  NAME_MAX_SIZE = 20

  def initialize
    super(color_map: ColorMap.new(
      debug: ANSI_DEBUG,
      info: ANSI_INFO,
      warn: ANSI_WARN,
      error: ANSI_ERROR,
      fatal: ANSI_FATAL
    ))
  end

  def call(log, logger)
    self.log = log
    self.logger = logger
    self.color = color_map[log.level]
    wrap_level(level, message, payload, exception)
  end

  def origin
    origin = log.name.truncate(NAME_MAX_SIZE).center(NAME_MAX_SIZE)
    colorize(origin, AnsiColors::TEXT_CYAN)
  end

  def message
    return unless log.message

    colorize(log.message)
  end

  def exception
    return unless log.exception

    clazz = "#{log.exception.class}\n"
    message = log.exception.message.chomp('')
    backtrace = clean_backtrace(log.exception.backtrace)

    "#{colorize(clazz, ANSI_FATAL)}#{colorize(message, ANSI_ERROR)}#{backtrace}"
  end

  private

  def clean_backtrace(backtrace)
    root_path = Rails.root.to_s
    backtrace = backtrace.select { |line| line.starts_with?(root_path) }
    backtrace = backtrace.map { |line| line.delete_prefix("#{root_path}/") }
    backtrace.compact.join("\n")
  end

  def colorize(text, tint = color)
    "#{tint}#{text}#{AnsiColors::CLEAR}"
  end

  def wrap_level(level, *items)
    prefix = "#{origin} #{colorize('>')} "
    continuation = "#{origin} #{colorize('#')} "
    items.map do |item|
      AnsiWrapper.wrap(item, 100, prefix, continuation)
    end.compact.join("\n") # FIXME: previously was: join(' ')
  end
end