santoku 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f2d8bfadb34d6d3a3afdfcc83ab1278c9a39e578
-  data.tar.gz: df7f809fa31474ab1eef2cbb3b75ce2541e0f903
+  metadata.gz: fbf7295813d610a5e5f36bc497af900d690b98c1
+  data.tar.gz: 95089cdb8cfe09bb57ad35c1fd2ce7a3894ec763
 SHA512:
-  metadata.gz: 3bbd5a78c001b6ba734b547c20964020560bf57bb75b5ab99d999946f974e0f8d41e388ad4f3a1430568ec57ad177f730a52d666a2d570888ab05bc63d8ad712
-  data.tar.gz: cb620bb5d996f0286047fd59d71ee456615e0241be082013c10a38e68d9e03b81ed6f917fb68f326c777726aa71ffb604733342e811f648136545ea916fc38d8
+  metadata.gz: 68f756ff452d1f66604752a69107445e30728b7e8478284de23fbb5ae106626d778213bd6bc1a78dd7ceb7bdc2a1501d7bb1bacfc6ec769fd81b4041be8e0466
+  data.tar.gz: 84c3d165ba29d80644a1319468814be213461219fe12b89884d348f78d9ef6f6bfe7654e4a2cc170e994fc199fe2653a5c2e162aa011bcaf84f139e028428c84

data/lib/santoku/helpers.rb

@@ -0,0 +1,90 @@
+require 'ridley'
+require 'net/ssh'
+require 'santoku/printers'
+
+class Santoku
+  class << self
+    def failed(reason)
+      failed_output_stream.push reason
+      print_fail
+    end
+
+    def succeeded(reason)
+      success_output_stream.push reason
+      print_success
+    end
+
+    def timed_out(reason)
+      timeout_output_stream.push reason
+      print_timeout
+    end
+
+    def timeout_output_stream
+      @timeout_output_stream ||= Array.new
+    end
+
+    def failed_output_stream
+      @failed_output_stream ||= Array.new
+    end
+
+    def success_output_stream
+      @success_output_stream ||= Array.new
+    end
+
+    def verbose_success?
+      @verbose_success
+    end
+
+    def invert?
+      @invert
+    end
+
+  private
+    def 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 ridley
+      @ridley ||= Ridley.new(
+        server_url: knife_config.chef_server_url,
+        client_name: knife_config.node_name,
+        client_key: knife_config.client_key
+      )
+    end
+
+    def ssh_exec!(ssh, command)
+      # I am not awesome enough to have made this method myself
+      # 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
+  end
+end

data/lib/santoku/printers.rb

@@ -0,0 +1,49 @@
+require 'colorize'
+
+class Santoku
+  class << self
+    def print_fail
+      print 'F'.red
+    end
+
+    def print_success
+      print '.'.green
+    end
+
+    def print_timeout
+      print '*'.yellow
+    end
+
+    def print_timeout_stream
+      timeout_output_stream.each do |output|
+        puts output.yellow
+      end
+    end
+
+    def print_failed_stream
+      failed_output_stream.each do |output|
+        puts output.red
+      end
+    end
+
+    def print_success_stream
+      success_output_stream.each do |output|
+        puts output.green
+      end
+    end
+
+    def print_summary
+      print "\n"
+      print_timeout_stream
+      print_failed_stream
+      print_success_stream if verbose_success?
+
+      puts "\n---------------------------------------\n"
+
+      puts "#{'Success'.green}: #{success_output_stream.count}"
+      puts "#{'Timed out or does not resolve'.yellow}: #{timeout_output_stream.count}"
+      puts "#{'Failed'.red}: #{failed_output_stream.count}"
+      puts "#{'Total'.blue}: #{success_output_stream.count + timeout_output_stream.count + failed_output_stream.count}"
+    end
+  end
+end

metadata

@@ -1,11 +1,10 @@
 --- !ruby/object:Gem::Specification
 name: santoku
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Steven De Coeyer
-- Mike Morraye
 autorequire: 
 bindir: bin
 cert_chain: []
@@ -76,6 +75,8 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/santoku.rb
+- lib/santoku/helpers.rb
+- lib/santoku/printers.rb
 - bin/santoku
 homepage: https://github.com/zhann/santoku
 licenses: