summaryrefslogtreecommitdiff
path: root/tool/colorize.rb
diff options
context:
space:
mode:
authornobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>2017-04-21 03:01:12 +0000
committernobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>2017-04-21 03:01:12 +0000
commitbdc6d41640f07d260b66eaaeea38145789227513 (patch)
treef35a75b2fd1550b09cc9aaaff59af047bde85528 /tool/colorize.rb
parent3ca61bb04e668e14da9525a7de3fc1167d1e1859 (diff)
ext/extmk.rb: colorize notes [Feature #13302]
* common.mk (build-ext): pass variables to colorize. * ext/extmk.rb: colorize notes with tool/colorize.rb. * tool/colorize.rb: extract from tool/generic_erb.rb. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@58426 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'tool/colorize.rb')
-rw-r--r--tool/colorize.rb41
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