require 'net/sftp'
require 'ptools'

module Middleman
  module Deploy
    module Methods
      class Sftp < Ftp
        def process
          puts "## Deploying via sftp to #{user}@#{host}:#{path}"

          # `nil` is a valid value for user and/or pass.
          Net::SFTP.start(host, user, password: pass, port: port) do |sftp|
            sftp.mkdir(path)

            Dir.chdir(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, filename, file_path)
          reply     = exception.message
          err_code  = reply[0, 3].to_i

          sftp.upload(filename, file_path) if err_code == 550
        end

        def upload_directory(sftp, filename)
          file_path = "#{path}/#{filename}"

          begin
            sftp.mkdir(file_path)
            puts "Created directory #{filename}"
          rescue
          end
        end

        def upload_file(sftp, filename)
          file_path = "#{path}/#{filename}"

          begin
            sftp.upload(filename, file_path)
          rescue Exception => exception
            handle_exception(exception, filename, file_path)
          end

          puts "Copied #{filename}"
        end
      end
    end
  end
end