nifty 0.2.5 → 0.2.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 5cdc153b695c7aff6d7f455493d363fff9e25b86
-  data.tar.gz: 391b28158089f86dadd7d65afd043f5ccd688401
+  metadata.gz: 8cf414409a82b584fa64445d7ab380746e8047b9
+  data.tar.gz: 8f3786014089406349e47db7be0fa96156279044
 SHA512:
-  metadata.gz: 7975505c83737b902cb1c1aba634c842e19c96fa2c3eb998c7a2807ec3160ed2e90a0fc844db911ac7828f5331ff7af7490d3cda6393d9b49c09c555260fd776
-  data.tar.gz: ea370f4d5d5d872423eb84a111601e7614d3724ba4aa26f34c03244b93349b4c2f79fed015d171271535d8714c8cf67519ece310dc6a70d2c71f1414f9a650d2
+  metadata.gz: 01d3d5df7891b4d48cabf47e1f0210f7f474062ae2fe03e4329e94eb6ce210834c5c11260f76cade2391c4d106d0f6a453297764004b413851f7da7e9c1364d4
+  data.tar.gz: a033fda42899d9a49e4bbeb0294f0bdae0e02343affc74ffcd82a9769fb93f880e0c23e6e351d4dedec7b6bc44e068e01779e8f929d0a5e7ca7b442b5e507c6c

data/lib/nifty/backend.rb

@@ -27,6 +27,14 @@ class Nifty::Backend
       nil
     end
 
+    # Returns hash from which new CLI command for backend is constructed
+    #
+    # @abstract
+    # @return [Hash, nil] hash from which new CLI command for backend is constructed
+    def migrate_options
+      nil
+    end
+
     # Routine run before any events are processed
     #
     # @abstract
@@ -41,6 +49,13 @@ class Nifty::Backend
     def post(parameters)
     end
 
+
+    # Prepares backend for integration with NIFTY
+    #
+    # @param [Hash] parameters to help the migration
+    def migrate(parameters)
+    end
+
     # Creates specific event for the backend
     #
     # @abstract

data/lib/nifty/backends/opennebula.rb

@@ -4,6 +4,14 @@ require 'chronic_duration'
 
 # OpenNebula backend
 class Nifty::Backends::Opennebula < Nifty::Backend
+  VMCATCHER_APPLIANCE_ID = 'VMCATCHER_EVENT_DC_IDENTIFIER'
+  VMCATCHER_APPLIANCE_VERSION = 'VMCATCHER_EVENT_HV_VERSION'
+  VMCATCHER_APPLIANCE_TITLE = 'VMCATCHER_EVENT_DC_TITLE'
+  VMCATCHER_APPLIANCE_DESCRIPTION = 'VMCATCHER_EVENT_DC_DESCRIPTION'
+  VMCATCHER_APPLIANCE_OS_ARCH = 'VMCATCHER_EVENT_SL_ARCH'
+  VMCATCHER_APPLIANCE_OS_DISTRIBUTION = 'VMCATCHER_EVENT_SL_OS'
+  VMCATCHER_APPLIANCE_OS_VERSION = 'VMCATCHER_EVENT_SL_OSVERSION'
+
   class << self
     # @see Nifty::Backend#backend?
     def backend?
@@ -66,6 +74,24 @@ class Nifty::Backends::Opennebula < Nifty::Backend
       }
     end
 
+    # @see Nifty::Backend#migrate_options
+    def migrate_options
+      { :secret => {
+          :type => :string,
+          :desc => 'Pair of username and password in form of \'username:password\' for accessing OpenNebula'
+        },
+        :endpoint => {
+          :type => :string,
+          :desc => 'OpenNebula\'s XML RPC endpoint'
+        },
+        :"api-call-timeout" => {
+          :required => true,
+          :type => :string,
+          :desc => 'How long will NIFTY wait for image/template operations to finish in OpenNebula'
+        }
+      }
+    end
+
     # @see Nifty::Backend#pre
     def pre(parameters)
       Nifty::Backends::Utils::Opennebula::Handler.api_call_timeout = ChronicDuration.parse(parameters[:'api-call-timeout'], keep_zero: true)
@@ -86,6 +112,48 @@ class Nifty::Backends::Opennebula < Nifty::Backend
       fail Nifty::Errors::BackendError, ex
     end
 
+    def migrate(parameters)
+      client = Nifty::Backends::Utils::Opennebula::Helper.client(parameters[:secret], parameters[:endpoint])
+      image_handler = Nifty::Backends::Utils::Opennebula::ImageHandler.new(client)
+      template_handler = Nifty::Backends::Utils::Opennebula::TemplateHandler.new(client)
+
+      image_handler.unmanaged_images.each do |image|
+        image_name = image.name
+        logger.debug "Processing image #{image_name}"
+
+        os = Cloud::Appliance::Descriptor::Os.new
+        os.arch = image["TEMPLATE/#{VMCATCHER_APPLIANCE_OS_ARCH}"]
+        os.version = image["TEMPLATE/#{VMCATCHER_APPLIANCE_OS_VERSION}"]
+        os.distribution = image["TEMPLATE/#{VMCATCHER_APPLIANCE_OS_DISTRIBUTION}"]
+
+        appliance = Cloud::Appliance::Descriptor::Appliance.new os: os
+        appliance.identifier = image["TEMPLATE/#{VMCATCHER_APPLIANCE_ID}"]
+        appliance.version = image["TEMPLATE/#{VMCATCHER_APPLIANCE_VERSION}"]
+        appliance.title = image["TEMPLATE/#{VMCATCHER_APPLIANCE_TITLE}"]
+        appliance.description = image["TEMPLATE/#{VMCATCHER_APPLIANCE_DESCRIPTION}"]
+
+        image_template = Tilt::ERBTemplate.new(File.join(Nifty::GEM_DIR, 'templates', 'image.erb'))
+        image_template_update = image_template.render(Object.new, {name: image_name, appliance: appliance})
+        logger.debug("Updating image template for image #{image_name}")
+        Nifty::Backends::Utils::Opennebula::Helper.handle_opennebula_error { image.update(image_template_update, true) }
+
+        logger.debug("Looking for templates with image #{image_name}...")
+        template_handler.unmanaged_templates_with_image(image_name).each do |template|
+          template_name = template.name
+          logger.debug "Processing template #{template_name}"
+          template_template = Tilt::ERBTemplate.new(File.join(Nifty::GEM_DIR, 'templates', 'template.erb'))
+          template_template_update = template_template.render(Object.new, {name: template_name, appliance: appliance})
+          logger.debug("Updating template template for template #{template_name}")
+          Nifty::Backends::Utils::Opennebula::Helper.handle_opennebula_error { template.update(template_template_update, true) }
+        end
+      end
+
+      Nifty::ExitCodes::NO_ERROR_EXIT_CODE
+    rescue Nifty::Errors::Backends::OpennebulaError => ex
+      logger.error "Migration error: #{ex.message}"
+      Nifty::ExitCodes::MIGRATION_ERROR_EXIT_CODE
+    end
+
     private
 
     # Checks whether there are expired images that should be removed

data/lib/nifty/backends/utils/opennebula/image_handler.rb

@@ -43,6 +43,12 @@ class Nifty::Backends::Utils::Opennebula::ImageHandler < Nifty::Backends::Utils:
     pool.find_all { |image| image["TEMPLATE/#{ATTRIBUTE_EXPIRATION}"] }
   end
 
+  def unmanaged_images()
+    reload!
+
+    pool.find_all { |image| image["TEMPLATE/#{ATTRIBUTE_APPLIANCE_ID}"].nil? && image["TEMPLATE/#{Nifty::Backends::Opennebula::VMCATCHER_APPLIANCE_ID}"] }
+  end
+
   def images_by_version(version)
     reload!
 

data/lib/nifty/backends/utils/opennebula/template_handler.rb

@@ -23,7 +23,13 @@ class Nifty::Backends::Utils::Opennebula::TemplateHandler < Nifty::Backends::Uti
   def templates(appliance_id)
     reload!
 
-    pool.find_all{ |template| template["TEMPLATE/#{ATTRIBUTE_APPLIANCE_ID}"] == appliance_id }
+    pool.find_all { |template| template["TEMPLATE/#{ATTRIBUTE_APPLIANCE_ID}"] == appliance_id }
+  end
+
+  def unmanaged_templates_with_image(image_name)
+    reload!
+
+    pool.find_all { |template| template["TEMPLATE/DISK/IMAGE"] == image_name && template["TEMPLATE/#{ATTRIBUTE_APPLIANCE_ID}"].nil? }
   end
 
   # Returns template with given id if exists

data/lib/nifty/command_executioner.rb

@@ -114,6 +114,21 @@ def #{backend}_transfer_methods
   end
 end
 ^
+
+    if backend_clazz.respond_to? 'migrate'
+      options = backend_clazz.migrate_options
+      options.each do |name, parameters|
+        parameters[:default] = Nifty::Settings[backend][name.to_s] unless parameters[:default]
+        method_option name, parameters
+      end
+
+      class_eval %Q^
+desc '#{backend}-migrate', 'Prepares #{backend.inspect} for integration with NIFTY'
+def #{backend}_migrate
+  migrate('#{backend}', options)
+end
+^
+    end
   end
 
   private
@@ -138,6 +153,16 @@ end
     exit processor.process_events
   end
 
+  def migrate(backend, options)
+    parameters = options.to_hash.deep_symbolize_keys
+    init_log parameters
+
+    backend = Nifty::Backends.const_get(backend.camelize)
+    logger.debug "Selected backend '#{backend.inspect}'"
+
+    exit backend.migrate(parameters)
+  end
+
   # Inits logging according to the settings
   #
   # @param [Hash] parameters

data/lib/nifty/exit_codes.rb

@@ -5,4 +5,5 @@ class Nifty::ExitCodes
   EVENT_PREPARATION_ERROR_EXIT_CODE = 3
   NO_TRANSFER_METHOD_ERROR_EXIT_CODE = 4
   TRANSFER_METHOD_ERROR_EXIT_CODE = 5
+  MIGRATION_ERROR_EXIT_CODE = 6
 end

data/lib/nifty/version.rb

@@ -1,3 +1,3 @@
 module Nifty
-  VERSION = "0.2.5"
+  VERSION = "0.2.6"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: nifty
 version: !ruby/object:Gem::Version
-  version: 0.2.5
+  version: 0.2.6
 platform: ruby
 authors:
 - Michal Kimle
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-02-02 00:00:00.000000000 Z
+date: 2016-02-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler