# Colorize output (based on greeneggs (c) 2009 Michael Fleet)
# TODO: Make own gem (assigned to rking)
module Bacon
  COLORS    = {'F' => 31, 'E' => 35, 'M' => 33, '.' => 32}
  USE_COLOR = !(ENV['NO_PRY_COLORED_BACON'] == 'true') && Pry::Helpers::BaseHelpers.use_ansi_codes?

  module TestUnitOutput
    def handle_requirement(description)
      error = yield

      if error.empty?
        print colorize_string('.')
      else
        print colorize_string(error[0..0])
      end
    end

    def handle_summary
      puts
      puts ErrorLog if Backtraces

      out = "%d tests, %d assertions, %d failures, %d errors" %
        Counter.values_at(:specifications, :requirements, :failed, :errors)

      if Counter.values_at(:failed, :errors).inject(:+) > 0
        puts colorize_string(out, 'F')
      else
        puts colorize_string(out, '.')
      end
    end

    def colorize_string(text, color = nil)
      if USE_COLOR
        "\e[#{ COLORS[color || text] }m#{ text }\e[0m"
      else
        text
      end
    end
  end
end

# Reset toplevel binding at the beginning of each test case.
module Bacon
  class Context
    alias _real_it it
    def it(description, &block)
      Pry.toplevel_binding = nil
      _real_it(description, &block)
    end
  end
end