module Vagrant
  module Plugin
    module Remote
      class Communicator
        # This module enables Communicator for server mode
        module Remote

          # Add an attribute accesor for the client
          # when applied to the Communicator class
          def self.prepended(klass)
            klass.class_eval do
              attr_accessor :client
            end
          end

          def initialize(machine)
            @logger = Log4r::Logger.new("vagrant::remote::communicator")
            @logger.debug("initializing communicator with remote baackend")
            @machine = machine
            @client = machine.client.communicate
          end

          def ready?
            @logger.debug("remote communicator, checking if it's ready")
            @client.ready(@machine)
          end

          def wait_for_ready(time)
            @logger.debug("remote communicator, waiting for ready")
            @client.wait_for_ready(@machine, time)
          end

          def download(from, to)
            @logger.debug("remote communicator, downloading #{from} -> #{to}")
            @client.download(@machine, from, to)
          end

          def upload(from, to)
            @logger.debug("remote communicator, uploading #{from} -> #{to}")
            @client.upload(@machine, from, to)
          end

          def execute(cmd, opts=nil, &block)
            @logger.debug("remote communicator, executing command")
            res = @client.execute(@machine, cmd, opts)
            yield :stdout, res.stdout if block_given?
            yield :stderr, res.stderr if block_given?
            res.exit_code
          end
  
          def sudo(cmd, opts=nil, &block)
            @logger.debug("remote communicator, executing (privileged) command")
            res = @client.privileged_execute(@machine, cmd, opts)
            yield :stdout, res.stdout if block_given?
            yield :stderr, res.stderr if block_given?
            res.exit_code
          end

          def test(cmd, opts=nil)
            @logger.debug("remote communicator, testing command")
            @client.test(@machine, cmd, opts)
          end

          def reset!
            @logger.debug("remote communicator, reseting")
            @client.reset(@machine)
          end
          
          def to_proto
            client.proto
          end
        end
      end
    end
  end
end