capistrano select git branch

cap_select_git_branch.rb

set :branch do

  branches=`git branch -r | sed "1 d"`.split
  branches.map { | b | b.gsub!("origin/","") }
  
  puts "What branch would you like to deploy from?"
  branches.each_index do | i |
    puts((i+1).to_s + ": " + branches[i])
  end

  branch_index = Capistrano::CLI.ui.ask('Branch to deploy from?:').to_i
  branches[branch_index-1]

end