file_transfer 0.0.1 → 0.0.2

data/lib/file_transfer/sftp.rb

@@ -5,15 +5,28 @@ module FileTransfer
     def initialize(options = {})
       options[:port] ||= 22
       super(options)
-      connect
     end
 
     def download(from_path, to_path)
-      @sftp.download! from_path, to_path
+      connect if closed?
+      timeout(300) do
+        @sftp.download! from_path, to_path
+      end
     end
 
     def upload(from_path, to_path)
-      @sftp.upload! from_path, to_path
+      connect if closed?
+      timeout(300) do
+        @sftp.upload! from_path, to_path
+      end
+    end
+
+    def close
+      if @sftp && !@sftp.closed?
+        timeout(30) do
+          @sftp.close
+        end
+      end
     end
 
     private
@@ -22,6 +35,10 @@ module FileTransfer
         @sftp = Net::SFTP.start(@host, @username, :password => @password, :port => @port)
       end
 
+      def closed?
+        !@sftp || @sftp.closed?
+      end
+
       def timeout(seconds, &block)
         Timeout.timeout(seconds.to_i) do
           block.call

data/lib/file_transfer/version.rb

@@ -1,3 +1,3 @@
 module FileTransfer # :nodoc:
-  VERSION = "0.0.1"  # :nodoc:
+  VERSION = "0.0.2"  # :nodoc:
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: file_transfer
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors: