require 'shellwords'

module Roger::Release::Finalizers
  
  # Finalizes the release by uploading your mockup with rsync to a remote server
  # 
  # @see RsyncFinalizer#initialize for options
  #
  class Rsync < Base
  
    # @param Hash options The options
    #
    # @option options String :rsync The Rsync command to run (default is "rsync")
    # @option options String :remote_path The remote path to upload to
    # @option options String :host The remote host to upload to
    # @option options String :username The remote username to upload to
    # @option options Boolean :ask Prompt the user before uploading (default is true)
    def initialize(options = {})
      @options = {
        :rsync => "rsync",
        :remote_path => "",
        :host => "",
        :username  => "",
        :ask => true
      }.update(options)
    end
  
    def call(release, options = {})
      options = @options.dup.update(options)
    
      # Validate options
      validate_options!(release, options)
      
      if !options[:ask] || (prompt("Do you wish to upload to #{options[:host]}? Type y[es]: ")) =~ /\Ay(es)?\Z/
        begin
          `#{@options[:rsync]} --version`
        rescue Errno::ENOENT
          raise RuntimeError, "Could not find rsync in #{@options[:rsync].inspect}"
        end
        
        local_path = release.build_path.to_s
        remote_path = options[:remote_path]
        
        local_path += "/" unless local_path =~ /\/\Z/
        remote_path += "/" unless remote_path =~ /\/\Z/
        
        release.log(self, "Starting upload of #{(release.build_path + "*")} to #{options[:host]}")
      
        command = "#{options[:rsync]} -az #{Shellwords.escape(local_path)} #{Shellwords.escape(options[:username])}@#{Shellwords.escape(options[:host])}:#{Shellwords.escape(remote_path)}"
        
        # Run r.js optimizer
        output = `#{command}`
        
        # Check if r.js succeeded
        unless $?.success?
          raise RuntimeError, "Rsync failed.\noutput:\n #{output}"
        end
      end
    end
    
    protected
    
    def validate_options!(release, options)
      must_have_keys = [:remote_path, :host, :username]
      if (options.keys & must_have_keys).size != must_have_keys.size
        release.log(self, "You must specify these options: #{(must_have_keys - options.keys).inspect}")
        raise "Missing keys: #{(must_have_keys - options.keys).inspect}"
      end
    end
    
    def prompt(question = 'Do you wish to continue?')
      print(question)
      return $stdin.gets.strip
    end
  
  end
end

Roger::Release::Finalizers.register(:rsync, Roger::Release::Finalizers::Rsync)