santoku 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 0738e1ba13ad70766617662b5e62054064e58183
+  data.tar.gz: ade98e3db326a6079ada9faf97ec0b14901882d6
+SHA512:
+  metadata.gz: d39af2df669f155aace2331d56080fe847296d868be5511114b30fcd55ee9b4a9ce9ca689fd907f9d9172d75eba44ce7905d3426a38af71f7ab16f61af4ffeda
+  data.tar.gz: b9e80ecff70535e561083d2e777ea269c8e077a5244da9865bfd8ed784adac996bdf2c3bbd9259cbecf2f5cfbcf99e98cb817a61fb1a669325999052dc5a73e2

data/bin/santoku

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+require './lib/santoku'
+
+command = ARGV[0]
+query = ARGV[1]
+Santoku.run(command, query)

data/lib/santoku.rb

@@ -0,0 +1,108 @@
+require 'colorize'
+require 'net/ssh'
+require 'peach'
+require 'ridley'
+require 'yaml'
+
+class Santoku
+  def self.run(command='uptime', query)
+    output_stream = Array.new
+    failed_output_stream = Array.new
+    timeout_output_stream = Array.new
+
+    ridley.search(:node, query).peach(5) do |node|
+      begin
+        timeout 10 do
+          Net::SSH.start(node.chef_id, 'root', paranoid: false, forward_agent: true) do |ssh|
+            output = ssh_exec!(ssh, command)
+            if output[2] != 0
+              failed_output_stream.push "#{node.chef_id} error #{output[2]}: #{output[1]}"
+              print 'F'.red
+            else
+              output_stream.push output[0]
+              print '.'.green
+            end
+          end
+        end
+
+      rescue TimeoutError
+        timeout_output_stream.push "#{node.chef_id}: connection timed out"
+        print '*'.yellow
+      rescue SocketError
+        timeout_output_stream.push "#{node.chef_id}: node does not resolve"
+        print '*'.yellow
+      rescue Errno::EHOSTUNREACH
+        timeout_output_stream.push "#{node.chef_id}: no route to host"
+        print '*'.yellow
+      rescue Exception => e
+        failed_output_stream.push "#{node.chef_id}: #{e.inspect}"
+        print 'F'.red
+      end
+    end
+
+    print "\n"
+
+    timeout_output_stream.each do |output|
+      puts output.yellow
+    end
+
+    failed_output_stream.each do |output|
+      puts output.red
+    end
+
+    puts "\n---------------------------------------\n"
+
+    puts "#{'Success'.green}: #{output_stream.count}"
+    puts "#{'Timed out or does not resolve'.yellow}: #{timeout_output_stream.count}"
+    puts "#{'Failed'.red}: #{failed_output_stream.count}"
+  end
+
+private
+  def self.knife_config
+    if ::File.exist?(File.expand_path("../knife.rb", __FILE__))
+      Ridley::Chef::Config.from_file(File.expand_path("../knife.rb", __FILE__))
+    elsif ::File.exist?("#{ENV['HOME']}/.chef/knife.rb")
+      Ridley::Chef::Config.from_file("#{ENV['HOME']}/.chef/knife.rb")
+    else
+      raise 'Could not find knife.rb in current directory or home '
+    end
+  end
+
+  def self.ridley
+    @ridley ||= Ridley.new(
+      server_url: knife_config.chef_server_url,
+      client_name: knife_config.node_name,
+      client_key: knife_config.client_key
+    )
+  end
+end
+
+def ssh_exec!(ssh, command)
+  # I am not awesome enough to have made this method myself
+  # I've just modified it a bit
+  # Originally submitted by 'flitzwald' over here: http://stackoverflow.com/a/3386375
+  stdout_data = ""
+  stderr_data = ""
+  exit_code = nil
+
+  ssh.open_channel do |channel|
+    channel.exec(command) do |ch, success|
+      unless success
+        abort "FAILED: couldn't execute command (ssh.channel.exec)"
+      end
+      channel.on_data do |ch,data|
+        stdout_data+=data
+      end
+
+      channel.on_extended_data do |ch,type,data|
+        stderr_data+=data
+      end
+
+      channel.on_request("exit-status") do |ch,data|
+        exit_code = data.read_long
+      end
+    end
+  end
+  ssh.loop
+  [stdout_data, stderr_data, exit_code]
+end

metadata

@@ -0,0 +1,89 @@
+--- !ruby/object:Gem::Specification
+name: santoku
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Steven De Coeyer
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-10-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: peach
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: ridley
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A gem to perform command over parrallel ssh connections on multiple chef
+  serverspec. Output is rspec-like.
+email: steven@banteng.be
+executables:
+- santoku
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/santoku.rb
+- bin/santoku
+homepage: https://github.com/zhann/santoku
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.3
+signing_key: 
+specification_version: 4
+summary: Parrallel ssh commands over chef servers with rspec-like output
+test_files: []