vagrant-simplecloud 0.0.1

data/lib/vagrant-simplecloud/actions/destroy.rb

@@ -0,0 +1,32 @@
+require 'vagrant-simplecloud/helpers/client'
+
+module VagrantPlugins
+  module SimpleCloud
+    module Actions
+      class Destroy
+        include Helpers::Client
+
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @client = client
+          @simple_client = simple_client
+          @logger = Log4r::Logger.new('vagrant::simplecloud::destroy')
+        end
+
+        def call(env)
+          # submit destroy droplet request
+          @simple_client.droplets.delete(id: @machine.id)
+          #@client.delete("/v2/droplets/#{@machine.id}")
+
+          env[:ui].info I18n.t('vagrant_simple_cloud.info.destroying')
+
+          # set the machine id to nil to cleanup local vagrant state
+          @machine.id = nil
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-simplecloud/actions/modify_provision_path.rb

@@ -0,0 +1,38 @@
+module VagrantPlugins
+  module SimpleCloud
+    module Actions
+      class ModifyProvisionPath
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @logger =
+            Log4r::Logger.new('vagrant::simplecloud::modify_provision_path')
+        end
+
+        def call(env)
+          # check if provisioning is enabled
+          enabled = true
+          enabled = env[:provision_enabled] if env.has_key?(:provision_enabled)
+          return @app.call(env) if !enabled
+
+          username = @machine.ssh_info()[:username]
+
+          # change ownership of the provisioning path recursively to the
+          # ssh user
+          #
+          # TODO submit patch to vagrant to set appropriate permissions
+          # based on ssh username
+          @machine.config.vm.provisioners.each do |provisioner|
+            cfg = provisioner.config
+            path = cfg.upload_path if cfg.respond_to? :upload_path
+            path = cfg.provisioning_path if cfg.respond_to? :provisioning_path
+            @machine.communicate.sudo("chown -R #{username} #{path}",
+              :error_check => false)
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-simplecloud/actions/power_off.rb

@@ -0,0 +1,33 @@
+require 'vagrant-simplecloud/helpers/client'
+#TODO: --force
+module VagrantPlugins
+  module SimpleCloud
+    module Actions
+      class PowerOff
+        include Helpers::Client
+
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @client = client
+          @simple_client = simple_client
+          @logger = Log4r::Logger.new('vagrant::simplecloud::power_off')
+        end
+
+        def call(env)
+          # submit power off droplet request
+          result = @simple_client.post("/v2/droplets/#{@machine.id}/actions", {:type => 'power_off'})
+
+          # wait for request to complete
+          env[:ui].info I18n.t('vagrant_simple_cloud.info.powering_off')
+          @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-simplecloud/actions/power_on.rb

@@ -0,0 +1,36 @@
+require 'vagrant-simplecloud/helpers/client'
+
+module VagrantPlugins
+  module SimpleCloud
+    module Actions
+      class PowerOn
+        include Helpers::Client
+
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @client = client
+          @simple_client = simple_client
+          @logger = Log4r::Logger.new('vagrant::simplecloud::power_on')
+        end
+
+        def call(env)
+          # submit power on droplet request
+          result = JSON.parse(@simple_client.droplet_actions.power_on(droplet_id: @machine.id.to_s))
+          #result = @client.post("/v2/droplets/#{@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)
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,56 @@
+require 'vagrant-simplecloud/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
+          @simple_client = simple_client
+          @logger = Log4r::Logger.new('vagrant::simplecloud::rebuild')
+        end
+
+        def call(env)
+          # look up image id
+          image_id = @simple_client.images.find(id: @machine.provider_config.image).id
+
+          # submit rebuild request
+          result = @simple_client.post("/v2/droplets/#{@machine.id}/actions", {
+            :type => 'rebuild',
+            :image => image_id
+          })
+          # simple cloud api has not return region_slug in response
+          #image_id = @simple_client.images.find(id: @machine.provider_config.image).id
+          #env[:ui].info "#{@machine.id.to_s}, #{image_id}"
+          #result = JSON.parse(@simple_client.droplet_actions.rebuild(droplet_id: @machine.id, image: image_id))
+
+          # 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-simplecloud/actions/reload.rb

@@ -0,0 +1,33 @@
+require 'vagrant-simplecloud/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::simplecloud::reload')
+        end
+
+        def call(env)
+          # submit reboot droplet request
+          result = @client.post("/v2/droplets/#{@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'])
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end
+
+

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

@@ -0,0 +1,53 @@
+require 'vagrant-simplecloud/helpers/client'
+
+module VagrantPlugins
+  module SimpleCloud
+    module Actions
+      class SetupKey
+        include Helpers::Client
+
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @client = client
+          @simple_client = simple_client
+          @logger = Log4r::Logger.new('vagrant::simplecloud::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
+            # assigns existing ssh key id to env for use by other commands
+            @simple_client.ssh_keys.all().each do |key|
+                if key["name"] == ssh_key_name
+                    env[:ssh_key_id] = key["id"]
+                    env[:ui].info I18n.t('vagrant_simple_cloud.info.using_key', {
+                      :name => ssh_key_name
+                    })
+                    @app.call(env)
+                    return
+                end
+            end
+            env[:ssh_key_id] = create_ssh_key(ssh_key_name, env)
+            @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)
+          ssh_key = DropletKit::SSHKey.new(name: name, public_key: pub_key)
+          result = @simple_client.ssh_keys.create(ssh_key)
+          env[:ui].info I18n.t('vagrant_simple_cloud.info.creating_key', {
+            :name => name
+          })
+          result.id
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-simplecloud/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::simplecloud::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 /redhat/
+                @machine.communicate.execute('yum install sudo')
+                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
+            when /debian/
+                @machine.communicate.execute('apt-get install sudo')
+                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-simplecloud/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::simplecloud::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-simplecloud/actions/shut_down.rb

@@ -0,0 +1,34 @@
+require 'vagrant-simplecloud/helpers/client'
+
+module VagrantPlugins
+  module SimpleCloud
+    module Actions
+      class ShutDown
+        include Helpers::Client
+
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @client = client
+          @simple_client = simple_client
+          @logger = Log4r::Logger.new('vagrant::simplecloud::shut_down')
+        end
+
+        def call(env)
+          # submit shutdown droplet request
+          result = @simple_client.post("/v2/droplets/#{@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['droplet']['event_id'])
+          @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-simplecloud/actions/sync_folders.rb

@@ -0,0 +1,89 @@
+require 'vagrant/util/subprocess'
+require 'vagrant/util/which'
+
+module VagrantPlugins
+  module SimpleCloud
+    module Actions
+      class SyncFolders
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @logger = Log4r::Logger.new('vagrant::simplecloud::sync_folders')
+        end
+
+        def call(env)
+          ssh_info = @machine.ssh_info
+
+          @machine.config.vm.synced_folders.each do |id, data|
+            next if data[:disabled]
+
+            if @machine.guest.capability?(:rsync_installed)
+              installed = @machine.guest.capability(:rsync_installed)
+              if !installed
+                can_install = @machine.guest.capability?(:rsync_install)
+                raise Vagrant::Errors::RSyncNotInstalledInGuest if !can_install
+                @machine.ui.info I18n.t("vagrant.rsync_installing")
+                @machine.guest.capability(:rsync_install)
+              end
+            end
+
+            hostpath  = File.expand_path(data[:hostpath], env[:root_path])
+            guestpath = data[:guestpath]
+
+            # make sure there is a trailing slash on the host path to
+            # avoid creating an additional directory with rsync
+            hostpath = "#{hostpath}/" if hostpath !~ /\/$/
+
+            # on windows rsync.exe requires cygdrive-style paths
+            if Vagrant::Util::Platform.windows?
+              hostpath = hostpath.gsub(/^(\w):/) { "/cygdrive/#{$1}" }
+            end
+
+            env[:ui].info I18n.t('vagrant_simple_cloud.info.rsyncing', {
+              :hostpath => hostpath,
+              :guestpath => guestpath
+            })
+
+            # create the guest path
+            @machine.communicate.sudo("mkdir -p #{guestpath}")
+            @machine.communicate.sudo(
+              "chown -R #{ssh_info[:username]} #{guestpath}")
+
+            key = ssh_info[:private_key_path]
+            key = key[0] if key.is_a?(Array)
+
+            # attach to vagrant's default rsync opetions
+            # http://docs.vagrantup.com/v2/synced-folders/rsync.html
+            args = ["--verbose", "--archive", "--delete", "-z", "--copy-links", *Array(data[:rsync__args])]
+            exclude = [".vagrant/", *Array(data[:rsync__exclude])]
+
+            # rsync over to the guest path using the ssh info
+            command = [
+              "rsync",
+              *args.map{|e|[e]}.flatten,
+              *exclude.map{|e|["--exclude", e]}.flatten,
+              "-e", "ssh -p #{ssh_info[:port]} -o StrictHostKeyChecking=no -i '#{key}'",
+              hostpath,
+              "#{ssh_info[:username]}@#{ssh_info[:host]}:#{guestpath}"]
+
+            # we need to fix permissions when using rsync.exe on windows, see
+            # http://stackoverflow.com/questions/5798807/rsync-permission-denied-created-directories-have-no-permissions
+            if Vagrant::Util::Platform.windows?
+              command.insert(1, "--chmod", "ugo=rwX")
+            end
+
+            r = Vagrant::Util::Subprocess.execute(*command)
+            if r.exit_code != 0
+              raise Errors::RsyncError,
+                :guestpath => guestpath,
+                :hostpath => hostpath,
+                :stderr => r.stderr
+            end
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end