vagrant-simple_cloud 1.0.1 → 1.0.2

data/lib/vagrant-simple_cloud/actions/power_on.rb

@@ -0,0 +1,48 @@
+require 'vagrant-simple_cloud/helpers/client'
+
+module VagrantPlugins
+  module SimpleCloud
+    module Actions
+      class PowerOn
+        include Helpers::Client
+
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @client = client
+          @logger = Log4r::Logger.new('vagrant::simple_cloud::power_on')
+        end
+
+        def call(env)
+          # submit power on droplet request
+          result = @client.post("/v2/vps/#{@machine.id}/actions", {
+            :type => 'power_on'
+          })
+
+          # wait for request to complete
+          env[:ui].info I18n.t('vagrant_simple_cloud.info.powering_on') 
+          @client.wait_for_event(env, result['action']['id'])
+
+          # refresh droplet state with provider
+          Provider.droplet(@machine, :refresh => true)
+
+          # wait for ssh to be ready
+          switch_user = @machine.provider_config.setup?
+          user = @machine.config.ssh.username
+          @machine.config.ssh.username = 'root' if switch_user
+
+          retryable(:tries => 120, :sleep => 10) do
+            next if env[:interrupted]
+            raise 'not ready' if !@machine.communicate.ready?
+          end
+
+          @machine.config.ssh.username = user
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end
+
+

data/lib/vagrant-simple_cloud/actions/rebuild.rb

@@ -0,0 +1,49 @@
+require 'vagrant-simple_cloud/helpers/client'
+
+module VagrantPlugins
+  module SimpleCloud
+    module Actions
+      class Rebuild
+        include Helpers::Client
+        include Vagrant::Util::Retryable
+
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @client = client
+          @logger = Log4r::Logger.new('vagrant::simple_cloud::rebuild')
+        end
+
+        def call(env)
+
+          # submit rebuild request
+          result = @client.post("/v2/vps/#{@machine.id}/actions", {
+            :type => 'rebuild',
+            :image => @machine.provider_config.image
+          })
+
+          # wait for request to complete
+          env[:ui].info I18n.t('vagrant_simple_cloud.info.rebuilding')
+          @client.wait_for_event(env, result['action']['id'])
+
+          # refresh droplet state with provider
+          Provider.droplet(@machine, :refresh => true)
+
+          # wait for ssh to be ready
+          switch_user = @machine.provider_config.setup?
+          user = @machine.config.ssh.username
+          @machine.config.ssh.username = 'root' if switch_user
+
+          retryable(:tries => 120, :sleep => 10) do
+            next if env[:interrupted]
+            raise 'not ready' if !@machine.communicate.ready?
+          end
+
+          @machine.config.ssh.username = user
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-simple_cloud/actions/reload.rb

@@ -0,0 +1,45 @@
+require 'vagrant-simple_cloud/helpers/client'
+
+module VagrantPlugins
+  module SimpleCloud
+    module Actions
+      class Reload
+        include Helpers::Client
+
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @client = client
+          @logger = Log4r::Logger.new('vagrant::simple_cloud::reload')
+        end
+
+        def call(env)
+          # submit reboot droplet request
+          result = @client.post("/v2/vps/#{@machine.id}/actions", {
+            :type => 'reboot'
+          })
+
+          # wait for request to complete
+          env[:ui].info I18n.t('vagrant_simple_cloud.info.reloading')
+          @client.wait_for_event(env, result['action']['id'])
+
+          # wait for ssh to be ready
+          switch_user = @machine.provider_config.setup?
+          user = @machine.config.ssh.username
+          @machine.config.ssh.username = 'root' if switch_user
+
+          retryable(:tries => 120, :sleep => 10) do
+            next if env[:interrupted]
+            raise 'not ready' if !@machine.communicate.ready?
+          end
+
+          @machine.config.ssh.username = user
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end
+
+

data/lib/vagrant-simple_cloud/actions/setup_key.rb

@@ -0,0 +1,58 @@
+require 'vagrant-simple_cloud/helpers/client'
+
+module VagrantPlugins
+  module SimpleCloud
+    module Actions
+      class SetupKey
+        include Helpers::Client
+
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @client = client
+          @logger = Log4r::Logger.new('vagrant::simple_cloud::setup_key')
+        end
+
+        # TODO check the content of the key to see if it has changed
+        def call(env)
+          ssh_key_name = @machine.provider_config.ssh_key_name
+
+          begin
+            # assigns existing ssh key id to env for use by other commands
+            env[:ssh_key_id] = @client
+              .request('/v2/account/keys')
+              .find_id(:ssh_keys, :name => ssh_key_name)
+
+            env[:ui].info I18n.t('vagrant_simple_cloud.info.using_key', {
+              :name => ssh_key_name
+            })
+          rescue Errors::ResultMatchError
+            env[:ssh_key_id] = create_ssh_key(ssh_key_name, env)
+          end
+
+          @app.call(env)
+        end
+
+        private
+
+        def create_ssh_key(name, env)
+          # assumes public key exists on the same path as private key with .pub ext
+          path = @machine.config.ssh.private_key_path
+          path = path[0] if path.is_a?(Array)
+          path = File.expand_path(path, @machine.env.root_path)
+          pub_key = SimpleCloud.public_key(path)
+
+          env[:ui].info I18n.t('vagrant_simple_cloud.info.creating_key', {
+            :name => name
+          }) 
+
+          result = @client.post('/v2/account/keys', {
+            :name => name,
+            :public_key => pub_key
+          })
+          result['ssh_key']['id']
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-simple_cloud/actions/setup_sudo.rb

@@ -0,0 +1,48 @@
+module VagrantPlugins
+  module SimpleCloud
+    module Actions
+      class SetupSudo
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @logger = Log4r::Logger.new('vagrant::simple_cloud::setup_sudo')
+        end
+
+        def call(env)
+          # check if setup is enabled
+          return @app.call(env) unless @machine.provider_config.setup?
+
+          # override ssh username to root
+          user = @machine.config.ssh.username
+          @machine.config.ssh.username = 'root'
+
+          # check for guest name available in Vagrant 1.2 first
+          guest_name = @machine.guest.name if @machine.guest.respond_to?(:name)
+          guest_name ||= @machine.guest.to_s.downcase
+
+          case guest_name
+          when /debian/
+            if @machine.provider_config.image =~ /^debian-8/
+              env[:ui].info I18n.t('vagrant_simple_cloud.info.late_sudo_install_deb8')
+              @machine.communicate.execute(<<-'BASH')
+                if [ ! -x /usr/bin/sudo ] ; then apt-get update -y && apt-get install -y sudo ; fi
+              BASH
+            end
+          when /redhat/
+            env[:ui].info I18n.t('vagrant_simple_cloud.info.modifying_sudo')
+
+            # disable tty requirement for sudo
+            @machine.communicate.execute(<<-'BASH')
+              sed -i'.bk' -e 's/\(Defaults\s\+requiretty\)/# \1/' /etc/sudoers
+            BASH
+          end
+
+          # reset ssh username
+          @machine.config.ssh.username = user
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-simple_cloud/actions/setup_user.rb

@@ -0,0 +1,66 @@
+module VagrantPlugins
+  module SimpleCloud
+    module Actions
+      class SetupUser
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @logger = Log4r::Logger.new('vagrant::simple_cloud::setup_user')
+        end
+
+        def call(env)
+          # check if setup is enabled
+          return @app.call(env) unless @machine.provider_config.setup?
+
+          # check if a username has been specified
+          return @app.call(env) unless @machine.config.ssh.username
+
+          # override ssh username to root temporarily
+          user = @machine.config.ssh.username
+          @machine.config.ssh.username = 'root'
+
+          env[:ui].info I18n.t('vagrant_simple_cloud.info.creating_user', {
+            :user => user
+          })
+
+          # create user account
+          @machine.communicate.execute(<<-BASH)
+            if ! (grep ^#{user}: /etc/passwd); then
+              useradd -m -s /bin/bash #{user};
+            fi
+          BASH
+
+          # grant user sudo access with no password requirement
+          @machine.communicate.execute(<<-BASH)
+            if ! (grep #{user} /etc/sudoers); then
+              echo "#{user} ALL=(ALL:ALL) NOPASSWD: ALL" >> /etc/sudoers;
+            else
+              sed -i -e "/#{user}/ s/=.*/=(ALL:ALL) NOPASSWD: ALL/" /etc/sudoers;
+            fi
+          BASH
+
+          # create the .ssh directory in the users home
+          @machine.communicate.execute("su #{user} -c 'mkdir -p ~/.ssh'")
+
+          # add the specified key to the authorized keys file
+          path = @machine.config.ssh.private_key_path
+          path = path[0] if path.is_a?(Array)
+          path = File.expand_path(path, @machine.env.root_path)
+          pub_key = SimpleCloud.public_key(path)
+          @machine.communicate.execute(<<-BASH)
+            if ! grep '#{pub_key}' /home/#{user}/.ssh/authorized_keys; then
+              echo '#{pub_key}' >> /home/#{user}/.ssh/authorized_keys;
+            fi
+
+            chown -R #{user} /home/#{user}/.ssh;
+          BASH
+
+          # reset username
+          @machine.config.ssh.username = user
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-simple_cloud/actions/shut_down.rb

@@ -0,0 +1,35 @@
+require 'vagrant-simple_cloud/helpers/client'
+
+module VagrantPlugins
+  module SimpleCloud
+    module Actions
+      class ShutDown
+        include Helpers::Client
+
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @client = client
+          @logger = Log4r::Logger.new('vagrant::simple_cloud::shut_down')
+        end
+
+        def call(env)
+          # submit shutdown droplet request
+          result = @client.post("/v2/vps/#{@machine.id}/actions", {
+            :type => 'shutdown'
+          })
+
+          # wait for request to complete
+          env[:ui].info I18n.t('vagrant_simple_cloud.info.shutting_down')
+          @client.wait_for_event(env, result['action']['id'])
+
+          # refresh droplet state with provider
+          Provider.droplet(@machine, :refresh => true)
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end
+

data/lib/vagrant-simple_cloud/actions.rb

@@ -0,0 +1,164 @@
+require 'vagrant-simple_cloud/actions/check_state'
+require 'vagrant-simple_cloud/actions/create'
+require 'vagrant-simple_cloud/actions/destroy'
+require 'vagrant-simple_cloud/actions/shut_down'
+require 'vagrant-simple_cloud/actions/power_off'
+require 'vagrant-simple_cloud/actions/power_on'
+require 'vagrant-simple_cloud/actions/rebuild'
+require 'vagrant-simple_cloud/actions/reload'
+require 'vagrant-simple_cloud/actions/setup_user'
+require 'vagrant-simple_cloud/actions/setup_sudo'
+require 'vagrant-simple_cloud/actions/setup_key'
+require 'vagrant-simple_cloud/actions/modify_provision_path'
+
+module VagrantPlugins
+  module SimpleCloud
+    module Actions
+      include Vagrant::Action::Builtin
+
+      def self.destroy
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :not_created
+              env[:ui].info I18n.t('vagrant_simple_cloud.info.not_created')
+            else
+              b.use Call, DestroyConfirm do |env2, b2|
+                if env2[:result]
+                  b2.use Destroy
+                  b2.use ProvisionerCleanup if defined?(ProvisionerCleanup)
+                end
+              end
+            end
+          end
+        end
+      end
+
+      def self.ssh
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :active
+              b.use SSHExec
+            when :off
+              env[:ui].info I18n.t('vagrant_simple_cloud.info.off')
+            when :not_created
+              env[:ui].info I18n.t('vagrant_simple_cloud.info.not_created')
+            end
+          end
+        end
+      end
+
+      def self.ssh_run
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :active
+              b.use SSHRun
+            when :off
+              env[:ui].info I18n.t('vagrant_simple_cloud.info.off')
+            when :not_created
+              env[:ui].info I18n.t('vagrant_simple_cloud.info.not_created')
+            end
+          end
+        end
+      end
+
+      def self.provision
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :active
+              b.use Provision
+              b.use ModifyProvisionPath
+              b.use SyncedFolders
+            when :off
+              env[:ui].info I18n.t('vagrant_simple_cloud.info.off')
+            when :not_created
+              env[:ui].info I18n.t('vagrant_simple_cloud.info.not_created')
+            end
+          end
+        end
+      end
+
+      def self.up
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :active
+              env[:ui].info I18n.t('vagrant_simple_cloud.info.already_active')
+            when :off
+              b.use PowerOn
+              b.use provision
+            when :not_created
+              b.use SetupKey
+              b.use Create
+              b.use SetupSudo
+              b.use SetupUser
+              b.use provision
+            end
+          end
+        end
+      end
+
+      def self.halt
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :active
+              if env[:force_halt] 
+                b.use PowerOff
+              else
+                b.use ShutDown
+              end
+            when :off
+              env[:ui].info I18n.t('vagrant_simple_cloud.info.already_off')
+            when :not_created
+              env[:ui].info I18n.t('vagrant_simple_cloud.info.not_created')
+            end
+          end
+        end
+      end
+
+      def self.reload
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :active
+              b.use Reload
+              b.use provision
+            when :off
+              env[:ui].info I18n.t('vagrant_simple_cloud.info.off')
+            when :not_created
+              env[:ui].info I18n.t('vagrant_simple_cloud.info.not_created')
+            end
+          end
+        end
+      end
+
+      def self.rebuild
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :active, :off
+              b.use Rebuild
+              b.use SetupSudo
+              b.use SetupUser
+              b.use provision
+            when :not_created
+              env[:ui].info I18n.t('vagrant_simple_cloud.info.not_created')
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-simple_cloud/commands/list.rb

@@ -0,0 +1,93 @@
+require 'optparse'
+require 'vagrant-simple_cloud/helpers/client'
+
+module VagrantPlugins
+  module SimpleCloud
+    module Commands
+      class List < Vagrant.plugin('2', :command)
+        def self.synopsis
+          "list available images and regions from SimpleCloud"
+        end
+
+        def execute
+          @token = nil
+          
+          @opts = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant simplecloud-list [options] <images|regions|sizes> <token>'
+
+            o.on("-r", "--[no-]regions", "show the regions when listing images") do |r|
+              @regions = r
+            end
+            o.on("-h", "--help", "Displays help") do
+              puts o
+              return 0
+            end
+          end
+          
+          argv = parse_options(@opts)
+          @token = argv[1]
+          
+          if @token.nil?
+            usage
+            return 1
+          end
+          
+          case argv[0]
+          when "images"
+            result = query('/v2/images')
+            images = Array(result["images"])
+            if @regions
+              images_table = images.map do |image|
+                '%-50s %-20s %-20s %-50s' % ["#{image['distribution']} #{image['name']}", image['slug'], image['id'], image['regions'].join(', ')]
+              end
+              @env.ui.info I18n.t('vagrant_simple_cloud.info.images_with_regions', images: images_table.sort.join("\r\n"))
+            else
+              images_table = images.map do |image|
+                '%-50s %-30s %-30s' % ["#{image['distribution']} #{image['name']}", image['slug'], image['id']]
+              end
+              @env.ui.info I18n.t('vagrant_simple_cloud.info.images', images: images_table.sort.join("\r\n"))
+            end
+          when "regions"
+            result = query('/v2/regions')
+            regions = Array(result["regions"])
+            regions_table = regions.map { |region| '%-30s %-12s' % [region['name'], region['slug']] }
+            @env.ui.info I18n.t('vagrant_simple_cloud.info.regions', regions: regions_table.sort.join("\r\n"))
+          when "sizes"
+            result = query('/v2/sizes')
+            sizes = Array(result["sizes"])
+            sizes_table = sizes.map { |size| '%-15s %-15s %-12s' % ["#{size['memory']}", size['vcpus'], size['slug']] }
+            @env.ui.info I18n.t('vagrant_simple_cloud.info.sizes', sizes: sizes_table.sort_by{|s| s['memory']}.join("\r\n"))
+          else
+            usage
+            return 1
+          end
+
+          0
+        rescue Faraday::Error::ConnectionFailed, RuntimeError => e
+          @env.ui.error I18n.t('vagrant_simple_cloud.info.list_error', message: e.message)
+          1
+        end
+
+        def query(path)
+          connection = Faraday.new({
+            :url => "https://api.simplecloud.ru/"
+          })
+
+          result = connection.get(path, per_page: 100) do |req|
+            req.headers['Authorization'] = "Bearer #{@token}"
+          end
+
+          case result.status
+          when 200 then JSON.parse(result.body)
+          when 401 then raise("unauthorized access — is the token correct?")
+          else raise("call returned with status #{result.status}")
+          end
+        end
+        
+        def usage
+          @env.ui.info(@opts)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-simple_cloud/commands/rebuild.rb

@@ -0,0 +1,29 @@
+require 'optparse'
+
+module VagrantPlugins
+  module SimpleCloud
+    module Commands
+      class Rebuild < Vagrant.plugin('2', :command)
+
+        # Show description when `vagrant list-commands` is triggered
+        def self.synopsis
+          "plugin: vagrant-simple_cloud: destroys and ups the vm with the same ip address"
+        end
+
+        def execute
+          opts = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant rebuild [vm-name]'
+          end
+
+          argv = parse_options(opts)
+
+          with_target_vms(argv) do |machine|
+            machine.action(:rebuild)
+          end
+
+          0
+        end
+      end
+    end
+  end
+end