file_copy 0.0.1

data/lib/file_copy/copy.rb

@@ -0,0 +1,44 @@
+require 'net/ssh'
+require 'net/sftp'
+
+module BBFS
+  module FileCopy
+
+    # Creates ssh connection, assumes username and password
+    # or without password but with ssh keys.
+    def self.ssh_connect(username, password, server)
+      username = (username and username.length > 0) ? username : ENV['USER']
+      password = (password and password.length > 0) ? password : nil
+      port = 22 # 22 is a standart SSH port
+      raise "Undefined server" unless server
+      if (username and password)
+        Net::SSH.start(server, username,
+                       :password => password,
+                       :port => port)
+      elsif (username)
+        Net::SSH.start(server, username,
+                       :port => port)
+      else
+        raise "Undefined username"
+      end
+    end
+
+    # Simply copy map files using sftp server on dest_server.
+    # Note: stfp.upload support parallel uploads - default is 2.
+    # TODO(kolman): packing files, all, not all, determine by part of file.
+    def self.sftp_copy(username, password, server, files_map)
+      ssh = FileCopy.ssh_connect(username, password, server)
+      ssh.sftp.connect do |sftp|
+        uploads = files_map.map { |from,to|
+          # TODO(kolman): Create destination directory is not exists.
+          p "Copying #{from} to #{to}"
+          sftp.upload(from, to)
+        }
+        uploads.each { |u| u.wait }
+        p "Done."
+      end # sftp.connect
+    end # def initialize
+
+  end
+end
+

data/lib/file_copy.rb

@@ -0,0 +1,7 @@
+require_relative 'file_copy/copy'
+
+module BBFS
+  module FileCopy
+    VERSION = "0.0.1"
+  end
+end

data/test/file_copy/copy_spec.rb

@@ -0,0 +1,47 @@
+require_relative '../../lib/file_copy/copy.rb'
+
+module BBFS
+  module FileCopy
+    module Spec
+
+      describe 'FileCopy::ssh_connect' do
+        it 'should raise error when username not specified' do
+          ENV.stub(:[]).with(any_args()).and_return(nil)
+          expect { FileCopy::ssh_connect(nil, nil, 'a server') }.should raise_error "Undefined username"
+        end
+
+        it 'should raise error when server not specified' do
+          expect { FileCopy::ssh_connect('kuku', nil, nil) }.should raise_error "Undefined server"
+        end
+
+        it 'should try to connect if username is set explicitly' do
+          Net::SSH.should_receive(:start).with(any_args())
+          FileCopy::ssh_connect('kuku', nil, 'a server')
+        end
+        it 'should try to connect if username is set via ENV variable' do
+          Net::SSH.should_receive(:start).with(any_args())
+          ENV.stub(:[]).with("USER").and_return('kuku')
+          FileCopy::ssh_connect(nil, nil, 'a server')
+        end
+      end
+
+      # TODO(kolman): Very bad test, should rewrite and understand how to
+      # write test correctly.
+      describe 'FileCopy::sftp_copy' do
+        it 'should copy files' do
+          ssh_connection = double('Net::SSH::Connection')
+          FileCopy.should_receive(:ssh_connect).with(any_args()).and_return(ssh_connection)
+          sftp_session = double('Net::SFTP::Session')
+          ssh_connection.should_receive(:sftp).with(any_args()).and_return(sftp_session)
+          sftp_session.should_receive(:connect).with(any_args()).and_yield(sftp_session)
+          uploader = double('Operations::Upload')
+          uploader.should_receive(:wait).with(any_args()).and_return(true, true)
+          sftp_session.should_receive(:upload).with('a', 'b').and_return(uploader)
+          sftp_session.should_receive(:upload).with('c', 'd').and_return(uploader)
+          FileCopy::sftp_copy(nil, nil, nil, { 'a' => 'b', 'c' => 'd' })
+        end
+      end
+
+    end
+  end
+end

metadata

@@ -0,0 +1,48 @@
+--- !ruby/object:Gem::Specification
+name: file_copy
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Gena Petelko, Kolman Vornovitsky
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-04-08 00:00:00.000000000Z
+dependencies: []
+description: Library for copying files to remote servers.
+email: kolmanv@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/file_copy.rb
+- lib/file_copy/copy.rb
+- test/file_copy/copy_spec.rb
+homepage: http://github.com/kolmanv/bbfs
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.15
+signing_key: 
+specification_version: 3
+summary: Library for copying files remotely.
+test_files:
+- test/file_copy/copy_spec.rb