aeolus-image 0.0.1

data/lib/build_command.rb

@@ -0,0 +1,68 @@
+require 'imagefactory'
+
+module Aeolus
+  module Image
+    class BuildCommand < BaseCommand
+      attr_accessor :console
+      def initialize(opts={}, logger=nil)
+        super(opts, logger)
+        default = {
+          :template_str => '',
+          :template => '',
+          :target => [],
+          :image => '',
+          :build => ''
+        }
+        @options = default.merge(@options)
+        @console = ImageFactoryConsole.new()
+        @console.start
+      end
+
+      def run
+        if combo_implemented?
+          @options[:template_str] = read_file(@options[:template])
+          if @options[:template_str].nil?
+            puts "Cannot find specified file"
+            quit(1)
+          end
+
+          # Validate XML against TDL Schema
+          errors = validate_xml_document(File.dirname(__FILE__) + "/../examples/tdl.rng", @options[:template_str])
+          if errors.length > 0
+            puts "ERROR: The given Template does not conform to the TDL Schema, see below for specific details:"
+            errors.each do |error|
+              puts "- " + error.message
+            end
+            quit(1)
+          end
+
+          #This is a temporary hack in case the agent doesn't show up on bus immediately
+          sleep(5)
+          @console.build(@options[:template_str], @options[:target], @options[:image], @options[:build]).each do |adaptor|
+            puts ""
+            puts "Target Image: #{adaptor.image_id}"
+            puts "Image: #{adaptor.image}"
+            puts "Build: #{adaptor.build}"
+            puts "Status: #{adaptor.status}"
+            puts "Percent Complete: #{adaptor.percent_complete}"
+          end
+          quit(0)
+        end
+      end
+
+      def combo_implemented?
+        if @options[:template].empty? || @options[:target].empty?
+          puts "This combination of parameters is not currently supported"
+          quit(1)
+        end
+        true
+      end
+
+      def quit(code)
+        @console.shutdown
+        super
+      end
+
+    end
+  end
+end

data/lib/config_parser.rb

@@ -0,0 +1,212 @@
+require 'optparse'
+require 'logger'
+require 'base_command'
+require 'list_command'
+require 'build_command'
+require 'push_command'
+require 'import_command'
+require 'delete_command'
+
+module Aeolus
+  module Image
+    class ConfigParser
+      COMMANDS = %w(list build push import delete)
+      attr_accessor :options, :command, :args
+
+      def initialize(argv)
+        @args = argv
+        # Default options
+        @options = {}
+        parse
+      end
+
+      def process
+        # Check for command, then call appropriate Optionparser and initiate
+        # call to that class.
+        @command = @args.shift
+        # Eventually get the config file from user dir if it exists.
+        # File.expand_path("~")
+        if COMMANDS.include?(@command)
+          self.send(@command.to_sym)
+        else
+          @args << "-h"
+          puts "Valid command required: \n\n"
+          parse
+        end
+      end
+
+      private
+      def parse
+        @optparse ||= OptionParser.new do|opts|
+          opts.banner = "Usage: aeolus-image [#{COMMANDS.join('|')}] [general options] [command options]"
+
+          opts.separator ""
+          opts.separator "General options:"
+          opts.on('-u', '--user USERNAME', 'Conductor username') do |user|
+            @options[:user] = user
+          end
+          opts.on('-w', '--password PASSWORD', 'Conductor password') do |pw|
+            @options[:password] = pw
+          end
+          opts.on('-d', '--id ID', 'id for a given object') do |id|
+            @options[:id] = id
+          end
+          opts.on('-r', '--description NAME', 'description (e.g. "<image><name>MyImage</name></image>" or "/home/user/myImage.xml")') do |description|
+            @options[:description] = description
+          end
+          opts.on('-r', '--provider NAME1,NAME2',  Array,'name of specific provider (ie ec2-us-east1)') do |name|
+            @options[:provider] = name
+          end
+          opts.on('-I', '--image ID', 'ID of the base image, can be used in build and push commands, see examples') do |id|
+            @options[:image] = id
+          end
+          opts.on('-T', '--target TARGET1,TARGET2', Array, 'provider type (ec2, rackspace, rhevm, etc)') do |name|
+            @options[:target] = name
+          end
+          opts.on('-d', '--daemon', 'run as a background process') do
+            @options[:subcommand] = :images
+          end
+          opts.on( '-h', '--help', 'Get usage information for this tool') do
+            puts opts
+            exit(0)
+          end
+
+          opts.separator ""
+          opts.separator "List options:"
+          opts.on('-i', '--images', 'Retrieve a list of images') do
+            @options[:subcommand] = :images
+          end
+          opts.on('-b', '--builds ID', 'Retrieve the builds of an image') do |id|
+            @options[:subcommand] = :builds
+            @options[:id] = id
+          end
+          opts.on('-t', '--targetimages ID', 'Retrieve the target images from a build') do |id|
+            @options[:subcommand] = :targetimages
+            @options[:id] = id
+          end
+          opts.on('-P', '--providerimages ID', 'Retrieve the provider images from a target image') do |id|
+            @options[:subcommand] = :targetimages
+            @options[:id] = id
+          end
+          opts.on('-g', '--targets', 'Retrieve the values available for the --target parameter') do
+            @options[:subcommand] = :targets
+          end
+          opts.on('-p', '--providers', 'Retrieve the values available for the --provider parameter') do
+            @options[:subcommand] = :providers
+          end
+          opts.on('-a', '--accounts', 'Retrieve the values available for the --account parameter') do
+            @options[:subcommand] = :accounts
+          end
+
+          opts.separator ""
+          opts.separator "Build options:"
+          opts.on('-e', '--template FILE', 'path to file that contains template xml') do |file|
+            @options[:template] = file
+          end
+
+          opts.separator ""
+          opts.separator "Push options:"
+          opts.on('-B', '--build ID', 'push all target images for a build, to same providers as previously') do |id|
+            @options[:build] = id
+          end
+          opts.on('-A', '--account NAME', 'name of specific provider account to use for push') do |name|
+            @options[:account] = name
+          end
+
+          opts.separator ""
+          opts.separator "Delete options:"
+          opts.on('-m', '--targetimage ID', 'delete target image and its provider images') do |id|
+            @options[:targetimage] = id
+          end
+          opts.on('-D', '--providerimage ID', 'delete provider image') do |id|
+            @options[:providerimage] = id
+          end
+
+          opts.separator ""
+          opts.separator "List Examples:"
+          opts.separator "aeolus-image list --images                  # list available images"
+          opts.separator "aeolus-image list --builds $image_id        # list the builds of an image"
+          opts.separator "aeolus-image list --targetimages $build_id  # list the target images from a build"
+          opts.separator "aeolus-image list --targets                 # list the values available for the --target parameter"
+          opts.separator "aeolus-image list --providers               # list the values available for the --provider parameter"
+          opts.separator "aeolus-image list --accounts                # list the values available for the --account parameter"
+
+          opts.separator ""
+          opts.separator "Build examples:"
+          opts.separator "aeolus-image build --target ec2 --template my.tmpl  # build a new image for ec2 from the template"
+          opts.separator "aeolus-image build --image $image_id                # (NOT IMPLEMENTED) rebuild the image template and targets from latest build"
+          opts.separator %q{aeolus-image build --target ec2,rackspace \         # rebuild the image with a new template and set of targets
+                   --image $image_i \
+                   --template my.tmpl}
+
+          opts.separator ""
+          opts.separator "Push examples:"
+          opts.separator "aeolus-image push --provider ec2-us-east-1 --id $image_id         # initial push of an image build via image id to the specified provider"
+          opts.separator "aeolus-image push --build $build_id                               # push an image build via build id to the specified provider"
+          opts.separator "aeolus-image push --account $provider_account --build $build_id   # (NOT IMPLEMENTED) ditto, using a specific provider account"
+          opts.separator "aeolus-image push --image $image_id                               # (NOT IMPLEMENTED) push all the target images for the latest build"
+
+          opts.separator ""
+          opts.separator "Import examples:"
+          opts.separator "aeolus-image import --provider ec2-us-east-1 --target ec2 --id $ami_id # import an AMI from the specified provider"
+          opts.separator "aeolus-image import --provider ec2-us-east-1 --target ec2 --id $ami_id --description '<image><name>My Image</name></image>' # import an AMI from the specified provider"
+          opts.separator "aeolus-image import --provider ec2-us-east-1 --target ec2 --id $ami_id --description <path_to_xml_file> # import an AMI from the specified provider"
+
+          opts.separator ""
+          opts.separator "Delete examples: (DELETE CURRENTLY NOT IMPLEMENTED) "
+          opts.separator "aeolus-image delete --build $build_id               # deletes a build, updating latest/parent references as appropriate"
+          opts.separator "aeolus-image delete --targetimage $target_image     # deletes a target image and its provider images"
+          opts.separator "aeolus-image delete --providerimage $provider_image # deletes a provider image"
+        end
+
+        begin
+          @optparse.parse!(@args)
+        rescue OptionParser::InvalidOption
+          puts "Warning: Invalid option"
+          exit(1)
+        rescue OptionParser::MissingArgument => e
+          puts "Warning, #{e.message}"
+          exit(1)
+        end
+      end
+
+      # TODO: Remove all this boilerplate and replace with some metaprogramming,
+      # perhaps method_missing
+      def list
+        # TODO: Instantiate and call object matching command type, for example:
+        # l = ListCommand.new(@options)
+        # Each Command will call it's own internal method depending on the contents of the hash.
+        # For the list example above, that object would call a method 'images' based on the item
+        # @options[:subcommand] being :images, so internally that class may do something like:
+        # self.send(@options[:subcommand])
+        if @options[:subcommand].nil?
+          # TODO: Pull out Print Usage into seporate method, and print
+          puts "Could not find subcommand for list, run `./aeolus-image --help` for usage instructions"
+          exit(1)
+        else
+          list_command = ListCommand.new(@options)
+          list_command.send(@options[:subcommand])
+        end
+      end
+
+      def build
+        b = BuildCommand.new(@options)
+        b.run
+      end
+
+      def push
+        b = PushCommand.new(@options)
+        b.run
+      end
+
+      def import
+        import_command = ImportCommand.new(@options)
+        import_command.import_image
+      end
+
+      def delete
+        "Not implemented"
+      end
+    end
+  end
+end

data/lib/delete_command.rb

@@ -0,0 +1,9 @@
+module Aeolus
+  module Image
+    class DeleteCommand < BaseCommand
+      def initialize(opts={}, logger=nil)
+        super(opts, logger)
+      end
+    end
+  end
+end

data/lib/import_command.rb

@@ -0,0 +1,44 @@
+module Aeolus
+  module Image
+    class ImportCommand < BaseCommand
+      def initialize(opts={}, logger=nil)
+        super(opts, logger)
+        default = {
+          :image => '',
+          :build => '',
+          :id => '',
+          :description => '<image><name>' + @options[:id] + '</name></image>',
+          :target => '',
+          :provider => ''
+        }
+        @options = default.merge(@options)
+        @console = ImageFactoryConsole.new()
+        @console.start
+      end
+
+      def import_image
+        description = read_file(@options[:description])
+        if !description.nil?
+          @options[:description] = description
+        end
+        # TODO: Validate Description XML
+
+        #This is a temporary hack in case the agent doesn't show up on bus
+        #immediately
+        sleep(5)
+        import_map = @console.import_image(@options[:image], @options[:build], @options[:id], @options[:description], @options[:target].first, @options[:provider].first)
+        puts ""
+        puts "Target Image: " + import_map['target_image']
+        puts "Image: " + import_map['image']
+        puts "Build: " + import_map['build']
+        puts "Provider Image: " + import_map['provider_image']
+        quit(0)
+      end
+
+      def quit(code)
+        @console.shutdown
+        super
+      end
+    end
+  end
+end

data/lib/list_command.rb

@@ -0,0 +1,141 @@
+module Aeolus
+  module Image
+    class ListCommand < BaseCommand
+      def initialize(opts={}, logger=nil)
+        super(opts, logger)
+      end
+
+      def images
+        images = [["IMAGE ID", "LASTEST PUSHED BUILD", "NAME", "TARGET", "OS", "OS VERSION", "ARCH", "DESCRIPTION"]]
+        doc = Nokogiri::XML iwhd['/target_images'].get
+        # Check for any invalid data in iwhd
+        invalid_images = []
+        doc.xpath("/objects/object/key").each do |targetimage|
+          begin
+            build = iwhd["/target_images/" + targetimage.text + "/build"].get
+            image = iwhd["/builds/" + build + "/image"].get
+
+            if template_info = get_template_info(image, targetimage.text)
+              images << [image] + [lastest_pushed(image)] + template_info
+            else
+              images << [image] + [lastest_pushed(image)] +[get_image_name(image), iwhd["/target_images/" + targetimage + "/target"].get, "", "", "", ""]
+            end
+          rescue
+            invalid_images << targetimage.text
+          end
+        end
+        format_print(images)
+
+        unless invalid_images.empty?
+          puts "\nN.B. following images were not listed, aeolus-image encountered some invalid data in iwhd:"
+          puts invalid_images.join "\n"
+        end
+        quit(0)
+      end
+
+      def builds
+        doc = Nokogiri::XML iwhd['/builds'].get
+        doc.xpath("/objects/object/key").each do |build|
+          if iwhd['/builds/' + build.text + "/image"].get == @options[:id]
+            puts build.text
+          end
+        end
+        quit(0)
+      end
+
+      def targetimages
+        doc = Nokogiri::XML iwhd['/target_images'].get
+        doc.xpath("/objects/object/key").each do |target_image|
+          begin
+            if iwhd['/target_images/' + target_image.text + "/build"].get == @options[:id]
+              puts target_image.text
+            end
+          rescue RestClient::ResourceNotFound
+          end
+        end
+        quit(0)
+      end
+
+      def targets
+        targets = [["NAME", "TARGET CODE"]]
+        targets << ["Mock", "mock"]
+        targets << ["Amazon EC2", "ec2"]
+        targets << ["RHEV-M", "rhevm"]
+        targets << ["VMware vSphere", "vsphere"]
+        targets << ["Condor Cloud", "condor_cloud"]
+        format_print(targets)
+        quit(0)
+      end
+
+      def providers
+        print_values = [["NAME", "TYPE", "URL"]]
+
+        doc = Nokogiri::XML conductor['/providers'].get
+        doc.xpath("/providers/provider").each do |provider|
+          print_values << [provider.xpath("name").text, provider.xpath("provider_type").text, provider.xpath("url").text]
+        end
+
+        format_print(print_values)
+        quit(0)
+      end
+
+      def accounts
+        print_values = [["NAME", "PROVIDER", "PROVIDER TYPE"]]
+        doc = Nokogiri::XML conductor['/provider_accounts/'].get
+        doc.xpath("/provider_accounts/provider_account").each do |account|
+          print_values << [account.xpath("name").text, account.xpath("provider").text, account.xpath("provider_type").text]
+        end
+
+        format_print(print_values)
+        quit(0)
+      end
+
+      private
+      # Takes a 2D array of strings and neatly prints them to STDOUT
+      def format_print(print_values)
+        widths =  Array.new(print_values[0].size, 0)
+        print_values.each do |print_value|
+          widths = widths.zip(print_value).map! {|width, value| value.length > width ? value.length : width }
+        end
+
+        print_values.each do |print_value|
+          widths.zip(print_value) do |width, value|
+            printf("%-#{width + 5}s", value)
+          end
+          puts ""
+        end
+      end
+
+      def get_template_info(image, targetimage)
+        begin
+          template = Nokogiri::XML iwhd["/templates/" + iwhd["/target_images/" + targetimage + "/template"].get].get
+          [template.xpath("/template/name").text,
+            iwhd["/target_images/" + targetimage + "/target"].get,
+            template.xpath("/template/os/name").text,
+            template.xpath("/template/os/version").text,
+            template.xpath("/template/os/arch").text,
+            template.xpath("/template/description").text]
+        rescue
+        end
+      end
+
+      def get_image_name(image)
+        begin
+          template_xml = Nokogiri::XML iwhd["images/" + image].get
+          template_xml.xpath("/image/name").text
+        rescue
+          ""
+        end
+      end
+
+      def lastest_pushed(image)
+        begin
+          build = iwhd["/images/" + image + "/latest_build"].get
+          build.nil? ? "" : build
+        rescue
+          ""
+        end
+      end
+    end
+  end
+end