1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
ruby--ruby/tool/generic_erb.rb
nobu be8a9db524 colorize
* tool/generic_erb.rb: use VT100 sequence if tput does not work.

* tool/ifchange: ditto and add --color option.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@52220 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
2015-10-22 14:02:38 +00:00

55 lines
1.5 KiB
Ruby

# -*- coding: us-ascii -*-
require 'erb'
require 'optparse'
require 'fileutils'
$:.unshift(File.dirname(__FILE__))
require 'vpath'
vpath = VPath.new
timestamp = nil
output = nil
ifchange = nil
source = false
color = nil
opt = OptionParser.new do |o|
o.on('-t', '--timestamp[=PATH]') {|v| timestamp = v || true}
o.on('-o', '--output=PATH') {|v| output = v}
o.on('-c', '--[no-]if-change') {|v| ifchange = v}
o.on('-x', '--source') {source = true}
o.on('--color') {color = true}
vpath.def_options(o)
o.order!(ARGV)
end
unchanged = "unchanged"
updated = "updated"
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"
unchanged = "#{beg}#{colors["pass"] || "32;1"}m#{unchanged}#{reset}"
updated = "#{beg}#{colors["fail"] || "31;1"}m#{updated}#{reset}"
end
end
template = ARGV.shift or abort opt.to_s
erb = ERB.new(File.read(template), nil, '%-')
erb.filename = template
result = source ? erb.src : erb.result
if output
if ifchange and (vpath.open(output, "rb") {|f| f.read} rescue nil) == result
puts "#{output} #{unchanged}"
else
open(output, "wb") {|f| f.print result}
puts "#{output} #{updated}"
end
if timestamp
if timestamp == true
dir, base = File.split(output)
timestamp = File.join(dir, ".time." + base)
end
FileUtils.touch(timestamp)
end
else
print result
end