mirror of
https://github.com/pry/pry.git
synced 2022-11-09 12:35:05 -05:00
Only check for color once
This commit is contained in:
parent
f4b7ddf870
commit
0e221391b0
1 changed files with 3 additions and 7 deletions
|
@ -11,14 +11,10 @@ require 'open4'
|
|||
# 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}
|
||||
COLORS = {'F' => 31, 'E' => 35, 'M' => 33, '.' => 32}
|
||||
USE_COLOR = Pry::Helpers::BaseHelpers.use_ansi_codes?
|
||||
|
||||
module TestUnitOutput
|
||||
def handle_specification(name)
|
||||
@use_color = Pry::Helpers::BaseHelpers.use_ansi_codes?
|
||||
yield
|
||||
end
|
||||
|
||||
def handle_requirement(description)
|
||||
error = yield
|
||||
|
||||
|
@ -44,7 +40,7 @@ module Bacon
|
|||
end
|
||||
|
||||
def colorize_string(text, color = nil)
|
||||
if @use_color
|
||||
if USE_COLOR
|
||||
"\e[#{ COLORS[color || text] }m#{ text }\e[0m"
|
||||
else
|
||||
text
|
||||
|
|
Loading…
Reference in a new issue