rondy · March 30, 2012 22:32 · May 19, 2012 · May 19, 2012 · Mar 30, 2012
diff --git a/gistfile1.rb b/gistfile1.rb
@@ -25,7 +25,7 @@ def ask
   end
 
   def invalid_branch?(branch)
-    !git_branch.all.include? branch
+    !git_branch.exists? branch
   end
 
   def git_branch
@@ -42,6 +42,10 @@ def all
     branches.collect(&removing_asterisk_from_current_branch)
   end
 
+  def exists?(branch)
+    branches.include? branch
+  end
+
   private
 
   # => ["* master", "feature_1"]
@@ -52,4 +56,4 @@ def branches
   def removing_asterisk_from_current_branch
     lambda { |branch| branch.gsub(/^\* /,"") }
   end
-end
+end
diff --git a/gistfile1.rb b/gistfile1.rb
@@ -6,10 +6,10 @@
 end
 
 def prompt_branch_name
-  DeployBranchName.new.prompt
+  BranchedDeploy.new.prompt
 end
 
-class DeployBranchName
+class BranchedDeploy
   def prompt
     begin
       branch = ask

diff --git a/gistfile1.rb b/gistfile1.rb
@@ -0,0 +1,55 @@
+namespace :deploy do
+  task :set_branch do
+    set :branch, prompt_branch_name
+  end
+  before "deploy:update" , "deploy:set_branch"
+end
+
+def prompt_branch_name
+  DeployBranchName.new.prompt
+end
+
+class DeployBranchName
+  def prompt
+    begin
+      branch = ask
+    end while invalid_branch? branch
+    branch
+  end
+
+  private
+
+  # => * Type branch name to deploy: (master, feature_1) |master|
+  def ask
+    Capistrano::CLI.ui.ask("  * Type branch name to deploy (#{git_branch.all.join(', ')}):") { |q| q.default = git_branch.current }
+  end
+
+  def invalid_branch?(branch)
+    !git_branch.all.include? branch
+  end
+
+  def git_branch
+    @git_branch ||= GitBranch.new
+  end
+end
+
+class GitBranch
+  def current
+     branches.detect { |branch| branch.start_with? "*" }[2..-1]
+  end
+
+  def all
+    branches.collect(&removing_asterisk_from_current_branch)
+  end
+
+  private
+
+  # => ["* master", "feature_1"]
+  def branches
+    @branches ||= `git branch`.split("\n").map(&:strip)
+  end
+
+  def removing_asterisk_from_current_branch
+    lambda { |branch| branch.gsub(/^\* /,"") }
+  end
+end
No results found