le1t0-capistrano 2.5.18.002 → 2.5.18.003

data/CHANGELOG

@@ -1,3 +1,7 @@
+== 2.5.18.003 / May 18, 2010
+
+Merged in multistage support.
+
 == 2.5.18.002 / May 18, 2010
 
 Various fixes and additions:

data/VERSION

@@ -1 +1 @@
-2.5.18.002
+2.5.18.003

data/lib/capistrano/extensions.rb

@@ -1,3 +1,5 @@
+require 'capistrano/multistage'
+
 module Capistrano
   class ExtensionProxy #:nodoc:
     def initialize(config, mod)

data/lib/capistrano/multistage.rb

@@ -0,0 +1,61 @@
+require 'fileutils'
+
+unless Capistrano::Configuration.respond_to?(:instance)
+  abort "capistrano/ext/multistage requires Capistrano 2"
+end
+
+Capistrano::Configuration.instance.load do
+  location = fetch(:stage_dir, "config/deploy")
+
+  unless exists?(:stages)
+    set :stages, Dir["#{location}/*.rb"].map { |f| File.basename(f, ".rb") }
+  end
+
+  stages.each do |name| 
+    desc "Set the target stage to `#{name}'."
+    task(name) do 
+      set :stage, name.to_sym 
+      load "#{location}/#{stage}" 
+    end 
+  end 
+
+  on :load do
+    if stages.include?(ARGV.first)
+      # Execute the specified stage so that recipes required in stage can contribute to task list
+      find_and_execute_task(ARGV.first) if ARGV.any?{ |option| option =~ /-T|--tasks|-e|--explain/ }
+    else
+      # Execute the default stage so that recipes required in stage can contribute tasks
+      find_and_execute_task(default_stage) if exists?(:default_stage)
+    end
+  end
+
+  namespace :multistage do
+    desc "[internal] Ensure that a stage has been selected."
+    task :ensure do
+      if !exists?(:stage)
+        if exists?(:default_stage)
+          logger.important "Defaulting to `#{default_stage}'"
+          find_and_execute_task(default_stage)
+        else
+          abort "No stage specified. Please specify one of: #{stages.join(', ')} (e.g. `cap #{stages.first} #{ARGV.last}')"
+        end
+      end 
+    end
+
+    desc "Stub out the staging config files."
+    task :prepare do
+      FileUtils.mkdir_p(location)
+      stages.each do |name|
+        file = File.join(location, name + ".rb")
+        unless File.exists?(file)
+          File.open(file, "w") do |f|
+            f.puts "# #{name.upcase}-specific deployment configuration"
+            f.puts "# please put general deployment config in config/deploy.rb"
+          end
+        end
+      end
+    end
+  end
+
+  on :start, "multistage:ensure", :except => stages + ['multistage:prepare']
+end

metadata

@@ -6,8 +6,8 @@ version: !ruby/object:Gem::Version
   - 2
   - 5
   - 18
-  - 2
-  version: 2.5.18.002
+  - 3
+  version: 2.5.18.003
 platform: ruby
 authors: 
 - Le1t0
@@ -140,6 +140,7 @@ files:
 - lib/capistrano/errors.rb
 - lib/capistrano/extensions.rb
 - lib/capistrano/logger.rb
+- lib/capistrano/multistage.rb
 - lib/capistrano/processable.rb
 - lib/capistrano/recipes/compat.rb
 - lib/capistrano/recipes/deploy.rb