remoter 1.0.0.pre1

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Andrey Subbotin
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,17 @@
+= remoter
+
+A tool to simplify executing of shell command on several remote machines over SSH.
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2010 Andrey Subbotin. See LICENSE for details.

data/lib/remoter.rb

@@ -0,0 +1,28 @@
+module Remoter
+  
+  autoload :Base,       'remoter/base'
+  autoload :Session,    'remoter/session'  
+  autoload :UI,         'remoter/ui'
+
+  class RemoterError < StandardError
+    def self.status_code(code = nil)
+      define_method(:status_code) { code }
+    end
+  end
+  
+  class SCPError < RemoterError; status_code(10); end
+  class EXEError < RemoterError; status_code(11); end
+  
+  class << self
+    attr_writer :ui
+
+    def ui
+      @ui ||= UI.new
+    end
+
+    def read_file(file)
+      File.open(file, "rb") { |f| f.read }
+    end
+  end
+
+end

data/lib/remoter/base.rb

@@ -0,0 +1,77 @@
+# encoding: utf-8
+
+require 'thor'
+require 'nmap/parser'
+
+module Remoter
+  class Base < Thor
+    
+    def initialize(*)
+      super
+      the_shell = (options["no-color"] ? Thor::Shell::Basic.new : shell)
+      Remoter.ui = UI::Shell.new(the_shell)
+      Remoter.ui.debug! if options["verbose"]
+    end
+
+    class_option "no-color", :type => :boolean, :desc => "Disable colorization in output"
+    class_option "verbose",  :type => :boolean, :desc => "Enable verbose output mode", :aliases => "-v"
+    class_option "workers",  :type => :numeric, :desc => "Number of parallel SSH sessions", :aliases => "-w", :default => 5
+    class_option "username",  :type => :string, :desc => "Username to use for the SSH sessions", :aliases => "-u"
+
+    protected
+    
+      def self.beam(name, &block)
+        self.send :method_option, "targets", :type => :array, :alias => "-t", :required => true, :desc => 
+          "Can pass hostnames, IP addresses, networks, etc. Ex: some.agent.org, ouragents.com/24, 192.168.0.1; 10.0.0-255.1-254"
+        self.send :method_option, "exclude", :type => :array, :desc => 
+          "Exclude hosts/networks."
+        define_method name do
+          workers = []
+          items_per_worker = (hosts.count.to_f / options['workers'].to_f).ceil
+          hosts.each_slice(items_per_worker).to_a.each_with_index do |targets, worker_no|
+            Remoter.ui.debug "WORKER ##{worker_no}: #{targets.count}"
+            workers << Process.fork do 
+              Remoter.ui.debug "WORKER ##{worker_no}: Proceeding with #{targets.count} agents..."      
+              while (target = targets.shift)
+                Remoter.ui.debug "WORKER ##{worker_no}: PROCESSING #{target.addr}..."
+                begin
+                  send Base.body_method_name(name), Session.new(self, target)
+                rescue RemoterError => error
+                  Remoter.ui.error "WORKER ##{worker_no}: ERROR: #{error.message}"
+                  Remoter.ui.error "WORKER ##{worker_no}: FAILED for #{target.addr}... WILL RETRY"
+                  targets.push target
+                else
+                  Remoter.ui.confirm "WORKER ##{worker_no}: SUCEEDED for #{target.addr}"
+                end
+              end
+            end
+          end
+          workers.each { |pid| Process.wait(pid) }
+          Remoter.ui.confirm "ALL DONE. KEWL."
+        end
+        no_tasks do
+          define_method body_method_name(name), block
+        end
+      end
+      
+      def parser
+        @parser ||= Nmap::Parser.parsescan('nmap', nmap_args)
+      end
+  
+      def nmap_args
+        result = "-sP"
+        result += " --exclude #{options['exclude'].join(',')}" if options['exclude']
+        result += " #{options['targets'].join(' ')}"
+        result
+      end
+    
+      def hosts
+        @hosts ||= parser.hosts('up')
+      end
+      
+    private
+      
+      def self.body_method_name(beam_name) ; "#{beam_name.to_s}_body".to_sym ; end
+        
+  end
+end

data/lib/remoter/session.rb

@@ -0,0 +1,30 @@
+require 'shellwords'
+
+module Remoter
+  class Session
+    attr_reader :remoter
+    attr_reader :target
+    
+    def initialize(_remoter, _target)
+      @remoter = _remoter
+      @target = _target
+    end
+
+    def scp(from, to)
+      target_cmd = "scp -o StrictHostKeyChecking=no \"#{from}\" \"#{user_part}#{target.addr}:#{to.shellescape}\""
+      raise SCPError, "The following command failed: #{target_cmd}" unless system(target_cmd)
+    end
+    
+    def exe(cmd)
+      target_cmd = "ssh #{user_part}#{target.addr} -o StrictHostKeyChecking=no \"#{cmd}\""
+      raise EXEError, "The following command failed: #{target_cmd}" unless system(target_cmd)
+    end
+    
+  private
+  
+    def user_part
+      remoter.options['username'] ? "#{remoter.options['username']}@" : ''
+    end
+    
+  end
+end

data/lib/remoter/ui.rb

@@ -0,0 +1,56 @@
+module Remoter
+  class UI
+    def warn(message)
+    end
+
+    def debug(message)
+    end
+
+    def error(message)
+    end
+
+    def info(message)
+    end
+
+    def confirm(message)
+    end
+
+    class Shell < UI
+      attr_writer :shell
+
+      def initialize(shell)
+        @shell = shell
+        @quiet = false
+        @debug = ENV['DEBUG']
+      end
+
+      def debug(msg)
+        @shell.say(msg) if @debug && !@quiet
+      end
+
+      def info(msg)
+        @shell.say(msg) if !@quiet
+      end
+
+      def confirm(msg)
+        @shell.say(msg, :green) if !@quiet
+      end
+
+      def warn(msg)
+        @shell.say(msg, :yellow)
+      end
+
+      def error(msg)
+        @shell.say(msg, :red)
+      end
+
+      def be_quiet!
+        @quiet = true
+      end
+
+      def debug!
+        @debug = true
+      end
+    end
+  end
+end

data/lib/remoter/version.rb

@@ -0,0 +1,10 @@
+module Remoter
+  module Version
+    MAJOR = 1
+    MINOR = 0
+    PATCH = 0
+    BUILD = 'pre1'
+
+    STRING = [MAJOR, MINOR, PATCH, BUILD].compact.join('.')
+  end
+end

metadata

@@ -0,0 +1,100 @@
+--- !ruby/object:Gem::Specification 
+name: remoter
+version: !ruby/object:Gem::Version 
+  prerelease: true
+  segments: 
+  - 1
+  - 0
+  - 0
+  - pre1
+  version: 1.0.0.pre1
+platform: ruby
+authors: 
+- Andrey Subbotin
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-12-14 00:00:00 -08:00
+default_executable: []
+
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: nmap-parser
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  prerelease: false
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: thor
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  prerelease: false
+  version_requirements: *id002
+description: A tool to simplify executing of shell command on several remote machines over SSH.
+email: andrey@subbotin.me
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- LICENSE
+- README.rdoc
+- lib/remoter.rb
+- lib/remoter/base.rb
+- lib/remoter/session.rb
+- lib/remoter/ui.rb
+- lib/remoter/version.rb
+has_rdoc: true
+homepage: http://github.com/eploko/remoter
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">"
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 1
+      - 3
+      - 1
+      version: 1.3.1
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Run commands on multiple remote machines over SSH easily.
+test_files: []
+