lib/tap_dance/ui.rb
require 'rubygems/user_interaction'
module TapDance
class UI
attr_reader :log
class Logger < UI
LEVELS = %w(silent error warn detail confirm info debug)
attr_writer :shell
def initialize(options = {})
@level = ENV['DEBUG'] ? "debug" : "info"
end
def info(msg, newline = nil)
tell_me(msg, nil, newline) if level("info")
end
def confirm(msg, newline = nil)
tell_me(msg, :green, newline) if level("confirm")
end
def detail(msg, newline = nil)
tell_me(msg, :cyan, newline) if level("detail")
end
def warn(msg, newline = nil)
tell_me(msg, :yellow, newline) if level("warn")
end
def error(msg, newline = nil)
tell_me(msg, :red, newline) if level("error")
end
def debug(msg, newline = nil)
tell_me(msg, nil, newline) if level("debug")
end
def debug?
# needs to be false instead of nil to be newline param to other methods
level("debug")
end
def quiet?
LEVELS.index(@level) <= LEVELS.index("warn")
end
def ask(msg)
@shell.ask(msg)
end
def level=(level)
raise ArgumentError unless LEVELS.include?(level.to_s)
@level = level
end
def level(name = nil)
name ? LEVELS.index(name) <= LEVELS.index(@level) : @level
end
def trace(e, newline = nil)
msg = ["#{e.class}: #{e.message}", *e.backtrace].join("\n")
if debug?
tell_me(msg, nil, newline)
elsif @trace
STDERR.puts "#{msg}#{newline}"
end
end
def silence
old_level, @level = @level, "silent"
yield
ensure
@level = old_level
end
private
def tell_me(msg, color = nil, newline = nil)
if newline.nil?
logger(msg)
else
logger(msg, newline)
end
end
def logger(message, newline=(message.to_s !~ /( |\t)$/))
@log ||= ""
@log << message
@log << $/ if newline
end
end
class Shell < Logger
def initialize(options = {})
super
if options["no-color"] || !STDOUT.tty?
Thor::Base.shell = Thor::Shell::Basic
end
@shell = Thor::Base.shell.new
end
private
def tell_me(msg, color = nil, newline = nil)
msg = word_wrap(msg) if newline.is_a?(Hash) && newline[:wrap]
if msg.is_a? ShellResult
say(msg.out.chomp, color, newline)
say(msg.err.chomp, :red, newline)
else
say(msg, color, newline)
end
end
def say(msg, color = nil, newline = nil)
return if (msg.nil? || msg == "") && newline.nil?
line = if newline.nil? then [] else [newline] end
@shell.say(msg, color, *line)
logger(msg, *line)
end
def word_wrap(text, line_width = @shell.terminal_width)
text.split("\n").collect do |line|
line.length > line_width ? line.gsub(/(.{1,#{line_width}})(\s+|$)/, "\\1\n").strip : line
end * "\n"
end
end
end
end