Class: Ardb::CLI::CommandSet

Inherits:
Object
  • Object
show all
Defined in:
lib/ardb/cli/commands.rb

Instance Method Summary collapse

Constructor Details

#initialize(&unknown_cmd_block) ⇒ CommandSet

Returns a new instance of CommandSet.



228
229
230
231
232
233
# File 'lib/ardb/cli/commands.rb', line 228

def initialize(&unknown_cmd_block)
  @lookup    = Hash.new{ |h,k| unknown_cmd_block.call(k) }
  @names     = []
  @aliases   = {}
  @summaries = {}
end

Instance Method Details

#[](name) ⇒ Object



256
257
258
# File 'lib/ardb/cli/commands.rb', line 256

def [](name)
  @lookup[name]
end

#add(klass, name, *aliases) ⇒ Object



235
236
237
238
239
240
241
242
243
244
245
246
247
# File 'lib/ardb/cli/commands.rb', line 235

def add(klass, name, *aliases)
  begin
    cmd = klass.new
  rescue StandardError => err
    # don't add any commands you can't init
  else
    ([name] + aliases).each{ |n| @lookup[n] = cmd }
    @to_s = nil
    @names << name
    @aliases[name] = aliases.empty? ? '' : "(#{aliases.join(', ')})"
    @summaries[name] = cmd.summary.to_s.empty? ? '' : "# #{cmd.summary}"
  end
end

#remove(name) ⇒ Object



249
250
251
252
253
254
# File 'lib/ardb/cli/commands.rb', line 249

def remove(name)
  @lookup.delete(name)
  @names.delete(name)
  @aliases.delete(name)
  @to_s = nil
end

#sizeObject



260
261
262
# File 'lib/ardb/cli/commands.rb', line 260

def size
  @names.size
end

#to_sObject



264
265
266
267
268
269
270
271
# File 'lib/ardb/cli/commands.rb', line 264

def to_s
  max_name_size  = @names.map{ |n| n.size }.max || 0
  max_alias_size = @aliases.values.map{ |v| v.size }.max || 0

  @to_s ||= @names.map do |n|
    "#{n.ljust(max_name_size)} #{@aliases[n].ljust(max_alias_size)} #{@summaries[n]}"
  end.join("\n")
end