vos 0.0.4 → 0.1.0

data/lib/rsh/drivers/ssh.rb

@@ -1,147 +0,0 @@
-module Rsh
-  module Drivers
-    class Ssh < Abstract
-      def initialize options = {}
-        super
-        raise "ssh options not provided!" unless options[:ssh]
-        raise "invalid ssh options!" unless options[:ssh].is_a?(Hash)
-      end
-    
-      def upload_file from_local_path, to_remote_path
-        sftp.upload! from_local_path, fix_path(to_remote_path)
-      end
-
-      def download_file from_remote_path, to_local_path
-        File.open to_local_path, "w" do |out|
-          sftp.download! fix_path(from_remote_path), out #, :recursive => true          
-        end
-      end
-
-      def exist? remote_file_path
-        begin
-          fattrs = sftp.stat! fix_path(remote_file_path)
-          fattrs.directory? or fattrs.file? or fattrs.symlink?
-        rescue Net::SFTP::StatusException
-          false
-        end
-      end
-      alias_method :directory_exist?, :exist?
-      alias_method :file_exist?, :exist?
-
-      def remove_file remote_file_path
-        sftp.remove! fix_path(remote_file_path)
-      end
-
-      def exec command
-        remote do |ssh, sftp|
-          # somehow net-ssh doesn't executes ~/.profile, so we need to execute it manually
-          # command = ". ~/.profile && #{command}"
-
-          stdout, stderr, code, signal = hacked_exec! ssh, command
-
-          return code, stdout, stderr
-        end
-      end    
-    
-      def open      
-        ssh_options = self.options[:ssh].clone
-        host = options[:host] || raise('host not provided!')
-        user = ssh_options.delete(:user) || raise('user not provied!')
-        @ssh = Net::SSH.start(host, user, ssh_options)
-        @sftp = @ssh.sftp.connect
-      end
-    
-      def close
-        @ssh.close
-        # @sftp.close not needed
-        @ssh, @sftp = nil
-      end
-        
-      def create_directory path
-        remote do |ssh, sftp|          
-          sftp.mkdir! path
-          # exec "mkdir #{path}"
-        end        
-      end
-    
-      def remove_directory path
-        exec "rm -r #{path}"
-      end
-      
-      def upload_directory from_local_path, to_remote_path
-        remote do |ssh, sftp|
-          sftp.upload! from_local_path, fix_path(to_remote_path)
-        end
-      end
-      
-      def download_directory from_remote_path, to_local_path
-        remote do |ssh, sftp|
-          sftp.download! fix_path(from_remote_path), to_local_path, :recursive => true
-        end
-      end
-    
-      protected
-        def fix_path path
-          path.sub(/^\~/, home)
-        end
-        
-        def home
-          unless @home
-            command = 'cd ~; pwd'
-            code, stdout, stderr = exec command
-            raise "can't execute '#{command}'!" unless code == 0
-            @home = stdout.gsub("\n", '')    
-          end
-          @home
-        end
-      
-        # taken from here http://stackoverflow.com/questions/3386233/how-to-get-exit-status-with-rubys-netssh-library/3386375#3386375
-        def hacked_exec!(ssh, command, &block)
-          stdout_data = ""
-          stderr_data = ""
-          exit_code = nil
-          exit_signal = nil
-        
-          channel = ssh.open_channel do |channel|
-            channel.exec(command) do |ch, success|
-              raise "could not execute command: #{command.inspect}" unless success
-
-              channel.on_data{|ch2, data| stdout_data << data}
-              channel.on_extended_data{|ch2, type, data| stderr_data << data}
-              channel.on_request("exit-status"){|ch,data| exit_code = data.read_long}
-              channel.on_request("exit-signal"){|ch, data| exit_signal = data.read_long}
-            end          
-          end  
-        
-          channel.wait      
-        
-          [stdout_data, stderr_data, exit_code, exit_signal]
-        end
-    
-        def remote(&block)
-          if @ssh
-            block.call @ssh, @sftp
-          else            
-            # Rails.logger.info "Connecting to remote Hadoop #{options[:user]}@#{options[:host]}"          
-            begin
-              open
-              block.call @ssh, @sftp
-            ensure
-              close
-            end
-                              
-            # Net::SSH.start options[:host], options[:user], :config => true do |ssh|
-            #   ssh.sftp.connect do |sftp|
-            #     begin
-            #       @ssh, @sftp = ssh, sftp
-            #       block.call @ssh, @sftp
-            #     ensure
-            #       @ssh, @sftp = nil
-            #     end
-            #   end
-            # end
-          end
-        end
-    end
-  end
-end

data/lib/rsh/gems.rb

@@ -1,2 +0,0 @@
-gem 'net-ssh'
-gem 'net-sftp'

data/spec/abstract_driver/local_file

@@ -1 +0,0 @@
-some content

data/spec/abstract_driver.rb

@@ -1,82 +0,0 @@
-require 'spec_helper'
-
-shared_examples_for 'abstract driver' do
-  dir = "#{File.dirname __FILE__}/abstract_driver"
-  with_tmp_spec_dir dir, before: :each
-  
-  describe "io" do
-    before :each do
-      @local_dir = spec_dir      
-      @remote_dir = @driver.generate_tmp_dir_name    
-      
-      @driver.remove_directory @remote_dir if @driver.directory_exist? @remote_dir
-      @driver.create_directory @remote_dir
-    end
-
-    after :each do
-      @driver.remove_directory @remote_dir if @driver.directory_exist? @remote_dir
-    end
-  
-    describe "files" do  
-      before :each do
-        @local_file = "#{@local_dir}/local_file"
-        @check_file = "#{@local_dir}/check_file"
-        @remote_file = "#{@remote_dir}/remote_file"
-      end
-            
-      it "file_exist?" do
-        @driver.file_exist?(@remote_file).should be_false
-        @driver.upload_file(@local_file, @remote_file)
-        @driver.file_exist?(@remote_file).should be_true
-      end
-
-      it "upload & download file" do
-        @driver.upload_file(@local_file, @remote_file)
-        @driver.file_exist?(@remote_file).should be_true
-      
-        @driver.download_file(@remote_file, @check_file)
-        File.read(@local_file).should == File.read(@check_file)
-      end
-    
-      it "remove_file" do
-        @driver.upload_file(@local_file, @remote_file)
-        @driver.file_exist?(@remote_file).should be_true
-        @driver.remove_file(@remote_file)
-        @driver.file_exist?(@remote_file).should be_false
-      end
-    end
-    
-    describe 'directories' do
-      before :each do
-        @from_local, @remote_path, @to_local = "#{@local_dir}/dir", "#{@remote_dir}/upload", "#{@local_dir}/download"
-      end
-      
-      it "directory_exist?, create_directory, remove_directory" do
-        dir = "#{@remote_dir}/some_dir"
-        @driver.directory_exist?(dir).should be_false
-        @driver.create_directory(dir)
-        @driver.directory_exist?(dir).should be_true
-        @driver.remove_directory(dir)
-        @driver.directory_exist?(dir).should be_false
-      end
-    
-      it "upload_directory & download_directory" do
-        upload_path_check = "#{@remote_path}/dir2/file"
-        @driver.file_exist?(upload_path_check).should be_false
-        @driver.upload_directory(@from_local, @remote_path)
-        @driver.file_exist?(upload_path_check).should be_true
-
-        download_path_check = "#{@to_local}/dir2/file"
-        File.exist?(download_path_check).should be_false
-        @driver.download_directory(@remote_path, @to_local)
-        File.exist?(download_path_check).should be_true
-      end
-    end  
-  end
-  
-  describe "shell" do
-    it 'exec' do
-      @driver.exec("echo 'ok'").should == [0, "ok\n", ""]
-    end  
-  end
-end

data/spec/box_spec/local_file

@@ -1 +0,0 @@
-some content

data/spec/local_driver_spec.rb

@@ -1,9 +0,0 @@
-require 'abstract_driver'
-
-describe Rsh::Drivers::Local do
-  it_should_behave_like "abstract driver"    
-    
-  before :each do
-    @driver = Rsh::Drivers::Local.new
-  end
-end

data/spec/ssh_driver_spec.rb

@@ -1,15 +0,0 @@
-require 'abstract_driver'
-require 'yaml'
-
-describe Rsh::Drivers::Ssh do
-  it_should_behave_like "abstract driver"    
-
-  before :each do
-    @driver = Rsh::Drivers::Ssh.new config[:remote_driver]
-    @driver.open
-  end
-  
-  after :each do
-    @driver.close
-  end  
-end