vagrant-openstack-provider 0.1

data/lib/vagrant-openstack/action.rb

@@ -0,0 +1,123 @@
+require "pathname"
+
+require "vagrant/action/builder"
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      # Include the built-in modules so we can use them as top-level things.
+      include Vagrant::Action::Builtin
+
+      # This action is called to destroy the remote machine.
+      def self.action_destroy
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use Call, IsCreated do |env, b2|
+            if !env[:result]
+              b2.use MessageNotCreated
+              next
+            end
+
+            b2.use ConnectOpenstack
+            b2.use DeleteServer
+          end
+        end
+      end
+
+      # This action is called when `vagrant provision` is called.
+      def self.action_provision
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use Call, IsCreated do |env, b2|
+            if !env[:result]
+              b2.use MessageNotCreated
+              next
+            end
+
+            b2.use Provision
+            b2.use SyncFolders
+          end
+        end
+      end
+
+      # This action is called to read the SSH info of the machine. The
+      # resulting state is expected to be put into the `:machine_ssh_info`
+      # key.
+      def self.action_read_ssh_info
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use ConnectOpenstack
+          b.use ReadSSHInfo
+        end
+      end
+
+      # This action is called to read the state of the machine. The
+      # resulting state is expected to be put into the `:machine_state_id`
+      # key.
+      def self.action_read_state
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use ConnectOpenstack
+          b.use ReadState
+        end
+      end
+
+      def self.action_ssh
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use Call, IsCreated do |env, b2|
+            if !env[:result]
+              b2.use MessageNotCreated
+              next
+            end
+
+            b2.use SSHExec
+          end
+        end
+      end
+
+      def self.action_ssh_run
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use Call, IsCreated do |env, b2|
+            if !env[:result]
+              b2.use MessageNotCreated
+              next
+            end
+
+            b2.use SSHRun
+          end
+        end
+      end
+
+      def self.action_up
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use Call, IsCreated do |env, b2|
+            if env[:result]
+              b2.use MessageAlreadyCreated
+              next
+            end
+
+            b2.use ConnectOpenstack
+            b2.use Provision
+            b2.use SyncFolders
+            b2.use CreateServer
+          end
+        end
+      end
+
+      # The autoload farm
+      action_root = Pathname.new(File.expand_path("../action", __FILE__))
+      autoload :ConnectOpenstack, action_root.join("connect_openstack")
+      autoload :CreateServer, action_root.join("create_server")
+      autoload :DeleteServer, action_root.join("delete_server")
+      autoload :IsCreated, action_root.join("is_created")
+      autoload :MessageAlreadyCreated, action_root.join("message_already_created")
+      autoload :MessageNotCreated, action_root.join("message_not_created")
+      autoload :ReadSSHInfo, action_root.join("read_ssh_info")
+      autoload :ReadState, action_root.join("read_state")
+      autoload :SyncFolders, action_root.join("sync_folders")
+    end
+  end
+end

data/lib/vagrant-openstack/action/connect_openstack.rb

@@ -0,0 +1,30 @@
+require "fog"
+require "log4r"
+require "restclient"
+require "json"
+require_relative "../openstack_client"
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      # This action connects to Openstack, verifies credentials work, and
+      # puts the Openstack connection object into the `:openstack_compute` key
+      # in the environment.
+      class ConnectOpenstack
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new("vagrant_openstack::action::connect_openstack")
+        end
+
+        def call(env)
+          # Get the configs
+          config = env[:machine].provider_config
+          client = OpenstackClient::new()
+          env[:openstack_client] = client
+          client.authenticate(env)
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openstack/action/create_server.rb

@@ -0,0 +1,134 @@
+require "fog"
+require "log4r"
+require 'socket'
+require "timeout"
+
+require 'vagrant/util/retryable'
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      class CreateServer
+        include Vagrant::Util::Retryable
+
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new("vagrant_openstack::action::create_server")
+        end
+
+        def call(env)
+          config = env[:machine].provider_config
+          client = env[:openstack_client]
+
+          # Find the flavor
+          env[:ui].info(I18n.t("vagrant_openstack.finding_flavor"))
+          flavors = client.get_all_flavors(env)
+          flavor = find_matching(flavors, config.flavor)
+          raise Errors::NoMatchingFlavor if !flavor
+
+          # Find the image
+          env[:ui].info(I18n.t("vagrant_openstack.finding_image"))
+          images = client.get_all_images(env)
+          image = find_matching(images, config.image)
+          raise Errors::NoMatchingImage if !image
+
+          # Figure out the name for the server
+          server_name = config.server_name || env[:machine].name
+
+          # Output the settings we're going to use to the user
+          env[:ui].info(I18n.t("vagrant_openstack.launching_server"))
+          env[:ui].info(" -- Flavor       : #{flavor.name}")
+          env[:ui].info(" -- FlavorRef    : #{flavor.id}")
+          env[:ui].info(" -- Image        : #{image.name}")
+          env[:ui].info(" -- KeyPair      : #{config.keypair_name}")
+          env[:ui].info(" -- ImageRef     : #{image.id}")
+          env[:ui].info(" -- Disk Config  : #{config.disk_config}") if config.disk_config
+          env[:ui].info(" -- Network      : #{config.network}") if config.network
+          env[:ui].info(" -- Tenant       : #{config.tenant_name}")
+          env[:ui].info(" -- Name         : #{server_name}")
+
+          #TODO(julienvey) add metadata
+          #TODO(julienvey) add availability_zone
+          #TODO(julienvey) add disk_config
+
+          server_id = client.create_server(env, server_name, image.id, flavor.id, config.keypair_name)
+
+          #TODO(julienvey) Find a network if provided
+          #if config.network
+          #  network = find_matching(env[:openstack_network].networks, config.network)
+          #  options[:nics] = [{"net_id" => network.id}] if network
+          #end
+
+          # Store the ID right away so we can track it
+          env[:machine].id = server_id
+
+          # Wait for the server to finish building
+          env[:ui].info(I18n.t("vagrant_openstack.waiting_for_build"))
+          timeout(200) do
+            while client.get_server_details(env, server_id)['status'] != 'ACTIVE' do
+              sleep 3
+              @logger.debug("Waiting for server to be ACTIVE")
+            end
+          end
+
+          if config.floating_ip
+            env[:ui].info("Using floating IP #{config.floating_ip}")
+            client.add_floating_ip(env, server_id, config.floating_ip)
+          end
+
+          if !env[:interrupted]
+            # Clear the line one more time so the progress is removed
+            env[:ui].clear_line
+
+            # Wait for SSH to become available
+            host = env[:machine].provider_config.floating_ip
+            ssh_timeout = env[:machine].provider_config.ssh_timeout
+            sleep 240
+            if !port_open?(env, host, 22, ssh_timeout)
+              env[:ui].error(I18n.t("vagrant_openstack.timeout"))
+              raise Errors::SshUnavailable,
+                    :host    => host,
+                    :timeout => ssh_timeout
+            end
+
+            env[:ui].info(I18n.t("vagrant_openstack.ready"))
+          end
+
+          @app.call(env)
+        end
+
+        protected
+
+        def port_open?(env, ip, port, timeout)
+          start_time = Time.now
+          current_time = start_time
+          while (current_time - start_time) <= timeout
+            begin
+              env[:ui].info(I18n.t("vagrant_openstack.waiting_for_ssh"))
+              TCPSocket.new(ip, port)
+              return true
+            rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH
+              sleep 1
+            end
+            current_time = Time.now
+          end
+          return false
+        end
+
+        # This method finds a matching _thing_ in a collection of
+        # _things_. This works matching if the ID or NAME equals to
+        # `name`. Or, if `name` is a regexp, a partial match is chosen
+        # as well.
+        def find_matching(collection, name)
+          collection.each do |single|
+            return single if single.id == name
+            return single if single.name == name
+            return single if name.is_a?(Regexp) && name =~ single.name
+          end
+
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openstack/action/delete_server.rb

@@ -0,0 +1,26 @@
+require "log4r"
+
+module VagrantPlugins
+  module Openstack
+    module Action
+      # This deletes the running server, if there is one.
+      class DeleteServer
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_openstack::action::delete_server")
+        end
+
+        def call(env)
+          if env[:machine].id
+            env[:ui].info(I18n.t("vagrant_openstack.deleting_server"))
+            server = env[:openstack_compute].servers.get(env[:machine].id)
+            server.destroy
+            env[:machine].id = nil
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openstack/action/is_created.rb

@@ -0,0 +1,16 @@
+module VagrantPlugins
+  module Openstack
+    module Action
+      class IsCreated
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:result] = env[:machine].state.id != :not_created
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openstack/action/message_already_created.rb

@@ -0,0 +1,16 @@
+module VagrantPlugins
+  module Openstack
+    module Action
+      class MessageAlreadyCreated
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t("vagrant_openstack.already_created"))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openstack/action/message_not_created.rb

@@ -0,0 +1,16 @@
+module VagrantPlugins
+  module Openstack
+    module Action
+      class MessageNotCreated
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t("vagrant_openstack.not_created"))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,52 @@
+require "log4r"
+
+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
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_openstack::action::read_ssh_info")
+        end
+
+        def call(env)
+          env[:machine_ssh_info] = read_ssh_info(env)
+
+          @app.call(env)
+        end
+
+        def read_ssh_info(env)
+          client = env[:openstack_client]
+          machine = env[:machine]
+          config = env[:machine].provider_config
+          @logger.debug(config)
+          return nil if machine.id.nil?
+          begin
+            details = client.get_server_details(env, machine.id)
+          rescue Exception => e
+            # The machine can't be found
+            @logger.error(e)
+            @logger.info("Machine couldn't be found, assuming it got destroyed.")
+            machine.id = nil
+            return nil
+          end
+
+          for addr in details['addresses']['private']
+            if addr['OS-EXT-IPS:type'] == 'floating'
+              host = addr['addr']
+            end
+          end
+
+          return {
+            # Usually there should only be one public IP
+            :host => host,
+            :port => 22,
+            :username => config.ssh_username
+          }
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,40 @@
+require "log4r"
+
+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
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_openstack::action::read_state")
+        end
+
+        def call(env)
+          env[:machine_state_id] = read_state(env)
+
+          @app.call(env)
+        end
+
+        def read_state(env)
+          machine = env[:machine]
+          client = env[:openstack_client]
+          return :not_created if machine.id.nil?
+
+          # Find the machine
+          server = client.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
+
+          # Return the state
+          return server['status'].downcase.to_sym
+        end
+      end
+    end
+  end
+end