#!/usr/bin/env ruby
# colorize the output of rspec stories
# eg: stakeout "stories/all.rb | ./colorize.rb" stories/* stories/**/*
module Color
COLORS = { :clear => 0, :red => 31, :green => 32, :yellow => 33 }
def self.method_missing(color_name, *args)
color(color_name) + args.first + color(:clear)
end
def self.color(color)
"\e[#{COLORS[color.to_sym]}m"
end
end
# Growl stuff ---------
AUTOTEST_IMAGE_ROOT = "~/.autotest_images"
USE_GROWL = system 'which growlnotify'
def self.growl(title, msg, img, pri=0, sticky="")
system "growlnotify -n autotest --image #{img} -p #{pri} -m '#{title}: #{msg.strip}' #{sticky}" if USE_GROWL
true
end
def growl_success(msg); growl("PASS", msg, "#{AUTOTEST_IMAGE_ROOT}/pass.png", 2); end
def growl_fail(msg); growl("FAIL", msg, "#{AUTOTEST_IMAGE_ROOT}/fail.png", 2); end
def growl_pending(msg); growl("PENDING", msg, "#{AUTOTEST_IMAGE_ROOT}/pending.png", 2); end
# Summary -------------
# 14 scenarios: 13 succeeded, 0 failed, 1 pending
RESULTS_LINE = /(\d+)\s+scenarios:\s*(\d+)\s+succeeded?,\s*(\d+)\s+failed?(,\s*(\d+)\s+pending)?.*/
# colorize the summary
def colorize_summary(summary)
output = summary.slice(RESULTS_LINE)
succeeded, failed, pending = $~[2].to_i, $~[3].to_i, $~[5].to_i
growl_fail(summary) and return Color.red(summary) if failed > 0
growl_pending(summary) and return Color.yellow(summary) if pending > 0
growl_success(summary) and return Color.green(summary)
end
STDIN.each_line do |line|
colored_line = case line
when /FAILED/ then Color.red(line)
when /PENDING/ then Color.yellow(line)
when RESULTS_LINE then colorize_summary(line)
else line
end
puts colored_line
end
