require 'net/sftp' require 'ptools' module Middleman module Deploy module Methods class Sftp < Ftp def process puts "## Deploying via sftp to #{self.user}@#{self.host}:#{path}" # `nil` is a valid value for user and/or pass. Net::SFTP.start(self.host, self.user, :password => self.pass, :port => self.port) do |sftp| sftp.mkdir(self.path) Dir.chdir(self.server_instance.build_dir) do filtered_files.each do |filename| if File.directory?(filename) upload_directory(sftp, filename) else upload_file(sftp, filename) end end end end end protected def handle_exception(exception) reply = exception.message err_code = reply[0,3].to_i if err_code == 550 sftp.upload(filename, file_path) end end def upload_directory(sftp, filename) file_path = "#{self.path}/#{filename}" begin sftp.mkdir(file_path) puts "Created directory #{filename}" rescue end end def upload_file(sftp, filename) file_path = "#{self.path}/#{filename}" begin sftp.upload(filename, file_path) rescue Exception => exception handle_exception(exception, file_path) end puts "Copied #{filename}" end end end end end