innvent/matross

dep_included? 'foreman'

_cset(:foreman_user)  { user }
_cset :foreman_bin,   'bundle exec foreman'
_cset :foreman_procs, {}
_cset :rails_env,     ''

namespace :foreman do

  desc "Pre-setup, creates the shared upstart folder"
  task :pre_setup, :except => { :no_release => true } do
    run "mkdir -p #{shared_path}/upstart"
  end
  before "foreman:setup", "foreman:pre_setup"

  desc "Merges all partial Procfiles and defines a specific dotenv"
  task :setup, :except => { :no_release => true } do
    cmd = <<-EOF.gsub(/^\s+/, '')
      rm -f #{shared_path}/Procfile-matross;
      for file in #{current_path}/Procfile #{shared_path}/Procfile.*; do \
        [ -f $file ] && cat $file >> #{shared_path}/Procfile-matross;
      done;
      rm -f #{shared_path}/Procfile.*;
      cat <(echo \"RAILS_ENV=#{rails_env.to_s.shellescape}\") \
        $(test -f #{current_path}/.env && echo \"$_\") \
        $(test -f #{current_path}/.env-#{stage} && echo \"$_\") > \
        #{shared_path}/.env-matross;
    EOF
    run cmd, :shell => "/bin/bash"
  end
  before "foreman:export", "foreman:setup"

  desc "Export the Procfile to Ubuntu's upstart scripts"
  task :export, :except => { :no_release => true } do
    matross_path = "#{shared_path}/matross"
    run "mkdir -p #{matross_path}"
    upload File.expand_path("../templates/foreman", __FILE__), matross_path,
           :via => :scp, :recursive => true

    # By default spawn one instance of every process
    procs = {}
    capture("cat #{current_path}/Procfile-matross").split("\n").each { |line|
      process = line[/^([A-Za-z0-9_]+):\s*(.+)$/, 1]
      procs[process] = 1
    }
    procs.merge!(foreman_procs)

    proc_list = ""
    procs.each { |process, num|
      proc_list << ',' unless proc_list.empty?
      proc_list << "#{process}=#{num}"
    }
    proc_list = " -c #{proc_list}" unless proc_list.empty?

    run "cd #{current_path} && #{foreman_bin} export upstart #{shared_path}/upstart "\
      "-f #{current_path}/Procfile-matross "\
      "-a #{application} "\
      "-u #{foreman_user} "\
      "-l #{shared_path}/log "\
      "-t #{matross_path}/foreman "\
      "-e #{shared_path}/.env-matross"\
      << proc_list
    run "cd #{shared_path}/upstart && #{sudo} cp * /etc/init/"
  end
  before "deploy:restart", "foreman:export"

  desc "Symlink configuration scripts"
  task :symlink, :except => { :no_release => true } do
    run "ln -nfs #{shared_path}/Procfile-matross #{current_path}/Procfile-matross"
    run "ln -nfs #{shared_path}/.env-matross #{current_path}/.env-matross"
  end
  after "foreman:setup", "foreman:symlink"

  desc "Symlink upstart logs to application shared/log"
  task :log, :except => { :no_release => true } do
    run <<-EOF.gsub(/^s+/, '')
    ls #{shared_path}/upstart -1 | while read line; do
      logname=$(basename $line | sed "s/.conf$/.log/");
      #{sudo} touch /var/log/upstart/${logname} &&
      #{sudo} chmod o+r /var/log/upstart/${logname} &&
      ln -nfs /var/log/upstart/${logname} #{shared_path}/log/${logname};
    done
    EOF
  end
  after "foreman:export", "foreman:log"

  desc "Stop services"
  task :stop, :except => { :no_release => true } do
    run "#{sudo} stop #{application}"
  end

  desc "Restart services"
  task :restart, :except => { :no_release => true } do
    run "#{sudo} start #{application} || #{sudo} restart #{application}"
  end
  after "deploy:restart", "foreman:restart"

  desc "Remove upstart scripts"
  task :remove, :except => { :no_release => true } do
    run "cd #{shared_path}/upstart && rm -f Procfile*"
    run "cd /etc/init/ && #{sudo} rm #{application}*"
  end
end