screwcap 0.3 → 0.3.1

data/Manifest.txt

@@ -5,11 +5,14 @@ History.txt
 Manifest.txt
 README.rdoc
 Rakefile
+TODO
 bin/screwcap
 lib/exts.rb
 lib/screwcap.rb
 lib/screwcap/base.rb
 lib/screwcap/deployer.rb
+lib/screwcap/message_logger.rb
+lib/screwcap/runner.rb
 lib/screwcap/sequence.rb
 lib/screwcap/server.rb
 lib/screwcap/task.rb
@@ -28,6 +31,7 @@ test/config/expect.rb
 test/config/gateway.rb
 test/config/no_server.rb
 test/config/rails_tasks.rb
+test/config/real_world_recipe.rb
 test/config/simple_recipe.rb
 test/config/undefined_command_set.rb
 test/config/undefined_item.rb

data/Rakefile

@@ -11,7 +11,7 @@ Hoe.plugin :newgem
 # Generate all the Rake tasks
 # Run 'rake -T' to see list of generated tasks (from gem root directory)
 $hoe = Hoe.spec 'screwcap' do
-  self.version  = '0.3'
+  self.version  = '0.3.1'
   self.developer 'Grant Ammons', 'grant@pipelinedeals.com'
   self.rubyforge_name       = self.name # TODO this is default value
   self.extra_deps         = [['net-ssh','>= 2.0.23'],['net-ssh-gateway','>=1.0.1'], ['net-scp','>=1.0.4']]

data/TODO

@@ -0,0 +1 @@
+* validate if a gateway exists if declared with server

data/lib/screwcap/message_logger.rb

@@ -0,0 +1,35 @@
+module MessageLogger
+  def self.included(base)
+    base.extend(ClassMethods)
+  end
+
+  module ClassMethods
+    def log(msg, options = {})
+      logmsg(msg, $stdout, options)
+    end
+
+    def errorlog(msg, options = {})
+      logmsg(msg, $stderr, options)
+    end
+
+    private
+
+    def logmsg(msg, output, options)
+      return if @options and @options[:silent] == true
+      if @options and not @options[:nocolor] == true
+        case options[:color]
+        when :blue
+          output << "\033[0;36m#{msg}#{"\033[0m" if options[:clear]}"
+        when :bluebold
+          output << "\033[1;36m#{msg}#{"\033[0m" if options[:clear]}"
+        when :red
+          output << "\033[0;31m#{msg}#{"\033[0m" if options[:clear]}"
+        when :green
+          output << "\033[0;32m#{msg}#{"\033[0m" if options[:clear]}"
+        end
+      else
+        output << msg
+      end
+    end
+  end
+end

data/lib/screwcap/runner.rb

@@ -0,0 +1,107 @@
+class Runner
+  include MessageLogger
+
+  def self.execute! task, options
+    @task = task; @options = options
+    threads = []
+    @task.__servers.each do |_server|
+      _server.__addresses.each do |_address|
+        if task.__options[:parallel] == false
+          execute_on(_server, _address)
+        else
+          threads << Thread.new(_server, _address) { |server, address| execute_on(server, address) }
+        end
+      end
+    end
+    threads.each {|t| t.join }
+  end
+
+  private
+
+  def self.execute_on(server, address) 
+    begin
+      log "\n*** BEGIN executing task #{@task.__name} on #{server.name} with address #{address}\n", :color => :blue
+
+      server.__with_connection_for(address) do |ssh|
+        failed_command = nil
+        @task.__commands.each do |command|
+          if run_command(ssh, address, server, command) != 0 and command[:onfailure]
+            failed_command = command
+            break
+          end
+        end
+        if failed_command
+          @task.__commands = []
+          @task.send(failed_command[:onfailure])
+          @task.__commands.each { |command| run_command(ssh, address, server, command) }
+        end
+      end
+    rescue Net::SSH::AuthenticationFailed => e
+      raise Net::SSH::AuthenticationFailed, "Authentication failed for server named #{server.name}.  Please check your authentication credentials."
+    #rescue Exception => e
+    #  errorlog "    F: (#{address}): #{e}", :color => :red
+    #ensure
+    #  log "*** END executing task #{@task.__name} on #{server.name} with address #{address}\n\n", :color => :blue
+    end
+  end
+
+  # courtesy of flitzwald on stackoverflow
+  # http://stackoverflow.com/questions/3386233/how-to-get-exit-status-with-rubys-netssh-library
+  def self.ssh_exec!(ssh, command)
+    stdout_data = ""
+    stderr_data = ""
+    exit_code = nil
+    exit_signal = nil
+    ssh.open_channel do |channel|
+      channel.exec(command) do |ch, success|
+        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
+  
+        channel.on_request("exit-signal") do |ch, data|
+          exit_signal = data.read_long
+        end
+      end
+    end
+    ssh.loop
+    [stdout_data, stderr_data, exit_code, exit_signal]
+  end
+
+
+  def self.run_command(ssh, address, server, command)
+    if command[:type] == :remote
+      log "    I: (#{address}):  #{command[:command]}\n", :color => :green
+      stdout, stderr, exit_code, exit_signal = ssh_exec! ssh, command[:command]
+      log("    O: (#{address}):  #{stdout}", :color => :green) unless stdout.blank?
+      errorlog("    O: (#{address}):  #{stderr}", :color => :red) unless stderr.blank?
+      errorlog("    E: (#{address}): #{command[:command]} return exit code: #{exit_code}\n", :color => :red) if exit_code != 0
+      return exit_code
+    elsif command[:type] == :local
+      ret = `#{command[:command]}`
+      if $?.to_i == 0
+        log "    I: (local):  #{command[:command]}\n", :color => :blue
+        log "    O: (local):  #{ret}\n", :color => :blue
+      else
+        log "    I: (local):  #{command[:command]}\n", :color => :blue
+        errorlog "    O: (local):  #{ret}\n", :color => :red
+        errorlog("    E: (local): #{command[:command]} return exit code: #{$?}\n", :color => :red) if $? != 0
+      end
+      return $?
+    elsif command[:type] == :scp
+      log "    I: (#{address}): SCP #{command[:local]} to #{server.__user}@#{address}:#{command[:remote]}\n", :color => :green
+      server.__upload_to!(address, command[:local], command[:remote])
+
+      # this will need to be improved to allow for :onfailure
+      return 0
+    end
+
+  end
+end

data/lib/screwcap.rb

@@ -17,7 +17,7 @@ require 'screwcap/sequence'
 require 'screwcap/deployer'
 
 module Screwcap
-  VERSION='0.3'
+  VERSION='0.3.1'
 
   class TaskNotFound < RuntimeError; end
   class NoServersDefined < Exception; end

data/screwcap.gemspec

@@ -6,7 +6,7 @@ require 'bundler/version'
  
 Gem::Specification.new do |s|
   s.name        = "screwcap"
-  s.version     = "0.3"
+  s.version     = "0.3.1"
   s.platform    = Gem::Platform::RUBY
   s.author     = "Grant Ammons"
   s.email       = ["grant@pipelinedealsco.com"]

data/test/config/real_world_recipe.rb

@@ -0,0 +1,15 @@
+addresses = []
+addresses << "ec2-174-129-166-79.compute-1.amazonaws.com"
+#addresses << "ec2-174-129-60-243.compute-1.amazonaws.com"
+#addresses << "ec2-204-236-255-140.compute-1.amazonaws.com"
+#gateway :dev_gateway, :address => "dev.pipelinedealsco.com", :user => "root", :keys => "~/.ssh/dev_key"
+server :staging, :addresses => addresses, :user => "root", :keys => "~/.ssh/pipeline_key"
+
+command_set :dofail do
+  run "echo 'failing yeah'"
+end
+
+task_for :task, :server => :staging, :parallel => false do
+  run "ruby -e 'Kernel.exit(0)'", :onfailure => :dofail
+  run "echo 'happy days'"
+end

metadata

@@ -1,12 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: screwcap
 version: !ruby/object:Gem::Version 
-  hash: 13
+  hash: 17
   prerelease: false
   segments: 
   - 0
   - 3
-  version: "0.3"
+  - 1
+  version: 0.3.1
 platform: ruby
 authors: 
 - Grant Ammons
@@ -118,11 +119,14 @@ files:
 - Manifest.txt
 - README.rdoc
 - Rakefile
+- TODO
 - bin/screwcap
 - lib/exts.rb
 - lib/screwcap.rb
 - lib/screwcap/base.rb
 - lib/screwcap/deployer.rb
+- lib/screwcap/message_logger.rb
+- lib/screwcap/runner.rb
 - lib/screwcap/sequence.rb
 - lib/screwcap/server.rb
 - lib/screwcap/task.rb
@@ -141,6 +145,7 @@ files:
 - test/config/gateway.rb
 - test/config/no_server.rb
 - test/config/rails_tasks.rb
+- test/config/real_world_recipe.rb
 - test/config/simple_recipe.rb
 - test/config/undefined_command_set.rb
 - test/config/undefined_item.rb