namespace :staff_bar do

desc "Place a BRANCH file with the current branch in the current release path"
task :set_current_branch do
  on roles(:app) do
    within release_path do
      execute :echo, "\"#{fetch(:branch)}\" >> BRANCH"
    end
  end
end

end

after 'deploy:finished', 'staff_bar:set_current_branch'