96 lines
2.3 KiB
Ruby
96 lines
2.3 KiB
Ruby
|
module Bundler
|
||
|
|
module UI
|
||
|
|
class Shell
|
||
|
|
LEVELS = %w(silent error warn confirm info debug)
|
||
|
|
|
||
|
|
attr_writer :shell
|
||
|
|
|
||
|
|
def initialize(options = {})
|
||
|
|
if options["no-color"] || !STDOUT.tty?
|
||
|
|
Thor::Base.shell = Thor::Shell::Basic
|
||
|
|
end
|
||
|
|
@shell = Thor::Base.shell.new
|
||
|
|
@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 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)
|
||
|
|
return unless debug?
|
||
|
|
msg = ["#{e.class}: #{e.message}", *e.backtrace].join("\n")
|
||
|
|
tell_me(msg, nil, newline)
|
||
|
|
end
|
||
|
|
|
||
|
|
def silence
|
||
|
|
old_level, @level = @level, "silent"
|
||
|
|
yield
|
||
|
|
ensure
|
||
|
|
@level = old_level
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
# valimism
|
||
|
|
def tell_me(msg, color = nil, newline = nil)
|
||
|
|
msg = word_wrap(msg) if newline.is_a?(Hash) && newline[:wrap]
|
||
|
|
if newline.nil?
|
||
|
|
@shell.say(msg, color)
|
||
|
|
else
|
||
|
|
@shell.say(msg, color, newline)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def strip_leading_spaces(text)
|
||
|
|
spaces = text[/\A\s+/, 0]
|
||
|
|
spaces ? text.gsub(/#{spaces}/, '') : text
|
||
|
|
end
|
||
|
|
|
||
|
|
def word_wrap(text, line_width = @shell.terminal_width)
|
||
|
|
strip_leading_spaces(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
|