vagrant-openstack-illuin-provider 0.12.0

data/lib/vagrant-openstack-illuin-provider/action/message.rb

@@ -0,0 +1,19 @@
+require 'vagrant-openstack-illuin-provider/action/abstract_action'
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      class Message < AbstractAction
+        def initialize(app, _env, message)
+          @app = app
+          @message = message
+        end
+
+        def execute(env)
+          env[:ui].info(@message)
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openstack-illuin-provider/action/provision.rb

@@ -0,0 +1,60 @@
+require 'log4r'
+
+require 'vagrant/action/builder'
+
+require 'vagrant-openstack-illuin-provider/action/abstract_action'
+require 'vagrant-openstack-illuin-provider/action/read_ssh_info'
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      include Vagrant::Action::Builtin
+
+      class ProvisionWrapper < AbstractAction
+        def initialize(app, env)
+          @app    = app
+          @env    = env
+          @logger = Log4r::Logger.new('vagrant_openstack::action::provision_wrapper')
+        end
+
+        def execute(env)
+          @logger.info 'Run provisioning'
+          InternalProvisionWrapper.new(@app, @env).call(@env)
+          @app.call(env)
+        end
+      end
+
+      class InternalProvisionWrapper < Vagrant::Action::Builtin::Provision
+        def initialize(app, env)
+          @logger = Log4r::Logger.new('vagrant_openstack::action::internal_provision_wrapper')
+          super app, env
+        end
+
+        def run_provisioner(env)
+          if env[:provisioner].is_a?(Vagrant.plugin('2').manager.provisioners[:shell])
+            handle_shell_meta_args(env)
+          end
+          env[:provisioner].provision
+        end
+
+        private
+
+        def handle_shell_meta_args(env)
+          config = env[:provisioner].config
+          args = config.args.nil? ? [] : [config.args].flatten
+          config.args = []
+          @logger.info "Shell provisioner args: #{args}"
+          args.each do |arg|
+            if '@@ssh_ip@@'.eql? arg
+              ssh_info = VagrantPlugins::Openstack::Action.get_ssh_info(env)
+              @logger.info "Replace meta-arg #{arg} by value #{ssh_info[:host]}"
+              config.args << ssh_info[:host]
+            else
+              config.args << arg
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openstack-illuin-provider/action/read_ssh_info.rb

@@ -0,0 +1,74 @@
+require 'log4r'
+
+require 'vagrant-openstack-illuin-provider/config_resolver'
+require 'vagrant-openstack-illuin-provider/utils'
+require 'vagrant-openstack-illuin-provider/action/abstract_action'
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      # This action reads the SSH info for the machine and puts it into the
+      # `:machine_ssh_info` key in the environment.
+
+      class ReadSSHInfo < AbstractAction
+        def initialize(app, _env, resolver = ConfigResolver.new, utils = Utils.new)
+          @app    = app
+          @logger = Log4r::Logger.new('vagrant_openstack::action::read_ssh_info')
+          @resolver = resolver
+          @utils = utils
+        end
+
+        def execute(env)
+          @logger.info 'Reading SSH info'
+          server_id = env[:machine].id.to_sym
+          SSHInfoHolder.instance.tap do |holder|
+            holder.synchronize do
+              holder.ssh_info[server_id] = read_ssh_info(env) if holder.ssh_info[server_id].nil?
+              env[:machine_ssh_info] = holder.ssh_info[server_id]
+            end
+          end
+          @app.call(env)
+        end
+
+        private
+
+        def read_ssh_info(env)
+          config = env[:machine].provider_config
+          env[:ui].warn('SSH is disabled in the provider config. The action you are attempting is likely to fail') if config.ssh_disabled
+          hash = {
+            host: @utils.get_ip_address(env),
+            port: @resolver.resolve_ssh_port(env),
+            username: @resolver.resolve_ssh_username(env)
+          }
+          if env[:machine].config.ssh.insert_key
+            hash[:private_key_path] = "#{env[:machine].data_dir}/#{get_keypair_name(env)}" unless config.keypair_name || config.public_key_path
+          end
+          # Should work silently when https://github.com/mitchellh/vagrant/issues/4637 is fixed
+          hash[:log_level] = 'ERROR'
+          hash
+        end
+
+        def get_keypair_name(env)
+          env[:openstack_client].nova.get_server_details(env, env[:machine].id)['key_name']
+        end
+      end
+
+      class SSHInfoHolder < Mutex
+        include Singleton
+
+        #
+        # Keys are machine ids
+        #
+        attr_accessor :ssh_info
+
+        def initialize
+          @ssh_info = {}
+        end
+      end
+
+      def self.get_ssh_info(env)
+        SSHInfoHolder.instance.ssh_info[env[:machine].id.to_sym]
+      end
+    end
+  end
+end

data/lib/vagrant-openstack-illuin-provider/action/read_state.rb

@@ -0,0 +1,43 @@
+require 'log4r'
+
+require 'vagrant-openstack-illuin-provider/action/abstract_action'
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      # This action reads the state of the machine and puts it in the
+      # `:machine_state_id` key in the environment.
+      class ReadState < AbstractAction
+        def initialize(app, _env)
+          @app    = app
+          @logger = Log4r::Logger.new('vagrant_openstack::action::read_state')
+        end
+
+        def execute(env)
+          env[:machine_state_id] = read_state(env)
+          @app.call(env)
+        end
+
+        def read_state(env)
+          machine = env[:machine]
+          return :not_created if machine.id.nil?
+
+          # Find the machine
+          server = env[:openstack_client].nova.get_server_details(env, machine.id)
+          if server.nil? || server['status'] == 'DELETED'
+            # The machine can't be found
+            @logger.info('Machine not found or deleted, assuming it got destroyed.')
+            machine.id = nil
+            return :not_created
+          end
+
+          if !server['OS-EXT-STS:task_state'].nil?
+            server['OS-EXT-STS:task_state'].downcase.to_sym
+          else
+            server['status'].downcase.to_sym
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openstack-illuin-provider/action/resume.rb

@@ -0,0 +1,24 @@
+require 'vagrant-openstack-illuin-provider/action/abstract_action'
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      class Resume < AbstractAction
+        def initialize(app, _env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_openstack::action::resume_server')
+        end
+
+        def execute(env)
+          if env[:machine].id
+            @logger.info "Resuming suspended VM #{env[:machine].id}..."
+            env[:ui].info I18n.t('vagrant.actions.vm.resume.resuming')
+            env[:openstack_client].nova.resume_server(env, env[:machine].id)
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openstack-illuin-provider/action/snapshot_cleanup.rb

@@ -0,0 +1,32 @@
+require 'vagrant-openstack-illuin-provider/action/abstract_action'
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      class SnapshotCleanup < AbstractAction
+        def initialize(app, _env)
+          @app = app
+        end
+
+        def call(env)
+          nova = env[:openstack_client].nova
+          machine_snapshots = nova.list_snapshots(env, env[:machine].id)
+
+          snapshots_to_clean = machine_snapshots.select do |s|
+            s.metadata.key?('vagrant_snapshot')
+          end
+
+          @app.call env
+
+          unless snapshots_to_clean.empty?
+            env[:ui].info("Deleting Vagrant snapshots: #{snapshots_to_clean.map(&:name)}")
+          end
+
+          snapshots_to_clean.each do |s|
+            nova.delete_snapshot(env, s.id)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openstack-illuin-provider/action/snapshot_delete.rb

@@ -0,0 +1,32 @@
+require 'vagrant-openstack-illuin-provider/action/abstract_action'
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      class SnapshotDelete < AbstractAction
+        def initialize(app, _env)
+          @app = app
+        end
+
+        def call(env)
+          nova = env[:openstack_client].nova
+          machine_snapshots = nova.list_snapshots(env, env[:machine].id)
+
+          snapshot = machine_snapshots.find { |s| s.name == env[:snapshot_name] }
+
+          unless snapshot.nil?
+            env[:ui].info(I18n.t('vagrant.actions.vm.snapshot.deleting',
+                                 name: snapshot.name))
+
+            nova.delete_snapshot(env, snapshot.id)
+
+            env[:ui].info(I18n.t('vagrant.actions.vm.snapshot.deleted',
+                                 name: snapshot.name))
+          end
+
+          @app.call env
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openstack-illuin-provider/action/snapshot_list.rb

@@ -0,0 +1,22 @@
+require 'vagrant-openstack-illuin-provider/action/abstract_action'
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      class SnapshotList < AbstractAction
+        def initialize(app, _env)
+          @app = app
+        end
+
+        def call(env)
+          nova = env[:openstack_client].nova
+          machine_snapshots = nova.list_snapshots(env, env[:machine].id)
+
+          env[:machine_snapshot_list] = machine_snapshots.map(&:name)
+
+          @app.call env
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openstack-illuin-provider/action/snapshot_restore.rb

@@ -0,0 +1,29 @@
+require 'vagrant-openstack-illuin-provider/action/abstract_action'
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      class SnapshotRestore < AbstractAction
+        def initialize(app, _env)
+          @app = app
+        end
+
+        def call(env)
+          nova = env[:openstack_client].nova
+          machine_snapshots = nova.list_snapshots(env, env[:machine].id)
+
+          snapshot = machine_snapshots.find { |s| s.name == env[:snapshot_name] }
+
+          unless snapshot.nil?
+            env[:ui].info(I18n.t('vagrant.actions.vm.snapshot.restoring',
+                                 name: snapshot.name))
+
+            nova.restore_snapshot(env, env[:machine].id, snapshot.id)
+          end
+
+          @app.call env
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openstack-illuin-provider/action/snapshot_save.rb

@@ -0,0 +1,51 @@
+require 'vagrant-openstack-illuin-provider/action/abstract_action'
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      class SnapshotSave < AbstractAction
+        def initialize(app, _env, retry_interval = 3)
+          @app = app
+          @retry_interval = retry_interval
+        end
+
+        def call(env)
+          nova = env[:openstack_client].nova
+          config = env[:machine].provider_config
+
+          env[:ui].info(I18n.t('vagrant.actions.vm.snapshot.saving',
+                               name: env[:snapshot_name]))
+
+          nova.create_snapshot(
+            env,
+            env[:machine].id, env[:snapshot_name])
+
+          image = nova.list_snapshots(env, env[:machine].id).find { |i| i.name == env[:snapshot_name] }
+
+          timeout(config.server_create_timeout, Errors::Timeout) do
+            loop do
+              image_status = nova.get_image_details(env, image.id)
+
+              break if image_status['status'] == 'ACTIVE'
+
+              unless image_status['progress'].nil?
+                env[:ui].clear_line
+                env[:ui].report_progress(image_status['progress'], 100, false)
+              end
+
+              sleep @retry_interval
+            end
+          end
+
+          # Clear progress output.
+          env[:ui].clear_line
+
+          env[:ui].success(I18n.t('vagrant.actions.vm.snapshot.saved',
+                                  name: env[:snapshot_name]))
+
+          @app.call env
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openstack-illuin-provider/action/start_server.rb

@@ -0,0 +1,24 @@
+require 'log4r'
+
+require 'vagrant-openstack-illuin-provider/action/abstract_action'
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      class StartServer < AbstractAction
+        def initialize(app, _env)
+          @app    = app
+          @logger = Log4r::Logger.new('vagrant_openstack::action::start_server')
+        end
+
+        def execute(env)
+          if env[:machine].id
+            env[:ui].info(I18n.t('vagrant_openstack.starting_server'))
+            env[:openstack_client].nova.start_server(env, env[:machine].id)
+          end
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openstack-illuin-provider/action/stop_server.rb

@@ -0,0 +1,25 @@
+require 'log4r'
+
+require 'vagrant-openstack-illuin-provider/action/abstract_action'
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      class StopServer < AbstractAction
+        def initialize(app, _env)
+          @app    = app
+          @logger = Log4r::Logger.new('vagrant_openstack::action::stop_server')
+        end
+
+        def execute(env)
+          if env[:machine].id
+            @logger.info "Stopping server #{env[:machine].id}..."
+            env[:ui].info(I18n.t('vagrant_openstack.stopping_server'))
+            env[:openstack_client].nova.stop_server(env, env[:machine].id)
+          end
+          @app.call(env)
+        end
+      end
+    end
+  end
+end