summaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
authorelijah <elijah@riseup.net>2012-11-24 20:22:45 -0800
committerelijah <elijah@riseup.net>2012-11-24 20:22:45 -0800
commit051675c61937f184c555bac3af07be182f0c6acd (patch)
treefd8f900185b0b82b39f71d7b1ec860a20356d295 /lib
parentba80599d9230b63b71bfeca45fc89c7f69987723 (diff)
fixed broken `leap list`
Diffstat (limited to 'lib')
-rw-r--r--lib/leap_cli/commands/list.rb149
-rw-r--r--lib/leap_cli/config/manager.rb3
2 files changed, 89 insertions, 63 deletions
diff --git a/lib/leap_cli/commands/list.rb b/lib/leap_cli/commands/list.rb
index 033c95f..5f455c0 100644
--- a/lib/leap_cli/commands/list.rb
+++ b/lib/leap_cli/commands/list.rb
@@ -1,80 +1,103 @@
-module LeapCli
- module Commands
+require 'command_line_reporter'
- desc 'List nodes and their classifications'
- long_desc 'Prints out a listing of nodes, services, or tags.'
- arg_name 'filter'
- command :list do |c|
- c.flag 'print', :desc => 'What attributes to print (optional)'
- c.action do |global_options,options,args|
- if options['print']
- print_node_properties(manager.filter(args), options['print'])
+module LeapCli; module Commands
+
+ desc 'List nodes and their classifications'
+ long_desc 'Prints out a listing of nodes, services, or tags.'
+ arg_name 'filter'
+ command :list do |c|
+ c.flag 'print', :desc => 'What attributes to print (optional)'
+ c.action do |global_options,options,args|
+ puts
+ if options['print']
+ print_node_properties(manager.filter(args), options['print'])
+ else
+ if args.any?
+ NodeTable.new(manager.filter(args)).run
else
- if args.any?
- print_config_table(:nodes, manager.filter(args))
- else
- print_config_table(:services, manager.services)
- print_config_table(:tags, manager.tags)
- print_config_table(:nodes, manager.nodes)
- end
+ TagTable.new('SERVICES', manager.services).run
+ TagTable.new('TAGS', manager.tags).run
+ NodeTable.new(manager.nodes).run
end
end
end
+ end
- private
+ private
- def self.print_node_properties(nodes, properties)
- node_list = manager.nodes
- properties = properties.split(',')
- max_width = nodes.keys.inject(0) {|max,i| [i.size,max].max}
- nodes.keys.sort.each do |node_name|
- value = properties.collect{|prop| node_list[node_name][prop]}.join(', ')
- printf("%#{max_width}s %s\n", node_name, value)
- end
+ def self.print_node_properties(nodes, properties)
+ node_list = manager.nodes
+ properties = properties.split(',')
+ max_width = nodes.keys.inject(0) {|max,i| [i.size,max].max}
+ nodes.keys.sort.each do |node_name|
+ value = properties.collect{|prop| node_list[node_name][prop]}.join(', ')
+ printf("%#{max_width}s %s\n", node_name, value)
end
+ puts
+ end
- def self.print_config_table(type, object_list)
- style = {:border_x => '-', :border_y => ':', :border_i => '-', :width => 60}
-
- if type == :services
- t = table do
- self.style = style
- self.headings = ['SERVICE', 'NODES']
- list = object_list.keys.sort
- list.each do |name|
- add_row [name, object_list[name].node_list.keys.join(', ')]
- add_separator unless name == list.last
- end
+ class TagTable
+ include CommandLineReporter
+ def initialize(heading, tag_list)
+ @heading = heading
+ @tag_list = tag_list
+ end
+ def run
+ tags = @tag_list.keys.sort
+ max_width = [20, (tags+[@heading]).inject(0) {|max,i| [i.size,max].max}].max
+ table :border => false do
+ row :header => true, :color => 'cyan' do
+ column @heading, :align => 'right', :width => max_width
+ column "NODES", :width => HighLine::SystemExtensions.terminal_size.first - max_width - 2, :padding => 2
end
- puts t
- puts "\n\n"
- elsif type == :tags
- t = table do
- self.style = style
- self.headings = ['TAG', 'NODES']
- list = object_list.keys.sort
- list.each do |name|
- add_row [name, object_list[name].node_list.keys.join(', ')]
- add_separator unless name == list.last
+ tags.each do |tag|
+ row do
+ column tag
+ column @tag_list[tag].node_list.keys.sort.join(', ')
end
end
- puts t
- puts "\n\n"
- elsif type == :nodes
- t = table do
- self.style = style
- self.headings = ['NODE', 'SERVICES', 'TAGS']
- list = object_list.keys.sort
- list.each do |name|
- services = object_list[name]['services'] || []
- tags = object_list[name]['tags'] || []
- add_row [name, services.to_a.join(', '), tags.to_a.join(', ')]
- add_separator unless name == list.last
+ end
+ vertical_spacing
+ end
+ end
+
+ #
+ # might be handy: HighLine::SystemExtensions.terminal_size.first
+ #
+ class NodeTable
+ include CommandLineReporter
+ def initialize(node_list)
+ @node_list = node_list
+ end
+ def run
+ rows = @node_list.keys.sort.collect do |node_name|
+ [node_name, @node_list[node_name].services.sort.join(', '), @node_list[node_name].tags.sort.join(', ')]
+ end
+ unless rows.any?
+ puts Paint["no results", :red]
+ puts
+ return
+ end
+ padding = 2
+ max_node_width = [20, (rows.map{|i|i[0]} + ["NODES"] ).inject(0) {|max,i| [i.size,max].max}].max
+ max_service_width = (rows.map{|i|i[1]} + ["SERVICES"]).inject(0) {|max,i| [i.size+padding+padding,max].max}
+ max_tag_width = (rows.map{|i|i[2]} + ["TAGS"] ).inject(0) {|max,i| [i.size,max].max}
+ table :border => false do
+ row :header => true, :color => 'cyan' do
+ column "NODES", :align => 'right', :width => max_node_width
+ column "SERVICES", :width => max_service_width, :padding => 2
+ column "TAGS", :width => max_tag_width
+ end
+ rows.each do |r|
+ row do
+ column r[0]
+ column r[1]
+ column r[2]
end
end
- puts t
end
+ vertical_spacing
end
-
end
-end
+
+end; end
diff --git a/lib/leap_cli/config/manager.rb b/lib/leap_cli/config/manager.rb
index 3d09e09..39dbcd2 100644
--- a/lib/leap_cli/config/manager.rb
+++ b/lib/leap_cli/config/manager.rb
@@ -229,6 +229,9 @@ module LeapCli
end
# inherit from tags
+ if node.vagrant?
+ node['tags'] = (node['tags'] || []).to_a + ['local']
+ end
if node['tags']
node['tags'].to_a.sort.each do |node_tag|
tag = @tags[node_tag]