vagrant-niftycloud 0.1.0.dev

data/lib/vagrant-niftycloud/action/connect_niftycloud.rb

@@ -0,0 +1,56 @@
+# -*- coding: utf-8 -*-
+require "NIFTY"
+require "log4r"
+NIFTY::LOG.level = Logger::DEBUG
+
+module VagrantPlugins
+  module NiftyCloud
+    module Action
+      # This action connects to NiftyCloud, verifies credentials work, and
+      # puts the NiftyCloud connection object into the `:niftycloud_compute` key
+      # in the environment.
+      class ConnectNiftyCloud
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_niftycloud::action::connect_niftycloud")
+        end
+
+        def call(env)
+          # Get the zone we're going to booting up in
+          zone = env[:machine].provider_config.zone
+
+          # Get the configs
+          zone_config     = env[:machine].provider_config.get_zone_config(zone)
+
+          # Build the fog config
+          niftycloud_config = {
+            :access_key => zone_config.access_key_id,
+            :secret_key => zone_config.secret_access_key
+          }
+
+          # 例外の定義は以下参照
+          # http://cloud.nifty.com/api/sdk/rdoc/
+          begin
+            @logger.info("Connecting to NiftyCloud...")
+            env[:niftycloud_compute] = NIFTY::Cloud::Base.new(niftycloud_config)
+          rescue NIFTY::ConfigurationError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudConfigurationError,
+              :message => e.message
+          rescue NIFTY::ArgumentError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudArgumentError,
+              :message => e.message
+          rescue NIFTY::ResponseFormatError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudResponseFormatError,
+              :message => e.message
+          rescue NIFTY::ResponseError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudResponseError,
+              :code    => e.error_code,
+              :message => e.error_message
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,18 @@
+module VagrantPlugins
+  module NiftyCloud
+    module Action
+      # This can be used with "Call" built-in to check if the machine
+      # is created and branch in the middleware.
+      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-niftycloud/action/message_already_created.rb

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

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

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

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

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

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

@@ -0,0 +1,52 @@
+require "log4r"
+
+module VagrantPlugins
+  module NiftyCloud
+    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_niftycloud::action::read_ssh_info")
+        end
+
+        def call(env)
+          env[:machine_ssh_info] = read_ssh_info(env[:niftycloud_compute], env[:machine])
+
+          @app.call(env)
+        end
+
+        def read_ssh_info(niftycloud, machine)
+          return nil if machine.id.nil?
+
+          # Find the machine
+          # 例外の定義は以下参照
+          # http://cloud.nifty.com/api/sdk/rdoc/
+          begin
+            server = niftycloud.describe_instances(:instance_id => machine.id).reservationSet.item.first.instancesSet.item.first
+            # Read the DNS info
+            return {
+              :host => server.ipAddress,
+              :port => 22
+            }
+          rescue NIFTY::ConfigurationError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudConfigurationError,
+              :message => e.message
+          rescue NIFTY::ArgumentError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudArgumentError,
+              :message => e.message
+          rescue NIFTY::ResponseFormatError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudResponseFormatError,
+              :message => e.message
+          rescue NIFTY::ResponseError => e
+            # The machine can't be found
+            @logger.info("Machine couldn't be found, assuming it got destroyed.")
+            machine.id = nil
+            return nil
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,56 @@
+require "log4r"
+
+module VagrantPlugins
+  module NiftyCloud
+    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_niftycloud::action::read_state")
+        end
+
+        def call(env)
+          env[:machine_state_id] = read_state(env[:niftycloud_compute], env[:machine])
+
+          @app.call(env)
+        end
+
+        def read_state(niftycloud, machine)
+          return :not_created if machine.id.nil?
+
+          # Find the machine
+          # 例外の定義は以下参照
+          # http://cloud.nifty.com/api/sdk/rdoc/
+          begin
+            server = niftycloud.describe_instances(:instance_id => machine.id).reservationSet.item.first.instancesSet.item.first
+
+            state = server.instanceState.name
+            case state
+            when 'suspending'
+              @logger.info("Machine not found or terminated, assuming it got destroyed.")
+              machine.id = nil
+              return :not_created
+            else
+              @logger.debug("Server State #{state.to_sym}")
+              return state.to_sym
+            end
+          rescue NIFTY::ConfigurationError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudConfigurationError,
+              :message => e.message
+          rescue NIFTY::ArgumentError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudArgumentError,
+              :message => e.message
+          rescue NIFTY::ResponseFormatError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudResponseFormatError,
+              :message => e.message
+          rescue NIFTY::ResponseError => e
+            machine.id = nil
+            return :not_created
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-niftycloud/action/resume_instance.rb

@@ -0,0 +1,56 @@
+# -*- coding: utf-8 -*-
+require "log4r"
+
+module VagrantPlugins
+  module NiftyCloud
+    module Action
+      # This resume the running instance.
+      class ResumeInstance
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_niftycloud::action::resume_instance")
+        end
+
+        def call(env)
+
+          # 例外の定義は以下参照
+          # http://cloud.nifty.com/api/sdk/rdoc/
+          begin
+            env[:ui].info(I18n.t("vagrant_niftycloud.resuming"))
+
+            # 起動直後等、resume処理できないステータスの場合一旦待つ
+            server = env[:niftycloud_compute].describe_instances(:instance_id => env[:machine].id).reservationSet.item.first.instancesSet.item.first
+            while server.instanceState.name == 'pending'
+              sleep 5
+              server = env[:niftycloud_compute].describe_instances(:instance_id => env[:machine].id).reservationSet.item.first.instancesSet.item.first
+            end
+
+            if server.instanceState.name != 'running'
+              env[:niftycloud_compute].start_instances(:instance_id => env[:machine].id)
+              while server.instanceState.name != 'running'
+                sleep 5
+                server = env[:niftycloud_compute].describe_instances(:instance_id => env[:machine].id).reservationSet.item.first.instancesSet.item.first
+              end
+            end
+
+            @app.call(env)
+          rescue NIFTY::ConfigurationError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudConfigurationError,
+              :message => e.message
+          rescue NIFTY::ArgumentError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudArgumentError,
+              :message => e.message
+          rescue NIFTY::ResponseFormatError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudResponseFormatError,
+              :message => e.message
+          rescue NIFTY::ResponseError => e
+            ui.error("Could not locate server '#{env[:machine].id}'.  Please verify it was provisioned in the current zone.")
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudResponseError,
+              :code    => e.error_code,
+              :message => e.error_message
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-niftycloud/action/run_instance.rb

@@ -0,0 +1,131 @@
+# -*- coding: utf-8 -*-
+require "log4r"
+require 'vagrant/util/retryable'
+
+module VagrantPlugins
+  module NiftyCloud
+    module Action
+      # This runs the configured instance.
+      class RunInstance
+        include Vagrant::Util::Retryable
+
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_niftycloud::action::run_instance")
+        end
+
+        def call(env)
+          # Initialize metrics if they haven't been
+          env[:metrics] ||= {}
+
+          # Get the zone we're going to booting up in
+          zone = env[:machine].provider_config.zone
+
+          # Get the configs
+          # TODO 開発フェーズなのでinstance_idを固定にしている
+          # Vagrantfileで設定できるようにすべきか、その場合Vagrantfileを共有している環境では同じIDでサーバ立てるとエラーになることを考慮する
+          instance_id              = 'test2'
+          zone_config              = env[:machine].provider_config.get_zone_config(zone)
+          image_id                 = zone_config.image_id
+          zone                     = zone_config.zone
+          instance_type            = zone_config.instance_type
+          key_name                 = zone_config.key_name,
+          firewall                 = zone_config.firewall,
+          user_data                = zone_config.user_data
+
+          # Launch!
+          env[:ui].info(I18n.t("vagrant_niftycloud.launching_instance"))
+          env[:ui].info(" -- Server Type: #{instance_type}")
+          env[:ui].info(" -- ImageId: #{image_id}")
+          env[:ui].info(" -- Zone: #{zone}") if zone
+          env[:ui].info(" -- Key Name: #{key_name}") if key_name
+          env[:ui].info(" -- User Data: yes") if user_data
+          env[:ui].info(" -- Firewall: #{firewall.inspect}") if !firewall.empty?
+          env[:ui].info(" -- User Data: #{user_data}") if user_data
+
+          options = {
+            :instance_id              => instance_id,
+            :availability_zone        => zone,
+            :instance_type            => instance_type,
+            :image_id                 => image_id,
+            :key_name                 => zone_config.key_name,
+            :password                 => 'password',
+            :user_data                => user_data,
+            :accounting_type          => 2, #従量課金
+            :disable_api_termination  => false #APIから即terminate可
+          }
+
+          if !firewall.empty?
+            options[:security_group] = firewall
+          end
+
+          # 例外の定義は以下参照
+          # http://cloud.nifty.com/api/sdk/rdoc/
+          begin
+            # インスタンス立ち上げ開始
+            server = env[:niftycloud_compute].run_instances(options).instancesSet.item.first
+          rescue NIFTY::ConfigurationError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudConfigurationError,
+              :message => e.message
+          rescue NIFTY::ArgumentError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudArgumentError,
+              :message => e.message
+          rescue NIFTY::ResponseFormatError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudResponseFormatError,
+              :message => e.message
+          rescue NIFTY::ResponseError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudResponseError,
+              :code    => e.error_code,
+              :message => e.error_message
+          end
+            
+          # リトライ回数。サーバステータスがrunningになるまで5秒のintervalでdescribe_instancesを実行するので
+          # タイムアウト秒数/5を上限回数とする
+          tries = zone_config.instance_ready_timeout / 5
+          count = 0
+          retryable(:on => Errors::InstanceReadyTimeout, :tries => tries) do
+            env[:ui].info(I18n.t("vagrant_niftycloud.waiting_for_ready"))
+            while server.instanceState.name != 'running'
+              next if env[:interrupted]
+
+              count += 1 
+              sleep 5
+              server = env[:niftycloud_compute].describe_instances(:instance_id => instance_id).reservationSet.item.first.instancesSet.item.first
+              if count > tries
+                # Delete the instance
+                terminate(env)
+                # Notify the user
+                raise Errors::InstanceReadyTimeout, timeout: zone_config.instance_ready_timeout
+              end
+            end
+          end
+          
+          # Immediately save the ID since it is created at this point.
+          env[:machine].id = instance_id
+
+          # Terminate the instance if we were interrupted
+          terminate(env) if env[:interrupted]
+
+          @app.call(env)
+        end
+
+        def recover(env)
+          return if env["vagrant.error"].is_a?(Vagrant::Errors::VagrantError)
+
+          if env[:machine].provider.state.id != :not_created
+            # Undo the import
+            terminate(env)
+          end
+        end
+
+        def terminate(env)
+          destroy_env = env.dup
+          destroy_env.delete(:interrupted)
+          destroy_env[:config_validate] = false
+          destroy_env[:force_confirm_destroy] = true
+          env[:action_runner].run(Action.action_destroy, destroy_env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-niftycloud/action/suspend_instance.rb

@@ -0,0 +1,56 @@
+# -*- coding: utf-8 -*-
+require "log4r"
+
+module VagrantPlugins
+  module NiftyCloud
+    module Action
+      # This suspend the running instance.
+      class SuspendInstance
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_niftycloud::action::suspend_instance")
+        end
+
+        def call(env)
+
+          # 例外の定義は以下参照
+          # http://cloud.nifty.com/api/sdk/rdoc/
+          begin
+            env[:ui].info(I18n.t("vagrant_niftycloud.suspending"))
+
+            # 起動直後等、stop処理できないステータスの場合一旦待つ
+            server = env[:niftycloud_compute].describe_instances(:instance_id => env[:machine].id).reservationSet.item.first.instancesSet.item.first
+            while server.instanceState.name == 'pending'
+              sleep 5
+              server = env[:niftycloud_compute].describe_instances(:instance_id => env[:machine].id).reservationSet.item.first.instancesSet.item.first
+            end
+
+            if server.instanceState.name != 'stopped'
+              env[:niftycloud_compute].stop_instances(:instance_id => env[:machine].id, :force => false)
+              while server.instanceState.name != 'stopped'
+                sleep 5
+                server = env[:niftycloud_compute].describe_instances(:instance_id => env[:machine].id).reservationSet.item.first.instancesSet.item.first
+              end
+            end
+
+            @app.call(env)
+          rescue NIFTY::ConfigurationError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudConfigurationError,
+              :message => e.message
+          rescue NIFTY::ArgumentError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudArgumentError,
+              :message => e.message
+          rescue NIFTY::ResponseFormatError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudResponseFormatError,
+              :message => e.message
+          rescue NIFTY::ResponseError => e
+            raise VagrantPlugins::NiftyCloud::Errors::NiftyCloudResponseError,
+              :code    => e.error_code,
+              :message => e.error_message
+            ui.error("Could not locate server '#{env[:machine].id}'.  Please verify it was provisioned in the current zone.")
+          end
+        end
+      end
+    end
+  end
+end