tinyssh 0.0.1

data/bin/tinyssh

@@ -0,0 +1,72 @@
+#!/usr/bin/env ruby
+
+require 'optparse'
+require 'ostruct'
+require 'rubygems'
+require 'highline/import'
+require 'net/ssh'
+require 'net/scp'
+
+require 'lib/colorize'
+require 'lib/tinysshoptparse'
+require 'lib/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 options.host_file != nil
+      File.open(options.host_file).each_line do |line|
+        HOST << line.gsub("\n", '')
+      end
+    else
+      TinysshOptparse.parse(['-h'])
+      puts 'ss'
+      exit
+    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)
+
+        if session.class == String
+          OUTPUT[index] = session % host
+          next
+        end
+
+        case ARGV[0] 
+        when 'r', 'run' then OUTPUT[index] = session.run(ARGV[1])
+        when 'u', 'up', 'upload' then OUTPUT[index] = session.up(ARGV[1], ARGV[2])
+        when 'd', 'dl', 'download' then OUTPUT[index] = session.dl(ARGV[1], ARGV[2])
+        end
+      }
+    end
+
+    T.each do |t|
+      t.join
+    end
+
+    OUTPUT.keys.sort.each do |k|
+      print OUTPUT[k]
+    end
+
+    if options.output_file != nil
+      File.open(options.output_file, 'w') do |f|
+        OUTPUT.keys.sort.each do |k|
+          f.write(OUTPUT[k])
+        end
+      end
+    end
+
+  else
+    TinysshOptparse.parse(['-h'])
+    exit
+  end
+end
+
+main(TinysshOptparse.parse(ARGV))

data/lib/colorize.rb

@@ -0,0 +1,16 @@
+#!/usr/bin/env ruby
+
+class String
+  # colorization
+  def colorize(text, color_code)
+    "\e[#{color_code}m#{text}\e[0m"
+  end
+
+  def red
+    colorize(self, 31)
+  end
+
+  def green
+    colorize(self, 32)
+  end
+end

data/lib/tinyssh.rb

@@ -0,0 +1,41 @@
+include Net::SSH
+
+module Net::SSH
+  def conn(*options)
+    begin
+      return Net::SSH.start(*options)
+    rescue Timeout::Error
+      return "\n#{OUTPUT_TITLE}\nHost : %s [ " + "Connecting Timeout".red + " ] !\n"
+    rescue Errno::EHOSTUNREACH
+      return "\n#{OUTPUT_TITLE}\nHost : %s [ " + "Unreachable".red + " ] !\n"
+    rescue Errno::ECONNREFUSED
+      return "\n#{OUTPUT_TITLE}\nHost : %s [ " + "Connection Refused".red + " ] !\n"
+    rescue Net::SSH::AuthenticationFailed
+      return "\n#{OUTPUT_TITLE}\nHost : %s [ " + "Authentication Failure".red + " ] !\n"
+    end
+  end
+end
+
+class Net::SSH::Connection::Session
+  def run(cmd)
+    output = self.exec!(cmd)
+    return "\n#{output}#{OUTPUT_TITLE}\nExecute \"#{cmd}\" at #{host}: [ ", "OK".green, " ]\n"
+  rescue
+    return "\n#{OUTPUT_TITLE}\nExecute \"#{cmd}\" at #{host}: [ ", "FAIL".red, " ] !\n"
+  end
+
+  def up(src, dst)
+    self.scp.upload!(src, dst, :recursive => true)
+    return "\n#{OUTPUT_TITLE}\nUpload #{host}:#{src} to #{dst}: [ ", "OK".green, " ] !\n"
+  rescue
+    return "\n#{OUTPUT_TITLE}\nUpload #{host}:#{src} to #{dst}: [ ", "FAIL".red, " ] !\n"
+  end
+
+  def dl(src, dst)
+    self.scp.download!(src, dst, :recursive => true)
+    return "\n#{OUTPUT_TITLE}\nDownload #{host}:#{src} to #{dst}: [ ", "OK".green, " ] !\n"
+  rescue
+    return "\n#{OUTPUT_TITLE}\nDownload #{host}:#{src} to #{dst}: [ ", "FAIL".red, " ] !\n"
+  end
+end
+

data/lib/tinysshoptparse.rb

@@ -0,0 +1,62 @@
+class TinysshOptparse
+  def self.parse(args)
+    options = OpenStruct.new
+    options.user        = 'root'
+    options.port        = '59878'
+    options.num         = '0'
+    options.keys        = ['/root/.ssh/id_rsa']
+    options.timeout     = 5
+    options.host_file   = nil
+    options.output_file = nil
+
+    opts = OptionParser.new do |opts|
+      opts.banner = "\nUsage: tinyssh.rb run|upload|download <position parameters> [option]"
+
+      opts.separator ""
+      opts.separator "Specific options:"
+
+      opts.on("-u", "--user [user]", 
+              "Specify a user for accessing remote hosts.(default: root)") do |user|
+        options.user = user
+      end
+
+      opts.on("-p", "--port [port]", 
+              "Port to connect to on the remote host(default: 59878)") do |port|
+        options.port = port
+      end
+
+      opts.on("-t", "--timeout [time]", 
+              "Set the command timeout to seconds(default: 0)") do |timeout|
+        options.timeout = timeout.to_i
+      end
+
+      opts.on("-l", "--list-file [file]", 
+              "File containing a list of hosts' IP Address") do |file|
+        options.host_file = file
+      end
+
+      opts.on("-o", "--output-file [file]", 
+              "Output result to log file") do |output_file|
+        options.output_file = output_file
+      end
+
+      opts.on("-i", "--identity-keys [keys]", 
+              "identity (private) keys for authentication(default: /root/.ssh/id_rsa)") do |keys|
+        options.keys = keys
+      end
+
+      opts.on("-n", "--number-of-thread [number]", 
+              "Specify the number of concurrent jobs for each execution") do |num|
+        options.num = num
+      end
+
+      opts.on_tail("-h", "--help", "Show this message") do
+        puts opts
+        exit
+      end
+    end
+
+    opts.parse!(args)
+    options
+  end
+end

metadata

@@ -0,0 +1,68 @@
+--- !ruby/object:Gem::Specification 
+name: tinyssh
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- William Herry
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-08-02 00:00:00 Z
+dependencies: []
+
+description: run ssh command, upload/download file/directory on/to muti host
+email: WilliamHerryChina@Gmail.com
+executables: 
+- tinyssh
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/colorize.rb
+- lib/tinysshoptparse.rb
+- lib/tinyssh.rb
+- bin/tinyssh
+homepage: https://rubygems.org/gems/tinyssh
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: Tiny SSH
+test_files: []
+