if defined?(Capistrano)
  Capistrano::Configuration.instance(:must_exist).load do

    namespace :foreman do
      desc <<-DESC
        Export the Procfile to upstart.  Will use sudo if available.

        You can override any of these defaults by setting the variables shown below.

        set :foreman_format,      "upstart"
        set :foreman_location,    "/etc/init"
        set :foreman_procfile,    "Procfile"
        set :foreman_app,         application
        set :foreman_user,        user
        set :foreman_log,         "\#{shared_path}/log"
        set :foreman_concurrency, false
      DESC
      task :export, :roles => :app do
        bundle_cmd          = fetch(:bundle_cmd, "bundle")
        foreman_format      = fetch(:foreman_format, "upstart")
        foreman_location    = fetch(:foreman_location, "/etc/init")
        foreman_procfile    = fetch(:foreman_procfile, "Procfile")
        foreman_app         = fetch(:foreman_app, application)
        foreman_user        = fetch(:foreman_user, user)
        foreman_port        = fetch(:foreman_port, 5000)
        foreman_log         = fetch(:foreman_log, "#{shared_path}/log")
        foreman_concurrency = fetch(:foreman_concurrency, false)

        args = ["#{foreman_format} #{foreman_location}"]
        args << "-f #{foreman_procfile}"
        args << "-a #{foreman_app}"
        args << "-u #{foreman_user}"
        args << "-l #{foreman_log}"
        args << "-p #{foreman_port}"
        args << "-c #{foreman_concurrency}" if foreman_concurrency
        run "cd #{current_path} && #{bundle_cmd} exec foreman export #{args.join(' ')}"
      end

      desc "Start the application services"
      task :start, :roles => :app do
        if foreman_format == 'upstart'
          run "#{sudo} start #{application}"
        elsif foreman_format == 'monit'
          run "monit start -g #{application}"
        end
      end

      desc "Stop the application services"
      task :stop, :roles => :app do
        if foreman_format == 'upstart'
          run "#{sudo} stop #{application}"
        elsif foreman_format == 'monit'
          run "monit stop -g #{application}"
        end
      end

      desc "Restart the application services"
      task :restart, :roles => :app do
        if foreman_format == 'upstart'
          run "#{sudo} start #{application} || #{sudo} restart #{application}"
        elsif foreman_format == 'monit'
          run "monit restart -g #{application}"
        end
      end
    end
  end
end