network.rb in testlab-0.8.0

- old
+ new

@@ -18,185 +18,28 @@
 #
 ################################################################################
 
 # NETWORKS
 ###########
-desc 'Manage networks'
-arg_name 'Describe arguments to network here'
-command :network do |c|
+build_lab_commands(:network, TestLab::Network) do |c|
 
-  c.desc 'Network ID or Name'
-  c.arg_name 'network'
-  c.flag [:n, :name]
-
-  # NETWORK CREATE
-  #################
-  c.desc 'Create a network'
-  c.long_desc <<-EOF
-Create a network.  The network is created.
-EOF
-  c.command :create do |create|
-    create.action do |global_options, options, args|
-      if options[:name].nil?
-        help_now!('a name is required') if options[:name].nil?
-      else
-        network = @testlab.networks.select{ |c| c.id.to_sym == options[:name].to_sym }.first
-        network.nil? and raise TestLab::TestLabError, "We could not find the network you screateplied!"
-
-        network.create
-      end
-    end
-  end
-
-  # NETWORK DESTROY
-  #################
-  c.desc 'Destroy a network'
-  c.long_desc <<-EOF
-Destroy a network.  The network is destroyed.
-EOF
-  c.command :destroy do |destroy|
-    destroy.action do |global_options, options, args|
-      if options[:name].nil?
-        help_now!('a name is required') if options[:name].nil?
-      else
-        network = @testlab.networks.select{ |c| c.id.to_sym == options[:name].to_sym }.first
-        network.nil? and raise TestLab::TestLabError, "We could not find the network you supplied!"
-
-        network.destroy
-      end
-    end
-  end
-
-  # NETWORK UP
-  #############
-  c.desc 'Up a network'
-  c.long_desc <<-EOF
-Up a network.  The network is started and brought online.
-EOF
-  c.command :up do |up|
-    up.action do |global_options, options, args|
-      if options[:name].nil?
-        help_now!('a name is required') if options[:name].nil?
-      else
-        network = @testlab.networks.select{ |c| c.id.to_sym == options[:name].to_sym }.first
-        network.nil? and raise TestLab::TestLabError, "We could not find the network you supplied!"
-
-        network.up
-      end
-    end
-  end
-
-  # NETWORK DOWN
-  ###############
-  c.desc 'Down a network'
-  c.long_desc <<-EOF
-Down a network.  The network is stopped taking it offline.
-EOF
-  c.command :down do |down|
-    down.action do |global_options, options, args|
-      if options[:name].nil?
-        help_now!('a name is required') if options[:name].nil?
-      else
-        network = @testlab.networks.select{ |c| c.id.to_sym == options[:name].to_sym }.first
-        network.nil? and raise TestLab::TestLabError, "We could not find the network you supplied!"
-
-        network.down
-      end
-    end
-  end
-
-  # NETWORK SETUP
-  ################
-  c.desc 'Setup a network'
-  c.long_desc <<-EOF
-Setup a network.  The network is created, started and provisioned.
-EOF
-  c.command :setup do |setup|
-    setup.action do |global_options, options, args|
-      if options[:name].nil?
-        help_now!('a name is required') if options[:name].nil?
-      else
-        network = @testlab.networks.select{ |c| c.id.to_sym == options[:name].to_sym }.first
-        network.nil? and raise TestLab::TestLabError, "We could not find the network you supplied!"
-
-        network.setup
-      end
-    end
-  end
-
-  # NETWORK TEARDOWN
-  ###################
-  c.desc 'Teardown a network'
-  c.long_desc <<-EOF
-Teardown a network.  The network is offlined and destroyed.
-EOF
-  c.command :teardown do |teardown|
-    teardown.action do |global_options, options, args|
-      if options[:name].nil?
-        help_now!('a name is required') if options[:name].nil?
-      else
-        network = @testlab.networks.select{ |c| c.id.to_sym == options[:name].to_sym }.first
-        network.nil? and raise TestLab::TestLabError, "We could not find the network you supplied!"
-
-        network.teardown
-      end
-    end
-  end
-
-  # NETWORK BUILD
-  ################
-  c.desc 'Build a network'
-  c.long_desc <<-EOF
-Attempts to build up the network.  TestLab will attempt to create, online and provision the network.
-
-The network is taken through the following phases:
-
-Create -> Up -> Setup
-EOF
-  c.command :build do |build|
-    build.action do |global_options, options, args|
-      if options[:name].nil?
-        help_now!('a name is required') if options[:name].nil?
-      else
-        network = @testlab.networks.select{ |c| c.id.to_sym == options[:name].to_sym }.first
-        network.nil? and raise TestLab::TestLabError, "We could not find the network you supplied!"
-
-        network.build
-      end
-    end
-  end
-
   # NETWORK STATUS
   #################
   c.desc 'Display the status of networks'
   c.long_desc <<-EOF
 Displays the status of all networks or single/multiple networks if supplied via the ID parameter.
 EOF
   c.command :status do |status|
     status.action do |global_options, options, args|
-      if options[:name].nil?
-        networks = Array.new
+      networks = iterate_objects_by_name(options[:name], TestLab::Network).delete_if{ |network| network.node.dead? }
 
-        if options[:name].nil?
-          # No ID supplied; show everything
-          networks = TestLab::Network.all
-        else
-          # ID supplied; show just those items
-          names = options[:name].split(',')
-          networks = TestLab::Network.find(names)
-          (networks.nil? || (networks.count == 0)) and raise TestLab::TestLabError, "We could not find any of the networks you supplied!"
+      if (networks.count == 0)
+        @testlab.ui.stderr.puts("You either have no networks defined or dead nodes!".yellow)
+      else
+        ZTK::Report.new(:ui => @testlab.ui).list(networks, TestLab::Network::STATUS_KEYS) do |network|
+          OpenStruct.new(network.status)
         end
-
-        networks = networks.delete_if{ |network| network.node.dead? }
-
-        if (networks.count == 0)
-          @testlab.ui.stderr.puts("You either have no networks defined or dead nodes!".yellow)
-        else
-          ZTK::Report.new(:ui => @testlab.ui).list(networks, TestLab::Network::STATUS_KEYS) do |network|
-            OpenStruct.new(network.status)
-          end
-        end
       end
     end
   end
 
   # ROUTES
@@ -207,54 +50,45 @@
     # ROUTE ADD
     ############
     route.desc 'Add routes to lab networks'
     route.command :add do |add|
       add.action do |global_options,options,args|
-        help_now!('a name is required') if options[:name].nil?
-
-        network = @testlab.networks.select{ |c| c.id.to_sym == options[:name].to_sym }.first
-        network.nil? and raise TestLab::TestLabError, "We could not find the network you supplied!"
-
-        p = TestLab::Provisioner::Route.new({}, @ui)
-        p.on_network_setup(network)
-        @testlab.ui.stdout.puts("Added routes successfully!".green.bold)
-        @testlab.ui.stdout.puts %x(netstat -nr | grep '#{network.node.ip}').strip
+        iterate_objects_by_name(options[:name], TestLab::Network) do |network|
+          p = TestLab::Provisioner::Route.new({}, @ui)
+          p.on_network_setup(network)
+          @testlab.ui.stdout.puts("Added routes successfully!".green.bold)
+          @testlab.ui.stdout.puts %x(netstat -nr | grep '#{network.node.ip}').strip
+        end
       end
     end
 
     # ROUTE DEL
     ############
     route.desc 'Delete routes to lab networks'
     route.command :del do |del|
       del.action do |global_options,options,args|
-        help_now!('a name is required') if options[:name].nil?
-
-        network = @testlab.networks.select{ |c| c.id.to_sym == options[:name].to_sym }.first
-        network.nil? and raise TestLab::TestLabError, "We could not find the network you supplied!"
-
-        p = TestLab::Provisioner::Route.new({}, @ui)
-        p.on_network_teardown(network)
-        @testlab.ui.stdout.puts("Deleted routes successfully!".red.bold)
-        @testlab.ui.stdout.puts %x(netstat -nr | grep '#{network.node.ip}').strip
+        iterate_objects_by_name(options[:name], TestLab::Network) do |network|
+          p = TestLab::Provisioner::Route.new({}, @ui)
+          p.on_network_teardown(network)
+          @testlab.ui.stdout.puts("Deleted routes successfully!".red.bold)
+          @testlab.ui.stdout.puts %x(netstat -nr | grep '#{network.node.ip}').strip
+        end
       end
     end
 
     # ROUTE SHOW
     #############
     route.desc 'Show routes to lab networks'
     route.command :show do |show|
       show.action do |global_options,options,args|
-        help_now!('a name is required') if options[:name].nil?
-
-        network = @testlab.networks.select{ |c| c.id.to_sym == options[:name].to_sym }.first
-        network.nil? and raise TestLab::TestLabError, "We could not find the network you supplied!"
-
-        @testlab.ui.stdout.puts("TestLab routes:".green.bold)
-        case RUBY_PLATFORM
-        when /darwin/ then
-          @testlab.ui.stdout.puts %x(netstat -nrf inet | grep '#{network.node.ip}').strip
-        when /linux/ then
-          @testlab.ui.stdout.puts %x(netstat -nr | grep '#{network.node.ip}').strip
+        iterate_objects_by_name(options[:name], TestLab::Network) do |network|
+          @testlab.ui.stdout.puts("TestLab routes:".green.bold)
+          case RUBY_PLATFORM
+          when /darwin/ then
+            @testlab.ui.stdout.puts %x(netstat -nrf inet | grep '#{network.node.ip}').strip
+          when /linux/ then
+            @testlab.ui.stdout.puts %x(netstat -nr | grep '#{network.node.ip}').strip
+          end
         end
       end
     end
   end
 