lib/populus/remote_runner.rb in populus-0.0.3 vs lib/populus/remote_runner.rb in populus-0.0.4

@@ -1,6 +1,9 @@
 require 'specinfra'
+require 'erb'
+require 'tempfile'
+require 'fileutils'
 
 module Populus
   class RemoteRunner
     def initialize(backend, &run_it)
       @backend = backend
@@ -12,8 +15,31 @@
 
       res = @backend.run_command(command.join(" "))
       Populus.logger.debug("stdout:\n%s" % res.stdout)
       Populus.logger.debug("stderr:\n%s" % res.stderr)
       Populus.logger.info("Command exited: %d" % res.exit_status)
+    end
+
+    def create_file(path, template="", context=nil)
+      file = Tempfile.new(".populus-tmp")
+      content = ERB.new(template).result(context || binding)
+      file.write content
+      file.close
+      @backend.send_file(file.path, path)
+      Populus.logger.info("Created Successfully: %s" % path)
+
+      FileUtils.rm_f(file.path)
+    end
+
+    def upload_file(to_path, local: nil)
+      raise ArgumentError unless local
+      @backend.send_file(local, to_path)
+      Populus.logger.info("Upload Successfully: %s to %s" % [local, to_path])
+    end
+
+    def upload_dir(to_dir, local: nil)
+      raise ArgumentError unless local
+      @backend.send_directory(local, to_dir)
+      Populus.logger.info("Upload Directory Successfully: %s to %s" % [local, to_dir])
     end
   end
 end