fixed broken `leap list`

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]