rigger 0.2.10

data/.document

@@ -0,0 +1,5 @@
+lib/**/*.rb
+bin/*
+- 
+features/**/*.feature
+LICENSE.txt

data/.rspec

@@ -0,0 +1 @@
+--color

data/Gemfile

@@ -0,0 +1,14 @@
+source "http://rubygems.org"
+
+gem "net-ssh-multi", "1.0.1"
+gem "net-sftp", "2.0.5"
+gem "popen4", "0.1.2"
+
+group :development do
+  gem "rspec", "~> 2.3.0"
+  gem "bundler", "~> 1.0.0"
+  gem "jeweler", "~> 1.5.2"
+  gem "rcov", ">= 0"
+  gem "mocha", ">= 0.9.8"
+  gem "bourne", ">= 1.0"
+end

data/Gemfile.lock

@@ -0,0 +1,50 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    Platform (0.4.0)
+    bourne (1.0)
+      mocha (= 0.9.8)
+    diff-lcs (1.1.2)
+    git (1.2.5)
+    jeweler (1.5.2)
+      bundler (~> 1.0.0)
+      git (>= 1.2.5)
+      rake
+    mocha (0.9.8)
+      rake
+    net-sftp (2.0.5)
+      net-ssh (>= 2.0.9)
+    net-ssh (2.1.3)
+    net-ssh-gateway (1.0.1)
+      net-ssh (>= 1.99.1)
+    net-ssh-multi (1.0.1)
+      net-ssh (>= 1.99.2)
+      net-ssh-gateway (>= 0.99.0)
+    open4 (0.9.6)
+    popen4 (0.1.2)
+      Platform (>= 0.4.0)
+      open4 (>= 0.4.0)
+    rake (0.8.7)
+    rcov (0.9.9)
+    rspec (2.3.0)
+      rspec-core (~> 2.3.0)
+      rspec-expectations (~> 2.3.0)
+      rspec-mocks (~> 2.3.0)
+    rspec-core (2.3.1)
+    rspec-expectations (2.3.0)
+      diff-lcs (~> 1.1.2)
+    rspec-mocks (2.3.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  bourne (>= 1.0)
+  bundler (~> 1.0.0)
+  jeweler (~> 1.5.2)
+  mocha (>= 0.9.8)
+  net-sftp (= 2.0.5)
+  net-ssh-multi (= 1.0.1)
+  popen4 (= 0.1.2)
+  rcov
+  rspec (~> 2.3.0)

data/LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2011 James Golick
+
+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,19 @@
+= rigger
+
+Description goes here.
+
+== Contributing to rigger
+ 
+* Check out the latest master to make sure the feature hasn't been implemented or the bug hasn't been fixed yet
+* Check out the issue tracker to make sure someone already hasn't requested it and/or contributed it
+* Fork the project
+* Start a feature/bugfix branch
+* Commit and push until you are happy with your contribution
+* Make sure to add tests for it. This is important so I don't break it in a future version unintentionally.
+* Please try not to mess with the Rakefile, version, or history. If you want to have your own version, or is otherwise necessary, that is fine, but please isolate to its own commit so I can cherry-pick around it.
+
+== Copyright
+
+Copyright (c) 2011 James Golick. See LICENSE.txt for
+further details.
+

data/Rakefile

@@ -0,0 +1,51 @@
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'rake'
+
+require 'jeweler'
+Jeweler::Tasks.new do |gem|
+  # gem is a Gem::Specification... see http://docs.rubygems.org/read/chapter/20 for more options
+  gem.name = "rigger"
+  gem.homepage = "http://github.com/jamesgolick/rigger"
+  gem.license = "MIT"
+  gem.summary = %Q{}
+  gem.description = %Q{}
+  gem.email = "jamesgolick@gmail.com"
+  gem.authors = ["James Golick"]
+  # 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 'net-ssh-multi', '1.0.1'
+  gem.add_runtime_dependency 'popen4', '0.1.2'
+  #  gem.add_development_dependency 'rspec', '> 1.2.3'
+end
+Jeweler::RubygemsDotOrgTasks.new
+
+require 'rspec/core'
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new(:spec) do |spec|
+  spec.pattern = FileList['spec/**/*_spec.rb']
+end
+
+RSpec::Core::RakeTask.new(:rcov) do |spec|
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+end
+
+task :default => :spec
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "rigger #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/TODO

@@ -0,0 +1 @@
+* Make it easier to load recipes from installed gems or other files by hooking in to $LOAD_PATH.

data/VERSION

@@ -0,0 +1 @@
+0.2.10

data/bin/rig

@@ -0,0 +1,4 @@
+require "rigger"
+require "rigger/cli"
+
+Rigger::CLI.new.start

data/config/rig.rb

@@ -0,0 +1,22 @@
+require "stringio"
+
+set :current_path, "/var/www/fetlife"
+server :app, "james@jamesgolick.com"
+server :db, "fetlife@app1.dal.fetlife"
+
+namespace :fuck do
+  task :test, :depends => [] do
+    #run_task "fuck:you"
+    run "ls -la"
+  end
+
+  task :you, :single => true do
+    run "mkdir -p /home/james/fuckit"
+    put "config/rig.rb", "/home/james/asdf.txt"
+    run "ls -la"
+  end
+
+  task :this do
+    run "ls -lah"
+  end
+end

data/lib/rigger.rb

@@ -0,0 +1,4 @@
+module Rigger
+  class Error < RuntimeError; end
+  class CommandError < Error; end
+end

data/lib/rigger/cli.rb

@@ -0,0 +1,78 @@
+require "rigger/dsl"
+require "rigger/task_execution_service"
+require "optparse"
+
+module Rigger
+  class CLI
+    def initialize(args                           = ARGV,
+                   dsl                            = DSL,
+                   task_execution_service_factory = TaskExecutionService)
+      @args                           = args
+      @dsl                            = dsl
+      @task_execution_service_factory = task_execution_service_factory
+    end
+
+    def start
+      @config = @dsl.new
+      options = parse_options
+      load_builtin_recipes
+      load_config_file
+
+      if options[:display_tasks]
+        display_tasks
+      else
+        run_tasks
+      end
+    end
+
+    protected
+      def load_config_file
+        @config_file = locate_config_file
+        @config.load_from_file(@config_file)
+      end
+
+      def load_builtin_recipes
+        Dir[File.dirname(__FILE__) + "/recipes/**/*.rb"].each do |f|
+          @config.load_from_file f
+        end
+      end
+
+      def locate_config_file
+        location = possible_config_locations.detect { |l| File.exist?(l) }
+        location || raise_missing_config_file
+      end
+
+      def possible_config_locations
+        ["config/rig.rb"]
+      end
+
+      def raise_missing_config_file
+        raise "Couldn't find a config file in #{possible_config_locations.inspect}"
+      end
+
+      def run_tasks
+        executor = @task_execution_service_factory.new(@config)
+
+        @args.each do |task_name|
+          executor.call(task_name)
+        end
+      end
+
+      def parse_options
+        {}.tap do |options|
+          OptionParser.new do |opts|
+            opts.banner = "Usage: #{$0} [options]"
+            opts.on("-T", "--tasks", "Display tasks.") do
+              options[:display_tasks] = true
+            end
+          end.parse!
+        end
+      end
+
+      def display_tasks
+        @config.tasks.sort_by { |t| t.name }.each do |task|
+          puts "rig #{task.name}"
+        end
+      end
+  end
+end

data/lib/rigger/connection_set.rb

@@ -0,0 +1,13 @@
+require "net/ssh"
+
+module Rigger
+  class ConnectionSet
+    def initialize
+      @connections = {}
+    end
+
+    def call(server)
+      @connections[server] ||= Net::SSH.start(server.host, server.user)
+    end
+  end
+end

data/lib/rigger/dsl.rb

@@ -0,0 +1,64 @@
+require "rigger/task"
+require "rigger/server"
+
+module Rigger
+  class DSL
+    attr_reader :tasks, :servers
+
+    def initialize(server_factory = Server,
+                   task_factory   = Task,
+                   file           = File)
+      @server_factory    = server_factory
+      @task_factory      = task_factory
+      @file              = file
+      @tasks             = {}
+      @servers           = []
+      @current_namespace = []
+      @vars              = {}
+    end
+
+    def namespace(name, &block)
+      @current_namespace << name
+      yield
+      @current_namespace.pop
+    end
+
+    def server(role, host, options = {})
+      @servers << @server_factory.new(role, host, options)
+    end
+
+    def task(name, options = {}, &block)
+      full_name = (@current_namespace + [name]).join(":")
+      @tasks[full_name] = @task_factory.new(full_name, options, block)
+    end
+
+    def load_from_file(filename)
+      instance_eval(@file.read(filename), filename)
+    end
+
+    def locate_task(name)
+      @tasks[name.to_s]
+    end
+
+    def set(name, value)
+      @vars[name.to_s] = value
+    end
+
+    def get(name)
+      @vars[name.to_s] || missing_var(name)
+    end
+
+    def fetch(name, default)
+      @vars.fetch(name.to_s, default)
+    end
+
+    protected
+    def missing_task(name)
+      raise "Can't find a task called #{name}."
+    end
+
+    def missing_var(name)
+      raise "Can't find a variable named #{name}."
+    end
+  end
+end

data/lib/rigger/execution_strategy.rb

@@ -0,0 +1,57 @@
+module Rigger
+  module ExecutionStrategy
+    class BasicExecutionStrategy
+      def initialize(executor_factory = TaskExecutor)
+        @executor_factory = TaskExecutor
+      end
+
+      def call(task, servers, config, execution_service)
+        @executor_factory.new(task, servers, execution_service, config).call
+      end
+    end
+
+    class SerialExecutionStrategy < BasicExecutionStrategy
+      def call(task, servers, config, execution_service)
+        servers.each do |s|
+          super(task, [s], config, execution_service)
+        end
+      end
+
+      def appropriate_strategy_for?(task)
+        task.options[:serial]
+      end
+    end
+
+    class SingleExecutionStrategy < BasicExecutionStrategy
+      def call(task, servers, config, execution_service)
+        super(task, [servers.first], config, execution_service)
+      end
+
+      def appropriate_strategy_for?(task)
+        task.options[:single]
+      end
+    end
+
+    class Selector
+      def initialize(basic_strategy = BasicExecutionStrategy.new,
+                     strategies     = [SerialExecutionStrategy.new,
+                                       SingleExecutionStrategy.new])
+        @basic_strategy = basic_strategy
+        @strategies     = strategies
+      end
+
+      def call(task)
+        appropriate_strategies = @strategies.select do |s|
+          s.appropriate_strategy_for?(task)
+        end
+
+        if appropriate_strategies.length > 1
+          raise MultipleAppropriateStrategies, 
+            "More than one strategy was appropriate for #{task.name}."
+        else
+          appropriate_strategies.first || @basic_strategy
+        end
+      end
+    end
+  end
+end

data/lib/rigger/recipes/chef.rb

@@ -0,0 +1,21 @@
+namespace :chef do
+  task :write_json, :serial => true do
+    role = @current_servers.first.role
+    conf = get(:chef_config).merge(:role => role, :run_list => "role[#{role}]").to_json
+    run "mkdir -p #{get(:release_path)}/config"
+    chef_config =<<-_END_
+    cookbook_path    ["/var/chef/cookbooks",
+                      "#{get(:release_path)}/cookbooks"]
+    log_level         :info
+    file_cache_path  "/var/chef"
+    role_path        "#{get(:release_path)}/roles"
+    Chef::Log::Formatter.show_time = false
+_END_
+    put chef_config, get(:release_path) + "/config/chef-solo.rb"
+    put conf, get(:release_path) + "/config/chef.json"
+  end
+
+  task :run do
+    run "chef-solo --config #{get(:release_path)}/config/chef-solo.rb -j #{get(:release_path)}/config/chef.json"
+  end
+end

data/lib/rigger/recipes/deploy.rb

@@ -0,0 +1,56 @@
+namespace :deploy do
+  task :bootstrap do
+    set :shared_path,   "#{get(:deploy_to)}/shared"
+    set :releases_path, "#{get(:deploy_to)}/releases"
+    set :release_path,  "#{get(:deploy_to)}/releases/#{Time.now.utc.strftime("%Y%m%d%H%M%S")}"
+    set :current_path,  "#{get(:deploy_to)}/current"
+  end
+
+  task :setup do
+    run_task("deploy:bootstrap")
+
+    run "mkdir -p #{get(:deploy_to)}"
+    run "mkdir -p #{get(:shared_path)}"
+    run "mkdir -p #{get(:releases_path)}"
+  end
+
+  # ripped off mostly from capistrano
+  task :keep_releases do
+    keep          = fetch(:keep_releases, 5)
+    releases_path = get(:releases_path)
+    releases      = capture("cd #{releases_path} && ls -x").strip.split.sort
+
+    if keep >= releases.length
+      puts "no old releases to clean up"
+    else
+      puts "keeping #{keep} of #{releases.length} deployed releases"
+
+      directories = (releases - releases.last(keep)).map do |release|
+        File.join(releases_path, release)
+      end.join(" ")
+
+      run "rm -rf #{directories}"
+    end
+  end
+
+  task :rollback do
+    releases = capture("cd #{get(:releases_path)} && ls -x").strip.split.sort
+    run "rm -rf #{get(:release_path)}; true"
+
+    if releases.length > 1
+      run "ln -nfs #{File.join(get(:releases_path), releases[-2])} #{get(:current_path)}"
+    else
+      puts "    no releases to rollback to. skipping rollback of symlink..."
+    end
+
+    run_task "deploy:restart"
+  end
+
+  task :symlink do
+    run "ln -nfs #{get(:release_path)} #{get(:current_path)}"
+  end
+
+  task :restart, :role => :app, :serial => true do
+    run "sudo god restart #{get(:application)}"
+  end
+end

data/lib/rigger/recipes/sbt.rb

@@ -0,0 +1,26 @@
+namespace :sbt do
+  task :assembly do
+    run_locally("~/bin/sbt assembly")
+  end
+
+  task :read_properties do
+    file  = File.read(Dir.pwd + "/project/build.properties")
+    props = Hash[*file.split("\n").map { |p| p.split("=") }.flatten]
+
+    set :sbt_name, props["project.name"]
+    set :sbt_version, props["project.version"]
+    set :sbt_build_version, props["build.scala.versions"]
+    set :sbt_assembly_filename, "#{get(:sbt_name)}-assembly-#{get(:sbt_version)}.jar"
+    set :sbt_assembly_path, Dir.pwd + "/target/scala_#{get(:sbt_build_version)}/#{get(:sbt_assembly_filename)}"
+  end
+
+  task :update_code do
+    run "mkdir #{get(:release_path)}"
+
+    @current_servers.map do |server|
+      assembly_path = get(:sbt_assembly_path)
+      cmd = "rsync --progress -az --delete --rsh='ssh -p 22' #{assembly_path} #{server.connection_string}:#{get(:release_path)}/"
+      Thread.new { system(cmd) }
+    end.each { |t| t.join }
+  end
+end

data/lib/rigger/recipes/service.rb

@@ -0,0 +1,13 @@
+namespace :service do
+  task :package do
+    set :bundle_name, "#{get(:application)}-#{get(:sbt_version)}.tar.gz"
+    set :bundle_path, "target/#{get(:bundle_name)}"
+    run_locally "tar -czv --file=#{get(:bundle_path)} config cookbooks roles -C target/scala_#{get(:sbt_build_version)} #{get(:sbt_assembly_filename)}"
+  end
+
+  task :update_code do
+    run "mkdir #{get(:release_path)}"
+    put File.read(get(:bundle_path)), get(:release_path) + "/" + get(:bundle_name)
+    run "cd #{get(:release_path)} && tar zxvf #{get(:bundle_name)}"
+  end
+end

data/lib/rigger/server.rb

@@ -0,0 +1,15 @@
+module Rigger
+  class Server < Struct.new(:role, :connection_string, :options)
+    def connection
+      @connection ||= Net::SSH.start(host, user)
+    end
+
+    def host
+      connection_string.split("@").last
+    end
+
+    def user
+      connection_string.split("@").first
+    end
+  end
+end

data/lib/rigger/server_resolver.rb

@@ -0,0 +1,34 @@
+module Rigger
+  class ServerResolver
+    def initialize(config)
+      @config = config
+    end
+
+    def call(task)
+      roles = [*task.options[:role]].compact
+      only  = task.options.fetch(:only, {})
+
+      role_servers = roles.empty? ? @config.servers : from_roles(roles)
+      only_servers = only.empty?  ? @config.servers : from_only(only)
+
+      (role_servers & only_servers)
+    end
+
+    protected
+      def from_roles(roles)
+        roles.map do |role|
+          @config.servers.select do |server|
+            server.role == role
+          end
+        end.flatten
+      end
+
+      def from_only(only)
+        @config.servers.select do |server|
+          only.all? do |key, value|
+            server.options[key] == value
+          end
+        end
+      end
+  end
+end

data/lib/rigger/task.rb

@@ -0,0 +1,4 @@
+module Rigger
+  class Task < Struct.new(:name, :options, :block)
+  end
+end

data/lib/rigger/task_execution_service.rb

@@ -0,0 +1,26 @@
+require "rigger/connection_set"
+require "rigger/server_resolver"
+require "rigger/task_executor"
+require "rigger/execution_strategy"
+
+module Rigger
+  class TaskExecutionService
+    def initialize(config,
+                   server_resolver             = ServerResolver.new(config),
+                   execution_strategy_selector = ExecutionStrategy::Selector.new)
+      @config                      = config
+      @server_resolver             = server_resolver
+      @execution_strategy_selector = execution_strategy_selector
+    end
+
+    def call(task_name)
+      task    = @config.locate_task(task_name)
+      servers = @server_resolver.call(task)
+      
+      puts "  * executing '#{task_name}'"
+
+      strategy = @execution_strategy_selector.call(task)
+      strategy.call(task, servers, @config, self)
+    end
+  end
+end

data/lib/rigger/task_executor.rb

@@ -0,0 +1,195 @@
+require "net/ssh"
+require "net/sftp"
+require "popen4"
+
+module Rigger
+  class TaskExecutor
+      class SFTPTransferWrapper
+        attr_reader :operation
+
+        def initialize(session, &callback)
+          @sftp = session.sftp(false).connect  do |sftp|
+            @operation = callback.call(sftp)
+          end
+        end
+
+        def active?
+          @operation.nil? || @operation.active?
+        end
+
+        def [](key)
+          @operation[key]
+        end
+
+        def []=(key, value)
+          @operation[key] = value
+        end
+
+        def abort!
+          @operation.abort!
+        end
+      end
+
+    def initialize(task, servers, execution_service, config)
+      @task              = task
+      @current_servers   = servers
+      @execution_service = execution_service
+      @config            = config
+    end
+
+    def call
+      instance_eval(&@task.block)
+    end
+
+    def run(command)
+      execute(command, @current_servers) do |ch|
+        ch.on_data do |c, data|
+          data.split("\n").each do |line|
+            puts " ** [#{ch[:host]} :: stdout] #{line}"
+            $stdout.flush
+          end
+        end
+
+        ch.on_extended_data do |c, type, data|
+          data.split("\n").each do |line|
+            puts " ** [#{ch[:host]} :: stderr] #{line}"
+            $stderr.flush
+          end
+        end
+      end
+    end
+
+    def capture(command)
+      "".tap do |captured|
+        execute(command, [@current_servers.first]) do |ch|
+          ch.on_data do |c, data|
+            captured << data
+          end
+
+          ch.on_extended_data do |c, type, data|
+            data.split("\n").each do |line|
+              puts " ** [#{server.connection_string} :: stderr] #{line}"
+              $stderr.flush
+            end
+          end
+        end
+      end
+    end
+
+    def run_task(task_name)
+      @execution_service.call(task_name)
+    end
+
+    def run_locally(command)
+      puts "  * executing `#{command}` locally"
+      status = POpen4.popen4(command) do |stdout, stderr, stdin, pid|
+        stdout.each_line do |line|
+          puts " ** [locally :: stdout] #{line}"
+        end
+
+        stderr.each_line do |line|
+          puts " ** [locally :: stderr] #{line}"
+        end
+      end
+
+      if status && status.exitstatus == 0
+        puts "  * command finished"
+      else
+        raise CommandError, "Local command `#{command}` failed."
+      end
+    end
+
+    def get(name)
+      @config.get(name)
+    end
+
+    def set(name, value)
+      @config.set(name, value)
+    end
+
+    def fetch(name, default)
+      @config.fetch(name, default)
+    end
+
+    def put(data, path)
+      servers  = @current_servers.dup
+      channels = servers.map do |s|
+        callback = Proc.new do |channel, name, sent, total|
+          puts "[#{channel[:host]}] #{name}" if sent == 0
+        end
+
+        SFTPTransferWrapper.new(s.connection) do |sftp|
+          io = StringIO.new(data.respond_to?(:read) ? data.read : data)
+          sftp.upload(io, path, {}) do |status, we|
+            if status == :finish
+              sftp.close_channel
+            end
+          end
+        end
+      end
+
+      puts "  * transerring data to #{path}"
+
+      connections = servers.dup
+
+      failing_servers = []
+      errors          = []
+
+      loop do
+        connections.delete_if do |server|
+          begin
+            !server.connection.process(0.1) { |s| s.busy? }
+          rescue Net::SFTP::StatusException => e
+            failing_servers << server
+            errors << e.message
+          end
+        end
+        break if connections.empty?
+      end
+
+      if !failing_servers.empty?
+        raise CommandError, "Upload failed on #{failing_servers.map { |s| s.connection_string }.inspect} with #{errors.join(", ")}."
+      end
+
+      puts "  * finished"
+    end
+
+    protected
+      def execute(command, servers)
+        puts "  * executing `#{command}`"
+        puts "    servers: #{servers.map { |s| s.connection_string }.inspect}"
+        channels = servers.map do |server|
+          puts "    [#{server.connection_string}] executing command"
+          server.connection.open_channel do |ch|
+            ch.exec(command) do |ch, success|
+              ch[:host] = server.connection_string
+
+              yield ch
+
+              ch.on_request("exit-status") do |ch, data|
+                ch[:status] = data.read_long
+              end
+
+              ch.on_close do |ch|
+                ch[:closed] = true
+              end
+            end
+          end
+        end
+
+        connections = servers.map { |s| s.connection }
+
+        loop do
+          connections.delete_if { |ssh| !ssh.process(0.1) { |s| s.busy? } }
+          break if connections.empty?
+        end
+
+        failing_servers = channels.select { |ch| ch[:status] != 0 }
+        if failing_servers.empty?
+          puts "  * command finished"
+        else
+          raise CommandError, "Command `#{command}` failed on #{failing_servers.map { |ch| ch[:host] }.inspect}."
+        end
+      end
+    end
+end

data/spec/server_resolver_spec.rb

@@ -0,0 +1,56 @@
+require "spec_helper"
+require "rigger"
+require "rigger/server_resolver"
+
+describe "Rigger::ServerResolver" do
+  before do
+    @server1  = stub("Server", :options => {}, :role => :db)
+    @server2  = stub("Server", :options => {}, :role => :app)
+    @server3  = stub("Server", :options => {
+      :fuck => true
+    }, :role => :search)
+    @servers  = [@server1, @server2, @server3]
+    @config   = stub("Config", :servers => @servers)
+    @resolver = Rigger::ServerResolver.new(@config)
+  end
+
+  describe "when the task has no options" do
+    before do
+      @task = stub("Config", :options => {})
+    end
+
+    it "returns all the servers" do
+      @resolver.call(@task).should == @servers
+    end
+  end
+
+  describe "when the task specifies a role" do
+    before do
+      @task = stub("Config", :options => {:role => :app})
+    end
+
+    it "returns all the servers matching that role" do
+      @resolver.call(@task).should == [@server2]
+    end
+  end
+
+  describe "when the task specifies an option" do
+    before do
+      @task = stub("Config", :options => {:only => {:fuck => true}})
+    end
+
+    it "returns all the servers matching that role" do
+      @resolver.call(@task).should == [@server3]
+    end
+  end
+
+  describe "when no servers match" do
+    before do
+      @task = stub("Config", :options => {:only => {:FUCK => true}})
+    end
+
+    it "returns []" do
+      @resolver.call(@task).should == []
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,14 @@
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'rspec'
+require 'rigger'
+require 'mocha'
+require 'bourne'
+
+# Requires supporting files with custom matchers and macros, etc,
+# in ./support/ and its subdirectories.
+Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each {|f| require f}
+
+RSpec.configure do |config|
+  config.mock_with :mocha
+end

metadata

@@ -0,0 +1,268 @@
+--- !ruby/object:Gem::Specification 
+name: rigger
+version: !ruby/object:Gem::Version 
+  hash: 3
+  prerelease: 
+  segments: 
+  - 0
+  - 2
+  - 10
+  version: 0.2.10
+platform: ruby
+authors: 
+- James Golick
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-06-21 00:00:00 -07:00
+default_executable: rig
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 21
+        segments: 
+        - 1
+        - 0
+        - 1
+        version: 1.0.1
+  prerelease: false
+  type: :runtime
+  requirement: *id001
+  name: net-ssh-multi
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 5
+        segments: 
+        - 2
+        - 0
+        - 5
+        version: 2.0.5
+  prerelease: false
+  type: :runtime
+  requirement: *id002
+  name: net-sftp
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 31
+        segments: 
+        - 0
+        - 1
+        - 2
+        version: 0.1.2
+  prerelease: false
+  type: :runtime
+  requirement: *id003
+  name: popen4
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 2
+        - 3
+        - 0
+        version: 2.3.0
+  prerelease: false
+  type: :development
+  requirement: *id004
+  name: rspec
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 23
+        segments: 
+        - 1
+        - 0
+        - 0
+        version: 1.0.0
+  prerelease: false
+  type: :development
+  requirement: *id005
+  name: bundler
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id006 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 7
+        segments: 
+        - 1
+        - 5
+        - 2
+        version: 1.5.2
+  prerelease: false
+  type: :development
+  requirement: *id006
+  name: jeweler
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id007 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  prerelease: false
+  type: :development
+  requirement: *id007
+  name: rcov
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id008 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 43
+        segments: 
+        - 0
+        - 9
+        - 8
+        version: 0.9.8
+  prerelease: false
+  type: :development
+  requirement: *id008
+  name: mocha
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id009 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 15
+        segments: 
+        - 1
+        - 0
+        version: "1.0"
+  prerelease: false
+  type: :development
+  requirement: *id009
+  name: bourne
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id010 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 21
+        segments: 
+        - 1
+        - 0
+        - 1
+        version: 1.0.1
+  prerelease: false
+  type: :runtime
+  requirement: *id010
+  name: net-ssh-multi
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id011 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 31
+        segments: 
+        - 0
+        - 1
+        - 2
+        version: 0.1.2
+  prerelease: false
+  type: :runtime
+  requirement: *id011
+  name: popen4
+description: ""
+email: jamesgolick@gmail.com
+executables: 
+- rig
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE.txt
+- README.rdoc
+- TODO
+files: 
+- .document
+- .rspec
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.rdoc
+- Rakefile
+- TODO
+- VERSION
+- bin/rig
+- config/rig.rb
+- lib/rigger.rb
+- lib/rigger/cli.rb
+- lib/rigger/connection_set.rb
+- lib/rigger/dsl.rb
+- lib/rigger/execution_strategy.rb
+- lib/rigger/recipes/chef.rb
+- lib/rigger/recipes/deploy.rb
+- lib/rigger/recipes/sbt.rb
+- lib/rigger/recipes/service.rb
+- lib/rigger/server.rb
+- lib/rigger/server_resolver.rb
+- lib/rigger/task.rb
+- lib/rigger/task_execution_service.rb
+- lib/rigger/task_executor.rb
+- spec/server_resolver_spec.rb
+- spec/spec_helper.rb
+has_rdoc: true
+homepage: http://github.com/jamesgolick/rigger
+licenses: 
+- MIT
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.9.2
+signing_key: 
+specification_version: 3
+summary: ""
+test_files: 
+- spec/server_resolver_spec.rb
+- spec/spec_helper.rb