prep_kit 0.1.8 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8cd5f08170c8218f60129bdb66d94ad714ee2a16
-  data.tar.gz: ea53500e04585df087780ed5373829405c239be7
+  metadata.gz: 9191872a8674750052b1e662cdc945dd5d5625d5
+  data.tar.gz: 94ed4d91c37f711ea4c731e0c33fe00ff18321ed
 SHA512:
-  metadata.gz: b25f05441760238e487da7f3b3699ddf473a91441ac6df841dbcec6a6445daf58c8b80de8ee9f63d22b399f3de3ad851aae89b61ee73606a3daad1fe91a01fbc
-  data.tar.gz: 427de6267edb65f24ca6302ce27158a57add8edcb8caeb704599ee55ec740df7ffb5718457e3621c436797e22965145d006129ecc65007dd5f765915396d2cb6
+  metadata.gz: cc777bde408a35d80513a321d68c379331e05461a8d35443a7069f42b7543f121b82f0cf92ad67fc5bf5e003bbd2518e6daaee27de64263fe3337a93c577df3e
+  data.tar.gz: 6ece4b3b0a6e23dc5eeadffc10b1602c699f729e93d9706e7fc36618174f1da310714164683be26bac5e32ecb39a1f44d0b26dfcb0ec3a8a60f2f468f2287252

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    prep_kit (0.1.8)
+    prep_kit (0.1.9)
       colorize
       net-scp
       net-ssh

data/lib/prep_kit/remote.rb

@@ -1,16 +1,16 @@
 module PrepKit
   class Remote
     def initialize(host, user, options, &callback)
-      @ssh = create_ssh(host, user, options, &callback)
+      @ssh, @scp = create_connection(host, user, options, &callback)
     end
 
     def exec!(command)
-      @ssh.call(command)
+      @ssh.(command)
     end
 
     %i(upload! download!).each do |name|
       define_method(name) do |src, dst|
-        @ssh.call { |ssh| ssh.scp.public_send(name, src, dst) }
+        @scp.() { |scp| scp.public_send(name, src, dst) }
       end
     end
 
@@ -20,11 +20,23 @@ module PrepKit
 
     private
 
-    def create_ssh(host, user, options, &callback)
+    def create_connection(host, user, options, &callback)
+      connection = lambda do |&block|
+        Net::SSH.start(host, user, options) do |ssh|
+          block.(ssh)
+
+          ssh.loop
+        end
+      end
+
+      [create_ssh(connection, &callback), create_scp(connection)]
+    end
+
+    def create_ssh(connection, &callback)
       lambda do |command|
         status = nil
 
-        Net::SSH.start(host, user, options) do |ssh|
+        connection.() do |ssh|
           channel = ssh.open_channel do |chan|
             chan.exec command do |ch, success|
               raise RuntimeError command unless success
@@ -41,14 +53,17 @@ module PrepKit
                 status = data.read_long
               end
             end
-          end
 
-          channel.wait
-          ssh.loop
+            channel.wait
+          end
         end
 
         status
       end
     end
+
+    def create_scp(connection)
+      -> (&callback) { connection.() { |ssh| callback.(ssh.scp) } }
+    end
   end
 end

data/lib/prep_kit/version.rb

@@ -1,3 +1,3 @@
 module PrepKit
-  VERSION = "0.1.8"
+  VERSION = "0.1.9"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: prep_kit
 version: !ruby/object:Gem::Version
-  version: 0.1.8
+  version: 0.1.9
 platform: ruby
 authors:
 - daisuko