зеркало из https://github.com/github/ruby.git
166 строки
3.7 KiB
Ruby
166 строки
3.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "../vendored_thor"
|
|
|
|
module Bundler
|
|
module UI
|
|
class Shell
|
|
LEVELS = %w[silent error warn confirm info debug].freeze
|
|
|
|
attr_writer :shell
|
|
|
|
def initialize(options = {})
|
|
Thor::Base.shell = options["no-color"] ? Thor::Shell::Basic : nil
|
|
@shell = Thor::Base.shell.new
|
|
@level = ENV["DEBUG"] ? "debug" : "info"
|
|
@warning_history = []
|
|
end
|
|
|
|
def add_color(string, *color)
|
|
@shell.set_color(string, *color)
|
|
end
|
|
|
|
def info(msg = nil, newline = nil)
|
|
return unless info?
|
|
|
|
tell_me(msg || yield, nil, newline)
|
|
end
|
|
|
|
def confirm(msg = nil, newline = nil)
|
|
return unless confirm?
|
|
|
|
tell_me(msg || yield, :green, newline)
|
|
end
|
|
|
|
def warn(msg = nil, newline = nil, color = :yellow)
|
|
return unless warn?
|
|
return if @warning_history.include? msg
|
|
@warning_history << msg
|
|
|
|
tell_err(msg || yield, color, newline)
|
|
end
|
|
|
|
def error(msg = nil, newline = nil, color = :red)
|
|
return unless error?
|
|
|
|
tell_err(msg || yield, color, newline)
|
|
end
|
|
|
|
def debug(msg = nil, newline = nil)
|
|
return unless debug?
|
|
|
|
tell_me(msg || yield, nil, newline)
|
|
end
|
|
|
|
def info?
|
|
level("info")
|
|
end
|
|
|
|
def confirm?
|
|
level("confirm")
|
|
end
|
|
|
|
def warn?
|
|
level("warn")
|
|
end
|
|
|
|
def error?
|
|
level("error")
|
|
end
|
|
|
|
def debug?
|
|
level("debug")
|
|
end
|
|
|
|
def quiet?
|
|
level("quiet")
|
|
end
|
|
|
|
def ask(msg)
|
|
@shell.ask(msg)
|
|
end
|
|
|
|
def yes?(msg)
|
|
@shell.yes?(msg)
|
|
end
|
|
|
|
def no?
|
|
@shell.no?(msg)
|
|
end
|
|
|
|
def level=(level)
|
|
raise ArgumentError unless LEVELS.include?(level.to_s)
|
|
@level = level.to_s
|
|
end
|
|
|
|
def level(name = nil)
|
|
return @level unless name
|
|
unless index = LEVELS.index(name)
|
|
raise "#{name.inspect} is not a valid level"
|
|
end
|
|
index <= LEVELS.index(@level)
|
|
end
|
|
|
|
def trace(e, newline = nil, force = false)
|
|
return unless debug? || force
|
|
msg = "#{e.class}: #{e.message}\n#{e.backtrace.join("\n ")}"
|
|
tell_err(msg, nil, newline)
|
|
end
|
|
|
|
def silence(&blk)
|
|
with_level("silent", &blk)
|
|
end
|
|
|
|
def unprinted_warnings
|
|
[]
|
|
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 tell_err(message, color = nil, newline = nil)
|
|
return if @shell.send(:stderr).closed?
|
|
|
|
newline ||= message.to_s !~ /( |\t)\Z/
|
|
message = word_wrap(message) if newline.is_a?(Hash) && newline[:wrap]
|
|
|
|
color = nil if color && !$stderr.tty?
|
|
|
|
buffer = @shell.send(:prepare_message, message, *color)
|
|
buffer << "\n" if newline && !message.to_s.end_with?("\n")
|
|
|
|
@shell.send(:stderr).print(buffer)
|
|
@shell.send(:stderr).flush
|
|
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
|
|
|
|
def with_level(level)
|
|
original = @level
|
|
@level = level
|
|
yield
|
|
ensure
|
|
@level = original
|
|
end
|
|
end
|
|
end
|
|
end
|