#!/usr/bin/env ruby

# This script handles starting and stopping of resque workers

require File.dirname(__FILE__) + '/../config/environment'
require 'yaml'

rails_root = File.dirname(__FILE__) + '/..'
rails_env = ENV['RAILS_ENV'] || 'development'

worker_config = YAML.load_file(rails_root + '/config/resque_work.yml')[rails_env]

begin
  case ARGV[0]

  when 'start'
    count = worker_config['count'] || 1
    puts "#{count} workers being started"
    count.times do
      puts "Worker started." if Resque::Worker.new(worker_config['queue']).startup
    end
    puts "#{Resque.workers.length} workers running"
  when 'stop'
    workers = Resque.workers
    puts "#{workers.length} workers running"
    workers.each do |worker|
      worker.shutdown!
      worker.unregister_worker
    end
    puts "#{Resque.workers.length} workers running"
  when 'run'
  else
    puts "usage: start|stop|run"
  end
rescue => e
  HoptoadNotifier.notify(e)
  raise
end