diff options
Diffstat (limited to 'tool/colorize.rb')
-rw-r--r-- | tool/colorize.rb | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/tool/colorize.rb b/tool/colorize.rb new file mode 100644 index 0000000000..6b9f568020 --- /dev/null +++ b/tool/colorize.rb @@ -0,0 +1,41 @@ +class Colorize + def initialize(color = nil) + @colors = @reset = nil + if color or (color == nil && STDOUT.tty?) + if (/\A\e\[.*m\z/ =~ IO.popen("tput smso", "r", err: IO::NULL, &:read) rescue nil) + @beg = "\e[" + @colors = (colors = ENV['TEST_COLORS']) ? Hash[colors.scan(/(\w+)=([^:\n]*)/)] : {} + @reset = "#{@beg}m" + end + end + self + end + + DEFAULTS = {"pass"=>"32;1", "fail"=>"31;1", "skip"=>"33;1"} + + def decorate(str, name) + if @colors and color = (@colors[name] || DEFAULTS[name]) + "#{@beg}#{color}m#{str}#{@reset}" + else + str + end + end + + def pass(str) + decorate(str, "pass") + end + + def fail(str) + decorate(str, "fail") + end + + def skip(str) + decorate(str, "skip") + end +end + +if $0 == __FILE__ + colorize = Colorize.new + col = ARGV.shift + ARGV.each {|str| puts colorize.decorate(str, col)} +end |