Add epi_deploy:set_branch task for manually set the Capistrano :branc…

…h variable on deployment

lib/capistrano/epi_deploy.rb

@@ -1 +1 @@
-load File.join(File.dirname(__FILE__), 'tasks', 'multi_customers.rb')
+Dir.glob(File.join(File.dirname(__FILE__), 'tasks', '*.rb')).each { |f| require f }

lib/capistrano/tasks/branches.rb

@@ -0,0 +1,21 @@
+if File.exist?('config/initializers/version.rb')
+  require_relative 'config/initializers/version'
+end
+
+namespace :epi_deploy do
+  task :set_branch do
+    branch = if ENV['BRANCH']
+      ENV['BRANCH']
+    elsif Object.const_defined?('LATEST_RELEASE_TAG')
+      LATEST_RELEASE_TAG
+    end
+
+    if branch.nil?
+      raise 'Cannot determine commit to deploy as BRANCH environment variable is not set and LATEST_RELEASE_TAG constant in version.rb could not be found'
+    end
+
+    set :branch, branch
+  end
+end
+
+before 'deploy:starting', 'epi_deploy:set_branch'