1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
|
# frozen_string_literal: true
require_relative "../vendored_thor"
module Bundler
module UI
class Shell
LEVELS = %w[silent error warn confirm info debug].freeze
OUTPUT_STREAMS = [:stdout, :stderr].freeze
attr_writer :shell
attr_reader :output_stream
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 = []
@output_stream = :stdout
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?(msg)
@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 output_stream=(symbol)
raise ArgumentError unless OUTPUT_STREAMS.include?(symbol)
@output_stream = symbol
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 progress(&blk)
with_output_stream(:stderr, &blk)
end
def unprinted_warnings
[]
end
private
# valimism
def tell_me(msg, color = nil, newline = nil)
return tell_err(msg, color, newline) if output_stream == :stderr
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.match?(/( |\t)\Z/) if newline.nil?
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 = Thor::Terminal.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
def with_output_stream(symbol)
original = output_stream
self.output_stream = symbol
yield
ensure
@output_stream = original
end
end
end
end
|
