tinyssh 0.0.2 → 0.0.3

data/README.md

@@ -0,0 +1,60 @@
+tinyssh is a little ruby script to batch execute command, upload/download file or directory with ssh
+
+## Require
+you should have ruby installed and some support packages for ssh
+
+- ruby
+- rubygems
+- net-ssh
+- net-scp
+- highline
+
+you can install them by follow command if you use RPM base Linux
+
+```
+yum -y install ruby rubygems
+gem install net-ssh
+gem install net-scp
+gem install highline
+gem install tinyssh
+
+find / -name tinyssh # find where tinyssh installed
+ln -sv location_of_tinyssh /usr/local/bin # if you use rvm to manage multi version ruby, you don't need do this
+```
+
+## Usage
+
+```
+Usage: ruby tinyssh.rb run|upload|download <position parameters> [option]
+
+Specific options:
+    -u, --user [user]                Specify a user for accessing remote hosts.(default: root)
+    -p, --port [port]                Port to connect to on the remote host(default: 22)
+    -t, --timeout [time]             Set the command timeout to seconds(default: 0)
+    -l, --list-file [file]           File containing a list of hosts' IP Address
+    -o, --output-file [file]         Output result to log file
+    -i, --identity-key [key]         identity (private) key for authentication(default: /root/.ssh/id_rsa)
+    -n, --number-of-thread [number]  Specify the number of concurrent jobs for each execution
+    -h, --help                       Show this message
+```
+
+- Run command
+
+```
+ruby tinyssh.rb run "command" -l my_host.txt -u root -p 22 -t 10 -i my_keys -n 10
+```
+
+- Upload file or directory
+
+```
+ruby tinyssh.rb upload src_file_or_directory dst_directory -l my_host.txt -u root -p 22 -t 10 -i my_keys -n 10
+```
+
+- Download file or directory
+
+```
+ruby tinyssh.rb download src_file_or_directory dst_directory -l my_host.txt -u root -p 22 -t 10 -i my_keys -n 10
+```
+
+Note: `run`, `upload`, `download`, can be `r`, `u`, `d`
+Note: if you use key to authorize and your key is `/root/.ssh/id_rsa`, you don't need to specify `-i /root/.ssh/id_rsa`, but it will ask for password, just press enter

data/Rakefile

@@ -0,0 +1,8 @@
+require 'rake/testtask'
+
+Rake::TestTask.new do |t|
+  t.libs << 'test'
+end
+
+desc "Run tests"
+task :default => :test

data/bin/tinyssh

@@ -14,24 +14,28 @@ require 'tinyssh'
 T = []
 HOST = []
 OUTPUT = {}
-OUTPUT_TITLE = "-------------------------- EXECUTE STATUS --------------------------"
 
 def main(options)
-  if ['run', 'r', 'u', 'up', 'upload', 'd', 'dl', 'download'].include?(ARGV[0])
-    passphrase = ask("Enter passphrash: ") { |q| q.echo = false }
+  if ['run', 'r', 'u', 'up', 'upload', 'd', 'dl', 'download'].include?(ARGV[0]) and options.host_file != nil
+    File.open(options.host_file).each_line do |line|
+      HOST << line.gsub("\n", '')
+    end
 
-    if options.host_file != nil
-      File.open(options.host_file).each_line do |line|
-        HOST << line.gsub("\n", '')
-      end
+    if options.keys.length == 1
+      password = ask("Enter password: ") { |q| q.echo = false }
+    else
+      password = ''
+    end
+
+    if password == ''
+      passphrase = ask("Enter passphrash: ") { |q| q.echo = false }
     else
-      TinysshOptparse.parse(['-h'])
-      exit
+      passphrase = ''
     end
 
     HOST.each_with_index do |host, index|
       T << Thread.new {
-        session = Net::SSH.conn(host, options.user, :port => options.port, :keys => options.keys, :timeout => options.timeout, :passphrase => passphrase.chomp)
+        session = Net::SSH.conn(host, options.user, :port => options.port, :password => password, :keys => options.keys, :timeout => options.timeout, :passphrase => passphrase.chomp)
 
         if session.class == String
           OUTPUT[index] = session % host

data/lib/tinyssh.rb

@@ -8,6 +8,7 @@ include Net::SSH
 module Net::SSH
   def conn(*options)
     begin
+      puts 'try to connction'
       return Net::SSH.start(*options)
     rescue Timeout::Error
       return "\n#{OUTPUT_TITLE}\nHost : %s [ " + "Connecting Timeout".red + " ] !\n"
@@ -17,7 +18,7 @@ module Net::SSH
       return "\n#{OUTPUT_TITLE}\nHost : %s [ " + "Connection Refused".red + " ] !\n"
     rescue Net::SSH::AuthenticationFailed
       return "\n#{OUTPUT_TITLE}\nHost : %s [ " + "Authentication Failure".red + " ] !\n"
-    else
+    rescue
       return "\n#{OUTPUT_TITLE}\nHost : %s [ " + "Name or Server not know".red + " ] !\n"
     end
   end

data/lib/tinysshoptparse.rb

@@ -36,7 +36,6 @@ class TinysshOptparse
       opts.on("-l", "--list-file [file]", 
               "File containing a list of hosts' IP Address") do |file|
         options.host_file = file
-        return true
       end
 
       opts.on("-o", "--output-file [file]", 

data/test/test_tinyssh.rb

@@ -0,0 +1,25 @@
+require 'test/unit'
+
+require 'colorize'
+require 'tinysshoptparse'
+require 'tinyssh'
+
+OUTPUT_TITLE = "-------------------------- EXECUTE STATUS --------------------------"
+
+class TinysshTest < Test::Unit::TestCase
+  def test_colorize
+    assert_equal "\e[31mred\e[0m", "red".red
+    assert_equal "\e[32mgreen\e[0m", "green".green
+  end
+
+  def test_tinysshoptparse
+    assert_equal 'test.txt', TinysshOptparse.parse(['-l', 'test.txt']).host_file
+  end
+
+  def test_tinyssh
+    assert_equal "\n#{OUTPUT_TITLE}\nHost : %s [ " + "Connection Refused".red + " ] !\n",
+      Net::SSH.conn('127.0.0.1', 'root', :port => '222', :timeout => 2)
+    assert_equal "\n#{OUTPUT_TITLE}\nHost : %s [ " + "Connecting Timeout".red + " ] !\n",
+      Net::SSH.conn('192.168.8.9', 'root', :port => '22', :timeout => 2)
+  end
+end 

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: tinyssh
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 25
   prerelease: 
   segments: 
   - 0
   - 0
-  - 2
-  version: 0.0.2
+  - 3
+  version: 0.0.3
 platform: ruby
 authors: 
 - William Herry
@@ -30,6 +30,9 @@ files:
 - lib/colorize.rb
 - lib/tinysshoptparse.rb
 - lib/tinyssh.rb
+- README.md
+- Rakefile
+- test/test_tinyssh.rb
 - bin/tinyssh
 homepage: https://rubygems.org/gems/tinyssh
 licenses: []