vagrant-vmpooler 0.1.1 → 0.1.2

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

@@ -0,0 +1,103 @@
+require "vmfloaty/pooler"
+require "log4r"
+
+require 'vagrant/util/retryable'
+
+module VagrantPlugins
+  module Vmpooler
+    module Action
+      class CreateServer
+        include Vagrant::Util::Retryable
+
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_vmpooler::action::create_server")
+        end
+
+        def call(env)
+          # Get the configs
+          provider_config = env[:machine].provider_config
+
+          token = provider_config.token
+          url = provider_config.url
+          os = provider_config.os
+          ttl = provider_config.ttl
+          disk = provider_config.disk
+          verbose = provider_config.verbose
+
+          if !url
+            raise Errors::NoURLError,
+              :message => "There was no provided url to connect to vmpooler with"
+          end
+
+          if !token
+            env[:ui].warn("There was no token provided. All vms will only last 2 hours.")
+          end
+
+          if !os
+            raise Errors::NoURLError,
+              :message => "There was no operatingystem provided"
+          end
+
+          # Output the settings we're going to use to the user
+          env[:ui].info(I18n.t("vagrant_vmpooler.launching_server"))
+          env[:ui].info(" -- Vmpooler URL: #{url}")
+          env[:ui].info(" -- Vmpooler Verbose Mode: #{verbose}")
+          env[:ui].info(" -- Image: #{os}")
+          env[:ui].info(" -- Additional TTL: #{ttl}") if ttl
+          env[:ui].info(" -- Additional Disk: #{disk}") if disk
+
+          # Create the server
+          os_arr = {}
+          os_arr[os] = 1
+          server = Pooler.retrieve(verbose, os_arr, token, url)
+
+          # Store the ID right away so we can track it
+          # in this case it's the hostname
+          if server['ok'] == false
+            raise Errors::GetVMError,
+              :message => "Could not retrieve vm from pooler:\n #{server}"
+          end
+
+          server_name = server[os]["hostname"]
+          env[:machine].id = server_name
+
+          if ttl
+            response_body = Pooler.modify(verbose, url, server_name, token, ttl, nil)
+            if response_body['ok'] == false
+              env[:ui].warn(I18n.t("vagrant_vmpooler.errors.failed_ttl"))
+            end
+          end
+
+          if disk
+            response_body = Pooler.disk(verbose, url, server_name, token, disk)
+            if response_body['ok'] == false
+              env[:ui].warn(I18n.t("vagrant_vmpooler.errors.failed_disk_extend"))
+            end
+          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
+            env[:ui].info(I18n.t("vagrant_vmpooler.waiting_for_ssh"))
+            while true
+              begin
+                # If we're interrupted then just back out
+                break if env[:interrupted]
+                break if env[:machine].communicate.ready?
+              rescue Errno::ENETUNREACH, Errno::EHOSTUNREACH
+              end
+              sleep 2
+            end
+
+            env[:ui].info(I18n.t("vagrant_vmpooler.ready"))
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,50 @@
+require 'vagrant/util/retryable'
+require 'timeout'
+require 'vmfloaty/pooler'
+
+require "log4r"
+
+module VagrantPlugins
+  module Vmpooler
+    module Action
+      # This deletes the running server, if there is one.
+      class DeleteServer
+        include Vagrant::Util::Retryable
+
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_vmpooler::action::delete_server")
+        end
+
+        def call(env)
+          machine = env[:machine]
+          id = machine.id
+
+          if id
+            provider_config = machine.provider_config
+            token = provider_config.token
+            url = provider_config.url
+            verbose = provider_config.verbose
+
+            env[:ui].info(I18n.t("vagrant_vmpooler.deleting_server"))
+
+            os = []
+            os.push(id)
+            response_body = Pooler.delete(verbose, url, os, token)
+
+            if response_body[id]['ok'] == false
+              env[:ui].info(I18n.t("vagrant_vmpooler.not_deleted"))
+            else
+              env[:ui].info(I18n.t("vagrant_vmpooler.deleted"))
+              env[:machine].id = nil
+            end
+          else
+            env[:ui].info(I18n.t("vagrant_vmpooler.not_created"))
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmpooler/action/hard_reboot_server.rb

@@ -0,0 +1,24 @@
+require "log4r"
+
+module VagrantPlugins
+  module Vmpooler
+    module Action
+      # This hard reboots a running server, if there is one.
+      class HardRebootServer
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_vmpooler::action::hard_reboot_server")
+        end
+
+        def call(env)
+          if env[:machine].id
+            env[:ui].info(I18n.t("vagrant_vmpooler.hard_rebooting_server"))
+
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

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

data/lib/vagrant-vmpooler/action/is_paused.rb

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

data/lib/vagrant-vmpooler/action/is_suspended.rb

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

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

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

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

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

data/lib/vagrant-vmpooler/action/message_server_running.rb

@@ -0,0 +1,16 @@
+module VagrantPlugins
+  module Vmpooler
+    module Action
+      class MessageServerRunning
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t("vagrant_vmpooler.server_running", name: env[:machine].id))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmpooler/action/message_will_not_destroy.rb

@@ -0,0 +1,16 @@
+module VagrantPlugins
+  module Vmpooler
+    module Action
+      class MessageWillNotDestroy
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t("vagrant_vmpooler.will_not_destroy", name: env[:machine].name))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmpooler/action/pause_server.rb

@@ -0,0 +1,24 @@
+require "log4r"
+
+module VagrantPlugins
+  module Vmpooler
+    module Action
+      # This pauses a running server, if there is one.
+      class PauseServer
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_vmpooler::action::pause_server")
+        end
+
+        def call(env)
+          if env[:machine].id
+            env[:ui].info(I18n.t("vagrant_vmpooler.pausing_server"))
+
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,56 @@
+require "log4r"
+require "ipaddr"
+
+module VagrantPlugins
+  module Vmpooler
+    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_vmpooler::action::read_ssh_info")
+        end
+
+        def call(env)
+          env[:machine_ssh_info] = read_ssh_info(env[:machine])
+
+          @app.call(env)
+        end
+
+        def read_ssh_info(machine)
+          id = machine.id
+          return nil if id.nil?
+
+          provider_config = machine.provider_config
+          verbose = provider_config.verbose
+          url = provider_config.url
+          password = provider_config.password
+          os = provider_config.os
+
+          server = Pooler.query(verbose, url, id)
+          if server['ok'] == false
+            # The machine can't be found
+            @logger.info("Machine couldn't be found, assuming it got destroyed.")
+            machine.id = nil
+            return nil
+          else
+            if os =~ /win/i
+              username = 'Administrator'
+            else
+              username = 'root'
+            end
+
+            ssh_info = {
+              :host => id,
+              :username => username,
+              :password => password
+            }
+            ssh_info.merge! server[id]
+            return ssh_info
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,41 @@
+require "log4r"
+require 'vmfloaty/pooler'
+
+module VagrantPlugins
+  module Vmpooler
+    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_vmpooler::action::read_state")
+        end
+
+        def call(env)
+          env[:machine_state_id] = read_state(env[:machine])
+          @app.call(env)
+        end
+
+        def read_state(machine)
+          id = machine.id
+          return :not_created if id.nil?
+
+          provider_config = machine.provider_config
+          verbose = provider_config.verbose
+          url = provider_config.url
+
+          server = Pooler.query(verbose, url, id)
+          if server['ok'] == false
+            @logger.info(I18n.t("vagrant_vmpooler.not_created"))
+            machine.id = nil
+            return :not_created
+          end
+
+          return :created
+        end
+
+      end
+    end
+  end
+end

data/lib/vagrant-vmpooler/action/reboot_server.rb

@@ -0,0 +1,24 @@
+require "log4r"
+
+module VagrantPlugins
+  module Vmpooler
+    module Action
+      # This reboots a running server, if there is one.
+      class RebootServer
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_vmpooler::action::reboot_server")
+        end
+
+        def call(env)
+          if env[:machine].id
+            env[:ui].info(I18n.t("vagrant_vmpooler.rebooting_server"))
+
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end