namespace :foreman do
  desc <<-DESC
        Setup foreman configuration

        Configurable options are:

          set :foreman_roles, :all
          set :foreman_export_format, 'upstart'
          set :foreman_export_path, '/etc/init'
          set :foreman_flags, ''
          set :foreman_target_path, release_path
          set :foreman_app, -> { fetch(:application) }
          set :foreman_concurrency, 'web=2,worker=1' # default is not set
          set :foreman_log, -> { shared_path.join('log') }
          set :foreman_port, 3000 # default is not set
          set :foreman_user, 'www-data' # default is not set
    DESC

  task :setup do
    invoke :'foreman:export'
    invoke :'foreman:enable'
    invoke :'foreman:start'
  end

  desc "Enables service in systemd"
  task :enable do
    on roles fetch(:foreman_roles) do
      sudo :systemctl, "enable #{fetch(:foreman_app)}.target"
    end
  end

  desc "Disables service in systemd"
  task :disable do
    on roles fetch(:foreman_roles) do
      sudo :systemctl, "disable #{fetch(:foreman_app)}.target"
    end
  end


  desc "Export the Procfile to another process management format"
  task :export do
    on roles fetch(:foreman_roles) do |server|
      execute :mkdir, "-p", fetch(:foreman_export_path) unless test "[ -d #{fetch(:foreman_export_path)} ]"
      within fetch(:foreman_target_path, release_path) do

        options = {
          app: fetch(:foreman_app),
          log: fetch(:foreman_log)
        }
        options[:concurrency] = fetch(:foreman_concurrency) if fetch(:foreman_concurrency)
        options[:concurrency] = server.properties.foreman_concurrency if server.properties.foreman_concurrency
        options[:port] = fetch(:foreman_port) if fetch(:foreman_port)
        options[:user] = fetch(:foreman_user) if fetch(:foreman_user)

        sudo :foreman, 'export', fetch(:foreman_export_format), fetch(:foreman_export_path),
          options.map{ |k, v| "--#{k}='#{v}'" }, fetch(:foreman_flags)
      end
    end
  end

  desc "Start the application services"
  task :start do
    on roles fetch(:foreman_roles) do
      sudo :start, fetch(:foreman_app)
    end
  end

  desc "Stop the application services"
  task :stop do
    on roles fetch(:foreman_roles) do
      sudo :stop, fetch(:foreman_app)
    end
  end

  desc "Restart the application services"
  task :restart do
    on roles fetch(:foreman_roles) do
      sudo :restart, fetch(:foreman_app)
    end
  end

end

namespace :load do
  task :defaults do
    set :foreman_roles, :all
    set :foreman_export_format, 'upstart'
    set :foreman_export_path, '/etc/init'
    set :foreman_flags, ''
    set :foreman_app, -> { fetch(:application) }
    set :foreman_log, -> { shared_path.join('log') }
  end
end