Class: VMC::CLI
Class Method Summary
collapse
Instance Method Summary
collapse
Methods included from Spacing
#indented, #justify, #line, #lines, #spaced, #start_line, #tabular, #text_width, #trim_escapes
#ask, #handler, #input_state, #list_choices, #prompt, #show_default
Class Method Details
.client ⇒ Object
437
438
439
|
# File 'lib/vmc/cli.rb', line 437
def client
@@client
end
|
.client=(c) ⇒ Object
441
442
443
|
# File 'lib/vmc/cli.rb', line 441
def client=(c)
@@client = c
end
|
Instance Method Details
#check_logged_in ⇒ Object
71
72
73
74
75
76
77
78
79
80
81
82
|
# File 'lib/vmc/cli.rb', line 71
def check_logged_in
unless client.logged_in?
if force?
fail "Please log in with 'vmc login'."
else
line c("Please log in first to proceed.", :warning)
line
invoke :login
invalidate_client
end
end
end
|
#check_target ⇒ Object
65
66
67
68
69
|
# File 'lib/vmc/cli.rb', line 65
def check_target
unless client && client.target
fail "Please select a target with 'vmc target'."
end
end
|
#client(target = client_target) ⇒ Object
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
|
# File 'lib/vmc/cli.rb', line 391
def client(target = client_target)
return @@client if defined?(@@client) && @@client
return unless target
info = target_info(target)
token = info[:token] && CFoundry::AuthToken.from_hash(info)
@@client =
case info[:version]
when 2
fail "User switching not implemented for v2." if input[:proxy]
CFoundry::V2::Client.new(target, token)
when 1
CFoundry::V1::Client.new(target, token)
else
CFoundry::Client.new(target, token)
end
@@client.proxy = input[:proxy]
@@client.trace = input[:trace]
uri = URI.parse(target)
@@client.log = File.expand_path("#{LOGS_DIR}/#{uri.host}.log")
unless info.key? :version
info[:version] = @@client.version
save_target_info(info, target)
end
if org = info[:organization]
@@client.current_organization = @@client.organization(org)
end
if space = info[:space]
@@client.current_space = @@client.space(space)
end
@@client
rescue CFoundry::InvalidTarget
end
|
#client_target ⇒ Object
309
310
311
312
313
|
# File 'lib/vmc/cli.rb', line 309
def client_target
if File.exists?(target_file)
File.read(target_file).chomp
end
end
|
#color_enabled? ⇒ Boolean
225
226
227
|
# File 'lib/vmc/cli.rb', line 225
def color_enabled?
input[:color]
end
|
#debug? ⇒ Boolean
221
222
223
|
# File 'lib/vmc/cli.rb', line 221
def debug?
!!input[:debug]
end
|
#default_action ⇒ Object
57
58
59
60
61
62
63
|
# File 'lib/vmc/cli.rb', line 57
def default_action
if input[:version]
line "vmc #{VERSION}"
else
super
end
end
|
#ensure_config_dir ⇒ Object
315
316
317
318
|
# File 'lib/vmc/cli.rb', line 315
def ensure_config_dir
config = File.expand_path(VMC::CONFIG_DIR)
FileUtils.mkdir_p(config) unless File.exist? config
end
|
#err(msg, status = 1) ⇒ Object
256
257
258
259
|
# File 'lib/vmc/cli.rb', line 256
def err(msg, status = 1)
$stderr.puts c(msg, :error)
exit_status status
end
|
#execute(cmd, argv, global = {}) ⇒ Object
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
# File 'lib/vmc/cli.rb', line 141
def execute(cmd, argv, global = {})
if input[:help]
invoke :help, :command => cmd.name.to_s
else
wrap_errors do
@command = cmd
precondition
save_token_if_it_changes do
super
end
end
end
end
|
#fail(msg) ⇒ Object
261
262
263
|
# File 'lib/vmc/cli.rb', line 261
def fail(msg)
raise UserError, msg
end
|
#fail_unknown(display, name) ⇒ Object
432
433
434
|
# File 'lib/vmc/cli.rb', line 432
def fail_unknown(display, name)
fail("Unknown #{display} '#{name}'.")
end
|
#force? ⇒ Boolean
217
218
219
|
# File 'lib/vmc/cli.rb', line 217
def force?
input[:force]
end
|
#invalidate_client ⇒ Object
386
387
388
389
|
# File 'lib/vmc/cli.rb', line 386
def invalidate_client
@@client = nil
client
end
|
#log_error(e) ⇒ Object
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
|
# File 'lib/vmc/cli.rb', line 174
def log_error(e)
ensure_config_dir
msg = e.class.name
msg << ": #{e}" unless e.to_s.empty?
crash_file = File.expand_path(VMC::CRASH_FILE)
FileUtils.mkdir_p(File.dirname(crash_file))
File.open(crash_file, "w") do |f|
f.puts "Time of crash:"
f.puts " #{Time.now}"
f.puts ""
f.puts msg
f.puts ""
if e.respond_to?(:request_trace)
f.puts "<<<"
f.puts e.request_trace
end
if e.respond_to?(:response_trace)
f.puts e.response_trace
f.puts ">>>"
f.puts ""
end
vmc_dir = File.expand_path("../../../..", __FILE__) + "/"
e.backtrace.each do |loc|
if loc =~ /\/gems\//
f.puts loc.sub(/.*\/gems\//, "")
else
f.puts loc.sub(vmc_dir, "")
end
end
end
end
|
#name_list(xs) ⇒ Object
271
272
273
274
275
276
277
|
# File 'lib/vmc/cli.rb', line 271
def name_list(xs)
if xs.empty?
d("none")
else
xs.collect { |x| c(x.name, :name) }.join(", ")
end
end
|
#no_v2 ⇒ Object
378
379
380
|
# File 'lib/vmc/cli.rb', line 378
def no_v2
fail "Not implemented for v2." if v2?
end
|
#normalize_targets_info(info_by_url) ⇒ Object
346
347
348
349
350
351
352
|
# File 'lib/vmc/cli.rb', line 346
def normalize_targets_info(info_by_url)
info_by_url.reduce({}) do |hash, pair|
key, value = pair
hash[key] = value.is_a?(String) ? { :token => value } : value
hash
end
end
|
#one_of(*paths) ⇒ Object
300
301
302
303
304
305
306
307
|
# File 'lib/vmc/cli.rb', line 300
def one_of(*paths)
paths.each do |p|
exp = File.expand_path(p)
return exp if File.exist? exp
end
File.expand_path(paths.first)
end
|
#precondition ⇒ Object
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
# File 'lib/vmc/cli.rb', line 84
def precondition
check_target
check_logged_in
return unless v2?
unless client.current_organization
fail "Please select an organization with 'vmc target --ask-org'."
end
unless client.current_space
fail "Please select a space with 'vmc target --ask-space'."
end
end
|
#quiet? ⇒ Boolean
213
214
215
|
# File 'lib/vmc/cli.rb', line 213
def quiet?
input[:quiet]
end
|
#remove_target_info(target = client_target) ⇒ Object
372
373
374
375
376
|
# File 'lib/vmc/cli.rb', line 372
def remove_target_info(target = client_target)
ts = targets_info
ts.delete target
save_targets(ts)
end
|
#sane_target_url(url) ⇒ Object
279
280
281
282
283
284
285
286
287
288
289
290
|
# File 'lib/vmc/cli.rb', line 279
def sane_target_url(url)
unless url =~ /^https?:\/\//
begin
TCPSocket.new(url, Net::HTTP.https_default_port)
url = "https://#{url}"
rescue Errno::ECONNREFUSED, SocketError, Timeout::Error
url = "http://#{url}"
end
end
url.gsub(/\/$/, "")
end
|
#save_target_info(info, target = client_target) ⇒ Object
366
367
368
369
370
|
# File 'lib/vmc/cli.rb', line 366
def save_target_info(info, target = client_target)
ts = targets_info
ts[target] = info
save_targets(ts)
end
|
#save_targets(ts) ⇒ Object
358
359
360
361
362
363
364
|
# File 'lib/vmc/cli.rb', line 358
def save_targets(ts)
ensure_config_dir
File.open(File.expand_path(VMC::TOKENS_FILE), "w") do |io|
YAML.dump(ts, io)
end
end
|
#save_token_if_it_changes ⇒ Object
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
|
# File 'lib/vmc/cli.rb', line 156
def save_token_if_it_changes
return yield unless client && client.token
before_token = client.token
yield
after_token = client.token
return unless after_token
if before_token != after_token
info = target_info
info[:token] = after_token.
save_target_info(info)
end
end
|
#set_target(url) ⇒ Object
320
321
322
323
324
325
326
327
328
|
# File 'lib/vmc/cli.rb', line 320
def set_target(url)
ensure_config_dir
File.open(File.expand_path(VMC::TARGET_FILE), "w") do |f|
f.write(sane_target_url(url))
end
invalidate_client
end
|
#table(headers, rows) ⇒ Object
265
266
267
268
269
|
# File 'lib/vmc/cli.rb', line 265
def table(, rows)
tabular(
!quiet? && .collect { |h| h && b(h) },
*rows)
end
|
#target_info(target = client_target) ⇒ Object
354
355
356
|
# File 'lib/vmc/cli.rb', line 354
def target_info(target = client_target)
targets_info[target] || {}
end
|
#targets_info ⇒ Object
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
|
# File 'lib/vmc/cli.rb', line 330
def targets_info
new_toks = File.expand_path(VMC::TOKENS_FILE)
old_toks = File.expand_path(VMC::OLD_TOKENS_FILE)
info =
if File.exist? new_toks
YAML.load_file(new_toks)
elsif File.exist? old_toks
MultiJson.load(File.read(old_toks))
end
info ||= {}
normalize_targets_info(info)
end
|
#user_colors ⇒ Object
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
|
# File 'lib/vmc/cli.rb', line 233
def user_colors
return @user_colors if @user_colors
colors = File.expand_path(COLORS_FILE)
@user_colors = super.dup
@user_colors.each do |k, v|
if v == :blue
@user_colors[k] = :cyan
end
end
if File.exists?(colors)
YAML.load_file(colors).each do |k, v|
@user_colors[k.to_sym] = v.to_sym
end
end
@user_colors
end
|
#v2? ⇒ Boolean
382
383
384
|
# File 'lib/vmc/cli.rb', line 382
def v2?
client.version == 2
end
|
#verbose? ⇒ Boolean
229
230
231
|
# File 'lib/vmc/cli.rb', line 229
def verbose?
input[:verbose]
end
|
#wrap_errors ⇒ Object
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
|
# File 'lib/vmc/cli.rb', line 99
def wrap_errors
yield
rescue CFoundry::Timeout => e
err(e.message)
rescue Interrupt
exit_status 130
rescue Mothership::Error
raise
rescue UserError => e
log_error(e)
err e.message
rescue SystemExit
raise
rescue CFoundry::Forbidden, CFoundry::InvalidAuthToken => e
if !$vmc_asked_auth
$vmc_asked_auth = true
line
line c("Not authenticated! Try logging in:", :warning)
invoke :login
retry
end
log_error(e)
err "Denied: #{e.description}"
rescue Exception => e
log_error(e)
msg = e.class.name
msg << ": #{e}" unless e.to_s.empty?
msg << "\nFor more information, see #{VMC::CRASH_FILE}"
err msg
raise if debug?
end
|