ingenerator/vagrant-mirror

require 'date'
require 'fileutils'

# Executes a full sync from the host to the guest instance based on the configuration
# in the vagrantfile, copying new or changed files to the guest as required.
#
# @author Andrew Coulton < andrew@ingerator.com >
module Vagrant
  module Mirror
    module Middleware
      class Sync < Base

        protected

        # Synchronizes the folder pairs configured in the vagrantfile
        #
        # @param [Array] The folder pairs to synchronise
        # @param [Vagrant::Action::Environment] The environment
        def execute(mirrors, env)
          ui = env[:ui]
          ui.info("Beginning directory synchronisation at " + DateTime.now.iso8601)

          begin
            each_mirror(mirrors) do | host_path, guest_sf_path, mirror_config |
              ui.info("Synchronising for #{host_path}")

              # Create any required symlinks
              mirror_config[:symlinks].each do | origpath |
                # Copy the path so that it is not altered elsewhere, then strip any leading /
                relpath = origpath.dup
                relpath.sub!(/^\//, '')
                source = "#{guest_sf_path}/#{relpath}"
                target = "#{mirror_config[:guest_path]}/#{relpath}"

                # Find the parent directory - we have to do this with regexp as we don't have the
                # right filesystem to use File.expand
                dirs = /^(.*)(\/.+)$/.match(target)
                if (dirs)
                  target_dir = dirs[1]
                else
                  target_dir = '/'
                end

                # Create the host directory if required
                host_dir = "#{host_path}/#{relpath}"
                if ( ! File.exists?(host_dir))
                  FileUtils.mkdir_p(host_dir)
                end

                # Create the parent directory and create the symlink
                ui.info("Creating link from #{target} to #{source}")
                env[:vm].channel.sudo("rm -f #{target} && mkdir -p #{target_dir} && ln -s #{source} #{target}")
              end

              # Trigger the sync on the remote host
              ui.info("Synchronising from #{guest_sf_path} to #{mirror_config[:guest_path]}")
              rsync = Vagrant::Mirror::Rsync.new(env[:vm], guest_sf_path, host_path, mirror_config)
              rsync.run('/')
            end

          rescue RuntimeError => e
            # Pass through Vagrant errors
            if e.is_a? Vagrant::Errors::VagrantError
              raise
            end

            # Convert to a vagrant error descendant so that the box is not cleaned up
            raise Vagrant::Mirror::Errors::Error.new("Vagrant-mirror caught a #{e.class.name} - #{e.message}")
          end

          ui.success("Completed directory synchronisation at " + DateTime.now.iso8601)
        end

      end
    end
  end
end