class VagrantPlugins::DigitalOcean::Actions::SyncFolders

Public Class Methods

new(app, env) click to toggle source
# File lib/vagrant-digitalocean/actions/sync_folders.rb, line 8
def initialize(app, env)
  @app = app
  @machine = env[:machine]
  @logger = Log4r::Logger.new('vagrant::digitalocean::sync_folders')
end

Public Instance Methods

call(env) click to toggle source
# File lib/vagrant-digitalocean/actions/sync_folders.rb, line 14
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_digital_ocean.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