bundle-new/lib/bundler/ui/shell.rb

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