puppet-pssh 0.1

data/LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2012 Sergio Rubio <rubiojr@frameos.org>
+
+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.md

@@ -0,0 +1,17 @@
+# puppet-pssh 
+
+Puppet parallel-ssh integration
+
+# Install
+
+    gem install puppet-pssh
+
+# Usage
+
+    puppet-pssh --help
+
+# Copyright
+
+Copyright (c) 2012 Sergio Rubio. See LICENSE.txt for
+further details.
+

data/Rakefile

@@ -0,0 +1,35 @@
+require 'rubygems'
+require 'rake'
+require './lib/puppet-pssh.rb'
+
+require 'jeweler'
+Jeweler::Tasks.new do |gem|
+  # gem is a Gem::Specification... see http://docs.rubygems.org/read/chapter/20 for more options
+  gem.version = PuppetPSSH::VERSION
+  gem.name = "puppet-pssh"
+  gem.homepage = "http://github.com/rubiojr/puppet-pssh"
+  gem.license = "MIT"
+  gem.summary = %Q{Puppet parallel-ssh integration}
+  gem.description = %Q{Puppet parallel-ssh integration}
+  gem.email = "rubiojr@frameos.org"
+  gem.authors = ["Sergio Rubio"]
+  # Include your dependencies below. Runtime dependencies are required when using your gem,
+  # and development dependencies are only needed for development (ie running rake tasks, tests, etc)
+  gem.add_runtime_dependency 'colored'
+  gem.add_runtime_dependency 'excon'
+  gem.add_runtime_dependency 'net-dns'
+  gem.add_runtime_dependency 'json'
+  gem.add_runtime_dependency 'clamp'
+  gem.add_development_dependency 'rspec', '> 1.2.3'
+  gem.add_development_dependency 'jeweler'
+end
+Jeweler::RubygemsDotOrgTasks.new
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+task :default => :build

data/bin/puppet-pssh

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby
+require File.join(File.dirname(__FILE__), '/../lib/puppet-pssh.rb')
+PuppetPSSH::Driver.run

data/lib/puppet-pssh.rb

@@ -0,0 +1,152 @@
+#!/usr/bin/env ruby
+require 'clamp'
+require 'excon'
+require 'json'
+require 'logger'
+require 'colored'
+
+module PuppetPSSH
+
+  VERSION = "0.1"
+
+  if !defined? Log or Log.nil?
+    Log = Logger.new($stdout)
+    Log.formatter = proc do |severity, datetime, progname, msg|
+      if severity == "INFO"
+        "*".bold.cyan + " #{msg}\n"
+      else
+        severity = severity.red.bold if severity == 'ERROR'
+        severity = severity.yellow.bold if severity == 'WARN'
+        "#{Time.now.to_i} #{severity}: #{msg}\n"
+      end
+    end
+    Log.level = Logger::INFO unless (ENV["DEBUG"].eql? "yes" or ENV["DEBUG"].eql? 'true')
+    Log.debug "Initializing logger"
+  end
+
+
+  class BaseCommand < Clamp::Command
+    
+    option ["-m", "--match"], "REGEX", "Only the nodes matching the regex", :default => '.*'
+    option ["-p", "--puppetmaster"], "PUPPETMASTER", "Puppet master host", :default => 'puppet'
+    option "--puppetmaster-port", "PUPPETMASTER_PORT", "Puppet master port", :default => '8080'
+    option "--use-ssl", :flag, "Use SSL (https) to communicate with the puppetmaster", :default => false
+    option "--debug", :flag, "Print debugging output", :default => false do |o|
+      Log.level = Logger::DEBUG 
+    end
+
+    def get_nodes(puppetmaster)
+      url = "#{use_ssl? ? 'https' : 'http'}://#{puppetmaster}:#{puppetmaster_port}/nodes"
+      Log.debug "Puppet master host: #{puppetmaster}"
+      Log.debug "Puppet master url: #{url}"
+
+      nodes = []
+      begin
+        out = Excon.get url
+        JSON.parse(out.body).each do |n| 
+          next unless  n =~ /#{match}/
+          nodes << n 
+        end
+      rescue TypeError => e
+        raise Exception.new "Error retrieving node list from master host: #{puppetmaster}"
+      rescue Excon::Errors::SocketError => e
+        raise Exception.new "Could not connect to the puppet master host: #{puppetmaster}"
+      end
+      nodes
+    end
+
+  end
+
+  class List < BaseCommand 
+
+    def execute
+      begin
+        get_nodes(puppetmaster).each { |n| puts n }
+      rescue Exception => e
+        Log.error e.message
+        exit 1
+      end
+    end
+  end
+
+  # 
+  # Run an arbitrary command using parallel-ssh against all the nodes
+  # registered in the puppet master
+  #
+  # Needs pssh (parallel-ssh) installed.
+  #
+  class Run < BaseCommand 
+
+    parameter "COMMAND ...", "Command to run"
+    option "--nameserver", "DNS_SERVER", "Resolve node name using the given nameserver"
+    option "--pssh-path", "PSSH_PATH", "Parallel-ssh command path", :default => '/usr/bin/parallel-ssh'
+    option "--hostlist-path", "HOSTLIST_PATH", "Save host list to path", :default => '/tmp/puppet-pssh-run-hostlist'
+    option ["-H", "--hostlist-path"], "HOSTLIST_PATH", "Save host list to path", :default => '/tmp/puppet-pssh-run-hostlist'
+    option ["-o", "--node-output-path"], "NODE_OUTPUT_PATH", "Save host list to path", :default => '/tmp/'
+    option "--[no-]host-key-verify", :flag, "Verify SSH host key", :default => true
+
+    def execute
+      unless File.exist?(pssh_path)
+        Log.error "parallel-ssh command not found in #{pssh_path}."
+        Log.error "Install it or use --pssh-path argument."
+        exit 1
+      end
+
+      nodes = []
+      begin
+        nodes = get_nodes(puppetmaster)
+      rescue => e
+        Log.error e.message
+        exit 1
+      end
+
+      unless File.exist?(hostlist_path)
+        Log.info "Generating hostlist..."
+        Log.debug "Hostlist path: #{hostlist_path}"
+        # 
+        # Optionally resolve names using specific DNS server
+        #
+        unless nameserver.nil?
+          require 'net/dns'
+          Log.info "DNS Server: #{nameserver}"
+          Log.info "Resolving node names... (may take a while)"
+          res = Net::DNS::Resolver.new
+          res.nameservers = nameserver
+        end
+        #
+        File.open hostlist_path, 'w' do |f|
+          nodes.each do |i|
+            address = i
+            # try to resolve before writing the list
+            Log.debug "Adding #{address}"
+            unless nameserver.nil?
+              address = res.query(i).answer.first.address rescue next
+            end
+            f.puts "#{address} root"
+          end
+        end
+      else
+        Log.warn "Using cached hostlist in #{hostlist_path}"
+      end
+
+      $stdout.sync = true
+      command = "sleep `echo $[ ( $RANDOM % 30 )  + 1 ]`;" + command_list.join(' ')
+      Log.info "Node log output path: #{node_output_path}"
+      Log.info "Running command '#{command}' with parallel-ssh..."
+      ssh_opts = ''
+      unless host_key_verify?
+        Log.warn 'Disabled host key verification'
+        ssh_opts = '-o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null'
+      end
+      system "#{pssh_path} -p 40 -o #{node_output_path} -t 300 -h #{hostlist_path} -x '#{ssh_opts}' " + "'#{command} 2>&1'"
+    end
+  end
+
+  class Driver < Clamp::Command
+
+    subcommand "run", "Run an arbitrary command against the nodes", Run
+    subcommand "list", "List registered nodes", List 
+
+  end
+
+end

metadata

@@ -0,0 +1,165 @@
+--- !ruby/object:Gem::Specification
+name: puppet-pssh
+version: !ruby/object:Gem::Version
+  version: '0.1'
+  prerelease: 
+platform: ruby
+authors:
+- Sergio Rubio
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-10-20 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: colored
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: excon
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-dns
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: clamp
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>'
+      - !ruby/object:Gem::Version
+        version: 1.2.3
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>'
+      - !ruby/object:Gem::Version
+        version: 1.2.3
+- !ruby/object:Gem::Dependency
+  name: jeweler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Puppet parallel-ssh integration
+email: rubiojr@frameos.org
+executables:
+- puppet-pssh
+extensions: []
+extra_rdoc_files:
+- LICENSE.txt
+- README.md
+files:
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/puppet-pssh
+- lib/puppet-pssh.rb
+homepage: http://github.com/rubiojr/puppet-pssh
+licenses:
+- MIT
+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.24
+signing_key: 
+specification_version: 3
+summary: Puppet parallel-ssh integration
+test_files: []