require "rest-client"
require "json"
require "multi_json"

module WorkerArmy
  class Client
    class << self
      def push_job(job_class, data = {}, callback_url = nil, queue_prefix = 'queue', retry_count = 0)
        raise "No data" unless data
        raise "No job class provided" unless job_class
        
        if ENV['worker_army_endpoint']
          # puts "Using environment variables for config..."
          @config = { endpoint: ENV['worker_army_endpoint'] }
        else
          begin
            # puts "Using config in your home directory"
            @config = YAML.load(File.read("#{ENV['HOME']}/.worker_army.yml"))
          rescue Errno::ENOENT
            raise "worker_army.yml expected in ~/.worker_army.yml"
          end
        end

        worker_army_base_url = @config['endpoint']
        callback_url = "#{worker_army_base_url}/generic_callback" unless callback_url
        response = nil
        begin
          response = RestClient.post "#{worker_army_base_url}/jobs",
            data.merge(
              job_class: job_class,
              callback_url: "#{worker_army_base_url}/callback?callback_url=#{callback_url}",
              queue_prefix: queue_prefix
            ).to_json,
            :content_type => :json, :accept => :json
        rescue => e
          puts "Failed! Retrying (#{retry_count})..."
          retry_count += 1
          if retry_count < client_retry_count(@config)
            sleep (retry_count * 2)
            push_job(job_class, data, callback_url, queue_prefix, retry_count)
          end
        end
        if response and response.body and response.code == 200
          hash = JSON.parse(response.body)
          hash.merge(success: true)
        else
          { success: false }
        end
      end

      def client_retry_count(config)
        if ENV['worker_army_client_retry_count']
          return ENV['worker_army_client_retry_count'].to_i
        elsif config and config['client_retry_count']
          return config['client_retry_count'].to_i
        else
          return 10
        end
      end
    end
  end
end