lib/simple_provision/scp.rb in simple_provision-0.99.5 vs lib/simple_provision/scp.rb in simple_provision-0.99.6

@@ -1,17 +1,19 @@
+require 'net/scp'
+require 'byebug'
+
 module SimpleProvision
   class SCP
     FILENAME = "simpro.tar.gz"
 
-    def initialize(server, opts)
-      @server, @opts = server, opts
+    def initialize(username, host, opts)
+      @username, @host, @opts = username, host, opts
     end
 
-    def to_server
+    def copy_files
       create_local_archive
       scp_files_to_server
-      extract_remote_archive
       remove_local_archive
     end
 
     private
 
@@ -36,14 +38,14 @@
 
       system("cd tmp && tar -czf #{FILENAME} files/ scripts/")
     end
 
     def scp_files_to_server
-      @server.scp("tmp/#{FILENAME}", ".")
-    end
-
-    def extract_remote_archive
-      @server.ssh("tar -xzf #{FILENAME}")
+      Net::SCP.start(@host, @username) do |scp|
+        scp.upload!("tmp/#{FILENAME}", ".")
+      end
+    rescue Net::SSH::HostKeyMismatch
+      puts "Please run ssh #{@username}@#{@host} to verify the fingerprint first"
     end
 
     def remove_local_archive
       `rm -rf tmp`
     end