Class: WavefrontDisplay::Base

Inherits:
Object
  • Object
show all
Includes:
WavefrontCli::Constants
Defined in:
lib/wavefront-cli/display/base.rb

Overview

Print human-friendly output. If a command requires a dedicated handler to format its output, define a method with the same name as that which fetches the data, in a WavefrontDisplay class, extending this one.

We provide #long_output() and #multicolumn() methods to solve standard formatting problems. To use them, define a do_() method but rather than printing the output, have it call the method.

Constant Summary

Constants included from WavefrontCli::Constants

WavefrontCli::Constants::ALL_PAGE_SIZE, WavefrontCli::Constants::DEFAULT_CONFIG, WavefrontCli::Constants::DEFAULT_OPTS, WavefrontCli::Constants::HUMAN_TIME_FORMAT, WavefrontCli::Constants::HUMAN_TIME_FORMAT_MS

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(raw_response, options = {}) ⇒ Base

Returns a new instance of Base.

Parameters:

  • the data returned by the SDK response.

  • (defaults to: {})

    options from docopt



23
24
25
26
27
28
29
30
31
32
33
34
# File 'lib/wavefront-cli/display/base.rb', line 23

def initialize(raw_response, options = {})
  @raw = raw_response

  data = if raw_response.respond_to?(:items)
           raw_response.items
         else
           raw_response
         end

  @data = prioritize_keys(data, priority_keys)
  @options = options
end

Instance Attribute Details

#dataObject (readonly)

Returns the value of attribute data.



17
18
19
# File 'lib/wavefront-cli/display/base.rb', line 17

def data
  @data
end

#optionsObject (readonly)

Returns the value of attribute options.



17
18
19
# File 'lib/wavefront-cli/display/base.rb', line 17

def options
  @options
end

#rawObject (readonly)

Returns the value of attribute raw.



17
18
19
# File 'lib/wavefront-cli/display/base.rb', line 17

def raw
  @raw
end

Instance Method Details

#_prioritize_keys(data, keys) ⇒ Hash, Map

Move the given fields to the start of a Hash or Map

Parameters:

  • keys to float

Returns:



115
116
117
118
119
120
121
122
123
# File 'lib/wavefront-cli/display/base.rb', line 115

def _prioritize_keys(data, keys)
  keys.each.with_object(data.is_a?(Map) ? Map.new : {}) do |k, a|
    next unless data.key?(k)
    a[k] = data[k]
    data.delete(k)
  end.merge(data)
rescue NoMethodError
  data
end

#do_deleteObject



213
214
215
# File 'lib/wavefront-cli/display/base.rb', line 213

def do_delete
  puts "Deleted #{friendly_name} '#{options[:'<id>']}'."
end

#do_importObject



208
209
210
211
# File 'lib/wavefront-cli/display/base.rb', line 208

def do_import
  puts "Imported #{friendly_name}."
  long_output
end

#do_listObject

The following do_ methods are default handlers called following their namesake operation in the corresponding WavefrontCli class. They can be overriden in the inheriting class.



196
197
198
# File 'lib/wavefront-cli/display/base.rb', line 196

def do_list
  long_output
end

#do_list_briefObject



204
205
206
# File 'lib/wavefront-cli/display/base.rb', line 204

def do_list_brief
  multicolumn(:id, :name)
end

#do_list_fieldsObject



200
201
202
# File 'lib/wavefront-cli/display/base.rb', line 200

def do_list_fields
  multicolumn(*filter_fields_as_arr.map(&:to_sym))
end

#do_queriesObject



265
266
267
268
269
270
271
# File 'lib/wavefront-cli/display/base.rb', line 265

def do_queries
  if options[:brief]
    multicolumn(:condition)
  else
    multicolumn(:id, :condition)
  end
end

#do_searchObject



233
234
235
236
237
238
239
# File 'lib/wavefront-cli/display/base.rb', line 233

def do_search
  if data.empty?
    puts 'No matches.'
  else
    long_output
  end
end

#do_search_briefObject



221
222
223
224
225
226
227
228
229
230
231
# File 'lib/wavefront-cli/display/base.rb', line 221

def do_search_brief
  display_keys = ([:id] + options[:'<condition>'].map do |c|
    c.split(/\W/, 2).first.to_sym
  end).uniq

  if data.empty?
    puts 'No matches.'
  else
    multicolumn(*display_keys)
  end
end

#do_tag_addObject



241
242
243
# File 'lib/wavefront-cli/display/base.rb', line 241

def do_tag_add
  puts "Tagged #{friendly_name} '#{options[:'<id>']}'."
end

#do_tag_clearObject



249
250
251
# File 'lib/wavefront-cli/display/base.rb', line 249

def do_tag_clear
  puts "Cleared tags on #{friendly_name} '#{options[:'<id>']}'."
end

#do_tag_deleteObject



245
246
247
# File 'lib/wavefront-cli/display/base.rb', line 245

def do_tag_delete
  puts "Deleted tag from #{friendly_name} '#{options[:'<id>']}'."
end

#do_tag_setObject



253
254
255
# File 'lib/wavefront-cli/display/base.rb', line 253

def do_tag_set
  puts "Set tags on #{friendly_name} '#{options[:'<id>']}'."
end

#do_tagsObject



257
258
259
260
261
262
263
# File 'lib/wavefront-cli/display/base.rb', line 257

def do_tags
  if data.empty?
    puts "No tags set on #{friendly_name} '#{options[:'<id>']}'."
  else
    data.sort.each { |t| puts t }
  end
end

#do_undeleteObject



217
218
219
# File 'lib/wavefront-cli/display/base.rb', line 217

def do_undelete
  puts "Undeleted #{friendly_name} '#{options[:'<id>']}'."
end

#drop_fields(*keys) ⇒ Nil

Modify, in-place, the data structure to remove fields which we deem not of interest to the user.

Parameters:

  • keys you do not wish to be shown.

Returns:



279
280
281
282
283
284
285
# File 'lib/wavefront-cli/display/base.rb', line 279

def drop_fields(*keys)
  if data.is_a?(Array)
    data.each { |i| i.delete_if { |k, _v| keys.include?(k.to_sym) } }
  else
    data.delete_if { |k, _v| keys.include?(k.to_sym) }
  end
end

#filter_data(data, fields) ⇒ Array[Hash]

Returns modified version of data. Each hash will contain only the fields given in ‘fields`, in the given order.

Parameters:

Returns:

  • modified version of data. Each hash will contain only the fields given in ‘fields`, in the given order



73
74
75
76
77
# File 'lib/wavefront-cli/display/base.rb', line 73

def filter_data(data, fields)
  data.map! do |d|
    fields.each_with_object({}) { |f, a| a[f] = d[f] if d.key?(f) }
  end
end

#filter_fields_as_arrArray

Returns filter fields from -O option.

Returns:

  • filter fields from -O option



187
188
189
# File 'lib/wavefront-cli/display/base.rb', line 187

def filter_fields_as_arr
  options[:fields].split(',')
end

#friendly_nameObject

return [String] the name of the thing we’re operating on, like

'alert' or 'dashboard'.


180
181
182
183
# File 'lib/wavefront-cli/display/base.rb', line 180

def friendly_name
  self.class.name.split('::').last.gsub(/([a-z])([A-Z])/, '\\1 \\2')
      .downcase
end

#human_time(time, force_utc = false) ⇒ Object

Make a time human-readable. Automatically deals with epoch seconds and epoch milliseconds

param t [Integer, String] a timestamp. If it’s a string, it is

converted to an int.

param force_utc [Boolean] force output in UTC. Currently only

used for unit tests.

return [String] a human-readable timestamp

Raises:



317
318
319
320
321
322
323
324
325
326
327
328
329
# File 'lib/wavefront-cli/display/base.rb', line 317

def human_time(time, force_utc = false)
  raise ArgumentError unless time.is_a?(Numeric) || time.is_a?(String)

  return 'FOREVER' if time == -1

  str = time.to_s
  fmt, out_fmt = time_formats(str)
  # rubocop:disable Style/DateTime
  ret = DateTime.strptime(str, fmt).to_time
  # rubocop:enable Style/DateTime
  ret = force_utc ? ret.utc : ret.localtime
  ret.strftime(out_fmt)
end

#key_width(hash = {}, pad = 2) ⇒ Integer

Give it a key-value hash, and it will return the size of the first column to use when formatting that data.

Parameters:

  • (defaults to: {})

    the data for which you need a column width

  • (defaults to: 2)

    the number of spaces you want between columns

Returns:

  • length of longest key + pad



172
173
174
175
# File 'lib/wavefront-cli/display/base.rb', line 172

def key_width(hash = {}, pad = 2)
  return 0 if hash.keys.empty?
  hash.keys.map(&:size).max + pad
end

#long_output(fields = nil, modified_data = nil) ⇒ Object

Default display method for ‘describe’ and long-list methods. Wraps around #_two_columns() giving you the chance to modify

Parameters:

  • (defaults to: nil)

    a list of fields you wish to display. If this is nil, all fields are displayed.

  • (defaults to: nil)

    lets you modify @data in-line. If this is truthy, it is used. Passing modified_data means that any fields parameter is ignored.



135
136
137
138
139
140
141
142
143
# File 'lib/wavefront-cli/display/base.rb', line 135

def long_output(fields = nil, modified_data = nil)
  if data.empty? || modified_data&.empty?
    puts 'No data.'
  else
    require_relative 'printer/long'
    puts WavefrontDisplayPrinter::Long.new(data, fields, modified_data)
    pagination_line
  end
end

#multicolumn(*columns) ⇒ Object



145
146
147
148
149
# File 'lib/wavefront-cli/display/base.rb', line 145

def multicolumn(*columns)
  require_relative 'printer/terse'
  puts WavefrontDisplayPrinter::Terse.new(data, columns)
  pagination_line
end

#pagination_lineObject

if this is a section of a larger dataset, say so

rubocop:disable Metrics/AbcSize



154
155
156
157
158
159
160
161
162
# File 'lib/wavefront-cli/display/base.rb', line 154

def pagination_line
  return unless raw.respond_to?(:moreItems) && raw.moreItems == true

  enditem = raw.limit.positive? ? raw.offset + raw.limit - 1 : 0
  puts format('List shows items %d to %d. Use -o and -L for more.',
              raw.offset, enditem)
rescue StandardError
  puts 'List shows paginated output. Use -o and -L for more.'
end

#prioritize_keys(data, keys) ⇒ Object



105
106
107
108
# File 'lib/wavefront-cli/display/base.rb', line 105

def prioritize_keys(data, keys)
  return _prioritize_keys(data, keys) unless data.is_a?(Array)
  data.map { |e| _prioritize_keys(e, keys) }
end

#priority_keysObject

Keys which we wish to float to the top of descriptions and long listing objects. Subclasses may define their own.



101
102
103
# File 'lib/wavefront-cli/display/base.rb', line 101

def priority_keys
  i[id name]
end

#readable_time(*keys) ⇒ Object

Modify, in-place, the @data structure to make times human-readable. Automatically handles second and millisecond epoch times. Currently only operates on top-level keys.

param keys [Symbol, Array] the keys you wish to be

turned into readable times.

return [Nil]



295
296
297
# File 'lib/wavefront-cli/display/base.rb', line 295

def readable_time(*keys)
  keys.each { |k| data[k] = human_time(data[k]) if data.key?(k) }
end

#readable_time_arr(*keys) ⇒ Object

As for #readable_time, but when @data is an array. For instance in “firing” alerts



302
303
304
305
306
# File 'lib/wavefront-cli/display/base.rb', line 302

def readable_time_arr(*keys)
  data.map do |row|
    keys.each { |k| row[k] = human_time(row[k]) if row.key?(k) }
  end
end

#run(method) ⇒ Object

find the correct method to deal with the output of the user’s command.



39
40
41
42
43
44
45
46
47
48
49
50
51
# File 'lib/wavefront-cli/display/base.rb', line 39

def run(method)
  if method == 'do_list'
    run_list
  elsif method == 'do_search'
    run_search
  elsif respond_to?("#{method}_brief")
    send("#{method}_brief")
  elsif respond_to?(method)
    send(method)
  else
    long_output
  end
end

#run_error(method) ⇒ Object

Display classes can provide a do_method_code() method, which handles <code> errors when running do_method(). (Code is 404 etc.)

Parameters:

  • the error method we wish to call



92
93
94
95
96
# File 'lib/wavefront-cli/display/base.rb', line 92

def run_error(method)
  return unless respond_to?(method)
  send(method)
  exit 1
end

#run_listObject

Choose the correct list handler. The user can specifiy a long listing with the –long options.



56
57
58
59
60
61
62
63
64
65
# File 'lib/wavefront-cli/display/base.rb', line 56

def run_list
  if options[:long]
    @data = filter_data(data, filter_fields_as_arr) if options[:fields]
    do_list
  elsif options[:fields]
    do_list_fields
  else
    do_list_brief
  end
end

#run_searchObject

Choose the correct search handler. The user can specifiy a long listing with the –long options.



82
83
84
# File 'lib/wavefront-cli/display/base.rb', line 82

def run_search
  options[:long] ? do_search : do_search_brief
end

#time_formats(str) ⇒ String

How do we format a timestamp?

Parameters:

  • an epoch timestamp, as a string

Returns:

  • DateTime formatter, strptime formatter



335
336
337
338
339
340
341
342
343
# File 'lib/wavefront-cli/display/base.rb', line 335

def time_formats(str)
  if str =~ /^\d{13}$/
    ['%Q', HUMAN_TIME_FORMAT_MS]
  elsif str =~ /^\d{10}$/
    ['%s', HUMAN_TIME_FORMAT]
  else
    raise ArgumentError
  end
end